未来
2bd8b61e91
新功能: - 大纲/章节生成改为服务端后台任务，支持断线续传 - 后台任务队列排队执行，按用户排队（同用户串行不同用户并发） - 章节管理页面添加后台任务列表弹窗和进度面板 - 章节状态添加 pending（待处理）选项 - 集成json5容错解析器 + 上下文感知JSON修复 - SSE流式生成添加心跳保活，防止连接超时 - SSEPostClient添加credentials:include修复network error - 每章最大伏笔数从2调整为5 - 添加大纲读区伏笔的功能 Bug修复: - 修复AI生成JSON中未转义引号/中文标点/多对象属性值未合并 - 修复JSON非法转义字符清洗和中文引号处理 - 修复MCP插件TimeoutError/连接失败上下文清理 - MCP插件后台注册添加重试机制 - 续写模式添加缺失的mcp_references参数 - 修复Alembic迁移链分叉 - 使用torch CPU版本加速Docker构建
46 lines
2.2 KiB
Python
46 lines
2.2 KiB
Python
"""后台任务数据模型 - 用于长时间运行的AI生成任务"""
|
|
from sqlalchemy import Column, String, Integer, DateTime, Boolean, JSON, Text
|
|
from sqlalchemy.sql import func
|
|
from app.database import Base
|
|
import uuid
|
|
|
|
|
|
class BackgroundTask(Base):
|
|
"""后台任务表 - 追踪所有长时间运行的生成任务"""
|
|
__tablename__ = "background_tasks"
|
|
|
|
id = Column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
|
|
user_id = Column(String(100), nullable=False, index=True, comment="用户ID")
|
|
project_id = Column(String(36), nullable=False, index=True, comment="项目ID")
|
|
|
|
# 任务类型
|
|
task_type = Column(String(50), nullable=False, comment="任务类型: outline_new/outline_continue/outline_expand/chapter_generate/chapter_batch/wizard")
|
|
|
|
# 任务状态
|
|
status = Column(String(20), default="pending", comment="任务状态: pending/running/completed/failed/cancelled")
|
|
progress = Column(Integer, default=0, comment="进度百分比(0-100)")
|
|
status_message = Column(String(500), comment="当前状态消息")
|
|
|
|
# 任务输入/输出
|
|
task_input = Column(JSON, comment="任务输入参数(JSON)")
|
|
task_result = Column(JSON, comment="任务结果(JSON)")
|
|
error_message = Column(Text, comment="错误信息")
|
|
|
|
# 进度详情（用于前端展示实时进度）
|
|
progress_details = Column(JSON, comment="进度详情: {stage, message, word_count, etc.}")
|
|
|
|
# 取消支持
|
|
cancel_requested = Column(Boolean, default=False, comment="是否请求取消")
|
|
|
|
# 重试信息
|
|
retry_count = Column(Integer, default=0, comment="已重试次数")
|
|
max_retries = Column(Integer, default=3, comment="最大重试次数")
|
|
|
|
# 时间记录
|
|
created_at = Column(DateTime, server_default=func.now(), comment="创建时间")
|
|
started_at = Column(DateTime, comment="开始时间")
|
|
completed_at = Column(DateTime, comment="完成时间")
|
|
updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now(), comment="更新时间")
|
|
|
|
def __repr__(self):
|
|
return f"<BackgroundTask(id={self.id[:8]}, type={self.task_type}, status={self.status}, progress={self.progress}%)>" |