Skip to content

Commit 17f2707

Browse files
committed
feat: add mistral support
1 parent 9868555 commit 17f2707

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

scrapegraphai/graphs/abstract_graph.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -188,6 +188,10 @@ def handle_model(model_name, provider, token_key, default_token=8192):
188188

189189
if "claude-3-" in llm_params["model"]:
190190
return handle_model(llm_params["model"], "anthropic", "claude3")
191+
192+
if llm_params["model"].startswith("mistral"):
193+
model_name = llm_params["model"].split("/")[-1]
194+
return handle_model(model_name, "mistral", model_name)
191195

192196
# Instantiate the language model based on the model name (models that do not use the common interface)
193197
if "deepseek" in llm_params["model"]:

0 commit comments

Comments
 (0)