mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-10-04 01:10:36 +00:00
zhangtao
119964e6f9
fix: 修复文件下载和删除功能 fix: 修复Redis哈希获取方法返回值类型 fix: 修复权限检查逻辑 perf: 优化部门和服务详情查询性能 perf: 优化角色数据范围显示 style: 清理无用导入和注释 style: 统一CRUD方法命名 docs: 更新main.py中的命令说明 chore: 移动IP定位工具类位置 chore: 更新.gitignore忽略迁移版本文件
76 lines
3.0 KiB
Python
76 lines
3.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from typing import List, Dict, Optional, Any
|
|
|
|
from app.core.exceptions import CustomException
|
|
from app.api.v1.module_system.auth.schema import AuthSchema
|
|
from app.utils.ai_util import AIClient
|
|
from .schema import McpCreateSchema, McpUpdateSchema, McpOutSchema, ChatQuerySchema
|
|
from .param import McpQueryParam
|
|
from .crud import McpCRUD
|
|
|
|
|
|
class McpService:
|
|
"""MCP服务层"""
|
|
|
|
@classmethod
|
|
async def detail_service(cls, auth: AuthSchema, id: int) -> Dict[str, Any]:
|
|
"""详情"""
|
|
obj = await McpCRUD(auth).get_by_id_crud(id=id)
|
|
if not obj:
|
|
raise CustomException(msg='MCP 服务器不存在')
|
|
return McpOutSchema.model_validate(obj).model_dump()
|
|
|
|
@classmethod
|
|
async def list_service(cls, auth: AuthSchema, search: Optional[McpQueryParam] = None, order_by: Optional[List[Dict[str, str]]] = None) -> List[Dict[str, Any]]:
|
|
"""列表查询"""
|
|
if order_by:
|
|
order_by = eval(str(order_by))
|
|
obj_list = await McpCRUD(auth).get_list_crud(search=search.__dict__ if search else {}, order_by=order_by)
|
|
return [McpOutSchema.model_validate(obj).model_dump() for obj in obj_list]
|
|
|
|
@classmethod
|
|
async def create_service(cls, auth: AuthSchema, data: McpCreateSchema) -> Dict[str, Any]:
|
|
"""创建"""
|
|
obj = await McpCRUD(auth).get_by_name_crud(name=data.name)
|
|
if obj:
|
|
raise CustomException(msg='创建失败，MCP 服务器已存在')
|
|
obj = await McpCRUD(auth).create_crud(data=data)
|
|
return McpOutSchema.model_validate(obj).model_dump()
|
|
|
|
@classmethod
|
|
async def update_service(cls, auth: AuthSchema, id: int, data: McpUpdateSchema) -> Dict[str, Any]:
|
|
"""更新"""
|
|
obj = await McpCRUD(auth).get_by_id_crud(id=id)
|
|
if not obj:
|
|
raise CustomException(msg='更新失败，该数据不存在')
|
|
exist_obj = await McpCRUD(auth).get_by_name_crud(name=data.name)
|
|
if exist_obj and exist_obj.id != id:
|
|
raise CustomException(msg='更新失败，MCP 服务器名称重复')
|
|
obj = await McpCRUD(auth).update_crud(id=id, data=data)
|
|
return McpOutSchema.model_validate(obj).model_dump()
|
|
|
|
@classmethod
|
|
async def delete_service(cls, auth: AuthSchema, ids: List[int]) -> None:
|
|
"""删除"""
|
|
if len(ids) < 1:
|
|
raise CustomException(msg='删除失败，删除对象不能为空')
|
|
for id in ids:
|
|
obj = await McpCRUD(auth).get_by_id_crud(id=id)
|
|
if not obj:
|
|
raise CustomException(msg='删除失败，该数据不存在')
|
|
await McpCRUD(auth).delete_crud(ids=ids)
|
|
|
|
@classmethod
|
|
async def chat_query(cls, query: ChatQuerySchema):
|
|
"""处理聊天查询"""
|
|
# 创建MCP客户端实例
|
|
mcp_client = AIClient()
|
|
try:
|
|
# 处理消息
|
|
async for response in mcp_client.process(query.message):
|
|
yield response
|
|
finally:
|
|
# 确保关闭客户端连接
|
|
await mcp_client.close()
|