From d2d9e63176dfb6bff91787bbce60850b47e35f41 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 21 Feb 2024 20:32:39 -0800 Subject: [PATCH] test(test_custom_callback_input.py): fix test --- litellm/tests/test_custom_callback_input.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_custom_callback_input.py b/litellm/tests/test_custom_callback_input.py index 25d531fda..9ea1a3bb1 100644 --- a/litellm/tests/test_custom_callback_input.py +++ b/litellm/tests/test_custom_callback_input.py @@ -835,7 +835,7 @@ async def test_async_completion_azure_caching_streaming(): ) assert len(customHandler_caching.errors) == 0 assert ( - len(customHandler_caching.states) > 4 + len(customHandler_caching.states) == 4 ) # pre, post, streaming .., success, success