Skip to content

Commit 609316c

Browse files
committed
Update abstract_graph.py
1 parent 2053693 commit 609316c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scrapegraphai/graphs/abstract_graph.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ def handle_model(model_name, provider, token_key, default_token=8192):
147147
warnings.simplefilter("ignore")
148148
return init_chat_model(**llm_params)
149149

150-
known_models = ["gpt","openai", "azure_openai", "google_genai", "ollama", "oneapi", "nvidia", "groq", "google_vertexai", "bedrock", "mistralai", "hugging_face", "deepseek", "ernie", "fireworks"]
150+
known_models = ["chatgpt","gpt","openai", "azure_openai", "google_genai", "ollama", "oneapi", "nvidia", "groq", "google_vertexai", "bedrock", "mistralai", "hugging_face", "deepseek", "ernie", "fireworks"]
151151

152152
if llm_params["model"].split("/")[0] not in known_models and llm_params["model"].split("-")[0] not in known_models:
153153
raise ValueError(f"Model '{llm_params['model']}' is not supported")

0 commit comments

Comments
 (0)