From b55adde7dcef331cd7cee40d461b52788faa0968 Mon Sep 17 00:00:00 2001 From: Xinyuan Lu <74362153+xinyual@users.noreply.github.com> Date: Thu, 18 Apr 2024 13:50:42 +0800 Subject: [PATCH] fix filter fields, adding geo point and date_nanos (#285) * fix filter fields, adding geo point and date_nanos Signed-off-by: xinyual * apply spotless Signed-off-by: xinyual --------- Signed-off-by: xinyual --- src/main/java/org/opensearch/agent/tools/PPLTool.java | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/main/java/org/opensearch/agent/tools/PPLTool.java b/src/main/java/org/opensearch/agent/tools/PPLTool.java index c89e4138..0b52c12f 100644 --- a/src/main/java/org/opensearch/agent/tools/PPLTool.java +++ b/src/main/java/org/opensearch/agent/tools/PPLTool.java @@ -107,6 +107,7 @@ public class PPLTool implements Tool { static { ALLOWED_FIELDS_TYPE = new HashSet<>(); // from // https://github.com/opensearch-project/sql/blob/2.x/docs/user/ppl/general/datatypes.rst#data-types-mapping + // and https://opensearch.org/docs/latest/field-types/supported-field-types/index/ ALLOWED_FIELDS_TYPE.add("boolean"); ALLOWED_FIELDS_TYPE.add("byte"); ALLOWED_FIELDS_TYPE.add("short"); @@ -119,10 +120,12 @@ public class PPLTool implements Tool { ALLOWED_FIELDS_TYPE.add("keyword"); ALLOWED_FIELDS_TYPE.add("text"); ALLOWED_FIELDS_TYPE.add("date"); + ALLOWED_FIELDS_TYPE.add("date_nanos"); ALLOWED_FIELDS_TYPE.add("ip"); ALLOWED_FIELDS_TYPE.add("binary"); ALLOWED_FIELDS_TYPE.add("object"); ALLOWED_FIELDS_TYPE.add("nested"); + ALLOWED_FIELDS_TYPE.add("geo_point"); try { DEFAULT_PROMPT_DICT = loadDefaultPromptDict();