perf: batch DB fetch for tools in get_tools

Author: Algorithm5838

backend/open_webui/models/tools.py

@@ -151,6 +151,19 @@ async def get_tool_by_id(self, id: str, db: AsyncSession | None = None) -> ToolM
         except Exception:
             return None
 
+    async def get_tools_by_ids(self, ids: list[str], db: AsyncSession | None = None) -> dict[str, ToolModel]:
+        if not ids:
+            return {}
+        async with get_async_db_context(db) as db:
+            result = await db.execute(select(Tool).where(Tool.id.in_(ids)))
+            tools = result.scalars().all()
+            tool_ids = [t.id for t in tools]
+            grants_map = await AccessGrants.get_grants_by_resources('tool', tool_ids, db=db)
+            return {
+                t.id: await self._to_tool_model(t, access_grants=grants_map.get(t.id, []), db=db)
+                for t in tools
+            }
+
     async def get_tools(self, defer_content: bool = False, db: AsyncSession | None = None) -> list[ToolUserModel]:
         async with get_async_db_context(db) as db:
             stmt = select(Tool).order_by(Tool.updated_at.desc())

backend/open_webui/utils/tools.py

@@ -166,8 +166,12 @@ async def get_tools(request: Request, tool_ids: list[str], user: UserModel, extr
     # Get user's group memberships for access control checks
     user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id)}
 
+    # Single query for all local tools instead of one per tool_id
+    local_tool_ids = [tid for tid in tool_ids if not tid.startswith('server:')]
+    tools_by_id = await Tools.get_tools_by_ids(local_tool_ids)
+
     for tool_id in tool_ids:
-        tool = await Tools.get_tool_by_id(tool_id)
+        tool = tools_by_id.get(tool_id)
         if tool:
             # Check access control for local tools
             if (