CalvinBackup/invariant-gateway
1
0
Fork 0
mirror of https://github.com/invariantlabs-ai/invariant-gateway.git synced 2026-02-12 22:42:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change gemini-2.0-flash to gemini-2.5-flash

Browse Source
This commit is contained in:
Hemang
2025-11-06 11:06:31 +01:00
parent 6e7106ea26
commit a5fe02bdf2
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -160,7 +160,7 @@ import random
import os
base_url = "/api/v1/gateway/litellm/{add-your-dataset-name-here}"
EXAMPLE_MODELS = ["openai/gpt-4o", "gemini/gemini-2.0-flash", "anthropic/claude-3-5-haiku-20241022"]
EXAMPLE_MODELS = ["openai/gpt-4o", "gemini/gemini-2.5-flash", "anthropic/claude-3-5-haiku-20241022"]
model = random.choice(SAMPLE_MODELS)
base_url += "/" + model.split("/")[0] # append /gemini /openai or /anthropic.

2
tests/integration/litellm/test_chat_without_tool_call.py
View File

@@ -10,7 +10,7 @@ from litellm import completion
MODEL_API_KEYS = {
"openai/gpt-4o": "OPENAI_API_KEY",
"gemini/gemini-2.0-flash": "GEMINI_API_KEY",
"gemini/gemini-2.5-flash": "GEMINI_API_KEY",
"anthropic/claude-3-5-haiku-20241022": "ANTHROPIC_API_KEY",
}