BJ_ZTEITS-Developers / parking-cloud-yudao

Commit 6415422f7dfb823fe0c5c5bf494dde1d78062f72

Authored by cherishsince
1 parent 7a785b1e

修改 client 构造方法，制定传入 QianWenOptions

Inline Side-by-side
Showing 1 changed file with 5 additions and 5 deletions
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/chatqianwen/QianWenChatClient.java
  View file @6415422
@@ -34,15 +34,15 @@ public class QianWenChatClient implements ChatClient, StreamingChatClient { @@ -34,15 +34,15 @@ public class QianWenChatClient implements ChatClient, StreamingChatClient {
34 34
35 private QianWenApi qianWenApi; 35 private QianWenApi qianWenApi;
36 36
37 - private ChatOptions chatOptions; 37 + private QianWenOptions qianWenOptions;
38 38
39 public QianWenChatClient(QianWenApi qianWenApi) { 39 public QianWenChatClient(QianWenApi qianWenApi) {
40 this.qianWenApi = qianWenApi; 40 this.qianWenApi = qianWenApi;
41 } 41 }
42 42
43 - public QianWenChatClient(QianWenApi qianWenApi, ChatOptions chatOptions) { 43 + public QianWenChatClient(QianWenApi qianWenApi, QianWenOptions qianWenOptions) {
44 this.qianWenApi = qianWenApi; 44 this.qianWenApi = qianWenApi;
45 - this.chatOptions = chatOptions; 45 + this.qianWenOptions = qianWenOptions;
46 } 46 }
47 47
48 public final RetryTemplate retryTemplate = RetryTemplate.builder() 48 public final RetryTemplate retryTemplate = RetryTemplate.builder()
@@ -81,11 +81,11 @@ public class QianWenChatClient implements ChatClient, StreamingChatClient { @@ -81,11 +81,11 @@ public class QianWenChatClient implements ChatClient, StreamingChatClient {
81 81
82 private CompletionsRequest createRequest(Prompt prompt, boolean stream) { 82 private CompletionsRequest createRequest(Prompt prompt, boolean stream) {
83 // 两个都为null 则没有配置文件 83 // 两个都为null 则没有配置文件
84 - if (chatOptions == null && prompt.getOptions() == null) { 84 + if (qianWenOptions == null && prompt.getOptions() == null) {
85 throw new ChatException("ChatOptions 未配置参数!"); 85 throw new ChatException("ChatOptions 未配置参数!");
86 } 86 }
87 // 优先使用 Prompt 里面的 ChatOptions 87 // 优先使用 Prompt 里面的 ChatOptions
88 - ChatOptions options = chatOptions; 88 + ChatOptions options = qianWenOptions;
89 if (prompt.getOptions() != null) { 89 if (prompt.getOptions() != null) {
90 options = (ChatOptions) prompt.getOptions(); 90 options = (ChatOptions) prompt.getOptions();
91 } 91 }