优化知识库召回提示词

ask-data-ai/ask-data-ai-biz/src/main/java/com/ask/service/impl/ChatService.java

@@ -17,10 +17,9 @@ import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 import org.springframework.util.CollectionUtils;
 
-import java.util.Arrays;
-import java.util.List;
-import java.util.Set;
+import java.util.*;
 import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.atomic.AtomicInteger;
 import java.util.function.Function;
 import java.util.function.Predicate;
 import java.util.regex.Matcher;
@@ -88,9 +87,25 @@ public class ChatService {
      * @return 拼接后的字符串
      */
     public String convertDocumentsToString(List<Document> documents) {
+        if (documents == null || documents.isEmpty()) {
+            return "";
+        }
+
+        AtomicInteger index = new AtomicInteger(1);
         return documents.stream()
-                .map(Document::getText) // 提取每个Document的内容
-                .collect(Collectors.joining("\n")); // 用换行符拼接
+                .filter(Objects::nonNull)                          // 1. 过滤 null Document
+                .map(doc -> {
+                    String title = Optional.ofNullable(doc.getMetadata())
+                            .map(m -> String.valueOf(m.get("fileName")))
+                            .orElse("");            // 2. 标题 null-safe
+                    String text = Optional.ofNullable(doc.getText())
+                            .orElse("");             // 3. 正文 null-safe
+                    return String.format("[%d] %s\n%s",
+                            index.getAndIncrement(),
+                            title,
+                            text);
+                })
+                .collect(Collectors.joining("\n\n"));
     }