scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-13 08:52:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 174

refactor(anthropic): read the background drop setting through drop_policy

Browse source 
drop_policy already reads per-request drop_params, the global litellm.drop_params, and additional_drop_params, so the bridge uses it instead of its own copy. Adds a test for the global setting.
This commit is contained in:
mateo-berri 2026-11-06 14:22:21 -08:00
parent 2d243a3f4a
commit c3a12326df
2 changed files with 10 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
litellm/llms/anthropic/messages/responses_adapters/handler.py
View file

@ -25,7 +25,7 @@ from litellm.litellm_core_utils.prompt_templates.common_utils import (
prompt_cache_breakpoint_with_readable_mode,
with_prompt_cache_breakpoint,
)
from litellm.litellm_core_utils.tool_fields import drops_mcp_servers, tools_without_mcp_servers
from litellm.litellm_core_utils.tool_fields import drop_policy, drops_mcp_servers, tools_without_mcp_servers
from litellm.responses.utils import auto_reasoning_summary
from litellm.types.integrations.custom_logger import CONVERTED_STREAM_KEYS
from litellm.types.llms.anthropic import (
@ -172,13 +172,6 @@ def _forwarded_kwargs(extra_kwargs: Mapping[str, object] | None) -> Mapping[str,
return extra_kwargs or {}
def _drops_background(forwarded_kwargs: Mapping[str, object]) -> bool:
additional_drop_params: Final = forwarded_kwargs.get("additional_drop_params")
return drops_unsupported_params(forwarded_kwargs.get("drop_params")) or (
isinstance(additional_drop_params, list) and "background" in additional_drop_params
)
def _requests_background(forwarded_kwargs: Mapping[str, object]) -> bool:
extra_body: Final = as_str_mapping(forwarded_kwargs.get("extra_body"))
return forwarded_kwargs.get("background") is not None or (
@ -191,7 +184,7 @@ def _without_background(extra_body: Mapping[str, object]) -> Mapping[str, object
def _reject_background_mode(model: str, forwarded_kwargs: Mapping[str, object]) -> None:
if not _requests_background(forwarded_kwargs) or _drops_background(forwarded_kwargs):
if not _requests_background(forwarded_kwargs) or drop_policy(forwarded_kwargs).drops("background"):
return
provider: Final = forwarded_kwargs.get("custom_llm_provider")
raise litellm.UnsupportedParamsError(

12
tests/unit/llms/anthropic/messages/responses_adapters/test_responses_adapters_handler.py
View file

@ -706,13 +706,17 @@ def test_background_answers_400_without_drop_params(background_request: Mapping[
@pytest.mark.parametrize(
"drop_setting",
("drop_setting", "global_drop_params"),
[
pytest.param({"drop_params": True}, id="drop_params"),
pytest.param({"additional_drop_params": ["background"]}, id="additional_drop_params"),
pytest.param({"drop_params": True}, False, id="drop_params"),
pytest.param({"additional_drop_params": ["background"]}, False, id="additional_drop_params"),
pytest.param({}, True, id="litellm.drop_params"),
],
)
def test_dropped_background_never_reaches_the_responses_request(drop_setting: Mapping[str, object]) -> None:
def test_dropped_background_never_reaches_the_responses_request(
drop_setting: Mapping[str, object], global_drop_params: bool, monkeypatch: pytest.MonkeyPatch
) -> None:
monkeypatch.setattr(litellm, "drop_params", global_drop_params)
responses_kwargs: Final = _build_responses_kwargs(
model="openai/gpt-5.6",
extra_kwargs={