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

feat: add mistral support

Browse Source
This commit is contained in:
Federico Aguzzi 2024-08-06 17:51:50 +02:00
parent 9868555123
commit 17f2707313
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scrapegraphai/graphs/abstract_graph.py
View File

@ -188,6 +188,10 @@ class AbstractGraph(ABC):
if "claude-3-" in llm_params["model"]: if "claude-3-" in llm_params["model"]:
return handle_model(llm_params["model"], "anthropic", "claude3") return handle_model(llm_params["model"], "anthropic", "claude3")
if llm_params["model"].startswith("mistral"):
model_name = llm_params["model"].split("/")[-1]
return handle_model(model_name, "mistral", model_name)
# Instantiate the language model based on the model name (models that do not use the common interface) # Instantiate the language model based on the model name (models that do not use the common interface)
if "deepseek" in llm_params["model"]: if "deepseek" in llm_params["model"]: