feat(新功能):

fix(修复bug):
docs(文档变更):
refactor(重构):
test(增加测试): Agent generate test
This commit is contained in:
zhouchengrong
2025-03-13 12:04:14 +08:00
parent fc982f6f04
commit e94d5ac6a3
3 changed files with 91 additions and 0 deletions

View 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)