huangge1199/vue-pro
1
0
Fork 0
mirror of https://gitee.com/huangge1199_admin/vue-pro.git synced 2024-11-23 15:51:52 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

增加 QianWenOptions 类型判断，避免强转失败！

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

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

@ -6,6 +6,7 @@ import cn.iocoder.yudao.framework.ai.chat.*;
import cn.iocoder.yudao.framework.ai.chat.messages.MessageType; import cn.iocoder.yudao.framework.ai.chat.messages.MessageType;
import cn.iocoder.yudao.framework.ai.chat.prompt.ChatOptions; import cn.iocoder.yudao.framework.ai.chat.prompt.ChatOptions;
import cn.iocoder.yudao.framework.ai.chat.prompt.Prompt; import cn.iocoder.yudao.framework.ai.chat.prompt.Prompt;
import cn.iocoder.yudao.framework.ai.chatyiyan.YiYanOptions;
import cn.iocoder.yudao.framework.ai.chatyiyan.exception.YiYanApiException; import cn.iocoder.yudao.framework.ai.chatyiyan.exception.YiYanApiException;
import com.aliyun.broadscope.bailian.sdk.models.*; import com.aliyun.broadscope.bailian.sdk.models.*;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
@ -89,6 +90,10 @@ public class QianWenChatClient implements ChatClient, StreamingChatClient {
if (prompt.getOptions() != null) { if (prompt.getOptions() != null) {
options = (ChatOptions) prompt.getOptions(); options = (ChatOptions) prompt.getOptions();
} }
// Prompt 里面是一个 ChatOptions，用户可以随意传入，这里做一下判断
if (!(options instanceof QianWenOptions)) {
throw new ChatException("Prompt 传入的不是 QianWenOptions!");
}
QianWenOptions qianWenOptions = (QianWenOptions) options; QianWenOptions qianWenOptions = (QianWenOptions) options;
// 需要额外处理 // 需要额外处理
if (!stream) { if (!stream) {