From cea888b225021e5be845ee8af21cb6d672dc8f5c Mon Sep 17 00:00:00 2001
From: cherishsince <cherishsince@aliyun.com>
Date: Tue, 4 Jun 2024 17:39:02 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90=E4=BC=98=E5=8C=96=E3=80=91MidjourneyN?=
 =?UTF-8?q?otifyReqVO=20action=20=E5=92=8C=20status=20=E6=8E=A5=E5=8F=97?=
 =?UTF-8?q?=20=E6=94=B9=E4=B8=BA=20string?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ai/client/vo/MidjourneyNotifyReqVO.java   | 38 ++++++++++++++++---
 1 file changed, 32 insertions(+), 6 deletions(-)

diff --git a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/client/vo/MidjourneyNotifyReqVO.java b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/client/vo/MidjourneyNotifyReqVO.java
index 4398a8272..2f8c280f3 100644
--- a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/client/vo/MidjourneyNotifyReqVO.java
+++ b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/client/vo/MidjourneyNotifyReqVO.java
@@ -1,11 +1,16 @@
 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;
+    }
 }