scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 86

Merge fba7993979 into b618c7ad86

Browse source
This commit is contained in:
ryan-crabbe-berri 2026-09-07 13:46:04 -07:00 • committed by GitHub
commit 884abf5cc4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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

@ -69,7 +69,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."""