diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/chat/AiChatMessageServiceImpl.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/chat/AiChatMessageServiceImpl.java index 0ce0d15bda..399ca5a6e3 100644 --- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/chat/AiChatMessageServiceImpl.java +++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/chat/AiChatMessageServiceImpl.java @@ -270,9 +270,29 @@ public class AiChatMessageServiceImpl implements AiChatMessageService { }).doOnError(throwable -> { log.error("[sendChatMessageStream][userId({}) sendReqVO({}) 发生异常]", userId, sendReqVO, throwable); // 忽略租户,因为 Flux 异步无法透传租户 - TenantUtils.executeIgnore(() -> chatMessageMapper.updateById( - new AiChatMessageDO().setId(assistantMessage.getId()).setContent(throwable.getMessage()) - .setReasoningContent(reasoningContentBuffer.toString()))); + TenantUtils.executeIgnore(() -> { + // 如果有内容,则更新内容 + if (StrUtil.isNotEmpty(contentBuffer)) { + chatMessageMapper.updateById(new AiChatMessageDO().setId(assistantMessage.getId()) + .setContent(contentBuffer.toString()).setReasoningContent(reasoningContentBuffer.toString())); + } else { + // 否则,则进行删除 + chatMessageMapper.deleteById(assistantMessage.getId()); + } + }); + }).doOnCancel(() -> { + log.info("[sendChatMessageStream][userId({}) sendReqVO({}) 取消请求]", userId, sendReqVO); + // 忽略租户,因为 Flux 异步无法透传租户 + TenantUtils.executeIgnore(() -> { + // 如果有内容,则更新内容 + if (StrUtil.isNotEmpty(contentBuffer)) { + chatMessageMapper.updateById(new AiChatMessageDO().setId(assistantMessage.getId()) + .setContent(contentBuffer.toString()).setReasoningContent(reasoningContentBuffer.toString())); + } else { + // 否则,则进行删除 + chatMessageMapper.deleteById(assistantMessage.getId()); + } + }); }).onErrorResume(error -> Flux.just(error(ErrorCodeConstants.CHAT_STREAM_ERROR))); } diff --git a/yudao-module-ai/src/test/java/cn/iocoder/yudao/module/ai/framework/ai/core/model/chat/TongYiChatModelTests.java b/yudao-module-ai/src/test/java/cn/iocoder/yudao/module/ai/framework/ai/core/model/chat/TongYiChatModelTests.java index 7c62ec71b6..23bd5d9e04 100644 --- a/yudao-module-ai/src/test/java/cn/iocoder/yudao/module/ai/framework/ai/core/model/chat/TongYiChatModelTests.java +++ b/yudao-module-ai/src/test/java/cn/iocoder/yudao/module/ai/framework/ai/core/model/chat/TongYiChatModelTests.java @@ -85,8 +85,8 @@ public class TongYiChatModelTests { List messages = new ArrayList<>(); messages.add(new UserMessage("详细分析下,如何设计一个电商系统?")); DashScopeChatOptions options = DashScopeChatOptions.builder() -// .withModel("qwen3-235b-a22b-thinking-2507") - .withModel("qwen-max-2025-01-25") + .withModel("qwen3-235b-a22b-thinking-2507") +// .withModel("qwen-max-2025-01-25") .withEnableThinking(true) // 必须设置,否则会报错 .build();