From bdcf32dce8fc7e3afc53b0c3359ca178f121986d Mon Sep 17 00:00:00 2001 From: lt Date: Fri, 1 May 2026 14:33:00 +0800 Subject: [PATCH] feat: add Case, TaskLog, LLMConfig, SystemConfig models --- platform/backend/app/models.py | 109 +++++++++++++++++++++++++++++++++ 1 file changed, 109 insertions(+) diff --git a/platform/backend/app/models.py b/platform/backend/app/models.py index 976e9bb..32f8e9b 100644 --- a/platform/backend/app/models.py +++ b/platform/backend/app/models.py @@ -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, + }