scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

Use CI_CD_DEFAULT_GEMINI_MODEL env var in live gemini CI tests

Browse source 
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.
This commit is contained in:
ryan-crabbe-berri 2026-06-02 17:15:59 -07:00
parent f047b1571e
commit fba7993979
2 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/local_testing/test_batch_completions.py
View file

@ -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,
)

4
tests/unified_google_tests/test_gemini_interactions.py
View file

@ -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."""