功能更新

yj

2025-07-23 1225b6cbf0a028b765a0ab6d784bcb80459a67bb

 app/models/conversation.py

@@ -1,26 +1,31 @@
"""
对话记录模型
"""
from sqlalchemy import Column, String, Integer, DateTime, Text, Boolean

from sqlalchemy import Column, String, Integer, DateTime, Text, Boolean, UniqueConstraint
from sqlalchemy.sql import func
from .database import Base


class Conversation(Base):
    """对话记录表"""

    __tablename__ = "conversations"
    
    id = Column(Integer, primary_key=True, index=True, autoincrement=True)
    user_conversation_key = Column(String(200), unique=True, index=True, nullable=False, 
                                 comment="用户对话唯一键(fromUser+conversation_id)")
    from_user = Column(String(100), index=True, nullable=False, comment="发送用户微信ID")
    from_user = Column(
        String(100), index=True, nullable=False, comment="发送用户微信ID"
    )
    conversation_id = Column(String(100), nullable=False, comment="Dify对话ID")
    group = Column(String(100), nullable=False, comment="群组ID")
    hour = Column(String(20), nullable=False, comment="小时标识(YYYYMMDD_HH)")
    
    # 用户问题
    user_question = Column(Text, nullable=False, comment="用户提问内容")
    
    # AI回答
    ai_answer = Column(Text, nullable=True, comment="AI回答内容")
    # 对话内容（JSON格式）
    content = Column(
        Text,
        nullable=True,
        comment='对话内容JSON格式，如[{"user":"你好","ai":"你好有什么能帮助你的"}]',
    )
    
    # 消息状态
    is_processed = Column(Boolean, default=False, comment="是否已处理")
@@ -32,7 +37,15 @@
    sent_time = Column(DateTime, nullable=True, comment="发送时间")
    
    created_at = Column(DateTime, default=func.now(), comment="创建时间")
    updated_at = Column(DateTime, default=func.now(), onupdate=func.now(), comment="更新时间")
    updated_at = Column(
        DateTime, default=func.now(), onupdate=func.now(), comment="更新时间"
    )

    # 添加四字段组合的唯一性约束
    __table_args__ = (
        UniqueConstraint('from_user', 'conversation_id', 'group', 'hour',
                         name='uq_conversation_key'),
    )
    
    def __repr__(self):
        return f"<Conversation(from_user='{self.from_user}', conversation_id='{self.conversation_id}')>"
        return f"<Conversation(from_user='{self.from_user}', conversation_id='{self.conversation_id}', group='{self.group}', hour='{self.hour}')>"