From 50ab424d73c5cb68ab16900c76d8054eb921f0a7 Mon Sep 17 00:00:00 2001
From: Steph Milovic <stephanie.milovic@elastic.co>
Date: Thu, 21 Nov 2024 12:50:33 -0700
Subject: [PATCH] remove legacy prompt

---
 .../server/ai_assistant_service/helpers.ts    | 30 +++++++++++++++++++
 .../server/ai_assistant_service/index.ts      |  2 +-
 .../elastic_assistant/server/plugin.ts        |  6 +++-
 3 files changed, 36 insertions(+), 2 deletions(-)

diff --git a/x-pack/plugins/elastic_assistant/server/ai_assistant_service/helpers.ts b/x-pack/plugins/elastic_assistant/server/ai_assistant_service/helpers.ts
index 57b7745a89c78..e4fdc74143f6a 100644
--- a/x-pack/plugins/elastic_assistant/server/ai_assistant_service/helpers.ts
+++ b/x-pack/plugins/elastic_assistant/server/ai_assistant_service/helpers.ts
@@ -9,6 +9,8 @@ import type { ElasticsearchClient } from '@kbn/core-elasticsearch-server';
 import type { KibanaRequest } from '@kbn/core-http-server';
 import type { SavedObjectsClientContract } from '@kbn/core-saved-objects-api-server';
 import type { MlPluginSetup } from '@kbn/ml-plugin/server';
+import { DeleteByQueryRequest } from '@elastic/elasticsearch/lib/api/types';
+import { getResourceName } from '.';
 import { knowledgeBaseIngestPipeline } from '../ai_assistant_data_clients/knowledge_base/ingest_pipeline';
 import { GetElser } from '../types';
 
@@ -96,3 +98,31 @@ export const deletePipeline = async ({ esClient, id }: DeletePipelineParams): Pr
 
   return response.acknowledged;
 };
+
+export const removeLegacyQuickPrompt = async (esClient: ElasticsearchClient) => {
+  const deleteQuery: DeleteByQueryRequest = {
+    index: `${getResourceName('prompts')}-*`,
+    query: {
+      bool: {
+        must: [
+          {
+            term: {
+              name: 'ES|QL Query Generation',
+            },
+          },
+          {
+            term: {
+              prompt_type: 'quick',
+            },
+          },
+          {
+            term: {
+              is_default: true,
+            },
+          },
+        ],
+      },
+    },
+  };
+  return esClient.deleteByQuery(deleteQuery);
+};
diff --git a/x-pack/plugins/elastic_assistant/server/ai_assistant_service/index.ts b/x-pack/plugins/elastic_assistant/server/ai_assistant_service/index.ts
index 81ddd69fb67d3..233b5781ddf68 100644
--- a/x-pack/plugins/elastic_assistant/server/ai_assistant_service/index.ts
+++ b/x-pack/plugins/elastic_assistant/server/ai_assistant_service/index.ts
@@ -40,7 +40,7 @@ import { hasAIAssistantLicense } from '../routes/helpers';
 
 const TOTAL_FIELDS_LIMIT = 2500;
 
-function getResourceName(resource: string) {
+export function getResourceName(resource: string) {
   return `.kibana-elastic-ai-assistant-${resource}`;
 }
 
diff --git a/x-pack/plugins/elastic_assistant/server/plugin.ts b/x-pack/plugins/elastic_assistant/server/plugin.ts
index 4386b95c3fa7a..8950168fe9f48 100755
--- a/x-pack/plugins/elastic_assistant/server/plugin.ts
+++ b/x-pack/plugins/elastic_assistant/server/plugin.ts
@@ -25,7 +25,7 @@ import { RequestContextFactory } from './routes/request_context_factory';
 import { PLUGIN_ID } from '../common/constants';
 import { registerRoutes } from './routes/register_routes';
 import { appContextService } from './services/app_context';
-import { createGetElserId } from './ai_assistant_service/helpers';
+import { createGetElserId, removeLegacyQuickPrompt } from './ai_assistant_service/helpers';
 
 export class ElasticAssistantPlugin
   implements
@@ -109,6 +109,10 @@ export class ElasticAssistantPlugin
         this.getElserId = createGetElserId(this.mlTrainedModelsProvider);
       }
     });
+    removeLegacyQuickPrompt(core.elasticsearch.client.asInternalUser).then((res) => {
+      if (res?.total)
+        this.logger.info(`Removed ${res.total} legacy quick prompts from AI Assistant`);
+    });
 
     return {
       actions: plugins.actions,