fix importing Cache from litellm (#6219)

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

View file

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

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_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")