Skip to content

Commit 79db9b9

Browse files
committed
feat: update model tokens
1 parent 739b05a commit 79db9b9

File tree

1 file changed

+21
-6
lines changed

1 file changed

+21
-6
lines changed

scrapegraphai/helpers/models_tokens.py

Lines changed: 21 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -60,8 +60,10 @@
6060
"google_genai": {
6161
"gemini-pro": 128000,
6262
"gemini-1.5-flash-latest": 128000,
63-
"gemini-2.0-flash-latest": 128000,
6463
"gemini-1.5-pro-latest": 128000,
64+
"gemini-2.0-flash-latest": 1000000,
65+
"gemini-2.0-flash-exp": 1000000,
66+
"gemini-2.0-pro-exp": 2000000,
6567
"models/embedding-001": 2048,
6668
},
6769
"google_vertexai": {
@@ -148,8 +150,11 @@
148150
"groq": {
149151
"llama3-8b-8192": 8192,
150152
"llama3-70b-8192": 8192,
153+
"llama-3.1-8b-instant": 128000,
154+
"llama-3.3-70b-versatile": 128000,
151155
"mixtral-8x7b-32768": 32768,
152156
"gemma-7b-it": 8192,
157+
"gemma2-9b-it": 8192,
153158
"claude-3-haiku-20240307": 8192,
154159
},
155160
"toghetherai": {
@@ -208,7 +213,7 @@
208213
"mistralai": {
209214
"mistral-large-latest": 128000,
210215
"open-mistral-nemo": 128000,
211-
"codestral-latest": 32000,
216+
"codestral-latest": 256000,
212217
"mistral-embed": 8000,
213218
"open-mistral-7b": 32000,
214219
"open-mixtral-8x7b": 32000,
@@ -250,8 +255,11 @@
250255
"claude-3-haiku": 200000,
251256
},
252257
"deepseek": {
253-
"deepseek-chat": 28672,
254-
"deepseek-coder": 16384,
258+
"deepseek-chat": 128000,
259+
"deepseek-coder": 128000,
260+
"deepseek-v3": 128000,
261+
"deepseek-v3.1": 128000,
262+
"deepseek-r1": 128000,
255263
},
256264
"ernie": {
257265
"ernie-bot-turbo": 4096,
@@ -285,6 +293,8 @@
285293
"claude-3-opus-latest": 200000,
286294
"gemini-1.5-flash-8b": 128000,
287295
"gemini-1.5-flash": 128000,
296+
"gemini-2.0-flash": 1000000,
297+
"gemini-2.0-pro": 2000000,
288298
"open-mixtral-8x7b": 32000,
289299
"open-mixtral-8x22b": 64000,
290300
"claude-3-5-sonnet-latest": 200000,
@@ -304,11 +314,16 @@
304314
"gemini-2.0-flash-exp": 1000000,
305315
"grok-beta": 128000,
306316
"grok-2-latest": 128000,
317+
"grok-3": 1000000,
318+
"grok-3-mini": 1000000,
307319
},
308320
"togetherai": {"Meta-Llama-3.1-70B-Instruct-Turbo": 128000},
309321
"xai": {
310322
"grok-1": 8192,
311-
"grok-3": 128000,
312-
"grok-3-mini": 128000,
323+
"grok-2": 128000,
324+
"grok-2-latest": 128000,
325+
"grok-3": 1000000,
326+
"grok-3-mini": 1000000,
327+
"grok-beta": 128000,
313328
},
314329
}

0 commit comments

Comments
 (0)