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-04-06 18:48:33 -07:00
parent 460546956d
commit 6234982fd9
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

@ -167,7 +167,7 @@ class CompletionCustomHandler(
assert isinstance(kwargs["optional_params"], dict) assert isinstance(kwargs["optional_params"], dict)
assert isinstance(kwargs["litellm_params"], dict) assert isinstance(kwargs["litellm_params"], dict)
assert isinstance(kwargs["litellm_params"]["api_base"], str) assert isinstance(kwargs["litellm_params"]["api_base"], str)
assert isinstance(kwargs["cache_hit"], Optional[bool]) assert kwargs["cache_hit"] is None or isinstance(kwargs["cache_hit"], bool)
assert isinstance(kwargs["start_time"], (datetime, type(None))) assert isinstance(kwargs["start_time"], (datetime, type(None)))
assert isinstance(kwargs["stream"], bool) assert isinstance(kwargs["stream"], bool)
assert isinstance(kwargs["user"], (str, type(None))) assert isinstance(kwargs["user"], (str, type(None)))
@ -269,7 +269,7 @@ class CompletionCustomHandler(
assert isinstance(kwargs["litellm_params"]["api_base"], str) assert isinstance(kwargs["litellm_params"]["api_base"], str)
assert isinstance(kwargs["start_time"], (datetime, type(None))) assert isinstance(kwargs["start_time"], (datetime, type(None)))
assert isinstance(kwargs["stream"], bool) assert isinstance(kwargs["stream"], bool)
assert isinstance(kwargs["cache_hit"], Optional[bool]) assert kwargs["cache_hit"] is None or isinstance(kwargs["cache_hit"], bool)
assert isinstance(kwargs["user"], (str, type(None))) assert isinstance(kwargs["user"], (str, type(None)))
assert isinstance(kwargs["input"], (list, dict, str)) assert isinstance(kwargs["input"], (list, dict, str))
assert isinstance(kwargs["api_key"], (str, type(None))) assert isinstance(kwargs["api_key"], (str, type(None)))