feat（新功能）:

fix（修复bug）:
docs（文档变更）:
refactor（重构）:
test(增加测试): Agent generate test

app/api/api_agent_generate_image.py

@@ -1,6 +1,8 @@
+import io
 import logging
 
 from fastapi import APIRouter, HTTPException
+from starlette.responses import StreamingResponse
 
 from app.schemas.response_template import ResponseModel
 from app.service.generate_image.agent_generate import GenerateImage
@@ -11,10 +13,7 @@ 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 data
+    server = GenerateImage()
+    byte_stream = server.get_result(prompt)
+    # 返回流式响应
+    return StreamingResponse(byte_stream, media_type="image/png")