Treat dot as nested field in field_map for inference processor

Signed-off-by: Sanjana679 <[email protected]>
opensearch-project · Nov 9, 2023 · 7006c97 · 7006c97
1 parent cda2f82
commit 7006c97
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/main/java/org/opensearch/neuralsearch/processor/InferenceProcessor.java b/src/main/java/org/opensearch/neuralsearch/processor/InferenceProcessor.java
@@ -154,6 +154,16 @@ Map<String, Object> buildMapWithProcessorKeyAndOriginalValue(IngestDocument inge
         for (Map.Entry<String, Object> fieldMapEntry : fieldMap.entrySet()) {
             String originalKey = fieldMapEntry.getKey();
             Object targetKey = fieldMapEntry.getValue();
+
+            int nestedDotIndex = originalKey.indexOf('.');
+            if (nestedDotIndex != -1) {
+                Map<String, Object> temp = new LinkedHashMap<>();
+                temp.put(originalKey.substring(nestedDotIndex + 1), targetKey);
+                targetKey = temp;
+
+                originalKey = originalKey.substring(0, nestedDotIndex);
+            }
+
             if (targetKey instanceof Map) {
                 Map<String, Object> treeRes = new LinkedHashMap<>();
                 buildMapWithProcessorKeyAndOriginalValueForMapType(originalKey, targetKey, sourceAndMetadataMap, treeRes);