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

cherishsince
1 parent 6415422f
Showing 1 changed file with 5 additions and 0 deletions
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/chatqianwen/QianWenChatClient.java
@@ -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.prompt.ChatOptions;
 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 com.aliyun.broadscope.bailian.sdk.models.*;
 import lombok.extern.slf4j.Slf4j;
@@ -89,6 +90,10 @@ public class QianWenChatClient  implements ChatClient, StreamingChatClient {
         if (prompt.getOptions() != null) {
             options = (ChatOptions) prompt.getOptions();
         }
+        // Prompt 里面是一个 ChatOptions，用户可以随意传入，这里做一下判断
+        if (!(options instanceof QianWenOptions)) {
+            throw new ChatException("Prompt 传入的不是 QianWenOptions!");
+        }
         QianWenOptions qianWenOptions = (QianWenOptions) options;
         // 需要额外处理
         if (!stream) {