From c8a2c8369fdb7bfd992c64b6fbbe2aa83ccede73 Mon Sep 17 00:00:00 2001
From: cherishsince <cherishsince@aliyun.com>
Date: Wed, 10 Apr 2024 20:02:50 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=B5=8B=E8=AF=95=E5=90=8D?=
 =?UTF-8?q?=E7=A7=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../MidjourneyInteractionsTests.java          | 24 +++++++++----------
 1 file changed, 12 insertions(+), 12 deletions(-)

diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/midjourney/MidjourneyInteractionsTests.java b/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/midjourney/MidjourneyInteractionsTests.java
index 10c9a0923c..1ed3afc1d7 100644
--- a/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/midjourney/MidjourneyInteractionsTests.java
+++ b/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/midjourney/MidjourneyInteractionsTests.java
@@ -1,11 +1,11 @@
 package cn.iocoder.yudao.framework.ai.midjourney;
 
 import cn.hutool.core.io.FileUtil;
-import cn.iocoder.yudao.framework.ai.midjourney.interactions.MidjourneyInteractions;
-import cn.iocoder.yudao.framework.ai.midjourney.vo.Attachments;
-import cn.iocoder.yudao.framework.ai.midjourney.vo.Describe;
-import cn.iocoder.yudao.framework.ai.midjourney.vo.ReRoll;
-import cn.iocoder.yudao.framework.ai.midjourney.vo.UploadAttachmentsRes;
+import cn.iocoder.yudao.framework.ai.midjourney.api.MidjourneyInteractionsApi;
+import cn.iocoder.yudao.framework.ai.midjourney.api.req.AttachmentsReq;
+import cn.iocoder.yudao.framework.ai.midjourney.api.req.DescribeReq;
+import cn.iocoder.yudao.framework.ai.midjourney.api.req.ReRollReq;
+import cn.iocoder.yudao.framework.ai.midjourney.api.res.UploadAttachmentsRes;
 import com.alibaba.fastjson.JSON;
 import org.junit.Before;
 import org.junit.Test;
@@ -38,24 +38,24 @@ public class MidjourneyInteractionsTests {
 
     @Test
     public void mjImageTest() {
-        MidjourneyInteractions mjImagineInteractions = new MidjourneyInteractions(midjourneyConfig);
+        MidjourneyInteractionsApi mjImagineInteractions = new MidjourneyInteractionsApi(midjourneyConfig);
         mjImagineInteractions.imagine("童话里应该是什么样子？");
     }
 
 
     @Test
     public void reRollTest() {
-        MidjourneyInteractions mjImagineInteractions = new MidjourneyInteractions(midjourneyConfig);
-        mjImagineInteractions.reRoll(new ReRoll()
+        MidjourneyInteractionsApi mjImagineInteractions = new MidjourneyInteractionsApi(midjourneyConfig);
+        mjImagineInteractions.reRoll(new ReRollReq()
                 .setMessageId("1226165117448753243")
                 .setCustomId("MJ::JOB::upsample::3::2aeefbef-43e2-4057-bcf1-43b5f39ab6f7"));
     }
 
     @Test
     public void uploadAttachmentsTest() {
-        MidjourneyInteractions mjImagineInteractions = new MidjourneyInteractions(midjourneyConfig);
+        MidjourneyInteractionsApi mjImagineInteractions = new MidjourneyInteractionsApi(midjourneyConfig);
         UploadAttachmentsRes res = mjImagineInteractions.uploadAttachments(
-                new Attachments().setFileSystemResource(
+                new AttachmentsReq().setFileSystemResource(
                         new FileSystemResource(new File("/Users/fansili/Downloads/DSC01402.JPG")))
         );
         System.err.println(JSON.toJSONString(res));
@@ -63,8 +63,8 @@ public class MidjourneyInteractionsTests {
 
     @Test
     public void describeTest() {
-        MidjourneyInteractions mjImagineInteractions = new MidjourneyInteractions(midjourneyConfig);
-        mjImagineInteractions.describe(new Describe()
+        MidjourneyInteractionsApi mjImagineInteractions = new MidjourneyInteractionsApi(midjourneyConfig);
+        mjImagineInteractions.describe(new DescribeReq()
                 .setFileName("DSC01402.JPG")
                 .setFinalFileName("16826931-2873-45ec-8cfb-0ad81f1a075f/DSC01402.JPG")
         );