scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-25 01:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 81

test(vertex_ai): document isolated cache test patches

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-18 20:43:59 +00:00
parent 35a4536b14
commit 2aa7f4b5e4
1 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
tests/test_litellm/llms/vertex_ai/context_caching/test_vertex_ai_context_caching.py
View file

@ -26,7 +26,7 @@ class TestContextCachingEndpoints:
# Mock is_prompt_caching_valid_prompt to return True by default.
# This avoids token counting in unit tests. The min-token guard is
# tested explicitly in test_check_and_create_cache_skips_when_below_min_tokens.
self._token_check_patcher = patch(
self._token_check_patcher = patch( # test-quality-ok: isolate token validation for cache creation tests
"litellm.llms.vertex_ai.context_caching.vertex_ai_context_caching.is_prompt_caching_valid_prompt",
return_value=True,
)
@ -2153,11 +2153,11 @@ class TestContextCachingMultiRegionUrls:
}
with (
patch(
patch( # test-quality-ok: isolate message splitting for cache creation tests
"litellm.llms.vertex_ai.context_caching.vertex_ai_context_caching.separate_cached_messages",
return_value=(cached_messages, non_cached_messages),
),
patch(
patch( # test-quality-ok: isolate cache key generation for cache creation tests
"litellm.llms.vertex_ai.context_caching.vertex_ai_context_caching.local_cache_obj.get_cache_key",
return_value="test_cache_key",
),
@ -2167,7 +2167,7 @@ class TestContextCachingMultiRegionUrls:
"_get_token_and_url_context_caching",
return_value=("token", "https://test-url.com"),
),
patch(
patch( # test-quality-ok: isolate request transformation for cache creation tests
"litellm.llms.vertex_ai.context_caching.vertex_ai_context_caching.transform_openai_messages_to_gemini_context_caching",
return_value={"model": "gemini-1.5-pro", "contents": []},
),
@ -2211,11 +2211,11 @@ class TestContextCachingMultiRegionUrls:
}
with (
patch(
patch( # test-quality-ok: isolate message splitting for cache creation tests
"litellm.llms.vertex_ai.context_caching.vertex_ai_context_caching.separate_cached_messages",
return_value=(cached_messages, non_cached_messages),
),
patch(
patch( # test-quality-ok: isolate cache key generation for cache creation tests
"litellm.llms.vertex_ai.context_caching.vertex_ai_context_caching.local_cache_obj.get_cache_key",
return_value="test_cache_key",
),
@ -2225,7 +2225,7 @@ class TestContextCachingMultiRegionUrls:
"_get_token_and_url_context_caching",
return_value=("token", "https://test-url.com"),
),
patch(
patch( # test-quality-ok: isolate request transformation for cache creation tests
"litellm.llms.vertex_ai.context_caching.vertex_ai_context_caching.transform_openai_messages_to_gemini_context_caching",
return_value={"model": "gemini-1.5-pro", "contents": []},
),
@ -2261,11 +2261,11 @@ class TestContextCachingMultiRegionUrls:
non_cached_messages = [self.sample_messages[1]]
with (
patch(
patch( # test-quality-ok: isolate message splitting for cache reuse tests
"litellm.llms.vertex_ai.context_caching.vertex_ai_context_caching.separate_cached_messages",
return_value=(cached_messages, non_cached_messages),
),
patch(
patch( # test-quality-ok: isolate cache key generation for cache reuse tests
"litellm.llms.vertex_ai.context_caching.vertex_ai_context_caching.local_cache_obj.get_cache_key",
return_value="test_cache_key",
),