知识库对话添加文件引用

5f45381d · 林洋洋 · 18813de1 · 5f45381d
Commit 5f45381d authored Jul 30, 2025 by 林洋洋
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 1 deletion

ChatController.java ...-biz/src/main/java/com/ask/controller/ChatController.java +7 -1

No files found.
--- a/ask-data-ai/ask-data-ai-biz/src/main/java/com/ask/controller/ChatController.java
+++ b/ask-data-ai/ask-data-ai-biz/src/main/java/com/ask/controller/ChatController.java
@@ -187,7 +187,13 @@ public class ChatController {
                        contentBuilder.append(assistantMessage.getText());
                    }
                    return result;
-                }).doOnComplete(() -> {
+                })
+                .concatWith(Mono.fromSupplier(() -> {
+                    ChatResult result = new ChatResult();
+                    result.setMessage(reference);
+                    return result;
+                }))
+                .doOnComplete(() -> {
                    String fullResponse = contentBuilder.toString();
                    chatService.saveHistoryMemory(conversationId, new AssistantMessage(fullResponse));
                });