From ec32a0a7d7e03b8a0a2c0670d951317e17eba79b Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 13 Sep 2025 09:59:01 -0700 Subject: [PATCH] fix: get_base_completion_call_args --- tests/llm_translation/test_gemini.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/llm_translation/test_gemini.py b/tests/llm_translation/test_gemini.py index 47e3aaa814..8fcc85aba8 100644 --- a/tests/llm_translation/test_gemini.py +++ b/tests/llm_translation/test_gemini.py @@ -21,7 +21,7 @@ import json class TestGoogleAIStudioGemini(BaseLLMChatTest): def get_base_completion_call_args(self) -> dict: - return {"model": "gemini/gemini-2.0-flash"} + return {"model": "gemini/gemini-2.5-flash"} def get_base_completion_call_args_with_reasoning_model(self) -> dict: return {"model": "gemini/gemini-2.5-flash"}