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

fix(test_custom_callback_input.py): fix unit test for embedding logging

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-15 21:46:51 -07:00
parent 2cfafdc7c3
commit a4b0740c4a
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_custom_callback_input.py
View file

@ -198,7 +198,10 @@ class CompletionCustomHandler(
assert isinstance(kwargs["user"], (str, type(None))) assert isinstance(kwargs["user"], (str, type(None)))
assert ( assert (
isinstance(kwargs["input"], list) isinstance(kwargs["input"], list)
and isinstance(kwargs["input"][0], dict) and (
isinstance(kwargs["input"][0], dict)
or isinstance(kwargs["input"][0], str)
)
) or isinstance(kwargs["input"], (dict, str)) ) or isinstance(kwargs["input"], (dict, str))
assert isinstance(kwargs["api_key"], (str, type(None))) assert isinstance(kwargs["api_key"], (str, type(None)))
assert isinstance( assert isinstance(