fix ppl allow list bug (#512)

Signed-off-by: xinyual <[email protected]>
opensearch-project · Jan 31, 2025 · 1e82c0b · 1e82c0b
1 parent 9cb89f6
commit 1e82c0b
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/main/java/org/opensearch/agent/tools/PPLTool.java b/src/main/java/org/opensearch/agent/tools/PPLTool.java
@@ -489,13 +489,14 @@ private String constructTableInfoByPPLResultForSpark(Map<String, Object> schema,
         StringJoiner tableInfoJoiner = new StringJoiner("\n");
         for (String key : sortedKeys) {
             String line = "";
-            if (ALLOWED_FIELDS_TYPE.contains(fieldsToType.get(key))) {
+            if (ALLOWED_FIELD_TYPE_FOR_SPARK.contains(fieldsToType.get(key))) {
                 line = "- " + key + ": " + fieldsToType.get(key);
+                if (fieldsToSample.containsKey(key)) {
+                    line += " (" + fieldsToSample.get(key) + ")";
+                }
+                tableInfoJoiner.add(line);
             }
-            if (fieldsToSample.containsKey(key)) {
-                line += " (" + fieldsToSample.get(key) + ")";
-            }
-            tableInfoJoiner.add(line);
+
         }
         return tableInfoJoiner.toString();