feat（新功能）: 新增agent 工具,图片生成接口

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

app/api/api_generate_image.py

@@ -3,10 +3,11 @@ import logging
 
 from fastapi import APIRouter, BackgroundTasks, HTTPException
 
-from app.schemas.generate_image import GenerateImageModel, GenerateProductImageModel, GenerateSingleLogoImageModel, GenerateRelightImageModel, GenerateMultiViewModel, BatchGenerateProductImageModel, BatchGenerateRelightImageModel
+from app.schemas.generate_image import GenerateImageModel, GenerateProductImageModel, GenerateSingleLogoImageModel, GenerateRelightImageModel, GenerateMultiViewModel, BatchGenerateProductImageModel, BatchGenerateRelightImageModel, AgentTollGenerateImageModel
 from app.schemas.pose_transform import BatchPoseTransformModel
 from app.schemas.response_template import ResponseModel
 from app.service.generate_batch_image.service import start_product_batch_generate, start_relight_batch_generate, start_pose_transform_batch_generate
+from app.service.generate_image.service_agent_tool_generate_image import AgentToolGenerateImage
 from app.service.generate_image.service_generate_image import GenerateImage, infer_cancel as generate_image_infer_cancel
 from app.service.generate_image.service_generate_multi_view import GenerateMultiView, infer_cancel as generate_multi_view_cancel
 from app.service.generate_image.service_generate_product_image import GenerateProductImage, infer_cancel as generate_product_image_cancel
@@ -304,3 +305,49 @@ async def batch_generate_pose_transform(request_batch_item: BatchPoseTransformMo
         }
     """
     return await start_pose_transform_batch_generate(request_batch_item)
+
+
+"""agent tool"""
+
+
+@router.post("/agent_tool_generate_image")
+def agent_tool_generate_image(request_item: AgentTollGenerateImageModel, background_tasks: BackgroundTasks):
+    """
+        创建一个具有以下参数的请求体:
+        - **prompt**: 想要生成图片的描述词
+        - **category**: 生成图片的类别，sketch print 等等
+        - **gender**: 生成sketch专用，服装类别
+        - **version**: 使用模型版本 fast 或者 high
+        - **size**: 生成数量
+        - **version**: 使用模型版本 fast 或者 high
+
+
+        示例参数：
+        {
+          "prompt": "a single item of sketch of Wabi-sabi, skirt, tiered, 4k, white background",
+          "category": "sketch",
+          "gender": "male",
+          "size":2,
+          "version":"high"
+        }
+    """
+    try:
+        logger.info(f"agent_tool_generate_image request item is : @@@@@@:{request_item.dict()}")
+        request_data = request_item.dict()
+        service = AgentToolGenerateImage(request_data['version'])
+        image_url_list, clothing_category_list = service.get_result(
+            prompt=request_data['prompt'],
+            size=request_data['size'],
+            version=request_data['version'],
+            category=request_data['category'],
+            gender=request_data['gender']
+        )
+        data = {
+            "image_url_list": image_url_list,
+            "clothing_category_list": clothing_category_list
+        }
+        logger.info(f"agent_tool_generate_image response item is : @@@@@@:{data}")
+    except Exception as e:
+        logger.warning(f"agent_tool_generate_image Run Exception @@@@@@:{e}")
+        raise HTTPException(status_code=404, detail=str(e))
+    return ResponseModel(data=data)