From 37e75c542aced3d4c3ce47a0e1a98a6943e56e80 Mon Sep 17 00:00:00 2001
From: Pere Miquel Brull <peremiquelbrull@gmail.com>
Date: Wed, 23 Oct 2024 12:10:45 +0200
Subject: [PATCH] MINOR - Metapilot automator config (#18165)

* Prep config

* MINOR - MetaPilot config

---------

Co-authored-by: Chirag Madlani <12962843+chirag-madlani@users.noreply.github.com>
---
 .../entity/applications/appExtension.json     |  3 +-
 .../external/metaPilotAppConfig.json          | 35 +++----------------
 2 files changed, 7 insertions(+), 31 deletions(-)

diff --git a/openmetadata-spec/src/main/resources/json/schema/entity/applications/appExtension.json b/openmetadata-spec/src/main/resources/json/schema/entity/applications/appExtension.json
index 700aaf79765..4533a27c66c 100644
--- a/openmetadata-spec/src/main/resources/json/schema/entity/applications/appExtension.json
+++ b/openmetadata-spec/src/main/resources/json/schema/entity/applications/appExtension.json
@@ -11,7 +11,8 @@
       "type": "string",
       "enum": [
         "status",
-        "limits"
+        "limits",
+        "custom"
       ]
     }
   },
diff --git a/openmetadata-spec/src/main/resources/json/schema/entity/applications/configuration/external/metaPilotAppConfig.json b/openmetadata-spec/src/main/resources/json/schema/entity/applications/configuration/external/metaPilotAppConfig.json
index 50110033cf3..e3cb9c51681 100644
--- a/openmetadata-spec/src/main/resources/json/schema/entity/applications/configuration/external/metaPilotAppConfig.json
+++ b/openmetadata-spec/src/main/resources/json/schema/entity/applications/configuration/external/metaPilotAppConfig.json
@@ -20,42 +20,17 @@
       "$ref": "#/definitions/metaPilotAppType",
       "default": "MetaPilot"
     },
-    "descriptionDatabases": {
-      "title": "Databases for Automated Description Generation",
-      "description": "Services and Databases configured to get the descriptions from.",
-      "type": "array",
-      "items": {
-        "placeholder": "Search Databases",
-        "$ref": "../../../../type/entityReference.json",
-        "format": "autoComplete",
-        "autoCompleteType": "database_search_index"
-      }
+    "queryFilter": {
+      "description": "Query filter to be passed to ES. E.g., `{\"query\":{\"bool\":{\"must\":[{\"bool\":{\"should\":[{\"term\":{\"domain.displayName.keyword\":\"DG Anim\"}}]}}]}}}`. This is the same payload as in the Explore page.",
+      "type": "string"
     },
     "patchIfEmpty": {
       "title": "Patch Description If Empty",
       "description": "Patch the description if it is empty, instead of raising a suggestion",
       "type": "boolean",
       "default": false
-    },
-    "copilotDatabases": {
-      "title": "Databases for SQL Copilot",
-      "description": "Services and Databases configured to get enable the SQL Copilot.",
-      "type": "array",
-      "items": {
-        "placeholder": "Search Databases",
-        "$ref": "../../../../type/entityReference.json",
-        "format": "autoComplete",
-        "autoCompleteType": "database_search_index"
-      }
-    },
-    "defaultScope": {
-      "title": "Default Chatbot Database Scope",
-      "description": "Default database scope for the chatbot.",
-      "$ref": "../../../../type/entityReference.json",
-      "format": "autoComplete",
-      "autoCompleteType": "database_search_index",
-      "placeholder": "Search Databases"
     }
   },
-  "additionalProperties": false
+  "additionalProperties": false,
+  "required": ["queryFilter"]
 }