Skip to content

Commit 96c8157

Browse files
committed
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]>
1 parent 0065949 commit 96c8157

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: spring-ai-test/src/main/java/org/springframework/ai/test/vectorstore/BaseVectorStoreTests.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ protected void deleteById() {
111111
Map<String, Object> metadata = results.get(0).getMetadata();
112112
assertThat(normalizeValue(metadata.get("country"))).isEqualTo("BG");
113113
// the values are converted into Double
114-
assertThat(normalizeValue(metadata.get("year"))).isEqualTo("2023.0");
114+
assertThat(normalizeValue(metadata.get("year"))).containsAnyOf("2023", "2023.0");
115115

116116
vectorStore.delete(List.of(documents.get(2).getId()));
117117
});

0 commit comments

Comments
 (0)