From 9d10881f3d87c7e89dabf657ecc565ad0d42fb20 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 23 Jul 2024 07:55:42 -0700 Subject: [PATCH] test: re-run ci/cd --- litellm/tests/test_custom_callback_input.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/tests/test_custom_callback_input.py b/litellm/tests/test_custom_callback_input.py index eae0412d391..1dc43212178 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)))