From 0b4504723b12120568761ae6e8ef26965174f16a Mon Sep 17 00:00:00 2001
From: "github-actions[bot]" <github-actions[bot]@users.noreply.github.com>
Date: Fri, 31 Jan 2025 16:05:32 +0000
Subject: [PATCH] fix ppl allow list bug (#512)

Signed-off-by: xinyual <xinyual@amazon.com>
(cherry picked from commit 1e82c0bf0ea01430c93dd4d445049ec55ed6e00b)
Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
---
 src/main/java/org/opensearch/agent/tools/PPLTool.java | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/src/main/java/org/opensearch/agent/tools/PPLTool.java b/src/main/java/org/opensearch/agent/tools/PPLTool.java
index 84230cc5..1ec0a742 100644
--- 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();