phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(proxy/utils.py): add stronger typing for litellm params in failure call logging

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-22 21:31:21 -07:00
parent f581180e0d
commit 14c5d6ff03
3 changed files with 30 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/tests/test_custom_callback_input.py
View file

@ -234,6 +234,7 @@ class CompletionCustomHandler(
)
assert isinstance(kwargs["optional_params"], dict)
assert isinstance(kwargs["litellm_params"], dict)
assert isinstance(kwargs["litellm_params"]["metadata"], Optional[dict])
assert isinstance(kwargs["start_time"], (datetime, type(None)))
assert isinstance(kwargs["stream"], bool)
assert isinstance(kwargs["user"], (str, type(None)))