shutiaozhao127-boop
cfdeeb5e41
添加后端API和前端项目的初始目录结构，包括应用层模块、工具模块和模型定义 配置前后端的.dockerignore文件以排除开发环境文件 添加前端样式文件，包括字体、Tailwind CSS和主题配置 初始化Python虚拟环境中的类型标记文件
25 lines
747 B
Python
25 lines
747 B
Python
from datetime import datetime, timezone
|
|
|
|
from beanie import Document, PydanticObjectId
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class ChatMessage(BaseModel):
|
|
role: str # user / assistant
|
|
content: str
|
|
timestamp: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
class ChatSession(Document):
|
|
user_id: PydanticObjectId
|
|
report_id: PydanticObjectId
|
|
messages: list[ChatMessage] = Field(default_factory=list)
|
|
created_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
|
|
updated_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
|
|
|
|
class Settings:
|
|
name = "chat_sessions"
|
|
indexes = [
|
|
[("user_id", 1), ("report_id", 1)],
|
|
]
|