From cce606d2a83dd1e316a0d2b02127e04545d429db Mon Sep 17 00:00:00 2001
From: zhouchengrong <Zchengrong@yeah.net>
Date: Mon, 23 Dec 2024 10:19:51 +0800
Subject: [PATCH] =?UTF-8?q?fix=20design=20stream=20=E4=B8=8E=E6=AD=A3?=
 =?UTF-8?q?=E5=B8=B8=E7=89=88=E6=9C=AC=E5=8C=BA=E5=88=86=20=E9=83=A8?=
 =?UTF-8?q?=E7=BD=B2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/api/api_design.py | 4 ++--
 app/schemas/design.py | 5 +++++
 2 files changed, 7 insertions(+), 2 deletions(-)

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