Fix BaseVectorStoreTests

- Fix assertion of value which is in Double in some of the vector stores (Milvus, Pinecone for now)

Signed-off-by: Ilayaperumal Gopinathan <ilayaperumal.gopinathan@broadcom.com>
This commit is contained in:
Ilayaperumal Gopinathan
2025-02-26 22:26:36 +00:00
parent 0065949d5f
commit 96c8157cbe

View File

@@ -111,7 +111,7 @@ public abstract class BaseVectorStoreTests {
Map<String, Object> metadata = results.get(0).getMetadata();
assertThat(normalizeValue(metadata.get("country"))).isEqualTo("BG");
// the values are converted into Double
assertThat(normalizeValue(metadata.get("year"))).isEqualTo("2023.0");
assertThat(normalizeValue(metadata.get("year"))).containsAnyOf("2023", "2023.0");
vectorStore.delete(List.of(documents.get(2).getId()));
});