diff --git a/litellm/tests/test_custom_logger.py b/litellm/tests/test_custom_logger.py index d24cf5757..1d373e0da 100644 --- a/litellm/tests/test_custom_logger.py +++ b/litellm/tests/test_custom_logger.py @@ -317,12 +317,11 @@ def test_redis_cache_completion_stream(): response_2_content += chunk.choices[0].delta.content or "" 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}" + assert response1.id == response2.id, f"Response 1 != Response 2. Same params, Response 1{response1.id} != Response 2{response2.id}" litellm.success_callback = [] litellm._async_success_callback = [] litellm.cache = None except Exception as e: print(e) litellm.success_callback = [] - raise e - pytest.fail(f"Error occurred: {e}") \ No newline at end of file + raise e \ No newline at end of file