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

fix(mcp): add target_format param to _process_mcp_tools_to_openai_format

Browse source
This commit is contained in:
PRABHU KIRAN VANDRANKI 2026-05-05 14:46:47 -04:00
parent 850fe595ac
commit 0557321cc2
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
litellm/responses/mcp/litellm_proxy_mcp_handler.py
View file

@ -388,6 +388,7 @@ class LiteLLM_Proxy_MCP_Handler:
user_api_key_auth: Any,
mcp_tools_with_litellm_proxy: List[ToolParam],
litellm_trace_id: Optional[str] = None,
target_format: Literal["responses", "chat"] = "responses",
) -> tuple[List[Any], dict[str, str]]:
"""
Centralized method to process MCP tools through the complete pipeline.
@ -396,6 +397,8 @@ class LiteLLM_Proxy_MCP_Handler:
user_api_key_auth: User authentication info for access control
mcp_tools_with_litellm_proxy: ToolParam objects with server_url starting with "litellm_proxy"
litellm_trace_id: Optional trace ID for linking list_mcp_tools spend logs to parent request
target_format: "chat" for ChatCompletions format (nested ``function`` key),
"responses" for Responses API flat format. Default: "responses".
Returns:
List of tools in OpenAI format ready to be sent to the LLM
@ -411,7 +414,7 @@ class LiteLLM_Proxy_MCP_Handler:
)
openai_tools = LiteLLM_Proxy_MCP_Handler._transform_mcp_tools_to_openai(
deduplicated_mcp_tools
deduplicated_mcp_tools, target_format=target_format
)
return openai_tools, tool_server_map