Skip to content

Commit cb6b353

Browse files
committed
fix(models_tokens): incorrect provider names
1 parent de1ec25 commit cb6b353

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

scrapegraphai/helpers/models_tokens.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
"gpt-4o-mini":128000,
2020

2121
},
22-
"azure": {
22+
"azure_openai": {
2323
"gpt-3.5-turbo-0125": 16385,
2424
"gpt-3.5": 4096,
2525
"gpt-3.5-turbo": 16385,
@@ -38,7 +38,7 @@
3838
"gpt-4o": 128000,
3939
"gpt-4o-mini":128000,
4040
},
41-
"gemini": {
41+
"google_genai": {
4242
"gemini-pro": 128000,
4343
"gemini-1.5-flash-latest": 128000,
4444
"gemini-1.5-pro-latest": 128000,
@@ -121,7 +121,7 @@
121121
"claude-3-haiku-20240307": 200000,
122122
"claude-3-5-sonnet-20240620": 200000
123123
},
124-
"vertexai": {
124+
"google_vertexai": {
125125
"gemini-1.5-flash": 128000,
126126
"gemini-1.5-pro": 128000,
127127
"gemini-1.0-pro": 128000

0 commit comments

Comments
 (0)