xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-08-20 09:20:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: mcp-ui

Browse Source
This commit is contained in:
wangcham
2025-10-25 02:28:20 +00:00
parent 8345edd9f7
commit d86b884cab
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/provider/tools/loaders/mcp.py
+2 -1
View File
@@ -172,6 +172,7 @@ class MCPLoader(loader.ToolLoader):
server_entity: persistence_mcp.MCPServer | sqlalchemy.Row[persistence_mcp.MCPServer] | dict, server_entity: persistence_mcp.MCPServer | sqlalchemy.Row[persistence_mcp.MCPServer] | dict,
): ):
session = await self.init_runtime_mcp_session(server_entity) session = await self.init_runtime_mcp_session(server_entity)
await session.start()
self.sessions[server_entity.name] = session self.sessions[server_entity.name] = session
async def get_tools(self) -> list[resource_tool.LLMTool]: async def get_tools(self) -> list[resource_tool.LLMTool]:
@@ -188,7 +189,7 @@ class MCPLoader(loader.ToolLoader):
return name in [f.name for f in self._last_listed_functions] return name in [f.name for f in self._last_listed_functions]
async def invoke_tool(self, name: str, parameters: dict) -> typing.Any: async def invoke_tool(self, name: str, parameters: dict) -> typing.Any:
for server_name, session in self.sessions.items(): for session in self.sessions.values():
for function in session.functions: for function in session.functions:
if function.name == name: if function.name == name:
return await function.func(**parameters) return await function.func(**parameters)