From 99d5e7c5094875bd379f7cd98d05d387677eccae Mon Sep 17 00:00:00 2001
From: YunaiV <zhijiantianya@gmail.com>
Date: Tue, 11 Mar 2025 12:53:34 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90=E4=BE=9D=E8=B5=96=E5=8D=87=E7=BA=A7?=
 =?UTF-8?q?=E3=80=91AI=EF=BC=9Aspring-ai-alibaba-starter=20=E5=8D=87?=
 =?UTF-8?q?=E7=BA=A7=E5=88=B0=E6=9C=80=E6=96=B0=EF=BC=8C=E8=A7=A3=E5=86=B3?=
 =?UTF-8?q?=E6=B5=81=E5=BC=8F=E8=BF=94=E5=9B=9E=E6=8A=A5=E9=94=99=E7=9A=84?=
 =?UTF-8?q?=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 yudao-module-ai/yudao-spring-boot-starter-ai/pom.xml           | 2 +-
 .../java/cn/iocoder/yudao/framework/ai/core/util/AiUtils.java  | 3 +--
 .../iocoder/yudao/framework/ai/chat/TongYiChatModelTests.java  | 1 +
 .../yudao/framework/ai/image/TongYiImagesModelTest.java        | 3 +--
 4 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/pom.xml b/yudao-module-ai/yudao-spring-boot-starter-ai/pom.xml
index 2d1a0af3aa..5a10c137f9 100644
--- a/yudao-module-ai/yudao-spring-boot-starter-ai/pom.xml
+++ b/yudao-module-ai/yudao-spring-boot-starter-ai/pom.xml
@@ -49,7 +49,7 @@
             <!-- 通义千问 -->
             <groupId>com.alibaba.cloud.ai</groupId>
             <artifactId>spring-ai-alibaba-starter</artifactId>
-            <version>1.0.0-M5.1</version>
+            <version>${spring-ai.version}.1</version>
         </dependency>
         <dependency>
             <!-- 文心一言 -->
diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/util/AiUtils.java b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/util/AiUtils.java
index a96d305492..cb968c3ffc 100644
--- a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/util/AiUtils.java
+++ b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/util/AiUtils.java
@@ -22,8 +22,7 @@ public class AiUtils {
         // noinspection EnhancedSwitchMigration
         switch (platform) {
             case TONG_YI:
-                // TODO @芋艿：tongyi 暂时没 maxTokens 选项
-                return DashScopeChatOptions.builder().withModel(model).withTemperature(temperature).build();
+                return DashScopeChatOptions.builder().withModel(model).withTemperature(temperature).withMaxToken(maxTokens).build();
             case YI_YAN:
                 return QianFanChatOptions.builder().model(model).temperature(temperature).maxTokens(maxTokens).build();
             case ZHI_PU:
diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/chat/TongYiChatModelTests.java b/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/chat/TongYiChatModelTests.java
index c632acf20f..b51d556a36 100644
--- a/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/chat/TongYiChatModelTests.java
+++ b/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/chat/TongYiChatModelTests.java
@@ -28,6 +28,7 @@ public class TongYiChatModelTests {
                     .withModel("qwen1.5-72b-chat") // 模型
 //                    .withModel("deepseek-r1") // 模型（deepseek-r1）
 //                    .withModel("deepseek-v3") // 模型（deepseek-v3）
+//                    .withModel("deepseek-r1-distill-qwen-1.5b") // 模型（deepseek-r1-distill-qwen-1.5b）
                     .build());
 
     @Test
diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/TongYiImagesModelTest.java b/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/TongYiImagesModelTest.java
index cc62f02fb6..ad4faaa46b 100644
--- a/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/TongYiImagesModelTest.java
+++ b/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/TongYiImagesModelTest.java
@@ -3,7 +3,6 @@ package cn.iocoder.yudao.framework.ai.image;
 import com.alibaba.cloud.ai.dashscope.api.DashScopeImageApi;
 import com.alibaba.cloud.ai.dashscope.image.DashScopeImageModel;
 import com.alibaba.cloud.ai.dashscope.image.DashScopeImageOptions;
-import com.alibaba.dashscope.aigc.imagesynthesis.ImageSynthesis;
 import org.junit.jupiter.api.Disabled;
 import org.junit.jupiter.api.Test;
 import org.springframework.ai.image.ImageOptions;
@@ -25,7 +24,7 @@ public class TongYiImagesModelTest {
     public void imageCallTest() {
         // 准备参数
         ImageOptions options = DashScopeImageOptions.builder()
-                .withModel(ImageSynthesis.Models.WANX_V1)
+                .withModel("wanx-v1")
                 .withHeight(256).withWidth(256)
                 .build();
         ImagePrompt prompt = new ImagePrompt("中国长城!", options);