feat: add Case, TaskLog, LLMConfig, SystemConfig models

2026-05-01 14:33:00 +08:00
parent aa76f8203f
commit bdcf32dce8
1 changed files with 109 additions and 0 deletions
@@ -108,3 +108,112 @@ class PublishRecord(Base):

    # 关联查询
    # 可以添加外键关联到 Topic, 但这里保持简单
+
+
+class Case(Base):
+    __tablename__ = "cases"
+
+    id = Column(Integer, primary_key=True, index=True, autoincrement=True)
+    title = Column(String, nullable=False)
+    field = Column(String, nullable=False)  # 对应四大支柱或其子领域
+    summary = Column(Text)
+    key_metrics = Column(Text, nullable=True)  # 关键数据
+    date = Column(String, nullable=True)  # 日期或年份字符串
+    source = Column(String, nullable=True)
+    credibility_rating = Column(String, nullable=True)  # 如 "⭐⭐⭐"
+    china_applicability = Column(String, nullable=True)  # 如 "⭐⭐⭐⭐"
+    source_url = Column(String, nullable=True)
+    created_at = Column(DateTime(timezone=True), server_default=func.now())
+    updated_at = Column(DateTime(timezone=True), onupdate=func.now())
+
+    def to_dict(self):
+        return {
+            "id": self.id,
+            "title": self.title,
+            "field": self.field,
+            "summary": self.summary,
+            "key_metrics": self.key_metrics,
+            "date": self.date,
+            "source": self.source,
+            "credibility_rating": self.credibility_rating,
+            "china_applicability": self.china_applicability,
+            "source_url": self.source_url,
+            "created_at": self.created_at.isoformat() if self.created_at else None,
+            "updated_at": self.updated_at.isoformat() if self.updated_at else None,
+        }
+
+
+class TaskLog(Base):
+    __tablename__ = "task_logs"
+
+    id = Column(Integer, primary_key=True, index=True, autoincrement=True)
+    task_name = Column(String, nullable=False)  # collector, creator, research, outline, writer, optimizer
+    topic_id = Column(String, nullable=True)  # 关联选题ID
+    status = Column(String, nullable=False)  # started, completed, failed
+    message = Column(Text, nullable=True)
+    started_at = Column(DateTime(timezone=True), server_default=func.now())
+    finished_at = Column(DateTime(timezone=True), nullable=True)
+    duration = Column(Integer, nullable=True)  # 秒数
+
+    def to_dict(self):
+        return {
+            "id": self.id,
+            "task_name": self.task_name,
+            "topic_id": self.topic_id,
+            "status": self.status,
+            "message": self.message,
+            "started_at": self.started_at.isoformat() if self.started_at else None,
+            "finished_at": self.finished_at.isoformat() if self.finished_at else None,
+            "duration": self.duration,
+        }
+
+
+class LLMConfig(Base):
+    __tablename__ = "llm_configs"
+
+    id = Column(Integer, primary_key=True, index=True, autoincrement=True)
+    name = Column(String, unique=True, nullable=False)  # 如 default_expand
+    system_prompt = Column(Text, nullable=True)
+    user_prompt_template = Column(Text, nullable=False)  # 含占位符 {topic.get('title')} 等
+    temperature = Column(Float, default=0.7)
+    max_tokens = Column(Integer, default=2000)
+    model = Column(String, nullable=True)  # 后端模型名
+    is_active = Column(Boolean, default=True)
+    created_at = Column(DateTime(timezone=True), server_default=func.now())
+    updated_at = Column(DateTime(timezone=True), onupdate=func.now())
+
+    def to_dict(self):
+        return {
+            "id": self.id,
+            "name": self.name,
+            "system_prompt": self.system_prompt,
+            "user_prompt_template": self.user_prompt_template,
+            "temperature": self.temperature,
+            "max_tokens": self.max_tokens,
+            "model": self.model,
+            "is_active": self.is_active,
+            "created_at": self.created_at.isoformat() if self.created_at else None,
+            "updated_at": self.updated_at.isoformat() if self.updated_at else None,
+        }
+
+
+class SystemConfig(Base):
+    __tablename__ = "system_configs"
+
+    id = Column(Integer, primary_key=True, index=True, autoincrement=True)
+    key = Column(String, unique=True, nullable=False)
+    value = Column(Text, nullable=True)  # 可存储 JSON 字符串
+    description = Column(String, nullable=True)
+    created_at = Column(DateTime(timezone=True), server_default=func.now())
+    updated_at = Column(DateTime(timezone=True), onupdate=func.now())
+
+    def to_dict(self):
+        import json
+        return {
+            "id": self.id,
+            "key": self.key,
+            "value": json.loads(self.value) if self.value else None,
+            "description": self.description,
+            "created_at": self.created_at.isoformat() if self.created_at else None,
+            "updated_at": self.updated_at.isoformat() if self.updated_at else None,
+        }