fix: align vectorstore live retrieval

2026-07-05 10:53:45 +08:00
parent 5c71f5fc79
commit f2bae0382c
3 changed files with 56 additions and 4 deletions
@@ -67,6 +67,32 @@ class VectorSearchServiceTest {
        assertTrue(results.get(0).getMetadata().contains("spring.md"));
    }

+    @Test
+    void vectorStoreSearchUsesDistanceMetadataAsCompatibleScore() {
+        VectorStore vectorStore = mock(VectorStore.class);
+        ObjectProvider<VectorStore> provider = mock(ObjectProvider.class);
+        when(provider.getIfAvailable()).thenReturn(vectorStore);
+        when(vectorStore.similaritySearch(any(SearchRequest.class))).thenReturn(List.of(
+                Document.builder()
+                        .id("spring-doc")
+                        .text("spring content")
+                        .metadata(Map.of("distance", 0.5659486, "category", "api"))
+                        .score(0.4340514)
+                        .build()
+        ));
+
+        VectorSearchService service = new VectorSearchService();
+        setMode(service, "auto");
+        setVectorStore(service, provider);
+
+        List<VectorSearchService.SearchResult> results = service.searchSimilarDocuments("query", 3, null);
+
+        assertEquals(1, results.size());
+        assertEquals("similarity", results.get(0).getScoreLabel());
+        assertEquals(0.4340514, results.get(0).getRawScore(), 0.0001);
+        assertEquals(0.5659486f, results.get(0).getScore(), 0.0001);
+    }
+
    @Test
    void autoModeFallsBackToSdkWhenVectorStoreFails() {
        VectorStore vectorStore = mock(VectorStore.class);