huangge1199/vue-pro
1
0
Fork 0
mirror of https://gitee.com/huangge1199_admin/vue-pro.git synced 2024-11-26 17:21:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
cherishsince 2024-03-16 19:47:14 +08:00
parent 7a785b1ec0
commit 6415422f7d
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/chatqianwen/QianWenChatClient.java
View File

@ -34,15 +34,15 @@ public class QianWenChatClient implements ChatClient, StreamingChatClient {
private QianWenApi qianWenApi; private QianWenApi qianWenApi;
private ChatOptions chatOptions; private QianWenOptions qianWenOptions;
public QianWenChatClient(QianWenApi qianWenApi) { public QianWenChatClient(QianWenApi qianWenApi) {
this.qianWenApi = qianWenApi; this.qianWenApi = qianWenApi;
} }
public QianWenChatClient(QianWenApi qianWenApi, ChatOptions chatOptions) { public QianWenChatClient(QianWenApi qianWenApi, QianWenOptions qianWenOptions) {
this.qianWenApi = qianWenApi; this.qianWenApi = qianWenApi;
this.chatOptions = chatOptions; this.qianWenOptions = qianWenOptions;
} }
public final RetryTemplate retryTemplate = RetryTemplate.builder() public final RetryTemplate retryTemplate = RetryTemplate.builder()
@ -81,11 +81,11 @@ public class QianWenChatClient implements ChatClient, StreamingChatClient {
private CompletionsRequest createRequest(Prompt prompt, boolean stream) { private CompletionsRequest createRequest(Prompt prompt, boolean stream) {
// 两个都为null 则没有配置文件 // 两个都为null 则没有配置文件
if (chatOptions == null && prompt.getOptions() == null) { if (qianWenOptions == null && prompt.getOptions() == null) {
throw new ChatException("ChatOptions 未配置参数!"); throw new ChatException("ChatOptions 未配置参数!");
} }
// 优先使用 Prompt 里面的 ChatOptions // 优先使用 Prompt 里面的 ChatOptions
ChatOptions options = chatOptions; ChatOptions options = qianWenOptions;
if (prompt.getOptions() != null) { if (prompt.getOptions() != null) {
options = (ChatOptions) prompt.getOptions(); options = (ChatOptions) prompt.getOptions();
} }