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

test fix

Browse source
This commit is contained in:
Ishaan Jaffer 2025-10-11 10:05:22 -07:00
parent b2747211c8
commit 33a4676184
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tests/llm_translation/test_gemini.py
View file

@ -151,7 +151,12 @@ def test_gemini_context_caching_with_ttl():
]
result_no_ttl = transform_openai_messages_to_gemini_context_caching(
model="gemini-1.5-pro", messages=messages_no_ttl, cache_key="test-no-ttl"
model="gemini-1.5-pro",
messages=messages_no_ttl,
cache_key="test-no-ttl",
custom_llm_provider="gemini",
vertex_project=None,
vertex_location=None,
)
# Verify no TTL field is present when not specified
@ -201,7 +206,12 @@ def test_gemini_context_caching_with_ttl():
# Test transformation with mixed messages
result_mixed = transform_openai_messages_to_gemini_context_caching(
model="gemini-1.5-pro", messages=messages_mixed, cache_key="test-mixed-ttl"
model="gemini-1.5-pro",
messages=messages_mixed,
cache_key="test-mixed-ttl",
custom_llm_provider="gemini",
vertex_project=None,
vertex_location=None,
)
# Should pick up the first valid TTL