【优化】MidjourneyNotifyReqVO action 和 status 接受改为 string

cherishsince
1 parent 95d26c9b
Showing 1 changed file with 32 additions and 6 deletions
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/client/vo/MidjourneyNotifyReqVO.java
 package cn.iocoder.yudao.module.ai.client.vo;
  
-import cn.iocoder.yudao.module.ai.client.enums.MidjourneyTaskActionEnum;
-import cn.iocoder.yudao.module.ai.client.enums.MidjourneyTaskStatusEnum;
 import io.swagger.v3.oas.annotations.media.Schema;
 import lombok.Data;
  
+import java.util.List;
+
 /**
+ * Midjourney Proxy 通知回调
+ *
+ *  - Midjourney Proxy：通知回调 bean 是 com.github.novicezk.midjourney.support.Task
+ *  - 毫秒 api 通知回调文档地址：https://gpt-best.apifox.cn/doc-3530863
+ *
  * @author fansili
  * @time 2024/5/31 10:37
  * @since 1.0
@@ -16,10 +21,10 @@ public class MidjourneyNotifyReqVO {
     @Schema(description = "job id")
     private String id;
  
-    @Schema(description = "任务类型")
-    private MidjourneyTaskActionEnum action;
-    @Schema(description = "任务状态")
-    private MidjourneyTaskStatusEnum status;
+    @Schema(description = "任务类型 MidjourneyTaskActionEnum")
+    private String action;
+    @Schema(description = "任务状态 MidjourneyTaskStatusEnum")
+    private String status;
  
     @Schema(description = "提示词")
     private String prompt;
@@ -46,4 +51,25 @@ public class MidjourneyNotifyReqVO {
     @Schema(description = "失败原因")
     private String failReason;
  
+    @Schema(description = "任务完成后的可执行按钮")
+    private List<Button> buttons;
+
+    @Data
+    public static class Button {
+
+        @Schema(description = "MJ::JOB::upsample::1::85a4b4c1-8835-46c5-a15c-aea34fad1862 动作标识")
+        private String customId;
+
+        @Schema(description = "图标 emoji")
+        private String emoji;
+
+        @Schema(description = "Make Variations 文本")
+        private String label;
+
+        @Schema(description = "类型，系统内部使用")
+        private String type;
+
+        @Schema(description = "样式: 2（Primary）、3（Green）")
+        private String style;
+    }
 }