scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

[Fix] Update gemini-2.0-flash to gemini-2.5-flash in test_gemini integration tests

Browse source 
gemini-2.0-flash is no longer available to new users, causing 404 errors
in llm_translation_testing tests.
This commit is contained in:
Yuneng Jiang 2026-03-30 13:53:38 -07:00
parent 40d4e79a00
commit b39748daf2
No known key found for this signature in database
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/llm_translation/test_gemini.py
View file

@ -532,7 +532,7 @@ def test_gemini_with_grounding():
## Check streaming
response = completion(
model="gemini/gemini-2.0-flash",
model="gemini/gemini-2.5-flash",
messages=[{"role": "user", "content": "What is the capital of France?"}],
tools=tools,
stream=True,
@ -566,7 +566,7 @@ def test_gemini_with_empty_function_call_arguments():
}
]
response = completion(
model="gemini/gemini-2.0-flash",
model="gemini/gemini-2.5-flash",
messages=[{"role": "user", "content": "What is the capital of France?"}],
tools=tools,
)
@ -775,7 +775,7 @@ def test_gemini_tool_use():
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": "What's the weather like in Lima, Peru today?"},
],
"model": "gemini/gemini-2.0-flash",
"model": "gemini/gemini-2.5-flash",
"tools": [
{
"type": "function",