fix design stream 与正常版本区分 部署
This commit is contained in:
@@ -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):
|
||||
"""
|
||||
创建一个具有以下参数的请求体:
|
||||
示例参数:
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user