mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-08 22:21:35 +00:00
test_gemini_context_caching_with_ttl
This commit is contained in:
parent
41953185bf
commit
c467986b8b
1 changed files with 6 additions and 0 deletions
|
|
@ -97,6 +97,9 @@ def test_gemini_context_caching_with_ttl():
|
|||
model="gemini-1.5-pro",
|
||||
messages=messages_with_ttl,
|
||||
cache_key="test-ttl-cache-key",
|
||||
custom_llm_provider="gemini",
|
||||
vertex_project=None,
|
||||
vertex_location=None,
|
||||
)
|
||||
|
||||
# Verify TTL is properly included in the result
|
||||
|
|
@ -123,6 +126,9 @@ def test_gemini_context_caching_with_ttl():
|
|||
model="gemini-1.5-pro",
|
||||
messages=messages_invalid_ttl,
|
||||
cache_key="test-invalid-ttl",
|
||||
custom_llm_provider="gemini",
|
||||
vertex_project=None,
|
||||
vertex_location=None,
|
||||
)
|
||||
|
||||
# Verify invalid TTL is not included
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue