tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-24 20:29:31 +00:00
Code Issues 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:47:05 -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
litellm/tests/test_custom_callback_input.py
+4 -1
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(