From 10a74d02c15bc57515e9bd10ee97cf3aa83ce45e Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 27 Dec 2023 16:34:29 +0530 Subject: [PATCH] test(test_router_fallbacks.py): fix test to check previous models in pre_api_call not on success --- litellm/tests/test_router_fallbacks.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/litellm/tests/test_router_fallbacks.py b/litellm/tests/test_router_fallbacks.py index e601343e5..36baae346 100644 --- a/litellm/tests/test_router_fallbacks.py +++ b/litellm/tests/test_router_fallbacks.py @@ -44,13 +44,6 @@ class MyCustomHandler(CustomLogger): print(f"On Success") async def async_log_success_event(self, kwargs, response_obj, start_time, end_time): - print( - f"previous_models: {kwargs['litellm_params']['metadata']['previous_models']}" - ) - self.previous_models += len( - kwargs["litellm_params"]["metadata"]["previous_models"] - ) # {"previous_models": [{"model": litellm_model_name, "exception_type": AuthenticationError, "exception_string": }]} - print(f"self.previous_models: {self.previous_models}") print(f"On Success") def log_failure_event(self, kwargs, response_obj, start_time, end_time):