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

(test) redis caching init calbacks to be empty

Browse source
This commit is contained in:
ishaan-jaff 2023-12-09 11:32:14 -08:00
parent c1c81af9d4
commit 8cbd6010c8
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_caching.py
View file

@ -195,6 +195,8 @@ def test_redis_cache_completion():
def test_redis_cache_completion_stream(): def test_redis_cache_completion_stream():
try: try:
litellm.success_callback = [] litellm.success_callback = []
litellm._async_success_callback = []
litellm.callbacks = []
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}"}]