Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refac: safer tool server handling

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-03-11 15:22:51 -05:00
parent f2217da94e
commit 8da29566a1
2 changed files with 17 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/routers/tools.py
+9 -3
View File
@@ -88,9 +88,15 @@ async def get_tools(
# OpenAPI Tool Servers # OpenAPI Tool Servers
server_access_grants = {} server_access_grants = {}
for server in await get_tool_servers(request): for server in await get_tool_servers(request):
connection = request.app.state.config.TOOL_SERVER_CONNECTIONS[ server_idx = server.get("idx", 0)
server.get("idx", 0) connections = request.app.state.config.TOOL_SERVER_CONNECTIONS
] if server_idx >= len(connections):
log.warning(
f"Tool server index {server_idx} out of range "
f"(have {len(connections)} connections), skipping server {server.get('id')}"
)
continue
connection = connections[server_idx]
server_config = connection.get("config", {}) server_config = connection.get("config", {})
server_id = f"server:{server.get('id')}" server_id = f"server:{server.get('id')}"
backend/open_webui/utils/tools.py
+8 -5
View File
@@ -275,11 +275,14 @@ async def get_tools(
continue continue
tool_server_idx = tool_server_data.get("idx", 0) tool_server_idx = tool_server_data.get("idx", 0)
tool_server_connection = ( connections = request.app.state.config.TOOL_SERVER_CONNECTIONS
request.app.state.config.TOOL_SERVER_CONNECTIONS[ if tool_server_idx >= len(connections):
tool_server_idx log.warning(
] f"Tool server index {tool_server_idx} out of range "
) f"(have {len(connections)} connections), skipping server {server_id}"
)
continue
tool_server_connection = connections[tool_server_idx]
# Check access control for tool server # Check access control for tool server
if not has_connection_access( if not has_connection_access(