From 4b5e852c6fab3be18f5b851e6e0ffdb04269692d Mon Sep 17 00:00:00 2001
From: xinyual <xinyual@amazon.com>
Date: Fri, 31 Jan 2025 23:14:14 +0800
Subject: [PATCH] fix ppl allow list bug

Signed-off-by: xinyual <xinyual@amazon.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 79a6f137..3a88a2d2 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();