diff --git a/app/api/api_design.py b/app/api/api_design.py index f12e170..ee4a651 100644 --- a/app/api/api_design.py +++ b/app/api/api_design.py @@ -4,7 +4,7 @@ import os from fastapi import APIRouter, HTTPException, UploadFile, File, Form, BackgroundTasks -from app.schemas.design import DesignModel, DesignProgressModel, ModelProgressModel, DBGConfigModel +from app.schemas.design import DesignModel, DesignProgressModel, ModelProgressModel, DBGConfigModel, DesignStreamModel from app.schemas.response_template import ResponseModel from app.service.design.model_process_service import model_transpose from app.service.design_batch.service import start_design_batch_generate @@ -197,7 +197,7 @@ def design(request_data: DesignModel, background_tasks: BackgroundTasks): @router.post("/design_v2") -async def design_v2(request_data: DesignModel, background_tasks: BackgroundTasks): +async def design_v2(request_data: DesignStreamModel, background_tasks: BackgroundTasks): """ 创建一个具有以下参数的请求体: 示例参数: diff --git a/app/schemas/design.py b/app/schemas/design.py index dab80d2..98d0a29 100644 --- a/app/schemas/design.py +++ b/app/schemas/design.py @@ -4,6 +4,11 @@ from pydantic import BaseModel class DesignModel(BaseModel): objects: list[dict] process_id: str + + +class DesignStreamModel(BaseModel): + objects: list[dict] + process_id: str requestId: str