refactor(module): 重构模块结构，将示例模块和AI模块迁移至应用模块

- 删除原有的示例模块(module_example)和AI模块(module_ai) - 在应用模块(module_application)下新增AI子模块和示例子模块 - 更新相关路由配置，移除不再使用的模块路由 - 调整定时任务模块从监控模块迁移至应用模块 - 更新前端API路径和组件文件结构 - 修正数据库表名前缀，统一使用'app_'作为应用模块表前缀
2026-09-23 05:10:57 +00:00 · 2025-10-05 12:24:40 +08:00
parent 055e054173
commit d943d295d0
62 changed files with 1566 additions and 1571 deletions
@@ -0,0 +1,119 @@
+# -*- coding: utf-8 -*-
+
+from fastapi import APIRouter, Depends, Path, Body, WebSocket
+from fastapi.responses import JSONResponse, StreamingResponse
+
+from app.common.response import StreamResponse, SuccessResponse
+from app.common.request import PaginationService
+from app.core.base_params import PaginationQueryParam
+from app.core.dependencies import AuthPermission
+from app.core.router_class import OperationLogRoute
+from app.core.logger import logger
+from app.api.v1.module_system.auth.schema import AuthSchema
+from .param import McpQueryParam
+from .service import McpService
+from .schema import McpCreateSchema, McpUpdateSchema, ChatQuerySchema
+
+
+AIRouter = APIRouter(route_class=OperationLogRoute, prefix="/ai", tags=["MCP智能助手"])
+
+
+@AIRouter.post("/chat", summary="智能对话", description="与MCP智能助手进行对话")
+async def chat_controller(
+    query: ChatQuerySchema,
+    auth: AuthSchema = Depends(AuthPermission(["app:ai:chat"]))
+) -> StreamingResponse:
+    """智能对话接口"""
+    user_name = auth.user.name if auth.user else "未知用户"
+    logger.info(f"用户 {user_name} 发起智能对话: {query.message[:50]}...")
+    
+    async def generate_response():
+        try:
+            async for chunk in McpService.chat_query(query=query):
+                # 确保返回的是字节串
+                if chunk:
+                    yield chunk.encode('utf-8') if isinstance(chunk, str) else chunk
+        except Exception as e:
+            logger.error(f"流式响应出错: {str(e)}")
+            yield f"抱歉，处理您的请求时出现了错误: {str(e)}".encode('utf-8')
+    
+    return StreamResponse(generate_response(), media_type="text/plain; charset=utf-8")
+
+
+@AIRouter.get("/detail/{id}", summary="获取 MCP 服务器详情", description="获取 MCP 服务器详情")
+async def detail_controller(
+    id: int = Path(..., description="MCP ID"),
+    auth: AuthSchema = Depends(AuthPermission(["app:ai:query"]))
+) -> JSONResponse:
+    result_dict = await McpService.detail_service(auth=auth, id=id)
+    logger.info(f"获取 MCP 服务器详情成功 {id}")
+    return SuccessResponse(data=result_dict, msg="获取 MCP 服务器详情成功")
+
+
+@AIRouter.get("/list", summary="查询 MCP 服务器列表", description="查询 MCP 服务器列表")
+async def list_controller(
+    page: PaginationQueryParam = Depends(),
+    search: McpQueryParam = Depends(),
+    auth: AuthSchema = Depends(AuthPermission(["app:ai:query"]))
+) -> JSONResponse:
+    result_dict_list = await McpService.list_service(auth=auth, search=search, order_by=page.order_by)
+    result_dict = await PaginationService.paginate(data_list=result_dict_list, page_no=page.page_no, page_size=page.page_size)
+    logger.info(f"查询 MCP 服务器列表成功")
+    return SuccessResponse(data=result_dict, msg="查询 MCP 服务器列表成功")
+
+
+@AIRouter.post("/create", summary="创建 MCP 服务器", description="创建 MCP 服务器")
+async def create_controller(
+    data: McpCreateSchema,
+    auth: AuthSchema = Depends(AuthPermission(["app:ai:create"]))
+) -> JSONResponse:
+    result_dict = await McpService.create_service(auth=auth, data=data)
+    logger.info(f"创建 MCP 服务器成功: {result_dict}")
+    return SuccessResponse(data=result_dict, msg="创建 MCP 服务器成功")
+
+
+@AIRouter.put("/update/{id}", summary="修改 MCP 服务器", description="修改 MCP 服务器")
+async def update_controller(
+    data: McpUpdateSchema,
+    id: int = Path(..., description="MCP ID"),
+    auth: AuthSchema = Depends(AuthPermission(["app:ai:update"]))
+) -> JSONResponse:
+    result_dict = await McpService.update_service(auth=auth, id=id, data=data)
+    logger.info(f"修改 MCP 服务器成功: {result_dict}")
+    return SuccessResponse(data=result_dict, msg="修改 MCP 服务器成功")
+
+
+@AIRouter.delete("/delete", summary="删除 MCP 服务器", description="删除 MCP 服务器")
+async def delete_controller(
+    ids: list[int] = Body(..., description="ID列表"),
+    auth: AuthSchema = Depends(AuthPermission(["app:ai:delete"]))
+) -> JSONResponse:
+    await McpService.delete_service(auth=auth, ids=ids)
+    logger.info(f"删除 MCP 服务器成功: {ids}")
+    return SuccessResponse(msg="删除 MCP 服务器成功")
+
+
+@AIRouter.websocket("/ws/chat", name="WebSocket聊天")
+async def websocket_chat_controller(
+    websocket: WebSocket,
+):
+    """WebSocket聊天接口
+    
+    ws://127.0.0.1:8001/api/v1/ai/mcp/ws/chat
+    """
+    await websocket.accept()
+    try:
+        while True:
+            data = await websocket.receive_text()
+            # 流式发送响应
+            try:
+                async for chunk in McpService.chat_query(query=ChatQuerySchema(message=data)):
+                    if chunk:
+                        await websocket.send_text(chunk)
+            except Exception as e:
+                logger.error(f"处理聊天查询出错: {str(e)}")
+                await websocket.send_text(f"抱歉，处理您的请求时出现了错误: {str(e)}")
+    except Exception as e:
+        logger.error(f"WebSocket聊天出错: {str(e)}")
+    finally:
+        await websocket.close()