fix design stream 与正常版本区分 部署

This commit is contained in:
zhouchengrong
2024-12-23 10:19:51 +08:00
parent 4896975c5b
commit cce606d2a8
2 changed files with 7 additions and 2 deletions

View File

@@ -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):
"""
创建一个具有以下参数的请求体:
示例参数: