【解决todo】通义千问增加 model 和 temperature

cherishsince
1 parent 89db7883
Showing 2 changed files with 10 additions and 2 deletions
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/config/YudaoAiAutoConfiguration.java
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/tongyi/QianWenOptions.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());
@@ -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