From b39748daf2dbbd8db75619934ac31f3172c67aa6 Mon Sep 17 00:00:00 2001 From: Yuneng Jiang Date: Mon, 30 Mar 2026 13:53:38 -0700 Subject: [PATCH] [Fix] Update gemini-2.0-flash to gemini-2.5-flash in test_gemini integration tests gemini-2.0-flash is no longer available to new users, causing 404 errors in llm_translation_testing tests. --- tests/llm_translation/test_gemini.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/llm_translation/test_gemini.py b/tests/llm_translation/test_gemini.py index b10a7d699c2..1ad71d25a05 100644 --- a/tests/llm_translation/test_gemini.py +++ b/tests/llm_translation/test_gemini.py @@ -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",