diff --git a/litellm/cache.py b/litellm/gpt_cache.py similarity index 100% rename from litellm/cache.py rename to litellm/gpt_cache.py diff --git a/litellm/tests/data_map.txt b/litellm/tests/data_map.txt index a4cc414230..96c53e15ee 100644 Binary files a/litellm/tests/data_map.txt and b/litellm/tests/data_map.txt differ diff --git a/litellm/tests/test_caching.py b/litellm/tests/test_caching.py index 7a6ca08828..3e89398a96 100644 --- a/litellm/tests/test_caching.py +++ b/litellm/tests/test_caching.py @@ -59,12 +59,12 @@ def test_caching_with_models(): def test_gpt_cache(): # INIT GPT Cache # from gptcache import cache - from litellm.cache import completion + from litellm.gpt_cache import completion cache.init() cache.set_openai_key() - messages = [{"role": "user", "content": "what is litellm YC 22?"}] + messages = [{"role": "user", "content": "what is litellm YC paul graham, partner?"}] response2 = completion(model="gpt-3.5-turbo", messages=messages) response3 = completion(model="command-nightly", messages=messages) print(f"response2: {response2}")