Skip to content

Commit 4835fdb

Browse files
committed
fix: update default models and add more gemini models
1 parent d5b82d7 commit 4835fdb

File tree

1 file changed

+25
-3
lines changed

1 file changed

+25
-3
lines changed

gptme/llm/models.py

Lines changed: 25 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -192,6 +192,28 @@ class _ModelDictMeta(TypedDict):
192192
"price_output": 10,
193193
"supports_vision": True,
194194
},
195+
"gemini-2.5-flash-lite": {
196+
"context": 1_000_000,
197+
"max_output": 64_000,
198+
"price_input": 0.1,
199+
"price_output": 0.4,
200+
"supports_vision": True,
201+
},
202+
"gemini-2.5-flash": {
203+
"context": 1_048_576,
204+
"max_output": 65_536,
205+
"price_input": 0.3,
206+
"price_output": 2.5,
207+
"supports_vision": True,
208+
},
209+
"gemini-2.5-pro": {
210+
"context": 1_048_576,
211+
"max_output": 8192,
212+
# NOTE: at >200k context price is 2x for input and 1.5x for output
213+
"price_input": 1.25,
214+
"price_output": 10,
215+
"supports_vision": True,
216+
},
195217
},
196218
# https://api-docs.deepseek.com/quick_start/pricing
197219
"deepseek": {
@@ -331,9 +353,9 @@ def get_recommended_model(provider: Provider) -> str: # pragma: no cover
331353
elif provider == "openrouter":
332354
return "meta-llama/llama-3.1-405b-instruct"
333355
elif provider == "gemini":
334-
return "gemini-1.5-flash-latest"
356+
return "gemini-2.5-pro"
335357
elif provider == "anthropic":
336-
return "claude-3-7-sonnet-20250219"
358+
return "claude-sonnet-4-20250514"
337359
else:
338360
raise ValueError(f"Provider {provider} did not have a recommended model")
339361

@@ -344,7 +366,7 @@ def get_summary_model(provider: Provider) -> str: # pragma: no cover
344366
elif provider == "openrouter":
345367
return "meta-llama/llama-3.1-8b-instruct"
346368
elif provider == "gemini":
347-
return "gemini-1.5-flash-latest"
369+
return "gemini-2.5-flash"
348370
elif provider == "anthropic":
349371
return "claude-3-haiku-20240307"
350372
elif provider == "deepseek":

0 commit comments

Comments
 (0)