From c467986b8b298f07fc7aa50e3d5dca0bfbcee02b Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 11 Oct 2025 09:20:20 -0700 Subject: [PATCH] test_gemini_context_caching_with_ttl --- tests/llm_translation/test_gemini.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/llm_translation/test_gemini.py b/tests/llm_translation/test_gemini.py index 64e66f3914..9c07592fbe 100644 --- a/tests/llm_translation/test_gemini.py +++ b/tests/llm_translation/test_gemini.py @@ -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