diff --git a/litellm/tests/test_custom_callback_input.py b/litellm/tests/test_custom_callback_input.py index eae0412d39..1dc4321217 100644 --- a/litellm/tests/test_custom_callback_input.py +++ b/litellm/tests/test_custom_callback_input.py @@ -232,6 +232,7 @@ class CompletionCustomHandler( assert isinstance(kwargs["messages"], list) and isinstance( kwargs["messages"][0], dict ) + assert isinstance(kwargs["optional_params"], dict) assert isinstance(kwargs["litellm_params"], dict) assert isinstance(kwargs["start_time"], (datetime, type(None)))