diff --git a/vector-stores/spring-ai-milvus-store/src/main/java/org/springframework/ai/vectorstore/milvus/MilvusVectorStore.java b/vector-stores/spring-ai-milvus-store/src/main/java/org/springframework/ai/vectorstore/milvus/MilvusVectorStore.java index 8ef20793167..d2af691e428 100644 --- a/vector-stores/spring-ai-milvus-store/src/main/java/org/springframework/ai/vectorstore/milvus/MilvusVectorStore.java +++ b/vector-stores/spring-ai-milvus-store/src/main/java/org/springframework/ai/vectorstore/milvus/MilvusVectorStore.java @@ -379,8 +379,10 @@ public List doSimilaritySearch(SearchRequest request) { JsonObject metadata = new JsonObject(); try { metadata = (JsonObject) rowRecord.get(this.metadataFieldName); - // inject the distance into the metadata. - metadata.addProperty(DocumentMetadata.DISTANCE.value(), 1 - getResultSimilarity(rowRecord)); + if (metadata != null) { + // inject the distance into the metadata. + metadata.addProperty(DocumentMetadata.DISTANCE.value(), 1 - getResultSimilarity(rowRecord)); + } } catch (ParamException e) { // skip the ParamException if metadata doesn't exist for the custom