61 lines
2.4 KiB
Java
61 lines
2.4 KiB
Java
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);
|
|
}
|
|
}
|