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