Forráskód Böngészése

【增加】AI Image 列表兼容 Midjourney 操作按钮

cherishsince 11 hónapja
szülő
commit
4c20593211

+ 10 - 0
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/vo/AiImageRespVO.java

@@ -1,8 +1,10 @@
 package cn.iocoder.yudao.module.ai.controller.admin.image.vo;
 
+import cn.iocoder.yudao.module.ai.client.vo.MidjourneyNotifyReqVO;
 import io.swagger.v3.oas.annotations.media.Schema;
 import lombok.Data;
 
+import java.util.List;
 import java.util.Map;
 
 @Schema(description = "管理后台 - 绘画 Response VO")
@@ -45,4 +47,12 @@ public class AiImageRespVO {
     @Schema(description = "绘制参数")
     private Map<String, String> options;
 
+    @Schema(description = "绘画 response")
+    private MidjourneyNotifyReqVO response;
+
+    @Schema(description = "mj 进度")
+    private String progress;
+
+    @Schema(description = "mj buttons 按钮")
+    private List<MidjourneyNotifyReqVO.Button> buttons;
 }

+ 19 - 0
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/image/AiImageDO.java

@@ -1,6 +1,7 @@
 package cn.iocoder.yudao.module.ai.dal.dataobject.image;
 
 import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
+import cn.iocoder.yudao.module.ai.client.vo.MidjourneyNotifyReqVO;
 import cn.iocoder.yudao.module.ai.dal.dataobject.model.AiChatModelDO;
 import cn.iocoder.yudao.module.ai.enums.image.AiImageStatusEnum;
 import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
@@ -13,6 +14,7 @@ import lombok.Data;
 import org.springframework.ai.openai.OpenAiImageOptions;
 import org.springframework.ai.stabilityai.api.StabilityAiImageOptions;
 
+import java.util.List;
 import java.util.Map;
 
 /**
@@ -92,6 +94,23 @@ public class AiImageDO extends BaseDO {
     @TableField(typeHandler = JacksonTypeHandler.class)
     private Map<String, String> options;
 
+    /**
+     * 绘画 response
+     */
+    @TableField(typeHandler = JacksonTypeHandler.class)
+    private MidjourneyNotifyReqVO response;
+
+    /**
+     * mj 进度
+     */
+    private String progress;
+
+    /**
+     * mj buttons 按钮
+     */
+    @TableField(typeHandler = JacksonTypeHandler.class)
+    private List<MidjourneyNotifyReqVO.Button> buttons;
+
     // TODO @芋艿:再瞅瞅
     /**
      * midjourney proxy 关联的 job id