fix(AbstractGraph): LangChain warnings handling, Mistral tokens

Author: f-aguzzi

scrapegraphai/graphs/abstract_graph.py

@@ -5,6 +5,7 @@
 from abc import ABC, abstractmethod
 from typing import Optional
 import uuid
+import warnings
 from pydantic import BaseModel
 
 from langchain_community.chat_models import ErnieBotChat
@@ -144,7 +145,9 @@ def handle_model(model_name, provider, token_key, default_token=8192):
                 self.model_token = default_token
             llm_params["model_provider"] = provider
             llm_params["model"] = model_name
-            return init_chat_model(**llm_params)
+            with warnings.catch_warnings():
+                warnings.simplefilter("ignore")
+                return init_chat_model(**llm_params)
 
         if "azure" in llm_params["model"]:
             model_name = llm_params["model"].split("/")[-1]

scrapegraphai/helpers/models_tokens.py

@@ -145,7 +145,7 @@
         "cohere.embed-english-v3": 512,
         "cohere.embed-multilingual-v3": 512
     },
-    "mistral": {
+    "mistralai": {
         "mistral-large-latest": 128000,
         "open-mistral-nemo": 128000,
         "codestral-latest": 32000,