diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py
index 113d1636..8f367ceb 100644
--- a/scrapegraphai/helpers/models_tokens.py
+++ b/scrapegraphai/helpers/models_tokens.py
@@ -80,10 +80,12 @@
             "llama3.2:1b": 128000,
             "scrapegraph": 8192,
             "mistral": 8192,
+            "mistral-small": 128000,
+            "mistral-openorca": 32000,
+            "mistral-large": 128000,
             "grok-1": 8192,
             "llava": 4096,
             "mixtral:8x22b-instruct": 65536,
-            "mistral-openorca": 32000,
             "nomic-embed-text": 8192,
             "nous-hermes2:34b": 4096,
             "orca-mini": 2048,