From cb6b35397e56c6785553480200aa948053d9904b Mon Sep 17 00:00:00 2001
From: Federico Aguzzi <62149513+f-aguzzi@users.noreply.github.com>
Date: Mon, 12 Aug 2024 10:52:13 +0200
Subject: [PATCH] fix(models_tokens): incorrect provider names

---
 scrapegraphai/helpers/models_tokens.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py
index e32838f1..8f863a9c 100644
--- a/scrapegraphai/helpers/models_tokens.py
+++ b/scrapegraphai/helpers/models_tokens.py
@@ -19,7 +19,7 @@ models_tokens = {
         "gpt-4o-mini":128000,
 
     },
-    "azure": {
+    "azure_openai": {
         "gpt-3.5-turbo-0125": 16385,
         "gpt-3.5": 4096,
         "gpt-3.5-turbo": 16385,
@@ -38,7 +38,7 @@ models_tokens = {
         "gpt-4o": 128000,
         "gpt-4o-mini":128000,
     },
-    "gemini": {
+    "google_genai": {
         "gemini-pro": 128000,
         "gemini-1.5-flash-latest": 128000,
         "gemini-1.5-pro-latest": 128000,
@@ -121,7 +121,7 @@ models_tokens = {
         "claude-3-haiku-20240307": 200000,
         "claude-3-5-sonnet-20240620": 200000
     },
-    "vertexai": {
+    "google_vertexai": {
         "gemini-1.5-flash": 128000,
         "gemini-1.5-pro": 128000,
         "gemini-1.0-pro": 128000