From 17f2707313f65a1e96443b3c8a1f5137892f2c5a Mon Sep 17 00:00:00 2001
From: Federico Aguzzi <62149513+f-aguzzi@users.noreply.github.com>
Date: Tue, 6 Aug 2024 17:51:50 +0200
Subject: [PATCH] feat: add mistral support

---
 scrapegraphai/graphs/abstract_graph.py | 4 ++++
 1 file changed, 4 insertions(+)

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"]: