feat(新功能):
fix(修复bug): docs(文档变更): refactor(重构): test(增加测试): Agent generate test
This commit is contained in:
20
app/api/api_agent_generate_image.py
Normal file
20
app/api/api_agent_generate_image.py
Normal file
@@ -0,0 +1,20 @@
|
||||
import logging
|
||||
|
||||
from fastapi import APIRouter, HTTPException
|
||||
|
||||
from app.schemas.response_template import ResponseModel
|
||||
from app.service.generate_image.agent_generate import GenerateImage
|
||||
|
||||
router = APIRouter()
|
||||
logger = logging.getLogger()
|
||||
|
||||
|
||||
@router.get("/agent_generate_image")
|
||||
def generate_image(prompt: str):
|
||||
try:
|
||||
server = GenerateImage()
|
||||
data = server.get_result(prompt)
|
||||
except Exception as e:
|
||||
logger.warning(f"generate_image Run Exception @@@@@@:{e}")
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
return ResponseModel(data=data)
|
||||
@@ -12,6 +12,8 @@ from app.api import api_mannequins_edit
|
||||
from app.api import api_prompt_generation
|
||||
from app.api import api_recommendation
|
||||
from app.api import api_super_resolution
|
||||
from app.api import api_agent_generate_image
|
||||
|
||||
from app.api import api_test
|
||||
|
||||
router = APIRouter()
|
||||
@@ -30,3 +32,4 @@ router.include_router(api_query_image.router, tags=['api_query_image'], prefix="
|
||||
router.include_router(api_brand_dna.router, tags=['api_brand_dna'], prefix="/api")
|
||||
router.include_router(api_recommendation.router, tags=['api_recommendation'], prefix="/api")
|
||||
router.include_router(api_mannequins_edit.router, tags=['api_mannequins_edit'], prefix="/api")
|
||||
router.include_router(api_agent_generate_image.router, tags=['api_agent_generate_image'], prefix="/api")
|
||||
|
||||
Reference in New Issue
Block a user