[8.x] [One Discover] Add AIAssistant prompts to logs overview detail (e…

…lastic#204339) (elastic#204910) # Backport This will backport the following commits from `main` to `8.x`: - [[One Discover] Add AIAssistant prompts to logs overview detail (elastic#204339)](elastic#204339)  ### Questions ? Please refer to the [Backport tool documentation](https://github.com/sqren/backport)  --------- Co-authored-by: Marco Antonio Ghiani <[email protected]>
kibanamachine · Dec 19, 2024 · f93a9c9 · f93a9c9
1 parent 8a46374
commit f93a9c9
Show file tree

Hide file tree

Showing 5 changed files with 34 additions and 8 deletions.
diff --git a/src/plugins/discover/public/context_awareness/__mocks__/index.tsx b/src/plugins/discover/public/context_awareness/__mocks__/index.tsx
@@ -25,6 +25,7 @@ import { ProfileProviderServices } from '../profile_providers/profile_provider_s
 import { ProfilesManager } from '../profiles_manager';
 import { DiscoverEBTManager } from '../../services/discover_ebt_manager';
 import { createLogsContextServiceMock } from '@kbn/discover-utils/src/__mocks__';
+import { discoverSharedPluginMock } from '@kbn/discover-shared-plugin/public/mocks';
 
 export const createContextAwarenessMocks = ({
   shouldRegisterProviders = true,
@@ -181,5 +182,6 @@ export const createContextAwarenessMocks = ({
 const createProfileProviderServicesMock = () => {
   return {
     logsContextService: createLogsContextServiceMock(),
+    discoverShared: discoverSharedPluginMock.createStartContract(),
   } as ProfileProviderServices;
 };
diff --git a/...areness/profile_providers/observability/log_document_profile/accessors/get_doc_viewer.tsx b/...areness/profile_providers/observability/log_document_profile/accessors/get_doc_viewer.tsx
@@ -11,11 +11,18 @@ import { i18n } from '@kbn/i18n';
 import { UnifiedDocViewerLogsOverview } from '@kbn/unified-doc-viewer-plugin/public';
 import React from 'react';
 import type { DocumentProfileProvider } from '../../../../profiles';
+import { ProfileProviderServices } from '../../../profile_provider_services';
 
-export const getDocViewer: DocumentProfileProvider['profile']['getDocViewer'] =
-  (prev) => (params) => {
+export const createGetDocViewer =
+  (services: ProfileProviderServices): DocumentProfileProvider['profile']['getDocViewer'] =>
+  (prev) =>
+  (params) => {
     const prevDocViewer = prev(params);
 
+    const logsAIAssistantFeature = services.discoverShared.features.registry.getById(
+      'observability-logs-ai-assistant'
+    );
+
     return {
       ...prevDocViewer,
       docViewsRegistry: (registry) => {
@@ -25,7 +32,12 @@ export const getDocViewer: DocumentProfileProvider['profile']['getDocViewer'] =
             defaultMessage: 'Log overview',
           }),
           order: 0,
-          component: (props) => <UnifiedDocViewerLogsOverview {...props} />,
+          component: (props) => (
+            <UnifiedDocViewerLogsOverview
+              {...props}
+              renderAIAssistant={logsAIAssistantFeature?.render}
+            />
+          ),
         });
 
         return prevDocViewer.docViewsRegistry(registry);

diff --git a/...context_awareness/profile_providers/observability/log_document_profile/accessors/index.ts b/...context_awareness/profile_providers/observability/log_document_profile/accessors/index.ts
@@ -7,4 +7,4 @@
  * License v3.0 only", or the "Server Side Public License, v 1".
  */
 
-export { getDocViewer } from './get_doc_viewer';
+export { createGetDocViewer } from './get_doc_viewer';
diff --git a/...public/context_awareness/profile_providers/observability/log_document_profile/profile.tsx b/...public/context_awareness/profile_providers/observability/log_document_profile/profile.tsx
@@ -10,15 +10,15 @@
 import { DataTableRecord } from '@kbn/discover-utils';
 import { DocumentProfileProvider, DocumentType } from '../../../profiles';
 import { ProfileProviderServices } from '../../profile_provider_services';
-import { getDocViewer } from './accessors';
+import { createGetDocViewer } from './accessors';
 import { OBSERVABILITY_ROOT_PROFILE_ID } from '../consts';
 
 export const createObservabilityLogDocumentProfileProvider = (
   services: ProfileProviderServices
 ): DocumentProfileProvider => ({
   profileId: 'observability-log-document-profile',
   profile: {
-    getDocViewer,
+    getDocViewer: createGetDocViewer(services),
   },
   resolve: ({ record, rootContext }) => {
     if (rootContext.profileId !== OBSERVABILITY_ROOT_PROFILE_ID) {

diff --git a/...bservability_solution/logs_shared/public/components/log_ai_assistant/log_ai_assistant.tsx b/...bservability_solution/logs_shared/public/components/log_ai_assistant/log_ai_assistant.tsx
@@ -35,6 +35,12 @@ export const LogAIAssistant = ({
       return undefined;
     }
 
+    const message = doc.fields.find((field) => field.field === 'message')?.value[0];
+
+    if (!message) {
+      return undefined;
+    }
+
     return getContextualInsightMessages({
       message:
         'Can you explain what this log message means? Where it could be coming from, whether it is expected and whether it is an issue.',
@@ -53,6 +59,10 @@ export const LogAIAssistant = ({
 
     const message = doc.fields.find((field) => field.field === 'message')?.value[0];
 
+    if (!message) {
+      return undefined;
+    }
+
     return getContextualInsightMessages({
       message: `I'm looking at a log entry. Can you construct a Kibana KQL query that I can enter in the search bar that gives me similar log entries, based on the message field?`,
       instructions: JSON.stringify({
@@ -61,7 +71,9 @@ export const LogAIAssistant = ({
     });
   }, [getContextualInsightMessages, doc]);
 
-  return (
+  const hasAtLeastOnePrompt = Boolean(explainLogMessageMessages || similarLogMessageMessages);
+
+  return hasAtLeastOnePrompt ? (
     <EuiFlexGroup direction="column" gutterSize="m">
       {ObservabilityAIAssistantContextualInsight && explainLogMessageMessages ? (
         <EuiFlexItem grow={false}>
@@ -82,7 +94,7 @@ export const LogAIAssistant = ({
         </EuiFlexItem>
       ) : null}
     </EuiFlexGroup>
-  );
+  ) : null;
 };
 
 // eslint-disable-next-line import/no-default-export