CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-07-01 21:00:48 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: update abstract graph

Browse Source
This commit is contained in:
Marco Vinciguerra 2024-08-24 21:57:04 +02:00
parent ab21576435
commit 86fe5fcaf1
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
scrapegraphai/graphs/abstract_graph.py
View File

@ -141,7 +141,8 @@ class AbstractGraph(ABC):
known_models = {"chatgpt","gpt","openai", "azure_openai", "google_genai", known_models = {"chatgpt","gpt","openai", "azure_openai", "google_genai",
"ollama", "oneapi", "nvidia", "groq", "google_vertexai", "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: 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") raise ValueError(f"Model '{llm_params['model']}' is not supported")