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聊天接口和前端组件 优化代码结构和性能，修复已知问题
54 lines
1.7 KiB
Python
54 lines
1.7 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, UserBySchema
|
|
|
|
|
|
class ChatSessionCreateSchema(BaseModel):
|
|
"""新增聊天会话"""
|
|
|
|
title: str = Field(..., description="会话标题")
|
|
|
|
|
|
class ChatSessionUpdateSchema(BaseModel):
|
|
"""更新聊天会话"""
|
|
|
|
title: str | None = Field(None, description="会话标题")
|
|
|
|
|
|
class ChatSessionOutSchema(ChatSessionCreateSchema, BaseSchema, UserBySchema):
|
|
"""聊天会话响应模型"""
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
|
|
@dataclass
|
|
class ChatSessionQueryParam:
|
|
"""聊天会话查询参数"""
|
|
|
|
def __init__(
|
|
self,
|
|
title: str | None = Query(None, description="会话标题"),
|
|
status: 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"],
|
|
),
|
|
updated_time: list[str] | None = Query(
|
|
None,
|
|
description="更新时间范围",
|
|
examples=["2025-01-01 00:00:00", "2025-12-31 23:59:59"],
|
|
),
|
|
) -> None:
|
|
self.title = (QueueEnum.like.value, f"%{title}%") if title else None
|
|
self.status = (QueueEnum.eq.value, status) if status else None
|
|
|
|
if created_time and len(created_time) == 2:
|
|
self.created_time = (QueueEnum.between.value, (created_time[0], created_time[1]))
|
|
if updated_time and len(updated_time) == 2:
|
|
self.updated_time = (QueueEnum.between.value, (updated_time[0], updated_time[1]))
|