Neo
6f8f12314f
包含多个会话的累积代码变更： - backend: AI 聊天服务、触发器调度、认证增强、WebSocket、调度器最小间隔 - admin-web: ETL 状态页、任务管理、调度配置、登录优化 - miniprogram: 看板页面、聊天集成、UI 组件、导航更新 - etl: DWS 新任务（finance_area_daily/board_cache）、连接器增强 - tenant-admin: 项目初始化 - db: 19 个迁移脚本（etl_feiqiu 11 + zqyy_app 8） - packages/shared: 枚举和工具函数更新 - tools: 数据库工具、报表生成、健康检查 - docs: PRD/架构/部署/合约文档更新 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
109 lines
2.6 KiB
Python
109 lines
2.6 KiB
Python
"""
|
||
小程序 CHAT 模块 Pydantic 模型。
|
||
|
||
覆盖：对话历史列表、消息查看、发送消息、SSE 流式请求等场景。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from app.schemas.base import CamelModel
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 对话历史（CHAT-1）
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
class ChatHistoryItem(CamelModel):
|
||
"""对话历史列表项。"""
|
||
|
||
id: int
|
||
title: str
|
||
customer_name: str | None = None
|
||
last_message: str | None = None
|
||
timestamp: str # ISO 8601，最后消息时间
|
||
unread_count: int = 0
|
||
|
||
|
||
class ChatHistoryResponse(CamelModel):
|
||
"""CHAT-1 对话历史列表响应。"""
|
||
|
||
items: list[ChatHistoryItem]
|
||
total: int
|
||
page: int
|
||
page_size: int
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 消息查看（CHAT-2）
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
class ReferenceCard(CamelModel):
|
||
"""引用卡片，附加在 AI 回复消息中的结构化上下文数据。"""
|
||
|
||
type: str # 'customer' | 'record'
|
||
title: str
|
||
summary: str
|
||
data: dict[str, str] # 键值对详情
|
||
|
||
|
||
class ChatMessageItem(CamelModel):
|
||
"""对话消息项。"""
|
||
|
||
id: int
|
||
role: str # 'user' | 'assistant'
|
||
content: str
|
||
created_at: str # ISO 8601（统一字段名）
|
||
reference_card: ReferenceCard | None = None
|
||
|
||
|
||
class ChatMessagesResponse(CamelModel):
|
||
"""CHAT-2 对话消息列表响应。"""
|
||
|
||
chat_id: int
|
||
items: list[ChatMessageItem]
|
||
total: int
|
||
page: int
|
||
page_size: int
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 发送消息（CHAT-3）
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
class MessageBrief(CamelModel):
|
||
"""消息摘要（用于发送消息响应）。"""
|
||
|
||
id: int
|
||
content: str
|
||
created_at: str # ISO 8601
|
||
|
||
|
||
class SendMessageRequest(CamelModel):
|
||
"""CHAT-3 发送消息请求体。"""
|
||
|
||
content: str
|
||
|
||
|
||
class SendMessageResponse(CamelModel):
|
||
"""CHAT-3 发送消息响应（含用户消息和 AI 回复）。"""
|
||
|
||
user_message: MessageBrief
|
||
ai_reply: MessageBrief
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# SSE 流式（CHAT-4）
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
class ChatStreamRequest(CamelModel):
|
||
"""CHAT-4 SSE 流式请求体。"""
|
||
|
||
chat_id: int
|
||
content: str
|
||
source_page: str | None = None
|
||
page_context: dict | None = None
|