mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-26 06:19:04 +00:00
zhangtao
ea4e35fd14
refactor(代码生成): 重构代码生成相关组件和路由路径 feat(WebSocket): 添加WebSocket支持并配置Nginx代理 fix(用户管理): 移除超级管理员过滤并调整用户数据 feat(定时任务): 完善定时任务日志功能并添加相关API style(前端): 优化AI助手界面样式和提示文本 chore(初始化数据): 清理和更新系统初始化数据 perf(前端): 优化表格操作列宽度和按钮样式 docs(注释): 更新控制器注释以反映模块变更
62 lines
2.3 KiB
Python
62 lines
2.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from fastapi import APIRouter, Depends, WebSocket
|
|
from fastapi.responses import JSONResponse, StreamingResponse
|
|
|
|
from app.common.response import StreamResponse, SuccessResponse
|
|
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 .service import MCPService
|
|
from .schema import ChatQuerySchema
|
|
|
|
|
|
MCPRouter = APIRouter(route_class=OperationLogRoute, prefix="", tags=["MCP智能助手"])
|
|
|
|
|
|
@MCPRouter.post("/mcp/chat", summary="智能对话", description="与MCP智能助手进行对话")
|
|
async def chat_controller(
|
|
query: ChatQuerySchema,
|
|
auth: AuthSchema = Depends(AuthPermission(permissions=["ai:mcp:chat"]))
|
|
) -> StreamingResponse:
|
|
"""智能对话接口"""
|
|
logger.info(f"用户 {auth.user.name} 发起智能对话: {query.message[:50]}...")
|
|
|
|
async def generate_response():
|
|
try:
|
|
async for chunk in MCPService.chat_query(query.message):
|
|
# 确保返回的是字节串
|
|
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 StreamingResponse(generate_response(), media_type="text/plain; charset=utf-8")
|
|
|
|
|
|
@MCPRouter.websocket("/ws/mcp/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(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() |