diff --git a/scrapegraphai/graphs/abstract_graph.py b/scrapegraphai/graphs/abstract_graph.py index f07bcb10..7e16f644 100644 --- a/scrapegraphai/graphs/abstract_graph.py +++ b/scrapegraphai/graphs/abstract_graph.py @@ -188,6 +188,10 @@ class AbstractGraph(ABC): if "claude-3-" in llm_params["model"]: 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) if "deepseek" in llm_params["model"]: