Skip to content

Commit 86fe5fc

Browse files
committed
fix: update abstract graph
1 parent ab21576 commit 86fe5fc

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

scrapegraphai/graphs/abstract_graph.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,8 @@ def handle_model(model_name, provider, token_key, default_token=8192):
141141

142142
known_models = {"chatgpt","gpt","openai", "azure_openai", "google_genai",
143143
"ollama", "oneapi", "nvidia", "groq", "google_vertexai",
144-
"bedrock", "mistralai", "hugging_face", "deepseek", "ernie", "fireworks"}
144+
"bedrock", "mistralai", "hugging_face", "deepseek", "ernie",
145+
"fireworks", "claude-3-"}
145146

146147
if llm_params["model"].split("/")[0] not in known_models and llm_params["model"].split("-")[0] not in known_models:
147148
raise ValueError(f"Model '{llm_params['model']}' is not supported")
@@ -267,4 +268,4 @@ def _create_graph(self):
267268
def run(self) -> str:
268269
"""
269270
Abstract method to execute the graph and return the result.
270-
"""
271+
"""

0 commit comments

Comments
 (0)