From 96c8157cbed792241afcc3a9b2e2a21f476e0251 Mon Sep 17 00:00:00 2001 From: Ilayaperumal Gopinathan Date: Wed, 26 Feb 2025 22:26:36 +0000 Subject: [PATCH] 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 --- .../ai/test/vectorstore/BaseVectorStoreTests.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spring-ai-test/src/main/java/org/springframework/ai/test/vectorstore/BaseVectorStoreTests.java b/spring-ai-test/src/main/java/org/springframework/ai/test/vectorstore/BaseVectorStoreTests.java index 94b6f33f0..17b0604ba 100644 --- a/spring-ai-test/src/main/java/org/springframework/ai/test/vectorstore/BaseVectorStoreTests.java +++ b/spring-ai-test/src/main/java/org/springframework/ai/test/vectorstore/BaseVectorStoreTests.java @@ -111,7 +111,7 @@ public abstract class BaseVectorStoreTests { Map 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())); });