mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-07 08:26:10 +00:00
test(bedrock_mantle): capture the drop warning with caplog instead of patching the logger
This commit is contained in:
parent
a1e58aabe7
commit
8713ba8a52
1 changed files with 14 additions and 5 deletions
|
|
@ -400,17 +400,26 @@ class TestBedrockMantleResponsesWebSearch:
|
|||
)
|
||||
assert body["tools"] == [self._WEB_SEARCH_TOOL, function_tool]
|
||||
|
||||
def test_web_search_is_not_logged_as_dropped(self):
|
||||
from unittest.mock import patch
|
||||
|
||||
def test_web_search_is_not_logged_as_dropped(self, caplog):
|
||||
cfg = BedrockMantleResponsesAPIConfig()
|
||||
with patch("litellm.llms.bedrock_mantle.responses.transformation.verbose_logger.warning") as mock_warning:
|
||||
|
||||
def drop_warnings() -> list[str]:
|
||||
return [r.getMessage() for r in caplog.records if "dropping unsupported tool type" in r.getMessage()]
|
||||
|
||||
with caplog.at_level(logging.WARNING, logger="LiteLLM"):
|
||||
cfg.map_openai_params(
|
||||
response_api_optional_params={"tools": [{"type": "file_search"}]},
|
||||
model="openai.gpt-5.6-sol",
|
||||
drop_params=False,
|
||||
)
|
||||
assert len(drop_warnings()) == 1
|
||||
caplog.clear()
|
||||
cfg.map_openai_params(
|
||||
response_api_optional_params={"tools": [self._WEB_SEARCH_TOOL]},
|
||||
model="openai.gpt-5.6-sol",
|
||||
drop_params=False,
|
||||
)
|
||||
assert mock_warning.call_count == 0
|
||||
assert drop_warnings() == []
|
||||
|
||||
def test_hoisted_web_search_tool_survives(self):
|
||||
cfg = BedrockMantleResponsesAPIConfig()
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue