AI Assistant Management Plugin + Knowledge Base Management

.eslintrc.js

@@ -921,6 +921,7 @@ module.exports = {
         'x-pack/plugins/profiling/**/*.tsx',
         'x-pack/plugins/synthetics/**/*.tsx',
         'x-pack/plugins/ux/**/*.tsx',
+        'src/plugins/ai_assistant_management/**/*.tsx',
       ],
       rules: {
         '@kbn/telemetry/event_generating_elements_should_be_instrumented': 'error',
@@ -938,6 +939,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"],
+    "requiredBundles": ["kibanaReact"]
+  }
+}

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

@@ -0,0 +1,79 @@
+/*
+ * 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();
+
+  if (!startDeps.observabilityAIAssistant) return () => {};
+
+  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={{
+              application: coreStart.application,
+              http: coreStart.http,
+              notifications: coreStart.notifications,
+              observabilityAIAssistant: startDeps.observabilityAIAssistant,
+              uiSettings: coreStart.uiSettings,
+              serverless: startDeps.serverless,
+              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,34 @@
+/*
+ * 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 } from 'react';
+import type { ChromeBreadcrumb } from '@kbn/core-chrome-browser';
+import type { CoreStart, HttpSetup } from '@kbn/core/public';
+import type { ObservabilityAIAssistantPluginStart } from '@kbn/observability-ai-assistant-plugin/public';
+import type { StartDependencies } from '../plugin';
+
+export interface ContextValue extends StartDependencies {
+  application: CoreStart['application'];
+  http: HttpSetup;
+  notifications: CoreStart['notifications'];
+  observabilityAIAssistant: ObservabilityAIAssistantPluginStart;
+  setBreadcrumbs: (crumbs: ChromeBreadcrumb[]) => void;
+  uiSettings: CoreStart['uiSettings'];
+}
+
+export 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>;
+};

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

@@ -0,0 +1,30 @@
+/*
+ * 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 {
+  '@timestamp': string;
+  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], '@timestamp': entry['@timestamp'] });
+    }
+  }, [] as Array<{ categoryName: string; entries: KnowledgeBaseEntry[]; '@timestamp': string }>);
+}

src/plugins/ai_assistant_management/observability/public/helpers/test_helper.tsx

@@ -0,0 +1,89 @@
+/*
+ * 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 { createMemoryHistory } from 'history';
+import { QueryClient, QueryClientProvider } from '@tanstack/react-query';
+import { render as testLibRender } from '@testing-library/react';
+import { coreMock } from '@kbn/core/public/mocks';
+import { __IntlProvider as IntlProvider } from '@kbn/i18n-react';
+import translations from '@kbn/translations-plugin/translations/ja-JP.json';
+
+import { mockObservabilityAIAssistantService } from '@kbn/observability-ai-assistant-plugin/public';
+import { RouterProvider } from '@kbn/typed-react-router-config';
+import { AppContextProvider } from '../context/app_context';
+import { RedirectToHomeIfUnauthorized } from '../routes/components/redirect_to_home_if_unauthorized';
+import { aIAssistantManagementObservabilityRouter } from '../routes/config';
+
+export const coreStart = coreMock.createStart();
+
+const queryClient = new QueryClient({
+  defaultOptions: {
+    queries: {
+      retry: false,
+    },
+  },
+  logger: {
+    // eslint-disable-next-line no-console
+    log: console.log,
+    // eslint-disable-next-line no-console
+    warn: console.warn,
+    error: () => {},
+  },
+});
+
+export const render = (component: React.ReactNode, params?: { show: boolean }) => {
+  const history = createMemoryHistory();
+
+  return testLibRender(
+    // @ts-ignore
+    <IntlProvider locale="en-US" messages={translations.messages}>
+      <RedirectToHomeIfUnauthorized
+        coreStart={{
+          application: {
+            ...coreStart.application,
+            capabilities: {
+              // @ts-ignore
+              management: { show: true },
+              observabilityAIAssistant: {
+                show: params?.show ?? true,
+              },
+            },
+          },
+        }}
+      >
+        <AppContextProvider
+          value={{
+            http: coreStart.http,
+            application: coreStart.application,
+            notifications: coreStart.notifications,
+            observabilityAIAssistant: {
+              service: mockObservabilityAIAssistantService,
+              useGenAIConnectors: () => ({
+                loading: false,
+                selectConnector: () => {},
+                reloadConnectors: () => {},
+              }),
+            },
+            uiSettings: coreStart.uiSettings,
+            setBreadcrumbs: () => {},
+          }}
+        >
+          <QueryClientProvider client={queryClient}>
+            <RouterProvider
+              history={history}
+              router={aIAssistantManagementObservabilityRouter as any}
+            >
+              {component}
+            </RouterProvider>
+          </QueryClientProvider>
+        </AppContextProvider>
+      </RedirectToHomeIfUnauthorized>
+    </IntlProvider>
+  );
+};