phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test(test_custom_callback_input.py): fix assert

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-22 20:39:14 -08:00
parent 579dfc3013
commit fb958e5244
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_custom_callback_input.py
View file

@ -152,11 +152,11 @@ class CompletionCustomHandler(
## RESPONSE OBJECT ## RESPONSE OBJECT
assert isinstance( assert isinstance(
response_obj, response_obj,
Union[ (
litellm.ModelResponse, litellm.ModelResponse,
litellm.EmbeddingResponse, litellm.EmbeddingResponse,
litellm.ImageResponse, litellm.ImageResponse,
], ),
) )
## KWARGS ## KWARGS
assert isinstance(kwargs["model"], str) assert isinstance(kwargs["model"], str)