From 9bad721754fe8bbe2e5f459d0706e0fefac569f3 Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期日, 20 九月 2026 09:05:41 +0800
Subject: [PATCH] feat(mes): 添加智能质检报告功能与AI导入能力
---
yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/framework/ai/config/AiAutoConfiguration.java | 34 +++++++++++++++++++---------------
1 files changed, 19 insertions(+), 15 deletions(-)
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/framework/ai/config/AiAutoConfiguration.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/framework/ai/config/AiAutoConfiguration.java
index 27ceff0..8d3938a 100644
--- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/framework/ai/config/AiAutoConfiguration.java
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/framework/ai/config/AiAutoConfiguration.java
@@ -13,31 +13,23 @@
import org.springframework.ai.openai.OpenAiChatOptions;
import org.springframework.ai.openai.OpenAiEmbeddingModel;
import org.springframework.ai.openai.OpenAiEmbeddingOptions;
-import org.springframework.ai.retry.RetryUtils;
import org.springframework.ai.tokenizer.JTokkitTokenCountEstimator;
import org.springframework.ai.tokenizer.TokenCountEstimator;
-import org.springframework.ai.vectorstore.milvus.autoconfigure.MilvusServiceClientProperties;
-import org.springframework.ai.vectorstore.milvus.autoconfigure.MilvusVectorStoreProperties;
-import org.springframework.ai.vectorstore.qdrant.autoconfigure.QdrantVectorStoreProperties;
-import org.springframework.ai.vectorstore.redis.autoconfigure.RedisVectorStoreProperties;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
+import java.time.Duration;
+
/**
- * 鑺嬮亾 AI 鑷姩閰嶇疆
+ * 瓒呯骇绠＄悊鍛� AI 鑷姩閰嶇疆
*
* 浣跨敤 OpenAI 鍏煎鎺ュ彛瀵规帴閫氫箟鍗冮棶 DashScope
+ * 鍚戦噺搴撻厤缃 yudao.ai.vector-store 鑺傜偣锛屼笉鍐嶄緷璧� Spring AI 鐨� Milvus 鑷姩閰嶇疆灞炴��
*/
@Configuration
-@EnableConfigurationProperties({
- YudaoAiProperties.class,
- QdrantVectorStoreProperties.class,
- RedisVectorStoreProperties.class,
- MilvusVectorStoreProperties.class,
- MilvusServiceClientProperties.class
-})
+@EnableConfigurationProperties(YudaoAiProperties.class)
@Slf4j
public class AiAutoConfiguration {
@@ -68,13 +60,25 @@
// ========== 閫氫箟鍗冮棶 Chat锛堥�氳繃 OpenAI 鍏煎鎺ュ彛锛�==========
- public static OpenAiChatModel buildTongYiChatModel(String apiKey, String model) {
+ /**
+ * 鏋勫缓閫氫箟鍗冮棶 Chat 妯″瀷
+ *
+ * @param temperature 閲囨牱娓╁害锛屼负 null 鏃剁敤 0.7锛堜笌鍘嗗彶琛屼负涓�鑷达級
+ * @param maxTokens 鍗曟鍥炲鐨勬渶澶� token 鏁帮紝涓� null 鏃舵部鐢ㄦā鍨嬮粯璁ゅ�笺��
+ * 涓嶄紶浼氳闀挎枃妗ｇ殑杈撳嚭琚潤榛樻埅鏂紝琛ㄧ幇涓恒�孞SON 灏鹃儴缂哄け銆佽В鏋愬け璐ャ��
+ * @param timeout 鍗曟璋冪敤瓒呮椂銆備笉璁捐秴鏃剁殑璇濓紝妯″瀷鎺掗槦鏃惰姹備細涓�鐩存寕鍒� TCP 灞傝秴鏃�
+ */
+ public static OpenAiChatModel buildTongYiChatModel(String apiKey, String model,
+ Double temperature, Integer maxTokens,
+ Duration timeout) {
return OpenAiChatModel.builder()
.options(OpenAiChatOptions.builder()
.baseUrl(DASHSCOPE_BASE_URL)
.apiKey(apiKey)
.model(StrUtil.blankToDefault(model, "qwen-plus"))
- .temperature(0.7)
+ .temperature(temperature != null ? temperature : 0.7)
+ .maxTokens(maxTokens)
+ .timeout(timeout)
.build())
.toolCallingManager(ToolCallingManager.builder().build())
.build();
--
Gitblit v1.9.3