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

style: apply black formatting

Browse source
This commit is contained in:
Lance Hsu 2026-04-02 21:45:02 +08:00
parent 5b2b4abb6f
commit a87329d2a5
No known key found for this signature in database
GPG key ID: CFF816BB8560B8B8
2 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
litellm/proxy/_experimental/mcp_server/semantic_tool_filter.py
View file

@ -3,6 +3,7 @@ Semantic MCP Tool Filtering using semantic-router
Filters MCP tools semantically for /chat/completions and /responses endpoints.
"""
from typing import TYPE_CHECKING, Any, Dict, List, Optional
from litellm._logging import verbose_logger

9
litellm/proxy/hooks/mcp_semantic_filter/hook.py
View file

@ -4,6 +4,7 @@ Semantic Tool Filter Hook
Pre-call hook that filters MCP tools semantically before LLM inference.
Reduces context window size and improves tool selection accuracy.
"""
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
from litellm._logging import verbose_proxy_logger
@ -226,10 +227,14 @@ class SemanticToolFilterHook(CustomLogger):
# Separate MCP tools (prefixed) from non-MCP tools — only filter
# MCP tools, always pass non-MCP tools through untouched.
def _tool_name(t):
return t.get("name", "") if isinstance(t, dict) else getattr(t, "name", "")
return (
t.get("name", "") if isinstance(t, dict) else getattr(t, "name", "")
)
mcp_tools = [t for t in tools if is_tool_name_prefixed(_tool_name(t))]
non_mcp_tools = [t for t in tools if not is_tool_name_prefixed(_tool_name(t))]
non_mcp_tools = [
t for t in tools if not is_tool_name_prefixed(_tool_name(t))
]
if not mcp_tools:
return None