feat: add spring ai retrieval sidecar

2026-07-05 03:22:03 +08:00
parent 5197712719
commit b9ec07de57
21 changed files with 944 additions and 0 deletions
@@ -0,0 +1,60 @@
+package com.superbiz.agent.service;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.superbiz.agent.dto.ComparableRetrievalResult;
+import org.junit.jupiter.api.Test;
+import org.springframework.ai.document.Document;
+
+import java.util.Map;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+class RetrievalResultNormalizerTest {
+
+    private final RetrievalResultNormalizer normalizer = new RetrievalResultNormalizer(new ObjectMapper());
+
+    @Test
+    void fromCurrentParsesMetadataAndLabelsDistanceScore() {
+        VectorSearchService.SearchResult result = new VectorSearchService.SearchResult();
+        result.setId("vec-1");
+        result.setMetadata("{\"docId\":\"doc-1\",\"_source\":\"docs/api.md\",\"title\":\"API\",\"breadcrumb\":\"A > B\",\"category\":\"api\"}");
+        result.setContent("abcdef");
+        result.setScore(0.25f);
+
+        ComparableRetrievalResult comparable = normalizer.fromCurrent(result, 1, 3);
+
+        assertEquals("current", comparable.getPath());
+        assertEquals("docs/api.md", comparable.getSource());
+        assertEquals("doc-1", comparable.getDocId());
+        assertEquals("API", comparable.getTitle());
+        assertEquals("A > B", comparable.getBreadcrumb());
+        assertEquals("api", comparable.getCategory());
+        assertEquals("abc...", comparable.getContentPreview());
+        assertEquals("l2_distance", comparable.getScoreLabel());
+        assertEquals(0.25, comparable.getScoreValue(), 0.0001);
+    }
+
+    @Test
+    void fromSidecarNormalizesDocumentMetadataAndLabelsSimilarityScore() {
+        Document document = Document.builder()
+                .id("doc-vector")
+                .text("sidecar content")
+                .metadata(Map.of(
+                        "docId", "doc-2",
+                        "_source", "docs/sidecar.md",
+                        "title", "Sidecar",
+                        "breadcrumb", "Root > Sidecar",
+                        "category", "rag"
+                ))
+                .score(0.91)
+                .build();
+
+        ComparableRetrievalResult comparable = normalizer.fromSidecar(document, 2, 100);
+
+        assertEquals("sidecar", comparable.getPath());
+        assertEquals(2, comparable.getRank());
+        assertEquals("docs/sidecar.md", comparable.getSource());
+        assertEquals("similarity", comparable.getScoreLabel());
+        assertEquals(0.91, comparable.getScoreValue(), 0.0001);
+    }
+}