ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki 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
Show Stats 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["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))
assert isinstance(kwargs["api_key"], (str, type(None)))
assert isinstance(