mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-22 13:05:18 +00:00
zhangtao
211fddd6e0
这是一次综合性的重构更新，包含以下主要变更： 1. 升级Python版本到3.12，更新依赖配置 2. 替换旧的.j2模板为.jinja2格式，新增代码生成模板 3. 重构权限过滤策略，更新权限枚举与模型配置 4. 移除Prefect依赖，替换为自研拓扑并行执行引擎 5. 重构认证与上下文管理，拆分租户/请求上下文 6. 简化响应模型、CRUD与服务层代码 7. 清理废弃的支付网关模块，重构订单定时任务 8. 更新在线用户、监控等模块的接口与路由 9. 优化邮件模板与工具类，新增邮件模板文件 10. 修复数据库会话配置与类型提示
124 lines
3.8 KiB
Python
124 lines
3.8 KiB
Python
from typing import Annotated, Any
|
|
|
|
from fastapi import APIRouter, Depends, Path
|
|
from fastapi.responses import JSONResponse
|
|
|
|
from app.common.response import ResponseSchema, SuccessResponse
|
|
from app.core.base_params import PaginationQueryParam
|
|
from app.core.base_schema import AuthSchema
|
|
from app.core.dependencies import AuthPermission
|
|
from app.core.router_class import OperationLogRoute
|
|
|
|
from .schema import (
|
|
AiChatRequestSchema,
|
|
AiChatResponseSchema,
|
|
ChatSessionCreateSchema,
|
|
ChatSessionQueryParam,
|
|
ChatSessionUpdateSchema,
|
|
)
|
|
from .service import ChatService
|
|
|
|
ChatRouter = APIRouter(route_class=OperationLogRoute, prefix="/chat", tags=["AI管理", "AI对话"])
|
|
|
|
|
|
@ChatRouter.get(
|
|
"/detail/{session_id}",
|
|
summary="获取会话详情",
|
|
response_model=ResponseSchema[dict[str, Any]],
|
|
)
|
|
async def get_session_detail_controller(
|
|
session_id: Annotated[str, Path(description="会话ID")],
|
|
auth: Annotated[AuthSchema, Depends(AuthPermission(["module_ai:chat:detail"]))],
|
|
) -> JSONResponse:
|
|
service = ChatService(auth)
|
|
result = await service.get_session(session_id=session_id)
|
|
return SuccessResponse(data=result, msg="获取会话详情成功")
|
|
|
|
|
|
@ChatRouter.get(
|
|
"/list",
|
|
summary="查询会话列表",
|
|
response_model=ResponseSchema[dict],
|
|
)
|
|
async def get_session_list_controller(
|
|
page: Annotated[PaginationQueryParam, Depends()],
|
|
search: Annotated[ChatSessionQueryParam, Depends()],
|
|
auth: Annotated[AuthSchema, Depends(AuthPermission(["module_ai:chat:query"]))],
|
|
) -> JSONResponse:
|
|
service = ChatService(auth)
|
|
result_dict = await service.page(
|
|
page_no=page.page_no,
|
|
page_size=page.page_size,
|
|
search=search,
|
|
order_by=page.order_by,
|
|
)
|
|
return SuccessResponse(data=result_dict, msg="查询会话列表成功")
|
|
|
|
|
|
@ChatRouter.post(
|
|
"/create",
|
|
summary="创建会话",
|
|
response_model=ResponseSchema[dict[str, Any]],
|
|
)
|
|
async def create_session_controller(
|
|
data: ChatSessionCreateSchema,
|
|
auth: Annotated[AuthSchema, Depends(AuthPermission(["module_ai:chat:create"]))],
|
|
) -> JSONResponse:
|
|
service = ChatService(auth)
|
|
result = await service.create(data=data)
|
|
return SuccessResponse(data=result, msg="创建会话成功")
|
|
|
|
|
|
@ChatRouter.put(
|
|
"/update/{session_id}",
|
|
summary="更新会话",
|
|
response_model=ResponseSchema[None],
|
|
)
|
|
async def update_session_controller(
|
|
session_id: Annotated[str, Path(description="会话ID")],
|
|
data: ChatSessionUpdateSchema,
|
|
auth: Annotated[AuthSchema, Depends(AuthPermission(["module_ai:chat:update"]))],
|
|
) -> JSONResponse:
|
|
service = ChatService(auth)
|
|
await service.update(session_id=session_id, data=data)
|
|
return SuccessResponse(data=None, msg="更新会话成功")
|
|
|
|
|
|
@ChatRouter.delete(
|
|
"/delete",
|
|
summary="删除会话",
|
|
response_model=ResponseSchema[None],
|
|
)
|
|
async def delete_session_controller(
|
|
session_ids: list[str],
|
|
auth: Annotated[AuthSchema, Depends(AuthPermission(["module_ai:chat:delete"]))],
|
|
) -> JSONResponse:
|
|
service = ChatService(auth)
|
|
await service.delete(session_ids=session_ids)
|
|
return SuccessResponse(data=None, msg="删除会话成功")
|
|
|
|
|
|
@ChatRouter.post(
|
|
"/ai-chat",
|
|
summary="AI 对话（非流式）",
|
|
response_model=ResponseSchema[AiChatResponseSchema],
|
|
)
|
|
async def ai_chat_controller(
|
|
data: AiChatRequestSchema,
|
|
auth: Annotated[AuthSchema, Depends(AuthPermission(["module_ai:chat:query"]))],
|
|
) -> JSONResponse:
|
|
service = ChatService(auth)
|
|
result = await service.chat_non_stream(
|
|
message=data.message,
|
|
session_id=data.session_id,
|
|
)
|
|
return SuccessResponse(
|
|
data=AiChatResponseSchema(
|
|
response=result["response"],
|
|
session_id=result["session_id"],
|
|
function_calls=result.get("function_calls"),
|
|
action=result.get("action"),
|
|
),
|
|
msg="对话成功",
|
|
)
|