scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 83

fix: strip server prefix from OpenAPI tool names before filter check

Browse source 
For OpenAPI-backed MCP servers (spec_path), tools are stored with
already-prefixed names (e.g. "myserver-get_pet"). When filtering
against disallowed_tools, the unprefixed name must be used so that
entries like "get_pet" match correctly at listing time.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
gambletan 2026-03-16 21:49:58 +08:00
parent a90906f75a
commit cbe480ba87
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
litellm/proxy/_experimental/mcp_server/mcp_server_manager.py
View file

@ -1075,11 +1075,20 @@ class MCPServerManager:
tools = await self._fetch_tools_with_timeout(client, server.name)
# Filter tools based on allowed_tools / disallowed_tools config
# For OpenAPI servers (spec_path), tools already have the server
# prefix in their name (e.g. "myserver-get_pet"). Strip it before
# checking so that unprefixed disallowed_tools entries match.
if server.allowed_tools or server.disallowed_tools:
prefix = f"{server.name}-"
tools = [
tool
for tool in tools
if self.check_allowed_or_banned_tools(tool.name, server)
if self.check_allowed_or_banned_tools(
tool.name.removeprefix(prefix)
if server.spec_path
else tool.name,
server,
)
]
prefixed_or_original_tools = self._create_prefixed_tools(