From 24681c81c09022f584a57006f2534b5f74723414 Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期二, 30 六月 2026 09:27:31 +0800
Subject: [PATCH] 初始化项目
---
yudao-module-ai/src/test/java/cn/iocoder/yudao/module/ai/framework/ai/core/model/chat/YiYanChatModelTests.java | 46 ++++++++++++++++++++++++++++++----------------
1 files changed, 30 insertions(+), 16 deletions(-)
diff --git a/yudao-module-ai/src/test/java/cn/iocoder/yudao/module/ai/framework/ai/core/model/chat/YiYanChatModelTests.java b/yudao-module-ai/src/test/java/cn/iocoder/yudao/module/ai/framework/ai/core/model/chat/YiYanChatModelTests.java
index cb7be2a..068a062 100644
--- a/yudao-module-ai/src/test/java/cn/iocoder/yudao/module/ai/framework/ai/core/model/chat/YiYanChatModelTests.java
+++ b/yudao-module-ai/src/test/java/cn/iocoder/yudao/module/ai/framework/ai/core/model/chat/YiYanChatModelTests.java
@@ -1,62 +1,76 @@
package cn.iocoder.yudao.module.ai.framework.ai.core.model.chat;
+import cn.hutool.system.SystemUtil;
+import cn.iocoder.yudao.module.ai.framework.ai.core.model.yiyan.YiYanChatModel;
import org.junit.jupiter.api.Disabled;
import org.junit.jupiter.api.Test;
-import org.springaicommunity.qianfan.QianFanChatModel;
-import org.springaicommunity.qianfan.QianFanChatOptions;
-import org.springaicommunity.qianfan.api.QianFanApi;
import org.springframework.ai.chat.messages.Message;
import org.springframework.ai.chat.messages.UserMessage;
import org.springframework.ai.chat.model.ChatResponse;
import org.springframework.ai.chat.prompt.Prompt;
+import org.springframework.ai.deepseek.DeepSeekChatModel;
+import org.springframework.ai.deepseek.DeepSeekChatOptions;
+import org.springframework.ai.deepseek.api.DeepSeekApi;
import reactor.core.publisher.Flux;
import java.util.ArrayList;
import java.util.List;
+import java.util.Objects;
-// TODO @鑺嬭壙锛氱櫨搴﹀崈甯� API 鎻愪緵浜� V2 鐗堟湰锛岀洰鍓� Spring AI 涓嶅吋瀹癸紝鍙叧閿� <https://github.com/spring-projects/spring-ai/issues/2179> 杩涘睍
+import static cn.iocoder.yudao.module.ai.util.AiUtils.validateApiKey;
+
/**
- * {@link QianFanChatModel} 鐨勯泦鎴愭祴璇�
+ * {@link YiYanChatModel} 鐨勯泦鎴愭祴璇�
*
* @author fansili
*/
public class YiYanChatModelTests {
- private final QianFanChatModel chatModel = new QianFanChatModel(
- new QianFanApi("DGnyzREuaY7av7c38bOM9Ji2", "9aR8myflEOPDrEeLhoXv0FdqANOAyIZW"), // 瀵嗛挜
- QianFanChatOptions.builder()
- .model("ERNIE-4.5-8K-Preview")
- .build()
- );
+ private static final String API_KEY = SystemUtil.get("YIYAN_API_KEY",
+ "sk-xxxx"); // 鎸夐渶鏀规垚浣犵殑鏂囧績涓�瑷� API Key
+ private static final String MODEL = SystemUtil.get("YIYAN_MODEL",
+ YiYanChatModel.MODEL_DEFAULT);
+
+ private final YiYanChatModel chatModel = new YiYanChatModel(DeepSeekChatModel.builder()
+ .deepSeekApi(DeepSeekApi.builder()
+ .baseUrl(YiYanChatModel.BASE_URL)
+ .apiKey(API_KEY)
+ .build())
+ .options(DeepSeekChatOptions.builder()
+ .model(MODEL)
+ .build())
+ .build());
@Test
@Disabled
public void testCall() {
+ validateApiKey(API_KEY);
// 鍑嗗鍙傛暟
List<Message> messages = new ArrayList<>();
- // TODO @鑺嬭壙锛氭枃蹇冧竴瑷�锛屽彧瑕佸甫涓� system message 灏辨姤閿欙紝宸茬粡鍚勭娴嬭瘯锛屽緢鑾悕锛�
-// messages.add(new SystemMessage("浣犳槸涓�涓紭璐ㄧ殑鏂囪█鏂囦綔鑰咃紝鐢ㄦ枃瑷�鏂囨弿杩扮潃鍚勫煄甯傜殑浜烘枃椋庢櫙銆�"));
messages.add(new UserMessage("1 + 1 = 锛�"));
// 璋冪敤
ChatResponse response = chatModel.call(new Prompt(messages));
// 鎵撳嵃缁撴灉
System.out.println(response);
+ System.out.println(Objects.requireNonNull(response.getResult()).getOutput());
}
@Test
@Disabled
public void testStream() {
+ validateApiKey(API_KEY);
// 鍑嗗鍙傛暟
List<Message> messages = new ArrayList<>();
- // TODO @鑺嬭壙锛氭枃蹇冧竴瑷�锛屽彧瑕佸甫涓� system message 灏辨姤閿欙紝宸茬粡鍚勭娴嬭瘯锛屽緢鑾悕锛�
-// messages.add(new SystemMessage("浣犳槸涓�涓紭璐ㄧ殑鏂囪█鏂囦綔鑰咃紝鐢ㄦ枃瑷�鏂囨弿杩扮潃鍚勫煄甯傜殑浜烘枃椋庢櫙銆�"));
messages.add(new UserMessage("1 + 1 = 锛�"));
// 璋冪敤
Flux<ChatResponse> flux = chatModel.stream(new Prompt(messages));
// 鎵撳嵃缁撴灉
- flux.doOnNext(System.out::println).then().block();
+ flux.doOnNext(response -> {
+// System.out.println(response);
+ System.out.println(Objects.requireNonNull(response.getResult()).getOutput());
+ }).then().block();
}
}
--
Gitblit v1.9.3