xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-27 07:54:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: mcp server datastructure

Browse Source
This commit is contained in:
Junyan Qin
2025-11-04 16:13:03 +08:00
parent bc1fbfa190
commit c1c03f11b4
13 changed files with 155 additions and 208 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/api/http/service/mcp.py
+17 -3
View File
@@ -40,7 +40,6 @@ class RuntimeMCPServer:
self.session = RuntimeMCPSession(
self.mcp_server_entity.name, mixed_config, self.mcp_server_entity.enable, self.ap
)
await self.session.initialize()
await self.session.start()
async def _test_mcp_server_task(self, task_context: taskmgr.TaskContext):
@@ -102,14 +101,29 @@ class MCPService:
def __init__(self, ap: app.Application) -> None:
self.ap = ap
async def get_mcp_servers(self) -> list[dict]:
async def get_mcp_servers(self, contain_runtime_info: bool = False) -> list[dict]:
result = await self.ap.persistence_mgr.execute_async(sqlalchemy.select(persistence_mcp.MCPServer))
servers = result.all()
return [self.ap.persistence_mgr.serialize_model(persistence_mcp.MCPServer, server) for server in servers]
serialized_servers = [
self.ap.persistence_mgr.serialize_model(persistence_mcp.MCPServer, server) for server in servers
]
if contain_runtime_info:
for server in serialized_servers:
session = self.ap.tool_mgr.mcp_tool_loader.get_session(server['name'])
runtime_info = None
if session:
runtime_info = session.get_runtime_info_dict()
server['runtime_info'] = runtime_info if runtime_info else None
return serialized_servers
async def create_mcp_server(self, server_data: dict) -> str:
server_data['uuid'] = str(uuid.uuid4())
print('server_data:', server_data)
await self.ap.persistence_mgr.execute_async(sqlalchemy.insert(persistence_mcp.MCPServer).values(server_data))
result = await self.ap.persistence_mgr.execute_async(