194 lines
7.1 KiB
Python
194 lines
7.1 KiB
Python
"""
|
||
app/api/v1/fission.py — 第11环裂变路由（对齐上线版 split.js：一次出N套）
|
||
|
||
POST /fission 触发裂变(1源任务→一次LLM出N套完整笔记包)
|
||
GET /fission/{id} 查裂变进度(聚合 FissionNote 落库/生图状态)
|
||
GET /fission/{id}/notes 取 N 套完整笔记包(独立展示页用)
|
||
工作台内呈现，N 套笔记落 FissionNote 表，非扇出 GenerationTask。
|
||
"""
|
||
import json
|
||
import logging
|
||
from typing import Annotated
|
||
|
||
from fastapi import APIRouter, Depends
|
||
from fastapi.responses import FileResponse
|
||
from pydantic import BaseModel, Field
|
||
from sqlalchemy.orm import Session
|
||
|
||
from app.core.database import get_db
|
||
from app.core.response import ok, raise_business, raise_not_found
|
||
from app.middleware.workspace_guard import CurrentUser, require_write_permission
|
||
from app.models.fission import FissionTask, FissionNote
|
||
|
||
logger = logging.getLogger(__name__)
|
||
router = APIRouter(tags=["fission"])
|
||
|
||
|
||
class FissionCreate(BaseModel):
|
||
source_task_id: int
|
||
reference_level: str = Field(default="mid") # low/mid/high
|
||
fanout_count: int = Field(default=3, ge=1, le=5)
|
||
|
||
|
||
@router.post("/fission")
|
||
def create_fission_endpoint(
|
||
body: FissionCreate,
|
||
current_user: Annotated[CurrentUser, Depends(require_write_permission)] = None,
|
||
db: Session = Depends(get_db),
|
||
):
|
||
"""触发裂变：从源任务一次出 N 套完整笔记包。返回 fission_id。"""
|
||
from app.services.fission_service import create_fission
|
||
fission_id, _ = create_fission(
|
||
db, current_user, body.source_task_id, body.reference_level, body.fanout_count,
|
||
)
|
||
return ok({"fission_id": fission_id,
|
||
"reference_level": body.reference_level, "fanout_count": body.fanout_count})
|
||
|
||
|
||
@router.get("/fission/{fission_id}")
|
||
def get_fission_progress(
|
||
fission_id: int,
|
||
current_user: Annotated[CurrentUser, Depends(require_write_permission)] = None,
|
||
db: Session = Depends(get_db),
|
||
):
|
||
"""查裂变进度：聚合 FissionNote 落库/生图状态（x/N 完成）。"""
|
||
ft = db.query(FissionTask).filter(
|
||
FissionTask.id == fission_id,
|
||
FissionTask.workspace_id == current_user.workspace_id,
|
||
).first()
|
||
if not ft:
|
||
raise_not_found("裂变任务不存在")
|
||
|
||
notes = db.query(FissionNote).filter(
|
||
FissionNote.fission_id == fission_id,
|
||
FissionNote.workspace_id == current_user.workspace_id,
|
||
).all()
|
||
done = sum(1 for n in notes if n.status == "image_done")
|
||
failed = sum(1 for n in notes if n.status == "failed")
|
||
total = len(notes)
|
||
# FissionTask.status 由 pipeline 维护(generating→completed/failed)；此处只读不改，
|
||
# 避免 API 轮询与 worker 写状态打架。进度数 = 已落库套数的生图完成度。
|
||
return ok({
|
||
"fission_id": ft.id, "reference_level": ft.reference_level,
|
||
"fanout_count": ft.fanout_count, "status": ft.status,
|
||
"progress": {"done": done, "failed": failed, "total": total},
|
||
})
|
||
|
||
|
||
@router.get("/fission/{fission_id}/notes")
|
||
def get_fission_notes(
|
||
fission_id: int,
|
||
current_user: Annotated[CurrentUser, Depends(require_write_permission)] = None,
|
||
db: Session = Depends(get_db),
|
||
):
|
||
"""取 N 套完整笔记包（独立展示页用）：每套含文案+标签+imagePlan+图+分数。"""
|
||
ft = db.query(FissionTask).filter(
|
||
FissionTask.id == fission_id,
|
||
FissionTask.workspace_id == current_user.workspace_id,
|
||
).first()
|
||
if not ft:
|
||
raise_not_found("裂变任务不存在")
|
||
|
||
rows = db.query(FissionNote).filter(
|
||
FissionNote.fission_id == fission_id,
|
||
FissionNote.workspace_id == current_user.workspace_id,
|
||
).order_by(FissionNote.seq.asc()).all()
|
||
|
||
notes = []
|
||
for r in rows:
|
||
try:
|
||
note = json.loads(r.note_json) if r.note_json else {}
|
||
except (ValueError, TypeError):
|
||
note = {}
|
||
try:
|
||
images = json.loads(r.images_json) if r.images_json else []
|
||
except (ValueError, TypeError):
|
||
images = []
|
||
# 内部评分中间字段(_score/_passed/_block)不外泄，前端用平铺的 score/passed
|
||
for k in ("_score", "_passed", "_block"):
|
||
note.pop(k, None)
|
||
notes.append({
|
||
"seq": r.seq, "note": note, "images": images,
|
||
"score": r.score, "passed": bool(r.passed),
|
||
"needs_optimization": bool(r.needs_optimization),
|
||
"dimension": r.dimension, "status": r.status,
|
||
})
|
||
|
||
return ok({"fission_id": ft.id, "status": ft.status,
|
||
"fanout_count": ft.fanout_count, "notes": notes})
|
||
|
||
|
||
@router.get("/fission/{fission_id}/download")
|
||
def download_fission_package(
|
||
fission_id: int,
|
||
current_user: Annotated[CurrentUser, Depends(require_write_permission)] = None,
|
||
db: Session = Depends(get_db),
|
||
):
|
||
"""下载裂变结果：把 N 套裂变笔记打成 zip（文案+图片+发布清单）。"""
|
||
ft = db.query(FissionTask).filter(
|
||
FissionTask.id == fission_id,
|
||
FissionTask.workspace_id == current_user.workspace_id,
|
||
).first()
|
||
if not ft:
|
||
raise_not_found("裂变任务不存在")
|
||
|
||
rows = db.query(FissionNote).filter(
|
||
FissionNote.fission_id == fission_id,
|
||
FissionNote.workspace_id == current_user.workspace_id,
|
||
).order_by(FissionNote.seq.asc()).all()
|
||
if not rows:
|
||
raise_business("裂变结果为空，暂不能下载")
|
||
|
||
from app.services.fission_package_exporter import build_fission_package
|
||
try:
|
||
zip_path = build_fission_package(ft, rows)
|
||
except ValueError as exc:
|
||
raise_business(str(exc))
|
||
|
||
return FileResponse(
|
||
path=zip_path,
|
||
media_type="application/zip",
|
||
filename=f"fission-{fission_id}.zip",
|
||
)
|
||
|
||
|
||
@router.post("/fission/{fission_id}/notes/{seq}/retry-images")
|
||
def retry_fission_note_images_endpoint(
|
||
fission_id: int,
|
||
seq: int,
|
||
current_user: Annotated[CurrentUser, Depends(require_write_permission)] = None,
|
||
db: Session = Depends(get_db),
|
||
):
|
||
"""补图：对某一套只重生标了 error 的失败分镜（中转站波动兜底）。
|
||
|
||
异步丢 Celery；前端轮询 GET /fission/{id}/notes 看 images 更新。
|
||
"""
|
||
ft = db.query(FissionTask).filter(
|
||
FissionTask.id == fission_id,
|
||
FissionTask.workspace_id == current_user.workspace_id,
|
||
).first()
|
||
if not ft:
|
||
raise_not_found("裂变任务不存在")
|
||
|
||
fn = db.query(FissionNote).filter(
|
||
FissionNote.fission_id == fission_id,
|
||
FissionNote.workspace_id == current_user.workspace_id,
|
||
FissionNote.seq == seq,
|
||
).first()
|
||
if not fn:
|
||
raise_not_found("该套笔记不存在")
|
||
|
||
try:
|
||
images = json.loads(fn.images_json) if fn.images_json else []
|
||
except (ValueError, TypeError):
|
||
images = []
|
||
failed = [img.get("role") for img in images if img.get("error")]
|
||
if not failed:
|
||
return ok({"fission_id": fission_id, "seq": seq,
|
||
"queued": False, "message": "该套无失败分镜，无需补图"})
|
||
|
||
from app.workers.tasks import retry_fission_images
|
||
retry_fission_images.delay(fission_id, fn.id)
|
||
return ok({"fission_id": fission_id, "seq": seq, "note_id": fn.id,
|
||
"queued": True, "failed_roles": failed})
|