From 90f920f55b5f7f5350f5e295ac32e81ea1e39858 Mon Sep 17 00:00:00 2001
From: cherishsince <cherishsince@aliyun.com>
Date: Wed, 5 Jun 2024 17:15:06 +0800
Subject: [PATCH] 【优化】midjourney action 重置 buttons

---
 yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/AiImageController.java | 2 +-
 yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/image/AiImageServiceImpl.java         | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/AiImageController.java b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/AiImageController.java
index 443b522..578d60a 100644
--- a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/AiImageController.java
+++ b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/AiImageController.java
@@ -79,7 +79,7 @@ public class AiImageController {
     }
 
     @Operation(summary = "midjourney - action(放大、缩小、U1、U2...)")
-    @PostMapping("/midjourney/action")
+    @GetMapping("/midjourney/action")
     public CommonResult<Boolean> midjourneyAction(@RequestParam("id") Long imageId,
                                                   @RequestParam("customId") String customId) {
         return success(imageService.midjourneyAction(getLoginUserId(), imageId, customId));
diff --git a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/image/AiImageServiceImpl.java b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/image/AiImageServiceImpl.java
index e4909f9..e64ff5b 100644
--- a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/image/AiImageServiceImpl.java
+++ b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/image/AiImageServiceImpl.java
@@ -258,6 +258,7 @@ public class AiImageServiceImpl implements AiImageService {
         newImage.setProgress(null);
         newImage.setTaskId(submitRespVO.getResult());
         newImage.setErrorMessage(null);
+        newImage.setButtons(null);
         // 4.2、保存数据库
         imageMapper.insert(newImage);
         return Boolean.TRUE;
--
libgit2 0.21.4