scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 48

fix(test): make semantic-similarity propagation test a true regression guard

Browse source
This commit is contained in:
devin-ai-integration[bot] 2026-07-01 04:23:34 +00:00 • committed by GitHub
parent 1934d6e4cd
commit e6e7fb2666
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 10 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
tests/test_litellm/caching/test_caching.py
View file

@ -183,12 +183,17 @@ async def test_async_get_cache_filters_kwargs_like_sync():
@pytest.mark.asyncio
async def test_async_get_cache_propagates_semantic_similarity_metadata():
"""async_get_cache must copy semantic-similarity from the backend's metadata
copy back into the caller's original metadata dict."""
"""async_get_cache must receive a metadata COPY (not the original) and then
propagate semantic-similarity back via _update_metadata_from_cache_lookup_kwargs.
Without the fix, the backend receives the original dict directly and the
propagation step is skipped entirely."""
cache = Cache(type=LiteLLMCacheType.LOCAL)
received_metadata_objects: list = []
async def fake_get(key, **kwargs):
kwargs.get("metadata", {})["semantic-similarity"] = 0.95
md = kwargs.get("metadata", {})
received_metadata_objects.append(md)
md["semantic-similarity"] = 0.95
return None
backend = MagicMock()
@ -203,4 +208,6 @@ async def test_async_get_cache_propagates_semantic_similarity_metadata():
metadata=original_metadata,
)
assert len(received_metadata_objects) == 1
assert received_metadata_objects[0] is not original_metadata
assert original_metadata.get("semantic-similarity") == 0.95