81 lines
3.1 KiB
Python
81 lines
3.1 KiB
Python
"""
|
||
会话和消息模型
|
||
"""
|
||
import uuid
|
||
import datetime
|
||
from sqlalchemy import Column, String, DateTime, ForeignKey, Boolean, Integer, Text, JSON
|
||
from sqlalchemy.dialects.postgresql import UUID
|
||
from sqlalchemy.orm import relationship
|
||
|
||
from app.db import Base
|
||
|
||
|
||
class Conversation(Base):
|
||
"""会话表
|
||
|
||
会话类型说明：
|
||
- 草稿会话 (is_draft=True): 使用应用的当前草稿配置，用于开发和测试
|
||
- 发布会话 (is_draft=False): 使用应用的当前发布版本配置，用于生产环境
|
||
|
||
工作空间隔离：
|
||
- 每个会话属于一个工作空间（workspace_id）
|
||
- 同一个应用在不同工作空间有独立的会话记录
|
||
- 支持应用分享后的会话隔离
|
||
"""
|
||
__tablename__ = "conversations"
|
||
|
||
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4, index=True)
|
||
|
||
# 关联信息
|
||
app_id = Column(UUID(as_uuid=True), ForeignKey("apps.id"), nullable=False, comment="应用ID")
|
||
workspace_id = Column(UUID(as_uuid=True), ForeignKey("workspaces.id"), nullable=False, comment="工作空间ID")
|
||
user_id = Column(String, nullable=True, comment="用户ID（外部系统）")
|
||
|
||
# 会话信息
|
||
title = Column(String(255), comment="会话标题")
|
||
summary = Column(Text, comment="会话摘要")
|
||
|
||
# 会话类型：True=草稿会话（使用草稿配置），False=发布会话（使用发布配置）
|
||
is_draft = Column(Boolean, default=True, nullable=False, comment="是否为草稿会话")
|
||
|
||
# 配置快照：保存创建会话时的完整配置，用于审计和问题追溯
|
||
config_snapshot = Column(JSON, comment="配置快照（Agent配置、模型配置等）")
|
||
|
||
# 统计信息
|
||
message_count = Column(Integer, default=0, comment="消息数量")
|
||
|
||
# 状态
|
||
is_active = Column(Boolean, default=True, nullable=False, comment="是否活跃")
|
||
|
||
# 时间戳
|
||
created_at = Column(DateTime, default=datetime.datetime.now, comment="创建时间")
|
||
updated_at = Column(DateTime, default=datetime.datetime.now, onupdate=datetime.datetime.now, comment="更新时间")
|
||
|
||
# 关联关系
|
||
app = relationship("App", back_populates="conversations")
|
||
workspace = relationship("Workspace")
|
||
messages = relationship("Message", back_populates="conversation", cascade="all, delete-orphan")
|
||
|
||
|
||
class Message(Base):
|
||
"""消息表"""
|
||
__tablename__ = "messages"
|
||
|
||
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4, index=True)
|
||
|
||
# 关联信息
|
||
conversation_id = Column(UUID(as_uuid=True), ForeignKey("conversations.id"), nullable=False, comment="会话ID")
|
||
|
||
# 消息内容
|
||
role = Column(String(20), nullable=False, comment="角色: user/assistant/system")
|
||
content = Column(Text, nullable=False, comment="消息内容")
|
||
|
||
# 元数据（避免使用 metadata 保留字）
|
||
meta_data = Column(JSON, comment="消息元数据（如模型、token使用等）")
|
||
|
||
# 时间戳
|
||
created_at = Column(DateTime, default=datetime.datetime.now, comment="创建时间")
|
||
|
||
# 关联关系
|
||
conversation = relationship("Conversation", back_populates="messages")
|