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 34  
35 35 private QianWenApi qianWenApi;
36 36  
37   - private ChatOptions chatOptions;
  37 + private QianWenOptions qianWenOptions;
38 38  
39 39 public QianWenChatClient(QianWenApi qianWenApi) {
40 40 this.qianWenApi = qianWenApi;
41 41 }
42 42  
43   - public QianWenChatClient(QianWenApi qianWenApi, ChatOptions chatOptions) {
  43 + public QianWenChatClient(QianWenApi qianWenApi, QianWenOptions qianWenOptions) {
44 44 this.qianWenApi = qianWenApi;
45   - this.chatOptions = chatOptions;
  45 + this.qianWenOptions = qianWenOptions;
46 46 }
47 47  
48 48 public final RetryTemplate retryTemplate = RetryTemplate.builder()
... ... @@ -81,11 +81,11 @@ public class QianWenChatClient implements ChatClient, StreamingChatClient {
81 81  
82 82 private CompletionsRequest createRequest(Prompt prompt, boolean stream) {
83 83 // 两个都为null 则没有配置文件
84   - if (chatOptions == null && prompt.getOptions() == null) {
  84 + if (qianWenOptions == null && prompt.getOptions() == null) {
85 85 throw new ChatException("ChatOptions 未配置参数!");
86 86 }
87 87 // 优先使用 Prompt 里面的 ChatOptions
88   - ChatOptions options = chatOptions;
  88 + ChatOptions options = qianWenOptions;
89 89 if (prompt.getOptions() != null) {
90 90 options = (ChatOptions) prompt.getOptions();
91 91 }
... ...