From fba79939799de738ae25b7429ad9fb7bbe8b8e21 Mon Sep 17 00:00:00 2001 From: ryan-crabbe-berri Date: Tue, 2 Jun 2026 17:15:59 -0700 Subject: [PATCH] Use CI_CD_DEFAULT_GEMINI_MODEL env var in live gemini CI tests Replace hardcoded gemini/* model names at the live-call sites in test_gemini_interactions.py and test_batch_completions.py with the CI_CD_DEFAULT_GEMINI_MODEL env var (keeping the gemini/ prefix), so a model deprecation can be handled from CircleCI project settings instead of editing code. --- tests/local_testing/test_batch_completions.py | 5 ++++- tests/unified_google_tests/test_gemini_interactions.py | 4 +++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/tests/local_testing/test_batch_completions.py b/tests/local_testing/test_batch_completions.py index 95bfe5e6e2b..022dd70b870 100644 --- a/tests/local_testing/test_batch_completions.py +++ b/tests/local_testing/test_batch_completions.py @@ -72,7 +72,10 @@ def test_batch_completions_models(): def test_batch_completion_models_all_responses(): try: responses = batch_completion_models_all_responses( - models=["gemini/gemini-2.5-flash-lite", "claude-haiku-4-5-20251001"], + models=[ + f"gemini/{os.environ.get('CI_CD_DEFAULT_GEMINI_MODEL', 'gemini-2.5-flash')}", + "claude-haiku-4-5-20251001", + ], messages=[{"role": "user", "content": "write a poem"}], max_tokens=10, ) diff --git a/tests/unified_google_tests/test_gemini_interactions.py b/tests/unified_google_tests/test_gemini_interactions.py index 0e0719843f1..62e50ce49ad 100644 --- a/tests/unified_google_tests/test_gemini_interactions.py +++ b/tests/unified_google_tests/test_gemini_interactions.py @@ -16,7 +16,9 @@ class TestGeminiInteractions(BaseInteractionsTest): def get_model(self) -> str: """Return the Gemini model string.""" - return "gemini/gemini-2.5-flash" + return ( + f"gemini/{os.environ.get('CI_CD_DEFAULT_GEMINI_MODEL', 'gemini-2.5-flash')}" + ) def get_api_key(self) -> str: """Return the Gemini API key from environment."""