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