aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

4
app/api/api_design.py
View File

@@ -4,7 +4,7 @@ import os
from fastapi import APIRouter, HTTPException, UploadFile, File, Form, BackgroundTasks 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.schemas.response_template import ResponseModel
from app.service.design.model_process_service import model_transpose from app.service.design.model_process_service import model_transpose
from app.service.design_batch.service import start_design_batch_generate 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") @router.post("/design_v2")
async def design_v2(request_data: DesignModel, background_tasks: BackgroundTasks): async def design_v2(request_data: DesignStreamModel, background_tasks: BackgroundTasks):
""" """
创建一个具有以下参数的请求体: 创建一个具有以下参数的请求体:
示例参数： 示例参数：

5
app/schemas/design.py
View File

@@ -4,6 +4,11 @@ from pydantic import BaseModel
class DesignModel(BaseModel): class DesignModel(BaseModel):
objects: list[dict] objects: list[dict]
process_id: str process_id: str
class DesignStreamModel(BaseModel):
objects: list[dict]
process_id: str
requestId: str requestId: str