From 25f4436c8d85ce4f35b166a79f87616b858c2a50 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 8 Dec 2023 19:10:35 -0800 Subject: [PATCH] (test) caching reset success_callback --- litellm/tests/test_caching.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/litellm/tests/test_caching.py b/litellm/tests/test_caching.py index 1e1a7271b4..7653a07487 100644 --- a/litellm/tests/test_caching.py +++ b/litellm/tests/test_caching.py @@ -194,6 +194,7 @@ def test_redis_cache_completion(): def test_redis_cache_completion_stream(): try: + litellm.success_callback = [] litellm.set_verbose = True random_number = random.randint(1, 100000) # add a random number to ensure it's always adding / reading from cache messages = [{"role": "user", "content": f"write a one sentence poem about: {random_number}"}] @@ -214,9 +215,11 @@ def test_redis_cache_completion_stream(): print("\nresponse 1", response_1_content) print("\nresponse 2", response_2_content) assert response_1_content == response_2_content, f"Response 1 != Response 2. Same params, Response 1{response_1_content} != Response 2{response_2_content}" + litellm.success_callback = [] litellm.cache = None except Exception as e: print(e) + litellm.success_callback = [] raise e """