feat(rag): modularize knowledge retrieval pipeline

2026-07-06 17:06:05 +08:00
parent a375daead7
commit cf3333d607
38 changed files with 2981 additions and 1033 deletions
@@ -0,0 +1,49 @@
+package com.superbiz.agent.service;
+
+import com.superbiz.agent.dto.ContextPack;
+import com.superbiz.agent.dto.EvidencePostprocessResult;
+import com.superbiz.agent.dto.LookupResult;
+import com.superbiz.agent.dto.RetrievalTrace;
+import org.springframework.stereotype.Service;
+
+import java.util.List;
+
+/**
+ * Assembles evidence-first LookupResult instances.
+ */
+@Service
+public class LookupResultAssembler {
+
+    public LookupResult assemble(EvidencePostprocessResult evidence,
+                                 ContextPack contextPack,
+                                 RetrievalTrace retrievalTrace) {
+        boolean found = evidence != null && evidence.hasUsableEvidence();
+        return LookupResult.builder()
+                .found(found)
+                .evidenceBlocks(evidence != null ? evidence.getEvidenceBlocks() : List.of())
+                .evidenceCandidateCount(evidence != null ? evidence.getCandidateCount() : 0)
+                .evidenceBlockCount(evidence != null ? evidence.getEvidenceBlockCount() : 0)
+                .contextPack(contextPack)
+                .retrievalTrace(retrievalTrace)
+                .rerankTrace(evidence != null ? evidence.getRerankTrace() : null)
+                .relevanceLevel(evidence != null ? evidence.getRelevanceLevel() : null)
+                .completenessHint(evidence != null ? evidence.getCompletenessHint() : null)
+                .message(found ? null : "知识库未检索到可用证据，请结合日志、指标、告警继续排查")
+                .build();
+    }
+
+    public LookupResult deduped(LookupResult original, List<String> retrievedDomains, String docKey) {
+        return LookupResult.builder()
+                .found(false)
+                .message("文档已在本会话中检索过，无需重复召回: " + docKey)
+                .evidenceBlocks(List.of())
+                .evidenceCandidateCount(0)
+                .evidenceBlockCount(0)
+                .retrievalTrace(original.getRetrievalTrace())
+                .rerankTrace(original.getRerankTrace())
+                .relevanceLevel(original.getRelevanceLevel())
+                .completenessHint(original.getCompletenessHint())
+                .retrievedDomainsThisSession(retrievedDomains)
+                .build();
+    }
+}