Browse Source

对话不存在时,thow异常

cherishsince 1 year ago
parent
commit
97d4e56c81

+ 5 - 0
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/impl/ChatConversationServiceImpl.java

@@ -1,6 +1,8 @@
 package cn.iocoder.yudao.module.ai.service.impl;
 
+import cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil;
 import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils;
+import cn.iocoder.yudao.module.ai.ErrorCodeConstants;
 import cn.iocoder.yudao.module.ai.convert.ChatConversationConvert;
 import cn.iocoder.yudao.module.ai.dataobject.AiChatConversationDO;
 import cn.iocoder.yudao.module.ai.mapper.AiChatConversationMapper;
@@ -54,6 +56,9 @@ public class ChatConversationServiceImpl implements ChatConversationService {
     @Override
     public ChatConversationRes getConversation(Long id) {
         AiChatConversationDO aiChatConversationDO = aiChatConversationMapper.selectById(id);
+        if (aiChatConversationDO == null) {
+            throw ServiceExceptionUtil.exception(ErrorCodeConstants.AI_CHAT_CONTINUE_NOT_EXIST);
+        }
         return ChatConversationConvert.INSTANCE.covnertChatConversationRes(aiChatConversationDO);
     }