feat: 添加新的模型，删除后端数据库

server/api/generate.py

@@ -1,11 +1,9 @@
-"""无状态套图生成：请求自带采集数据，不落商品库。"""
+"""无状态套图生成：请求自带采集数据，任务存进程内注册表。"""
 from __future__ import annotations
 
-from fastapi import APIRouter, BackgroundTasks, Depends, HTTPException
+from fastapi import APIRouter, BackgroundTasks, HTTPException
 
 from config import get_settings
-from db import get_db
-from models import Suite
 from schemas import (
     GenerateRequest, PLATFORM_SPECS, SUPPORTED_TYPES, TONGYI_MODELS, RIGHTAPI_MODELS,
     SuiteCreateResponse, TextMaterial, resolve_provider,
@@ -14,6 +12,7 @@ from schemas import (
 from services.generator import run_suite
 from services.planner import generate_plan
 from services.prompt import type_name
+from services.tasks import create_task
 
 router = APIRouter(prefix="/api", tags=["generate"])
 
@@ -48,7 +47,6 @@ def texts_to_raw(texts: list[TextMaterial]) -> dict:
 async def generate_suite(
     req: GenerateRequest,
     background: BackgroundTasks,
-    db=Depends(get_db),
 ) -> SuiteCreateResponse:
     if not req.images:
         raise HTTPException(status_code=400, detail="未勾选任何图片，无法生成")
@@ -72,7 +70,6 @@ async def generate_suite(
                 })
         if not jobs:
             raise HTTPException(status_code=400, detail="方案中所有项的数量都是 0")
-        types = list(dict.fromkeys(j["kind"] for j in jobs))
     else:
         types = req.types or ["white_bg", "key_features", "lifestyle", "multi_scene"]
         bad = [t for t in types if t not in SUPPORTED_TYPES]
@@ -92,19 +89,20 @@ async def generate_suite(
         raise HTTPException(status_code=400, detail=f"不支持的模型: {model}（tongyi 支持: {TONGYI_MODELS}）")
     if provider_name == "rightapi" and model and model not in RIGHTAPI_MODELS:
         raise HTTPException(status_code=400, detail=f"不支持的模型: {model}（rightapi 支持: {RIGHTAPI_MODELS}）")
-    suite = Suite(
-        product_id=None,
-        style_set=req.style_set,
-        style_prompt=req.style_prompt,
-        requirements=req.requirements,
+
+    task = create_task(
+        status="pending",
         platform=req.platform,
         lang=spec["lang"],
         ratio=spec["ratio"],
-        types=types,
-        plan=jobs,
+        style_set=req.style_set,
+        style_prompt=req.style_prompt,
+        requirements=req.requirements,
         provider=provider_name,
         model=model,
+        total=len(jobs),
         context=texts_to_raw(req.texts),
+        plan=jobs,
         # 参考图池：main 组优先，其余组按序补充（variant 绑定靠 variant_name 匹配）
         ref_images=[
             {
@@ -115,12 +113,9 @@ async def generate_suite(
             for i in sorted(req.images, key=lambda x: 0 if x.group_key == "main" else 1)
         ],
     )
-    db.add(suite)
-    await db.commit()
-    await db.refresh(suite)
 
-    background.add_task(run_suite, str(suite.id))
-    return SuiteCreateResponse(suite_id=str(suite.id))
+    background.add_task(run_suite, task)
+    return SuiteCreateResponse(suite_id=task.id)
 
 
 @router.post("/plan", response_model=PlanResponse)