scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 71

test(adapter): tool_choice drop when only web_search tools remain

Browse source
This commit is contained in:
Vigilans 2026-04-20 03:48:11 +08:00
parent cb8b128e2f
commit e58f730be9
1 changed files with 59 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

59
tests/test_litellm/llms/anthropic/experimental_pass_through/adapters/test_anthropic_experimental_pass_through_adapters_transformation.py
View file

@ -2641,3 +2641,62 @@ def test_translate_openai_response_to_anthropic_with_polyfill_both_compaction_an
cm = result.get("context_management")
assert cm is not None
assert cm["applied_edits"][0]["type"] == "compact_20260112"
class TestTranslateToolsDropsToolChoice:
"""Test that tool_choice is dropped when only web_search tools remain."""
def setup_method(self):
self.adapter = LiteLLMAnthropicMessagesAdapter()
def test_tool_choice_removed_when_only_web_search_tools(self):
"""When all tools are web_search-type, tool_choice should be popped."""
new_kwargs: dict = {"tool_choice": "auto", "model": "gpt-4o"}
anthropic_request = {
"model": "claude-sonnet-4-20250514",
"max_tokens": 1024,
"messages": [{"role": "user", "content": "search the web"}],
"tools": [
{"type": "web_search_20260209", "name": "web_search"},
],
"tool_choice": {"type": "auto"},
}
self.adapter._translate_tools_to_openai(
anthropic_message_request=anthropic_request,
new_kwargs=new_kwargs,
)
assert "tool_choice" not in new_kwargs
assert new_kwargs.get("web_search_options") == {}
def test_tool_choice_preserved_when_regular_tools_exist(self):
"""When regular tools exist alongside web_search, tool_choice stays."""
new_kwargs: dict = {"tool_choice": "auto", "model": "gpt-4o"}
anthropic_request = {
"model": "claude-sonnet-4-20250514",
"max_tokens": 1024,
"messages": [{"role": "user", "content": "search and compute"}],
"tools": [
{"type": "web_search_20260209", "name": "web_search"},
{
"type": "custom",
"name": "get_weather",
"description": "Get weather",
"input_schema": {
"type": "object",
"properties": {"location": {"type": "string"}},
},
},
],
"tool_choice": {"type": "auto"},
}
self.adapter._translate_tools_to_openai(
anthropic_message_request=anthropic_request,
new_kwargs=new_kwargs,
)
assert "tool_choice" in new_kwargs
assert new_kwargs.get("web_search_options") == {}
assert "tools" in new_kwargs