From eef9ed1b3e3b39c7eebf9e596af3fc396bb88d6e Mon Sep 17 00:00:00 2001
From: cherishsince <cherishsince@aliyun.com>
Date: Sat, 27 Apr 2024 17:22:40 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90=E4=BC=98=E5=8C=96=E3=80=91=E8=87=AA?=
 =?UTF-8?q?=E5=8A=A8=E9=85=8D=E7=BD=AE=E4=BC=98=E5=8C=96=EF=BC=8C=E6=98=9F?=
 =?UTF-8?q?=E7=81=AB=E3=80=81=E4=B8=80=E8=A8=80=20=E5=A2=9E=E5=8A=A0=20top?=
 =?UTF-8?q?K=20topP=20maxToken=20Temperature=20=E5=8F=82=E6=95=B0=E8=BD=AC?=
 =?UTF-8?q?=E6=8D=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ai/config/YudaoAiAutoConfiguration.java    | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/config/YudaoAiAutoConfiguration.java b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/config/YudaoAiAutoConfiguration.java
index fa61eeb3f..b9be65407 100644
--- a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/config/YudaoAiAutoConfiguration.java
+++ b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/config/YudaoAiAutoConfiguration.java
@@ -30,13 +30,19 @@ public class YudaoAiAutoConfiguration {
     @ConditionalOnProperty(value = "yudao.ai.xinghuo.enable", havingValue = "true")
     public XingHuoChatClient xingHuoChatClient(YudaoAiProperties yudaoAiProperties) {
         YudaoAiProperties.XingHuoProperties xingHuoProperties = yudaoAiProperties.getXinghuo();
+        // 转换配置
+        XingHuoOptions xingHuoOptions = new XingHuoOptions();
+        xingHuoOptions.setChatModel(xingHuoProperties.getChatModel());
+        xingHuoOptions.setTopK(xingHuoProperties.getTopK());
+        xingHuoOptions.setTemperature(xingHuoProperties.getTemperature());
+        xingHuoOptions.setMaxTokens(xingHuoProperties.getMaxTokens());
         return new XingHuoChatClient(
                 new XingHuoApi(
                         xingHuoProperties.getAppId(),
                         xingHuoProperties.getAppKey(),
                         xingHuoProperties.getSecretKey()
                 ),
-                new XingHuoOptions().setChatModel(xingHuoProperties.getChatModel())
+                xingHuoOptions
         );
     }
 
@@ -44,6 +50,7 @@ public class YudaoAiAutoConfiguration {
     @ConditionalOnProperty(value = "yudao.ai.qianwen.enable", havingValue = "true")
     public QianWenChatClient qianWenChatClient(YudaoAiProperties yudaoAiProperties) {
         YudaoAiProperties.QianWenProperties qianWenProperties = yudaoAiProperties.getQianwen();
+        // 转换配置
         QianWenOptions qianWenOptions = new QianWenOptions();
         qianWenOptions.setTopK(qianWenProperties.getTopK());
         qianWenOptions.setTopP(qianWenProperties.getTopP());
@@ -62,6 +69,12 @@ public class YudaoAiAutoConfiguration {
     @ConditionalOnProperty(value = "yudao.ai.yiyan.enable", havingValue = "true")
     public YiYanChatClient yiYanChatClient(YudaoAiProperties yudaoAiProperties) {
         YudaoAiProperties.YiYanProperties yiYanProperties = yudaoAiProperties.getYiyan();
+        // 转换配置
+        YiYanOptions yiYanOptions = new YiYanOptions();
+        yiYanOptions.setTopK(yiYanProperties.getTopK());
+        yiYanOptions.setTopP(yiYanProperties.getTopP());
+        yiYanOptions.setTemperature(yiYanProperties.getTemperature());
+        yiYanOptions.setMaxOutputTokens(yiYanProperties.getMaxTokens());
         return new YiYanChatClient(
                 new YiYanApi(
                         yiYanProperties.getAppKey(),
@@ -69,6 +82,7 @@ public class YudaoAiAutoConfiguration {
                         yiYanProperties.getChatModel(),
                         yiYanProperties.getRefreshTokenSecondTime()
                 ),
-                new YiYanOptions().setMax_output_tokens(2048));
+                yiYanOptions
+        );
     }
 }
\ No newline at end of file