tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-10 05:15:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(test_custom_callback_router.py): fix test

Browse Source
This commit is contained in:
Krrish Dholakia
2023-12-13 19:06:02 -08:00
parent 73ecc012a9
commit cffd190887
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_custom_callback_router.py
+1 -1
View File
@@ -214,7 +214,7 @@ class CompletionCustomHandler(CustomLogger): # https://docs.litellm.ai/docs/obse
assert isinstance(kwargs['original_response'], (str, litellm.CustomStreamWrapper)) or inspect.isasyncgen(kwargs['original_response']) or inspect.iscoroutine(kwargs['original_response'])
assert isinstance(kwargs['additional_args'], (dict, type(None)))
assert isinstance(kwargs['log_event_type'], str)
assert isinstance(kwargs["cache_hit"], Optional[bool])
assert kwargs["cache_hit"] is None or isinstance(kwargs["cache_hit"], bool)
### ROUTER-SPECIFIC KWARGS
assert isinstance(kwargs["litellm_params"]["metadata"], dict)
assert isinstance(kwargs["litellm_params"]["metadata"]["model_group"], str)