From 3fdb8fd5c3e4e8dcbbcd8c5da3038e6d556dc9f4 Mon Sep 17 00:00:00 2001 From: cherishsince Date: Thu, 27 Jun 2024 09:48:53 +0800 Subject: [PATCH] =?UTF-8?q?=E3=80=90=E8=A7=A3=E5=86=B3todo=E3=80=91?= =?UTF-8?q?=E9=80=9A=E4=B9=89=E5=8D=83=E9=97=AE=E5=A2=9E=E5=8A=A0=20model?= =?UTF-8?q?=20=E5=92=8C=20temperature?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../framework/ai/config/YudaoAiAutoConfiguration.java | 2 ++ .../framework/ai/core/model/tongyi/QianWenOptions.java | 10 ++++++++-- 2 files changed, 10 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 10a16e93b..9b4100399 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 @@ -63,6 +63,8 @@ public class YudaoAiAutoConfiguration { YudaoAiProperties.QianWenProperties qianWenProperties = yudaoAiProperties.getQianwen(); // 转换配置 QianWenOptions qianWenOptions = new QianWenOptions(); + qianWenOptions.setModel(qianWenProperties.getModel().getModel()); + qianWenOptions.setTemperature(qianWenProperties.getTemperature()); // qianWenOptions.setTopK(qianWenProperties.getTopK()); TODO 芋艿：后续弄 qianWenOptions.setTopP(qianWenProperties.getTopP()); qianWenOptions.setMaxTokens(qianWenProperties.getMaxTokens()); diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/tongyi/QianWenOptions.java b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/tongyi/QianWenOptions.java index 07805cf11..6bc59c3b3 100644 --- a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/tongyi/QianWenOptions.java +++ b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/tongyi/QianWenOptions.java @@ -6,8 +6,6 @@ import lombok.experimental.Accessors; import java.util.List; -// TODO @fan：增加一个 model 参数 -// TODO @fan：增加一个 Temperature 参数 /** * 阿里云 千问 属性 * @@ -34,6 +32,14 @@ public class QianWenOptions implements ChatOptions { * 用于限制模型生成token的数量，max_tokens设置的是生成上限，并不表示一定会生成这么多的token数量。其中qwen1.5-14b-chat、qwen1.5-7b-chat、qwen-14b-chat和qwen-7b-chat最大值和默认值均为1500，qwen-1.8b-chat、qwen-1.8b-longcontext-chat和qwen-72b-chat最大值和默认值均为2000 */ private Integer maxTokens = 1500; + /** + * 模型 + */ + private String model; + /** + * temperature + */ + private Float temperature; // // 适配 ChatOptions