feat: 插件开发 ozon 端主体完成

server/api/generate.py

@@ -7,7 +7,7 @@ from config import get_settings
 from db import get_db
 from models import Suite
 from schemas import (
-    GenerateRequest, PLATFORM_SPECS, SUPPORTED_TYPES, SuiteCreateResponse, TextMaterial,
+    GenerateRequest, PLATFORM_SPECS, SUPPORTED_TYPES, TONGYI_MODELS, SuiteCreateResponse, TextMaterial,
     PlanRequest, PlanResponse, PlanItemOut,
 )
 from services.generator import run_suite
@@ -36,6 +36,10 @@ def texts_to_raw(texts: list[TextMaterial]) -> dict:
             raw["sellingPoints"] = t.content
         elif t.kind == "desc" and t.content:
             raw["desc"] = t.content
+        elif t.kind == "sales" and t.content:
+            raw["sales"] = t.content
+        elif t.kind == "shop" and t.content:
+            raw["shop"] = t.content
     return raw
 
 
@@ -80,15 +84,21 @@ async def generate_suite(
     spec = PLATFORM_SPECS[req.platform]
 
     settings = get_settings()
+    provider_name = req.provider or settings.image_provider
+    model = req.model
+    if provider_name == "tongyi" and model and model not in TONGYI_MODELS:
+        raise HTTPException(status_code=400, detail=f"不支持的模型: {model}（tongyi 支持: {TONGYI_MODELS}）")
     suite = Suite(
         product_id=None,
         style_set=req.style_set,
+        style_prompt=req.style_prompt,
         platform=req.platform,
         lang=spec["lang"],
         ratio=spec["ratio"],
         types=types,
         plan=jobs,
-        provider=req.provider or settings.image_provider,
+        provider=provider_name,
+        model=model,
         context=texts_to_raw(req.texts),
         # 参考图池：main 组优先，其余组按序补充（variant 绑定靠 variant_name 匹配）
         ref_images=[

server/api/suites.py

@@ -13,7 +13,7 @@ from sqlalchemy.ext.asyncio import AsyncSession
 from config import get_settings
 from db import get_db
 from models import Product, ProductAsset, Suite, SuiteImage, STATUS_OK
-from schemas import PLATFORM_SPECS, SUPPORTED_TYPES, SuiteCreateRequest, SuiteCreateResponse, SuiteImageOut, SuiteOut
+from schemas import PLATFORM_SPECS, SUPPORTED_TYPES, TONGYI_MODELS, SuiteCreateRequest, SuiteCreateResponse, SuiteImageOut, SuiteOut
 from services import storage
 from services.generator import run_suite
 
@@ -35,6 +35,7 @@ async def _suite_out(db: AsyncSession, suite: Suite) -> SuiteOut:
         ratio=suite.ratio,
         types=list(suite.types or []),
         provider=suite.provider,
+        model=suite.model,
         images=[
             SuiteImageOut(
                 type_id=i.type_id, name=i.name, url=i.stored_url or "",
@@ -75,6 +76,9 @@ async def create_suite(
     spec = PLATFORM_SPECS[req.platform]
 
     settings = get_settings()
+    provider_name = req.provider or settings.image_provider
+    if provider_name == "tongyi" and req.model and req.model not in TONGYI_MODELS:
+        raise HTTPException(status_code=400, detail=f"不支持的模型: {req.model}（tongyi 支持: {TONGYI_MODELS}）")
     suite = Suite(
         product_id=product.id,
         style_set=req.style_set,
@@ -82,7 +86,8 @@ async def create_suite(
         lang=spec["lang"],
         ratio=spec["ratio"],
         types=req.types,
-        provider=req.provider or settings.image_provider,
+        provider=provider_name,
+        model=req.model,
     )
     db.add(suite)
     await db.commit()