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

(test) custom logger + cache

Browse source
This commit is contained in:
ishaan-jaff 2023-12-11 22:46:49 -08:00
parent ab90a66a8e
commit e49af2272d
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_custom_logger.py
View file

@ -317,7 +317,7 @@ def test_redis_cache_completion_stream():
response_2_content += chunk.choices[0].delta.content or "" response_2_content += chunk.choices[0].delta.content or ""
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 response1.id == response2.id, f"Response 1 != Response 2. Same params, Response 1{response1.id} != Response 2{response2.id}"
litellm.success_callback = [] litellm.success_callback = []
litellm._async_success_callback = [] litellm._async_success_callback = []
litellm.cache = None litellm.cache = None
@ -325,4 +325,3 @@ def test_redis_cache_completion_stream():
print(e) print(e)
litellm.success_callback = [] litellm.success_callback = []
raise e raise e
pytest.fail(f"Error occurred: {e}")