feat: ui for mcp

.gitignore

@@ -44,5 +44,6 @@ test.py
 .venv/
 uv.lock
 /test
+plugins.bak
 coverage.xml
-.coverage
+.coverage

pkg/api/http/controller/groups/market.py

@@ -0,0 +1,143 @@
+from __future__ import annotations
+
+import quart
+import datetime
+
+from .. import group
+
+
+@group.group_class('market', '/api/v1/market')
+class MarketRouterGroup(group.RouterGroup):
+    async def initialize(self) -> None:
+        @self.route('/plugins', methods=['POST'], auth_type=group.AuthType.USER_TOKEN)
+        async def _() -> str:
+            """获取插件市场列表"""
+            # data = await quart.request.json
+            # page = data.get('page', 1)
+            # page_size = data.get('page_size', 10)
+            # query = data.get('query', '')
+            # sort_by = data.get('sort_by', 'stars')
+            # sort_order = data.get('sort_order', 'DESC')
+
+            # # 这里是获取插件列表的实现
+            # # 实际项目中这部分会连接到真实的插件市场API或数据库
+            # # 这里我们只是返回一些假数据作为示例
+
+            # # 模拟延迟
+            # import asyncio
+
+            # await asyncio.sleep(0.5)
+
+            # 返回结果
+            return self.success(data={'plugins': [], 'total': 0})
+
+        @self.route('/mcp', methods=['POST'], auth_type=group.AuthType.USER_TOKEN)
+        async def _() -> str:
+            """获取MCP服务器市场列表"""
+            data = await quart.request.json
+            page = data.get('page', 1)
+            page_size = data.get('page_size', 10)
+            query = data.get('query', '')
+            sort_by = data.get('sort_by', 'stars')
+            sort_order = data.get('sort_order', 'DESC')
+
+            # 这里是获取MCP服务器列表的实现
+            # 实际项目中这部分会连接到真实的MCP市场API或数据库
+            # 这里我们只是返回一些假数据作为示例
+
+            # 模拟延迟
+            import asyncio
+
+            await asyncio.sleep(0.5)
+
+            # 生成假数据
+            servers = []
+
+            # 只在有搜索关键词或排序时才返回数据
+            if query or sort_by:
+                now = datetime.datetime.now().isoformat()
+                yesterday = (datetime.datetime.now() - datetime.timedelta(days=1)).isoformat()
+
+                test_servers = [
+                    {
+                        'ID': 1,
+                        'CreatedAt': yesterday,
+                        'UpdatedAt': now,
+                        'DeletedAt': None,
+                        'name': 'Google Maps MCP',
+                        'author': 'langbot-community',
+                        'description': 'Google Maps integration for LangBot, providing geocoding and directions capabilities.',
+                        'repository': 'langbot-community/google-maps-mcp',
+                        'artifacts_path': '',
+                        'stars': 124,
+                        'downloads': 342,
+                        'status': 'initialized',
+                        'synced_at': now,
+                        'pushed_at': now,
+                        'version': '1.0.0',
+                    },
+                    {
+                        'ID': 2,
+                        'CreatedAt': yesterday,
+                        'UpdatedAt': now,
+                        'DeletedAt': None,
+                        'name': 'Weather MCP',
+                        'author': 'langbot-community',
+                        'description': 'Weather integration for LangBot, providing current weather and forecasts.',
+                        'repository': 'langbot-community/weather-mcp',
+                        'artifacts_path': '',
+                        'stars': 85,
+                        'downloads': 215,
+                        'status': 'initialized',
+                        'synced_at': now,
+                        'pushed_at': yesterday,
+                        'version': '1.1.0',
+                    },
+                    {
+                        'ID': 3,
+                        'CreatedAt': yesterday,
+                        'UpdatedAt': now,
+                        'DeletedAt': None,
+                        'name': 'Serper Search MCP',
+                        'author': 'langbot-developers',
+                        'description': 'Serper Search integration for LangBot, providing advanced web search capabilities.',
+                        'repository': 'langbot-developers/serper-search-mcp',
+                        'artifacts_path': '',
+                        'stars': 67,
+                        'downloads': 178,
+                        'status': 'initialized',
+                        'synced_at': now,
+                        'pushed_at': yesterday,
+                        'version': '0.9.0',
+                    },
+                ]
+
+                # 应用搜索过滤
+                if query:
+                    query = query.lower()
+                    servers = [
+                        s
+                        for s in test_servers
+                        if query in s['name'].lower()
+                        or query in s['description'].lower()
+                        or query in s['author'].lower()
+                    ]
+                else:
+                    servers = test_servers
+
+                # 应用排序
+                reverse = sort_order.upper() == 'DESC'
+                if sort_by == 'stars':
+                    servers = sorted(servers, key=lambda s: s['stars'], reverse=reverse)
+                elif sort_by == 'created_at':
+                    servers = sorted(servers, key=lambda s: s['CreatedAt'], reverse=reverse)
+                elif sort_by == 'pushed_at':
+                    servers = sorted(servers, key=lambda s: s['pushed_at'], reverse=reverse)
+
+                # 应用分页
+                start_idx = (page - 1) * page_size
+                end_idx = start_idx + page_size
+                servers = servers[start_idx:end_idx]
+
+            # 返回结果
+            return self.success(data={'servers': servers, 'total': len(servers)})

pkg/api/http/controller/groups/mcp.py

@@ -0,0 +1,196 @@
+from __future__ import annotations
+import time
+import traceback
+import uuid
+
+import quart
+import asyncio
+
+import sqlalchemy
+
+from pkg.entity.persistence.mcp import MCPServer
+
+from .....core import taskmgr
+from .. import group
+
+from sqlalchemy import insert
+
+@group.group_class('mcp', '/api/v1/mcp')
+class MCPRouterGroup(group.RouterGroup):
+    async def initialize(self) -> None:
+        @self.route('/servers', methods=['GET', 'POST'], auth_type=group.AuthType.USER_TOKEN)
+        async def _() -> str:
+            """获取MCP服务器列表"""
+            if quart.request.method == 'GET':
+                result = await self.ap.persistence_mgr.execute_async(
+                    sqlalchemy.select(MCPServer).order_by(MCPServer.created_at.desc())
+                )
+                raw_results = result.all()
+                servers = [self.ap.persistence_mgr.serialize_model(MCPServer, row) for row in raw_results]
+
+                servers_with_status = []
+                for server in servers:
+                    # 设置状态
+                    if server['enable']:
+                        status = 'enabled'
+                    else:
+                        status = 'disabled'
+
+                    # 构建 config 对象 (前端期望的格式)
+                    extra_args = server.get('extra_args', {})
+                    config = {
+                        'name': server['name'],
+                        'mode': server['mode'],
+                        'enable': server['enable'],
+                    }
+
+                    # 根据模式添加相应的配置
+                    if server['mode'] == 'sse':
+                        config['url'] = extra_args.get('url', '')
+                        config['headers'] = extra_args.get('headers', {})
+                        config['timeout'] = extra_args.get('timeout', 60)
+                    elif server['mode'] == 'stdio':
+                        config['command'] = extra_args.get('command', '')
+                        config['args'] = extra_args.get('args', [])
+                        config['env'] = extra_args.get('env', {})
+
+                    server_info = {
+                        'name': server['name'],
+                        'mode': server['mode'],
+                        'enable': server['enable'],
+                        'status': status,
+                        'tools': [],  # 暂时返回空数组，需要连接到MCP服务器才能获取工具列表
+                        'config': config,
+                    }
+                    servers_with_status.append(server_info)
+
+                return self.success(data={'servers': servers_with_status})
+
+            elif quart.request.method == 'POST':
+                data = await quart.request.json
+                data = data['source']
+                try:
+                # 检查服务器名称是否重复
+                    result = await self.ap.persistence_mgr.execute_async(
+                        sqlalchemy.select(MCPServer).where(MCPServer.name == data['name'])
+                    )
+                    if result.first() is not None:
+                        return self.http_status(400, -1, 'Server name already exists')
+
+                    # 创建新服务器配置
+                    new_server = {
+                        'uuid': str(uuid.uuid4()),
+                        'name': data['name'],
+                        'mode': 'sse',
+                        'enable': data.get('enable', False),
+                        'description': data.get('description',''),
+                        'extra_args': {
+                            'url':data.get('url',''),
+                            'headers':data.get('headers',{}),
+                            'timeout':data.get('timeout',60),
+                        },
+                    }
+
+                    await self.ap.persistence_mgr.execute_async(
+                        sqlalchemy.insert(MCPServer).values(new_server)
+                    )
+
+                    return self.success()
+
+                except Exception as e:
+                    print(traceback.format_exc())
+
+        @self.route('/servers/<server_name>', methods=['GET', 'PUT', 'DELETE'], auth_type=group.AuthType.USER_TOKEN)
+        async def _(server_name: str) -> str:
+            """获取、更新或删除MCP服务器配置"""
+            result = await self.ap.persistence_mgr.execute_async(
+                sqlalchemy.select(MCPServer).where(MCPServer.name == server_name)
+            )
+            server = result.first()
+            if server is None:
+                return self.http_status(404, -1, 'Server not found')
+
+            if quart.request.method == 'GET':
+                server_data = self.ap.persistence_mgr.serialize_model(MCPServer, server)
+                return self.success(data={'server': server_data})
+
+            elif quart.request.method == 'PUT':
+                data = await quart.request.json
+                update_data = {
+                    'enable': data.get('enable', server.enable),
+                    'description': data.get('description', server.description),
+                }
+
+                extra_args = server.extra_args or {}
+                if server.mode == 'sse':
+                    extra_args.update({
+                        'url': data.get('url', extra_args.get('url','')),
+                        'headers': data.get('headers', extra_args.get('headers',{})),
+                        'timeout': data.get('timeout', extra_args.get('timeout',60)),
+                    })
+                update_data['extra_args'] = extra_args
+
+                await self.ap.persistence_mgr.execute_async(
+                    sqlalchemy.update(MCPServer).where(MCPServer.name == server_name).values(update_data)
+                )
+
+                return self.success()
+
+            elif quart.request.method == 'DELETE':
+                await self.ap.persistence_mgr.execute_async(
+                    sqlalchemy.delete(MCPServer).where(MCPServer.name == server_name)
+                )
+                return self.success()
+
+        @self.route('/servers/<server_name>/test', methods=['POST'], auth_type=group.AuthType.USER_TOKEN)
+        async def _(server_name: str) -> str:
+            """测试MCP服务器连接"""
+            result = await self.ap.persistence_mgr.execute_async(
+                sqlalchemy.select(MCPServer).where(MCPServer.name == server_name)
+            )
+            server = result.first()
+            if server is None:
+                return self.http_status(404, -1, 'Server not found')
+
+            # 创建测试任务
+            ctx = taskmgr.TaskContext.new()
+            wrapper = self.ap.task_mgr.create_user_task(
+                self._test_mcp_server(server, ctx),
+                kind='mcp-operation',
+                name=f'mcp-test-{server_name}',
+                label=f'Testing MCP server {server_name}',
+                context=ctx,
+            )
+            return self.success(data={'task_id': wrapper.id})
+
+    async def _test_mcp_server(self, server: MCPServer, ctx: taskmgr.TaskContext):
+        """测试MCP服务器连接"""
+        try:
+            from .....provider.tools.loaders.mcp import RuntimeMCPSession
+
+            ctx.current_action = f'Testing connection to {server.name}'
+
+            # 创建临时会话进行测试
+            session = RuntimeMCPSession(server.name, {
+                'name': server.name,
+                'mode': server.mode,
+                'enable': server.enable,
+                'description': server.description,
+                'extra_args': server.extra_args or {},
+            }, self.ap)
+            await session.initialize()
+
+            # 获取工具列表作为测试
+            tools_count = len(session.functions)
+            ctx.current_action = f'Successfully connected. Found {tools_count} tools.'
+
+            # 关闭测试会话
+            await session.shutdown()
+
+            return {'status': 'success', 'tools_count': tools_count}
+
+        except Exception as e:
+            ctx.current_action = f'Connection test failed: {str(e)}'
+            raise e
+
+