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 <[email protected]>
spring-projects · Feb 26, 2025 · 96c8157 · 96c8157
1 parent 0065949
commit 96c8157
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...g-ai-test/src/main/java/org/springframework/ai/test/vectorstore/BaseVectorStoreTests.java b/...g-ai-test/src/main/java/org/springframework/ai/test/vectorstore/BaseVectorStoreTests.java
@@ -111,7 +111,7 @@ protected void deleteById() {
 			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()));
 		});