injectlib/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2025-11-25 03:15:06 +08: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

3
pkg/provider/tools/loaders/mcp.py
View File

@@ -172,6 +172,7 @@ class MCPLoader(loader.ToolLoader):
server_entity: persistence_mcp.MCPServer | sqlalchemy.Row[persistence_mcp.MCPServer] | dict,
):
session = await self.init_runtime_mcp_session(server_entity)
await session.start()
self.sessions[server_entity.name] = session
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]
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:
if function.name == name:
return await function.func(**parameters)