AI Assistant Management Plugin + Knowledge Base Management

.eslintrc.js

@@ -938,6 +938,7 @@ module.exports = {
         'x-pack/plugins/profiling/**/!(*.stories.tsx|*.test.tsx|*.storybook_decorator.tsx|*.mock.tsx)',
         'x-pack/plugins/synthetics/**/!(*.stories.tsx|*.test.tsx|*.storybook_decorator.tsx|*.mock.tsx)',
         'x-pack/plugins/ux/**/!(*.stories.tsx|*.test.tsx|*.storybook_decorator.tsx|*.mock.tsx)',
+        'src/plugins/ai_assistant_management/**/!(*.stories.tsx|*.test.tsx|*.storybook_decorator.tsx|*.mock.tsx)',
       ],
       rules: {
         '@kbn/i18n/strings_should_be_translated_with_i18n': 'warn',

.github/CODEOWNERS

@@ -10,6 +10,8 @@ packages/kbn-ace @elastic/platform-deployment-management
 x-pack/plugins/actions @elastic/response-ops
 x-pack/test/alerting_api_integration/common/plugins/actions_simulators @elastic/response-ops
 src/plugins/advanced_settings @elastic/appex-sharedux @elastic/platform-deployment-management
+src/plugins/ai_assistant_management/observability @elastic/obs-knowledge-team
+src/plugins/ai_assistant_management/selection @elastic/obs-knowledge-team
 x-pack/packages/ml/aiops_components @elastic/ml-ui
 x-pack/plugins/aiops @elastic/ml-ui
 x-pack/packages/ml/aiops_utils @elastic/ml-ui

.i18nrc.json

@@ -1,6 +1,8 @@
 {
   "paths": {
     "advancedSettings": "src/plugins/advanced_settings",
+    "aiAssistantManagementSelection": "src/plugins/ai_assistant_management/selection",
+    "aiAssistantManagementObservability": "src/plugins/ai_assistant_management/observability",
     "alerts": "packages/kbn-alerts/src",
     "alertsUIShared": "packages/kbn-alerts-ui-shared/src",
     "apmOss": "src/plugins/apm_oss",

docs/developer/plugin-list.asciidoc

@@ -28,6 +28,14 @@ allowing users to configure their advanced settings, also known
 as uiSettings within the code.
 
 
+|{kib-repo}blob/{branch}/src/plugins/ai_assistant_management/observability/README.md[aiAssistantManagementObservability]
+|The aiAssistantManagementObservability plugin manages the Ai Assistant for Observability management section.
+
+
+|{kib-repo}blob/{branch}/src/plugins/ai_assistant_management/selection/README.md[aiAssistantManagementSelection]
+|The aiAssistantManagementSelection plugin manages the Ai Assistant management section.
+
+
 |{kib-repo}blob/{branch}/src/plugins/bfetch/README.md[bfetch]
 |bfetch allows to batch HTTP requests and streams responses back.
 

package.json

@@ -134,6 +134,8 @@
     "@kbn/actions-plugin": "link:x-pack/plugins/actions",
     "@kbn/actions-simulators-plugin": "link:x-pack/test/alerting_api_integration/common/plugins/actions_simulators",
     "@kbn/advanced-settings-plugin": "link:src/plugins/advanced_settings",
+    "@kbn/ai-assistant-management-observability-plugin": "link:src/plugins/ai_assistant_management/observability",
+    "@kbn/ai-assistant-management-plugin": "link:src/plugins/ai_assistant_management/selection",
     "@kbn/aiops-components": "link:x-pack/packages/ml/aiops_components",
     "@kbn/aiops-plugin": "link:x-pack/plugins/aiops",
     "@kbn/aiops-utils": "link:x-pack/packages/ml/aiops_utils",

packages/deeplinks/management/deep_links.ts

@@ -27,6 +27,8 @@ export type IntegrationsDeepLinkId = IntegrationsAppId | FleetAppId | OsQueryApp
 // Management
 export type ManagementAppId = typeof MANAGEMENT_APP_ID;
 export type ManagementId =
+  | 'aiAssistantManagementSelection'
+  | 'aiAssistantManagementObservability'
   | 'api_keys'
   | 'cases'
   | 'cross_cluster_replication'

packages/default-nav/management/default_navigation.ts

@@ -119,6 +119,9 @@ export const defaultNavigation: ManagementNodeDefinition = {
             {
               link: 'management:dataViews',
             },
+            {
+              link: 'management:aiAssistantManagementSelection',
+            },
             {
               // Saved objects
               link: 'management:objects',

packages/kbn-optimizer/limits.yml

@@ -1,6 +1,8 @@
 pageLoadAssetSize:
   actions: 20000
   advancedSettings: 27596
+  aiAssistantManagementObservability: 19279
+  aiAssistantManagementSelection: 19146
   aiops: 10000
   alerting: 106936
   apm: 64385

packages/shared-ux/chrome/navigation/mocks/src/navlinks.ts

@@ -29,6 +29,8 @@ const allNavLinks: AppDeepLinkId[] = [
   'fleet',
   'integrations',
   'management',
+  'management:aiAssistantManagementSelection',
+  'management:aiAssistantManagementObservability',
   'management:api_keys',
   'management:cases',
   'management:cross_cluster_replication',

src/plugins/ai_assistant_management/observability/README.md

@@ -0,0 +1,3 @@
+# `aiAssistantManagementObservability` plugin
+
+The `aiAssistantManagementObservability` plugin manages the `Ai Assistant for Observability` management section.

src/plugins/ai_assistant_management/observability/jest.config.js

@@ -0,0 +1,18 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0 and the Server Side Public License, v 1; you may not use this file except
+ * in compliance with, at your election, the Elastic License 2.0 or the Server
+ * Side Public License, v 1.
+ */
+
+module.exports = {
+  preset: '@kbn/test',
+  rootDir: '../../../..',
+  roots: ['<rootDir>/src/plugins/ai_assistant_management/observability'],
+  coverageDirectory: '<rootDir>/target/kibana-coverage/jest/src/plugins/ai_assistant_management',
+  coverageReporters: ['text', 'html'],
+  collectCoverageFrom: [
+    '<rootDir>/src/plugins/ai_assistant_management/observability/{common,public,server}/**/*.{ts,tsx}',
+  ],
+};

src/plugins/ai_assistant_management/observability/kibana.jsonc

@@ -0,0 +1,13 @@
+{
+  "type": "plugin",
+  "id": "@kbn/ai-assistant-management-observability-plugin",
+  "owner": "@elastic/obs-knowledge-team",
+  "plugin": {
+    "id": "aiAssistantManagementObservability",
+    "server": false,
+    "browser": true,
+    "requiredPlugins": ["management"],
+    "optionalPlugins": ["actions", "home", "observabilityAIAssistant", "serverless", "spaces"],
+    "requiredBundles": ["kibanaReact"]
+  }
+}

src/plugins/ai_assistant_management/observability/public/app.tsx

@@ -0,0 +1,75 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0 and the Server Side Public License, v 1; you may not use this file except
+ * in compliance with, at your election, the Elastic License 2.0 or the Server
+ * Side Public License, v 1.
+ */
+
+import React from 'react';
+import ReactDOM from 'react-dom';
+import { RouteRenderer, RouterProvider } from '@kbn/typed-react-router-config';
+import { QueryClient, QueryClientProvider } from '@tanstack/react-query';
+import { I18nProvider } from '@kbn/i18n-react';
+import { i18n } from '@kbn/i18n';
+import { CoreSetup } from '@kbn/core/public';
+import { wrapWithTheme } from '@kbn/kibana-react-plugin/public';
+import { ManagementAppMountParams } from '@kbn/management-plugin/public';
+import { StartDependencies, AiAssistantManagementObservabilityPluginStart } from './plugin';
+import { aIAssistantManagementObservabilityRouter } from './routes/config';
+import { RedirectToHomeIfUnauthorized } from './routes/components/redirect_to_home_if_unauthorized';
+import { AppContextProvider } from './context/app_context';
+
+interface MountParams {
+  core: CoreSetup<StartDependencies, AiAssistantManagementObservabilityPluginStart>;
+  mountParams: ManagementAppMountParams;
+}
+
+export const mountManagementSection = async ({ core, mountParams }: MountParams) => {
+  const [coreStart, startDeps] = await core.getStartServices();
+  const { element, history, setBreadcrumbs } = mountParams;
+  const { theme$ } = core.theme;
+
+  coreStart.chrome.docTitle.change(
+    i18n.translate('aiAssistantManagementObservability.app.titleBar', {
+      defaultMessage: 'AI Assistant for Observability Settings',
+    })
+  );
+
+  const queryClient = new QueryClient();
+
+  ReactDOM.render(
+    wrapWithTheme(
+      <RedirectToHomeIfUnauthorized coreStart={coreStart}>
+        <I18nProvider>
+          <AppContextProvider
+            value={{
+              http: coreStart.http,
+              navigateToApp: coreStart.application.navigateToApp,
+              notifications: coreStart.notifications,
+              observabilityAIAssistant: startDeps.observabilityAIAssistant,
+              uiSettings: coreStart.uiSettings,
+              setBreadcrumbs,
+            }}
+          >
+            <QueryClientProvider client={queryClient}>
+              <RouterProvider
+                history={history}
+                router={aIAssistantManagementObservabilityRouter as any}
+              >
+                <RouteRenderer />
+              </RouterProvider>
+            </QueryClientProvider>
+          </AppContextProvider>
+        </I18nProvider>
+      </RedirectToHomeIfUnauthorized>,
+      theme$
+    ),
+    element
+  );
+
+  return () => {
+    coreStart.chrome.docTitle.reset();
+    ReactDOM.unmountComponentAtNode(element);
+  };
+};

src/plugins/ai_assistant_management/observability/public/constants.ts

@@ -0,0 +1,14 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0 and the Server Side Public License, v 1; you may not use this file except
+ * in compliance with, at your election, the Elastic License 2.0 or the Server
+ * Side Public License, v 1.
+ */
+
+export const REACT_QUERY_KEYS = {
+  GET_GENAI_CONNECTORS: 'get_genai_connectors',
+  GET_KB_ENTRIES: 'get_kb_entries',
+  CREATE_KB_ENTRIES: 'create_kb_entry',
+  IMPORT_KB_ENTRIES: 'import_kb_entry',
+};

src/plugins/ai_assistant_management/observability/public/context/app_context.tsx

@@ -0,0 +1,40 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0 and the Server Side Public License, v 1; you may not use this file except
+ * in compliance with, at your election, the Elastic License 2.0 or the Server
+ * Side Public License, v 1.
+ */
+
+import React, { createContext, useContext } from 'react';
+import { ChromeBreadcrumb } from '@kbn/core-chrome-browser';
+import type { CoreStart, HttpSetup } from '@kbn/core/public';
+import type { StartDependencies } from '../plugin';
+
+interface ContextValue extends StartDependencies {
+  http: HttpSetup;
+  navigateToApp: CoreStart['application']['navigateToApp'];
+  notifications: CoreStart['notifications'];
+  uiSettings: CoreStart['uiSettings'];
+  setBreadcrumbs: (crumbs: ChromeBreadcrumb[]) => void;
+}
+
+const AppContext = createContext<ContextValue>(null as any);
+
+export const AppContextProvider = ({
+  children,
+  value,
+}: {
+  value: ContextValue;
+  children: React.ReactNode;
+}) => {
+  return <AppContext.Provider value={value}>{children}</AppContext.Provider>;
+};
+
+export const useAppContext = () => {
+  const ctx = useContext(AppContext);
+  if (!ctx) {
+    throw new Error('"useAppContext" can only be called inside of AppContext.Provider!');
+  }
+  return ctx;
+};

src/plugins/ai_assistant_management/observability/public/helpers/categorize_entries.ts

@@ -0,0 +1,29 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0 and the Server Side Public License, v 1; you may not use this file except
+ * in compliance with, at your election, the Elastic License 2.0 or the Server
+ * Side Public License, v 1.
+ */
+
+import type { KnowledgeBaseEntry } from '@kbn/observability-ai-assistant-plugin/common/types';
+
+export interface KnowledgeBaseEntryCategory {
+  categoryName: string;
+  entries: KnowledgeBaseEntry[];
+}
+
+export function categorizeEntries({ entries }: { entries: KnowledgeBaseEntry[] }) {
+  return entries.reduce((acc, entry) => {
+    const categoryName = entry.labels.category ?? entry.id;
+
+    const index = acc.findIndex((item) => item.categoryName === categoryName);
+
+    if (index > -1) {
+      acc[index].entries.push(entry);
+      return acc;
+    } else {
+      return acc.concat({ categoryName, entries: [entry] });
+    }
+  }, [] as Array<{ categoryName: string; entries: KnowledgeBaseEntry[] }>);
+}

src/plugins/ai_assistant_management/observability/public/hooks/use_create_knowledge_base_entry.ts

@@ -0,0 +1,85 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0 and the Server Side Public License, v 1; you may not use this file except
+ * in compliance with, at your election, the Elastic License 2.0 or the Server
+ * Side Public License, v 1.
+ */
+
+import { IHttpFetchError, ResponseErrorBody } from '@kbn/core/public';
+import { i18n } from '@kbn/i18n';
+import type { KnowledgeBaseEntry } from '@kbn/observability-ai-assistant-plugin/common/types';
+import { useMutation, useQueryClient } from '@tanstack/react-query';
+import { useAppContext } from '../context/app_context';
+import { REACT_QUERY_KEYS } from '../constants';
+
+type ServerError = IHttpFetchError<ResponseErrorBody>;
+
+export function useCreateKnowledgeBaseEntry() {
+  const {
+    notifications: { toasts },
+    observabilityAIAssistant,
+  } = useAppContext();
+  const queryClient = useQueryClient();
+  const observabilityAIAssistantApi = observabilityAIAssistant?.callApi;
+
+  return useMutation<
+    void,
+    ServerError,
+    {
+      entry: Omit<
+        KnowledgeBaseEntry,
+        '@timestamp' | 'confidence' | 'is_correction' | 'public' | 'labels' | 'role'
+      >;
+    }
+  >(
+    [REACT_QUERY_KEYS.CREATE_KB_ENTRIES],
+    ({ entry }) => {
+      if (!observabilityAIAssistantApi) {
+        return Promise.reject('Error with observabilityAIAssistantApi: API not found.');
+      }
+
+      return observabilityAIAssistantApi?.(
+        'POST /internal/observability_ai_assistant/kb/entries/save',
+        {
+          signal: null,
+          params: {
+            body: {
+              ...entry,
+              role: 'user_entry',
+            },
+          },
+        }
+      );
+    },
+    {
+      onSuccess: (_data, { entry }) => {
+        toasts.addSuccess(
+          i18n.translate(
+            'aiAssistantManagementObservability.kb.addManualEntry.successNotification',
+            {
+              defaultMessage: 'Successfully created {name}',
+              values: { name: entry.id },
+            }
+          )
+        );
+
+        queryClient.invalidateQueries({
+          queryKey: [REACT_QUERY_KEYS.GET_KB_ENTRIES],
+          refetchType: 'all',
+        });
+      },
+      onError: (error, { entry }) => {
+        toasts.addError(new Error(error.body?.message ?? error.message), {
+          title: i18n.translate(
+            'aiAssistantManagementObservability.kb.addManualEntry.errorNotification',
+            {
+              defaultMessage: 'Something went wrong while creating {name}',
+              values: { name: entry.id },
+            }
+          ),
+        });
+      },
+    }
+  );
+}