1.0.0初始化源代码

backend/app/models/interaction.py

@@ -0,0 +1,68 @@
+"""
+互动记录模型
+"""
+from sqlalchemy import Column, Integer, String, DateTime, Enum, Text, ForeignKey, Boolean, Float
+from sqlalchemy.sql import func
+from sqlalchemy.orm import relationship
+import enum
+
+from .base import Base
+
+
+class InteractionType(str, enum.Enum):
+    """互动类型枚举"""
+    COMMENT = "comment"  # 评论
+    REPLY = "reply"  # 回复
+    LIKE = "like"  # 点赞
+    FAVORITE = "favorite"  # 收藏
+    SHARE = "share"  # 转发
+
+
+class InteractionStatus(str, enum.Enum):
+    """互动状态枚举"""
+    PENDING = "pending"  # 待执行
+    SUCCESS = "success"  # 成功
+    FAILED = "failed"  # 失败
+    RETRYING = "retrying"  # 重试中
+
+
+class InteractionRecord(Base):
+    """互动记录表"""
+    __tablename__ = "interaction_records"
+    
+    id = Column(Integer, primary_key=True, autoincrement=True, comment="记录 ID")
+    
+    # 关联信息
+    virtual_user_id = Column(Integer, ForeignKey("virtual_users.id"), nullable=False, index=True, comment="虚拟用户 ID")
+    virtual_user = relationship("VirtualUser", backref="interaction_records")
+    
+    news_id = Column(String(100), nullable=False, index=True, comment="新闻 ID")
+    news_title = Column(String(500), comment="新闻标题")
+    
+    # 互动内容
+    interaction_type = Column(Enum(InteractionType), nullable=False, comment="互动类型")
+    content = Column(Text, comment="互动内容（评论/回复的文本）")
+    target_comment_id = Column(String(100), comment="目标评论 ID（回复时使用）")
+    
+    # 执行状态
+    status = Column(Enum(InteractionStatus), default=InteractionStatus.PENDING, comment="执行状态")
+    retry_count = Column(Integer, default=0, comment="重试次数")
+    error_message = Column(Text, comment="错误信息（失败时）")
+    
+    # AI 相关信息
+    ai_model_used = Column(String(100), comment="使用的 AI 模型")
+    tokens_used = Column(Integer, default=0, comment="消耗的 Token 数")
+    prompt_content = Column(Text, comment="发送给 AI 的提示词")
+    ai_response = Column(Text, comment="AI 返回的内容")
+    
+    # 接口响应
+    api_response = Column(Text, comment="会会接口返回的原始响应")
+    api_request_id = Column(String(200), comment="接口请求 ID")
+    
+    # 时间戳
+    execution_time = Column(DateTime, server_default=func.now(), comment="执行时间")
+    created_at = Column(DateTime, server_default=func.now(), comment="创建时间")
+    updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now(), comment="更新时间")
+    
+    def __repr__(self):
+        return f"<InteractionRecord(id={self.id}, user_id={self.virtual_user_id}, type='{self.interaction_type}', status='{self.status}')>"