From 8d338aee78ee3c463fc0ab538ceac402a6314f53 Mon Sep 17 00:00:00 2001 From: Krish Dholakia Date: Mon, 7 Apr 2025 19:43:19 -0700 Subject: [PATCH] fix(databricks/chat/transformation.py): remove reasoning_effort from request (#9811) Fixes https://github.com/BerriAI/litellm/issues/9700#issuecomment-2784431995 --- litellm/llms/databricks/chat/transformation.py | 1 + tests/llm_translation/base_llm_unit_tests.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/litellm/llms/databricks/chat/transformation.py b/litellm/llms/databricks/chat/transformation.py index 09c87a9168..bdccf6b938 100644 --- a/litellm/llms/databricks/chat/transformation.py +++ b/litellm/llms/databricks/chat/transformation.py @@ -240,6 +240,7 @@ class DatabricksConfig(DatabricksBase, OpenAILikeChatConfig, AnthropicConfig): optional_params["thinking"] = AnthropicConfig._map_reasoning_effort( non_default_params.get("reasoning_effort") ) + optional_params.pop("reasoning_effort", None) ## handle thinking tokens self.update_optional_params_with_thinking_tokens( non_default_params=non_default_params, optional_params=mapped_params diff --git a/tests/llm_translation/base_llm_unit_tests.py b/tests/llm_translation/base_llm_unit_tests.py index dee85fde5c..a8402aaeee 100644 --- a/tests/llm_translation/base_llm_unit_tests.py +++ b/tests/llm_translation/base_llm_unit_tests.py @@ -1272,3 +1272,5 @@ class BaseAnthropicChatTest(ABC): reasoning_effort="high", ) assert optional_params["thinking"] == {"type": "enabled", "budget_tokens": 4096} + + assert "reasoning_effort" not in optional_params \ No newline at end of file