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)
|
||||
Reference in New Issue
Block a user