From dcd216e3457bdbbbc7b8dc27783866b748e322fa Mon Sep 17 00:00:00 2001
From: Marco Vinciguerra <mvincig11@gmail.com>
Date: Sat, 15 Jun 2024 19:51:20 +0200
Subject: [PATCH] fix: removed duplicate from ollama dictionary

Co-Authored-By: Matteo Vedovati <68272450+vedovati-matteo@users.noreply.github.com>
---
 scrapegraphai/helpers/models_tokens.py | 71 ++++++++++----------------
 1 file changed, 27 insertions(+), 44 deletions(-)

diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py
index 510777fe..ee20e9b9 100644
--- a/scrapegraphai/helpers/models_tokens.py
+++ b/scrapegraphai/helpers/models_tokens.py
@@ -35,50 +35,33 @@
         "gemini-1.5-pro-latest":128000,
         "models/embedding-001": 2048
     },
-
-    "ollama": {
-        "command-r": 12800,
-        "command-r-plus": 12800,
-        "codellama": 16000,
-        "dbrx": 32768,
-        "dbrx:instruct": 32768,
-        "deepseek-coder:33b": 16000,
-        "dolphin-mixtral": 32000,
-        "llama2": 4096,
-        "llama3": 8192,
-        "llama3:70b-instruct": 8192,
-        "llava": 4096,
-        "llava:34b": 4096,
-        "llava_next": 4096,
-        "mistral": 8192,
-        "falcon": 2048,
-        "codellama": 16000,
-        "dolphin-mixtral": 32000,
-        "mistral-openorca": 32000,
-        "stablelm-zephyr": 8192,
-        "command-r-plus": 12800,
-        "command-r": 12800,
-        "mistral:7b-instruct": 32768,
-        "mistral-openorca": 32000,
-        "mixtral:8x22b-instruct": 65536,
-        "nous-hermes2:34b": 4096,
-        "orca-mini": 2048,
-        "phi3:3.8b": 12800,
-        "phi3:14b": 12800,
-        "qwen:0.5b": 32000,
-        "qwen:1.8b": 32000,
-        "qwen:4b": 32000,
-        "qwen:14b": 32000,
-        "qwen:32b": 32000,
-        "qwen:72b": 32000,
-        "qwen:110b": 32000,
-        "stablelm-zephyr": 8192,
-        "wizardlm2:8x22b": 65536,
-        # embedding models
-        "nomic-embed-text": 8192,
-        "snowflake-arctic-embed:335m": 8192,
-        "snowflake-arctic-embed:l": 8192,
-        "mxbai-embed-large": 512,
+    "ollama": { "command-r": 12800, 
+               "codellama": 16000, 
+               "dbrx": 32768, 
+               "deepseek-coder:33b": 16000, 
+               "falcon": 2048, 
+               "llama2": 4096, 
+               "llama3": 8192, 
+               "scrapegraph": 8192, 
+               "llava": 4096, 
+               "mixtral:8x22b-instruct": 65536, 
+               "mistral-openorca": 32000, 
+               "nomic-embed-text": 8192, 
+               "nous-hermes2:34b": 4096, 
+               "orca-mini": 2048, 
+               "phi3:3.8b": 12800, 
+               "qwen:0.5b": 32000, 
+               "qwen:1.8b": 32000, 
+               "qwen:4b": 32000, 
+               "qwen:14b": 32000, 
+               "qwen:32b": 32000, 
+               "qwen:72b": 32000, 
+               "qwen:110b": 32000, 
+               "stablelm-zephyr": 8192, 
+               "wizardlm2:8x22b": 65536, 
+               # embedding models 
+               "snowflake-arctic-embed": 8192, 
+               "mxbai-embed-large": 512 
     },
     "oneapi": {
         "qwen-turbo": 16380