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

style: apply black formatting to hosted_vllm messages transformation

Browse source 
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
Denis Khachyan 2026-05-28 11:54:58 +03:00
parent 57085d958b
commit 1dab2fa1be
1 changed files with 16 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
litellm/llms/hosted_vllm/messages/transformation.py
View file

@ -103,23 +103,30 @@ class HostedVLLMAnthropicMessagesConfig(BaseAnthropicMessagesConfig):
return f"{base}/v1/messages"
# Anthropic built-in tool types that vLLM does not support in /v1/messages
_UNSUPPORTED_TOOL_TYPES = frozenset({
"web_search",
"computer",
"text_editor",
"bash",
})
_UNSUPPORTED_TOOL_TYPES = frozenset(
{
"web_search",
"computer",
"text_editor",
"bash",
}
)
def _strip_unsupported_tools(self, tools: Optional[List[Dict]]) -> Optional[List[Dict]]:
def _strip_unsupported_tools(
self, tools: Optional[List[Dict]]
) -> Optional[List[Dict]]:
"""Remove Anthropic built-in tools (web_search, computer_use, etc.) that
vLLM's /v1/messages endpoint does not understand."""
if not tools:
return tools
return [
t for t in tools
t
for t in tools
if not isinstance(t, dict)
or t.get("type", "") not in self._UNSUPPORTED_TOOL_TYPES
and not t.get("type", "").startswith(tuple(f"{p}_" for p in self._UNSUPPORTED_TOOL_TYPES))
and not t.get("type", "").startswith(
tuple(f"{p}_" for p in self._UNSUPPORTED_TOOL_TYPES)
)
]
def transform_anthropic_messages_request(