From 86fe5fcaf1a6ba28786678874378f07fba1db40f Mon Sep 17 00:00:00 2001 From: Marco Vinciguerra Date: Sat, 24 Aug 2024 21:57:04 +0200 Subject: [PATCH] fix: update abstract graph --- scrapegraphai/graphs/abstract_graph.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/scrapegraphai/graphs/abstract_graph.py b/scrapegraphai/graphs/abstract_graph.py index bfc1848f..555e8211 100644 --- a/scrapegraphai/graphs/abstract_graph.py +++ b/scrapegraphai/graphs/abstract_graph.py @@ -141,7 +141,8 @@ class AbstractGraph(ABC): known_models = {"chatgpt","gpt","openai", "azure_openai", "google_genai", "ollama", "oneapi", "nvidia", "groq", "google_vertexai", - "bedrock", "mistralai", "hugging_face", "deepseek", "ernie", "fireworks"} + "bedrock", "mistralai", "hugging_face", "deepseek", "ernie", + "fireworks", "claude-3-"} if llm_params["model"].split("/")[0] not in known_models and llm_params["model"].split("-")[0] not in known_models: raise ValueError(f"Model '{llm_params['model']}' is not supported") @@ -267,4 +268,4 @@ class AbstractGraph(ABC): def run(self) -> str: """ Abstract method to execute the graph and return the result. - """ + """ \ No newline at end of file