From cb05f824ce2f7a0c0898c67e6034af57c3ec3d2a Mon Sep 17 00:00:00 2001 From: ZuanZuan Date: Sat, 7 Sep 2024 17:35:03 +0800 Subject: [PATCH] Update abstract_graph.py --- scrapegraphai/graphs/abstract_graph.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scrapegraphai/graphs/abstract_graph.py b/scrapegraphai/graphs/abstract_graph.py index 0d02b6d4..7b161963 100644 --- a/scrapegraphai/graphs/abstract_graph.py +++ b/scrapegraphai/graphs/abstract_graph.py @@ -128,7 +128,7 @@ def _create_llm(self, llm_config: dict) -> object: return llm_params["model_instance"] known_providers = {"openai", "azure_openai", "google_genai", "google_vertexai", - "ollama", "oneapi", "nvidia", "groq", "anthropic" "bedrock", "mistralai", + "ollama", "oneapi", "nvidia", "groq", "anthropic", "bedrock", "mistralai", "hugging_face", "deepseek", "ernie", "fireworks"} split_model_provider = llm_params["model"].split("/", 1)