From 4e90a2541714f82edf3ba5792a6dddeffa4a6041 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 7 Aug 2024 17:52:40 -0700 Subject: [PATCH] fix all optional param tests --- litellm/tests/test_optional_params.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/litellm/tests/test_optional_params.py b/litellm/tests/test_optional_params.py index 2cd5c11492..d961190c29 100644 --- a/litellm/tests/test_optional_params.py +++ b/litellm/tests/test_optional_params.py @@ -301,7 +301,7 @@ def test_dynamic_drop_params(drop_params): optional_params = litellm.utils.get_optional_params( model="command-r", custom_llm_provider="cohere", - response_format="json", + response_format={"type": "json"}, drop_params=drop_params, ) else: @@ -309,7 +309,7 @@ def test_dynamic_drop_params(drop_params): optional_params = litellm.utils.get_optional_params( model="command-r", custom_llm_provider="cohere", - response_format="json", + response_format={"type": "json"}, drop_params=drop_params, ) pytest.fail("Expected to fail") @@ -389,7 +389,7 @@ def test_dynamic_drop_additional_params(drop_params): optional_params = litellm.utils.get_optional_params( model="command-r", custom_llm_provider="cohere", - response_format="json", + response_format={"type": "json"}, additional_drop_params=["response_format"], ) else: @@ -397,7 +397,7 @@ def test_dynamic_drop_additional_params(drop_params): optional_params = litellm.utils.get_optional_params( model="command-r", custom_llm_provider="cohere", - response_format="json", + response_format={"type": "json"}, ) pytest.fail("Expected to fail") except Exception as e: