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(databricks/chat/transformation.py): remove reasoning_effort from request (#9811)

Browse source 
Fixes https://github.com/BerriAI/litellm/issues/9700#issuecomment-2784431995
This commit is contained in:
Krish Dholakia 2025-04-07 19:43:19 -07:00 • committed by GitHub
parent 6bec3f2549
commit 5d8b6ab296
2 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/llms/databricks/chat/transformation.py
View file

@ -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

2
tests/llm_translation/base_llm_unit_tests.py
View file

@ -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