phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

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 11e3ee4411
commit 42a2a2fe94
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
assert isinstance(
response_obj,
Union[
(
litellm.ModelResponse,
litellm.EmbeddingResponse,
litellm.ImageResponse,
],
),
)
## KWARGS
assert isinstance(kwargs["model"], str)