mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-22 13:05:18 +00:00
zhangtao
aad10384d4
将AI模块从module_application迁移到module_ai目录 新增聊天会话和消息的CRUD、服务和控制器 实现WebSocket聊天接口和前端组件 优化代码结构和性能，修复已知问题
57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
from dataclasses import dataclass
|
|
|
|
from fastapi import Query
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
from app.common.enums import QueueEnum
|
|
from app.core.base_schema import BaseSchema
|
|
|
|
|
|
class ChatMessageCreateSchema(BaseModel):
|
|
"""新增聊天消息"""
|
|
|
|
session_id: int = Field(..., description="会话ID")
|
|
type: str = Field(..., description="消息类型: user/assistant")
|
|
content: str = Field(..., description="消息内容")
|
|
timestamp: int = Field(..., description="时间戳")
|
|
files: list[dict] | None = Field(None, description="关联的文件信息")
|
|
|
|
|
|
class ChatMessageUpdateSchema(BaseModel):
|
|
"""更新聊天消息"""
|
|
|
|
session_id: int | None = Field(None, description="会话ID")
|
|
type: str | None = Field(None, description="消息类型: user/assistant")
|
|
content: str | None = Field(None, description="消息内容")
|
|
timestamp: int | None = Field(None, description="时间戳")
|
|
files: list[dict] | None = Field(None, description="关联的文件信息")
|
|
|
|
|
|
class ChatMessageOutSchema(ChatMessageCreateSchema, BaseSchema):
|
|
"""聊天消息响应模型"""
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
|
|
@dataclass
|
|
class ChatMessageQueryParam:
|
|
"""聊天消息查询参数"""
|
|
|
|
def __init__(
|
|
self,
|
|
session_id: int | None = Query(None, description="会话ID"),
|
|
type: str | None = Query(None, description="消息类型"),
|
|
content: str | None = Query(None, description="消息内容"),
|
|
created_time: list[str] | None = Query(
|
|
None,
|
|
description="创建时间范围",
|
|
examples=["2025-01-01 00:00:00", "2025-12-31 23:59:59"],
|
|
),
|
|
) -> None:
|
|
self.session_id = (QueueEnum.eq.value, session_id) if session_id else None
|
|
self.type = (QueueEnum.eq.value, type) if type else None
|
|
self.content = (QueueEnum.like.value, f"%{content}%") if content else None
|
|
|
|
if created_time and len(created_time) == 2:
|
|
self.created_time = (QueueEnum.between.value, (created_time[0], created_time[1]))
|