Address comments

Signed-off-by: Yizhe Liu <[email protected]>
opensearch-project · Jan 8, 2025 · f8b811c · f8b811c
1 parent 89aabe5
commit f8b811c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/main/java/org/opensearch/neuralsearch/processor/InferenceProcessor.java b/src/main/java/org/opensearch/neuralsearch/processor/InferenceProcessor.java
@@ -153,9 +153,8 @@ public void execute(IngestDocument ingestDocument, BiConsumer<IngestDocument, Ex
 
     @VisibleForTesting
     void preprocessIngestDocument(IngestDocument ingestDocument) {
-        if (ingestDocument == null) return;
+        if (ingestDocument == null || ingestDocument.getSourceAndMetadata() == null) return;
         Map<String, Object> sourceAndMetadataMap = ingestDocument.getSourceAndMetadata();
-        if (sourceAndMetadataMap == null) return;
         Map<String, Object> unflattened = ProcessorDocumentUtils.unflattenJson(sourceAndMetadataMap);
         unflattened.forEach(ingestDocument::setFieldValue);
         sourceAndMetadataMap.keySet().removeIf(key -> key.contains("."));