phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(test) caching reset success_callback

Browse source
This commit is contained in:
ishaan-jaff 2023-12-08 19:10:35 -08:00
parent 25edcf6868
commit 25f4436c8d
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_caching.py
View file

@ -194,6 +194,7 @@ def test_redis_cache_completion():
def test_redis_cache_completion_stream(): def test_redis_cache_completion_stream():
try: try:
litellm.success_callback = []
litellm.set_verbose = True litellm.set_verbose = True
random_number = random.randint(1, 100000) # add a random number to ensure it's always adding / reading from cache 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}"}] 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 1", response_1_content)
print("\nresponse 2", response_2_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}" 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 litellm.cache = None
except Exception as e: except Exception as e:
print(e) print(e)
litellm.success_callback = []
raise e raise e
""" """