fix mapping (#2279) (#2281)

Signed-off-by: Jing Zhang <[email protected]> (cherry picked from commit f67eab0) Co-authored-by: Jing Zhang <[email protected]>
opensearch-project · Mar 26, 2024 · 67a645c · 67a645c
1 parent cbbff4b
commit 67a645c
Showing 1 changed file with 38 additions and 1 deletion.
diff --git a/common/src/main/java/org/opensearch/ml/common/CommonValue.java b/common/src/main/java/org/opensearch/ml/common/CommonValue.java
@@ -263,7 +263,44 @@ public class CommonValue {
                         + "\": {\"type\": \"date\", \"format\": \"strict_date_time||epoch_millis\"},\n"
                         + "      \""
                         + MLModel.GUARDRAILS_FIELD
-                        + "\" : {\"type\": \"flat_object\"},\n"
+                        + "\" : {\n" +
+                "          \"properties\": {\n" +
+                "            \"input_guardrail\": {\n" +
+                "              \"properties\": {\n" +
+                "                \"regex\": {\n" +
+                "                  \"type\": \"text\"\n" +
+                "                },\n" +
+                "                \"stop_words\": {\n" +
+                "                  \"properties\": {\n" +
+                "                    \"index_name\": {\n" +
+                "                      \"type\": \"text\"\n" +
+                "                    },\n" +
+                "                    \"source_fields\": {\n" +
+                "                      \"type\": \"text\"\n" +
+                "                    }\n" +
+                "                  }\n" +
+                "                }\n" +
+                "              }\n" +
+                "            },\n" +
+                "            \"output_guardrail\": {\n" +
+                "              \"properties\": {\n" +
+                "                \"regex\": {\n" +
+                "                  \"type\": \"text\"\n" +
+                "                },\n" +
+                "                \"stop_words\": {\n" +
+                "                  \"properties\": {\n" +
+                "                    \"index_name\": {\n" +
+                "                      \"type\": \"text\"\n" +
+                "                    },\n" +
+                "                    \"source_fields\": {\n" +
+                "                      \"type\": \"text\"\n" +
+                "                    }\n" +
+                "                  }\n" +
+                "                }\n" +
+                "              }\n" +
+                "            }\n" +
+                "          }\n" +
+                "        },\n"
                         + "      \""
                         + MLModel.CONNECTOR_FIELD
                         + "\": {" + ML_CONNECTOR_INDEX_FIELDS + "    }\n},"