Skip to content

Commit 1e82c0b

Browse files
authored
fix ppl allow list bug (#512)
Signed-off-by: xinyual <[email protected]>
1 parent 9cb89f6 commit 1e82c0b

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

src/main/java/org/opensearch/agent/tools/PPLTool.java

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -489,13 +489,14 @@ private String constructTableInfoByPPLResultForSpark(Map<String, Object> schema,
489489
StringJoiner tableInfoJoiner = new StringJoiner("\n");
490490
for (String key : sortedKeys) {
491491
String line = "";
492-
if (ALLOWED_FIELDS_TYPE.contains(fieldsToType.get(key))) {
492+
if (ALLOWED_FIELD_TYPE_FOR_SPARK.contains(fieldsToType.get(key))) {
493493
line = "- " + key + ": " + fieldsToType.get(key);
494+
if (fieldsToSample.containsKey(key)) {
495+
line += " (" + fieldsToSample.get(key) + ")";
496+
}
497+
tableInfoJoiner.add(line);
494498
}
495-
if (fieldsToSample.containsKey(key)) {
496-
line += " (" + fieldsToSample.get(key) + ")";
497-
}
498-
tableInfoJoiner.add(line);
499+
499500
}
500501
return tableInfoJoiner.toString();
501502

0 commit comments

Comments
 (0)