phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix importing Cache from litellm (#6219)

Browse source
This commit is contained in:
Ishaan Jaff 2024-10-15 08:47:23 +05:30 • committed by GitHub
parent d0a3052937
commit cda0a993e2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/caching/__init__.py Normal file
View file

@ -0,0 +1 @@
from .caching import Cache

7
tests/local_testing/test_caching.py
View file

@ -2291,3 +2291,10 @@ async def test_basic_rerank_caching(sync_mode, top_n_1, top_n_2, expect_cache_hi
assert response.results is not None assert response.results is not None
assert_response_shape(response, custom_llm_provider="cohere") assert_response_shape(response, custom_llm_provider="cohere")
def test_basic_caching_import():
from litellm.caching import Cache
assert Cache is not None
print("Cache imported successfully")