ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test(test_router_fallbacks.py): fix test to check previous models in pre_api_call not on success

Browse Source
This commit is contained in:
Krrish Dholakia 2023-12-27 16:34:29 +05:30
parent f6f8d1a9df
commit 10a74d02c1
1 changed files with 0 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
litellm/tests/test_router_fallbacks.py
View File

@ -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": <complete_traceback>}]}
print(f"self.previous_models: {self.previous_models}")
print(f"On Success")
def log_failure_event(self, kwargs, response_obj, start_time, end_time):