[Obs AI Assistant] Use built-in inference endpoints over specific models

x-pack/platform/packages/shared/kbn-ai-assistant/src/chat/welcome_message.tsx

@@ -12,7 +12,6 @@ import type { ActionConnector } from '@kbn/triggers-actions-ui-plugin/public';
 import { GenerativeAIForObservabilityConnectorFeatureId } from '@kbn/actions-plugin/common';
 import { isSupportedConnectorType } from '@kbn/inference-common';
 import { AssistantBeacon } from '@kbn/ai-assistant-icon';
-import { KnowledgeBaseState } from '@kbn/observability-ai-assistant-plugin/public';
 import type { UseKnowledgeBaseResult } from '../hooks/use_knowledge_base';
 import type { UseGenAIConnectorsResult } from '../hooks/use_genai_connectors';
 import { Disclaimer } from './disclaimer';
@@ -61,13 +60,6 @@ export function WelcomeMessage({
     if (isSupportedConnectorType(createdConnector.actionTypeId)) {
       connectors.reloadConnectors();
     }
-
-    if (
-      !knowledgeBase.status.value ||
-      knowledgeBase.status.value?.kbState === KnowledgeBaseState.NOT_INSTALLED
-    ) {
-      knowledgeBase.install();
-    }
   };
 
   const ConnectorFlyout = useMemo(

x-pack/platform/packages/shared/kbn-ai-assistant/src/chat/welcome_message_knowledge_base.tsx

@@ -23,6 +23,8 @@ import usePrevious from 'react-use/lib/usePrevious';
 import { WelcomeMessageKnowledgeBaseSetupErrorPanel } from './welcome_message_knowledge_base_setup_error_panel';
 import { UseKnowledgeBaseResult } from '../hooks';
 
+const inferenceId = '.elser-2-elasticsearch'; // TODO: remove hardcoded inferenceId
+
 const SettingUpKnowledgeBase = () => (
   <>
     <EuiText color="subdued" size="s">
@@ -56,7 +58,7 @@ const InspectKnowledgeBasePopover = ({
 
   const handleInstall = async () => {
     setIsPopoverOpen(false);
-    await knowledgeBase.install();
+    await knowledgeBase.install(inferenceId);
   };
 
   return knowledgeBase.status.value?.modelStats ? (
@@ -101,7 +103,7 @@ export function WelcomeMessageKnowledgeBase({
   }, [knowledgeBase.isInstalling, prevIsInstalling]);
 
   const install = async () => {
-    await knowledgeBase.install();
+    await knowledgeBase.install(inferenceId);
   };
 
   if (knowledgeBase.isInstalling) return <SettingUpKnowledgeBase />;

x-pack/platform/packages/shared/kbn-ai-assistant/src/hooks/use_knowledge_base.test.tsx

@@ -73,14 +73,19 @@ describe('useKnowledgeBase', () => {
 
     // Trigger setup
     act(() => {
-      result.current.install();
+      result.current.install('.elser-2-elasticsearch');
     });
 
     // Verify that the install was called
     await waitFor(() => {
       expect(mockCallApi).toHaveBeenCalledWith(
         'POST /internal/observability_ai_assistant/kb/setup',
         {
+          params: {
+            query: {
+              inference_id: '.elser-2-elasticsearch',
+            },
+          },
           signal: null,
         }
       );

x-pack/platform/packages/shared/kbn-ai-assistant/src/hooks/use_knowledge_base.tsx

@@ -20,7 +20,7 @@ export interface UseKnowledgeBaseResult {
   status: AbortableAsyncState<APIReturnType<'GET /internal/observability_ai_assistant/kb/status'>>;
   isInstalling: boolean;
   isPolling: boolean;
-  install: () => Promise<void>;
+  install: (inferenceId: string) => Promise<void>;
 }
 
 export function useKnowledgeBase(): UseKnowledgeBaseResult {
@@ -44,36 +44,44 @@ export function useKnowledgeBase(): UseKnowledgeBaseResult {
     }
   }, [isInstalling, statusRequest]);
 
-  const install = useCallback(async () => {
-    setIsInstalling(true);
-    try {
-      // Retry the setup with a maximum of 5 attempts
-      await pRetry(
-        async () => {
-          await service.callApi('POST /internal/observability_ai_assistant/kb/setup', {
-            signal: null,
-          });
-        },
-        {
-          retries: 5,
+  const install = useCallback(
+    async (inferenceId: string) => {
+      setIsInstalling(true);
+      try {
+        // Retry the setup with a maximum of 5 attempts
+        await pRetry(
+          async () => {
+            await service.callApi('POST /internal/observability_ai_assistant/kb/setup', {
+              params: {
+                query: {
+                  inference_id: inferenceId,
+                },
+              },
+              signal: null,
+            });
+          },
+          {
+            retries: 5,
+          }
+        );
+        if (ml.mlApi?.savedObjects.syncSavedObjects) {
+          await ml.mlApi.savedObjects.syncSavedObjects();
         }
-      );
-      if (ml.mlApi?.savedObjects.syncSavedObjects) {
-        await ml.mlApi.savedObjects.syncSavedObjects();
-      }
 
-      // Refresh status after installation
-      statusRequest.refresh();
-    } catch (error) {
-      notifications!.toasts.addError(error, {
-        title: i18n.translate('xpack.aiAssistant.errorSettingUpInferenceEndpoint', {
-          defaultMessage: 'Could not create inference endpoint',
-        }),
-      });
-    }
-  }, [ml, service, notifications, statusRequest]);
+        // Refresh status after installation
+        statusRequest.refresh();
+      } catch (error) {
+        notifications!.toasts.addError(error, {
+          title: i18n.translate('xpack.aiAssistant.errorSettingUpKnowledgeBase', {
+            defaultMessage: 'Could not setup knowledge base',
+          }),
+        });
+      }
+    },
+    [ml, service, notifications, statusRequest]
+  );
 
-  // poll the status if isPolling (inference endpoint is created but deployment is not ready)
+  // poll the status if isPolling
   useEffect(() => {
     if (!isPolling) {
       return;

x-pack/platform/packages/shared/kbn-ai-assistant/src/knowledge_base/select_knowledge_base_model.tsx

@@ -0,0 +1,50 @@
+/*
+ * 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; you may not use this file except in compliance with the Elastic License
+ * 2.0.
+ */
+
+import React, { useEffect, useMemo } from 'react';
+import { EuiSelect } from '@elastic/eui';
+
+export function SelectKnowledgeBaseModel({
+  onSelectInferenceId,
+  inferenceId,
+}: {
+  onSelectInferenceId: (inferenceId: string) => void;
+  inferenceId: string;
+}) {
+  const inferenceEndpoints: Array<{ id: string; label: string }> = useMemo(
+    () => [
+      {
+        label: 'ELSER v2 (English-only)',
+        id: '.elser-2-elasticsearch',
+      },
+      {
+        label: 'E5 Small (Multilingual)',
+        id: '.multilingual-e5-small-elasticsearch',
+      },
+    ],
+    []
+  );
+
+  useEffect(() => {
+    if (!inferenceId) {
+      onSelectInferenceId(inferenceEndpoints[0].id);
+    }
+  }, [inferenceId, inferenceEndpoints, onSelectInferenceId]);
+
+  return (
+    <EuiSelect
+      fullWidth
+      data-test-subj="observabilityAiAssistantKbModelSelect"
+      options={inferenceEndpoints.map(({ label, id }) => ({ value: id, text: label }))}
+      onChange={(event) => {
+        const inferenceEndpoint = inferenceEndpoints.find(({ id }) => id === event.target.value)!;
+        onSelectInferenceId(inferenceEndpoint.id);
+      }}
+      value={inferenceId}
+    />
+  );
+}

x-pack/platform/plugins/private/translations/translations/fr-FR.json

@@ -10175,7 +10175,6 @@
     "xpack.aiAssistant.couldNotFindConversationTitle": "Conversation introuvable",
     "xpack.aiAssistant.disclaimer.disclaimerLabel": "Ce chat est alimenté par une intégration avec votre fournisseur LLM. Il arrive que les grands modèles de langage (LLM) présentent comme correctes des informations incorrectes. Elastic prend en charge la configuration ainsi que la connexion au fournisseur LLM et à votre base de connaissances, mais n'est pas responsable des réponses fournies par le LLM.",
     "xpack.aiAssistant.emptyConversationTitle": "Nouvelle conversation",
-    "xpack.aiAssistant.errorSettingUpInferenceEndpoint": "Impossible de créer le point de terminaison d'inférence",
     "xpack.aiAssistant.errorUpdatingConversation": "Impossible de mettre à jour la conversation",
     "xpack.aiAssistant.executedFunctionFailureEvent": "impossible d'exécuter la fonction {functionName}",
     "xpack.aiAssistant.flyout.confirmDeleteButtonText": "Supprimer la conversation",
@@ -18632,6 +18631,9 @@
     "xpack.fleet.agentLogs.logLevelSelectText": "Niveau du log",
     "xpack.fleet.agentLogs.oldAgentWarningTitle": "La vue Logs requiert Elastic Agent 7.11 ou une version ultérieure. Pour mettre à niveau un agent, accédez au menu Actions ou {downloadLink} une version plus récente.",
     "xpack.fleet.agentLogs.openInDiscoverUiLinkText": "Ouvrir dans Discover",
+    "xpack.fleet.agentLogs.resetLogLevel.errorTitleText": "Erreur lors de la réinitialisation du niveau de logging de l'agent",
+    "xpack.fleet.agentLogs.resetLogLevel.successText": "Réinitialiser le niveau de logging de l'agent pour la politique",
+    "xpack.fleet.agentLogs.resetLogLevelLabelText": "Réinitialiser pour rétablir la politique",
     "xpack.fleet.agentLogs.searchPlaceholderText": "Rechercher dans les logs…",
     "xpack.fleet.agentLogs.selectLogLevel.errorTitleText": "Erreur lors de la mise à jour du niveau de logging de l'agent",
     "xpack.fleet.agentLogs.selectLogLevel.successText": "Modification du niveau de logging de l'agent en \"{logLevel}\"",
@@ -27807,6 +27809,7 @@
     "xpack.ml.dataVisualizer.pageHeader": "Data Visualizer (Visualiseur de données)",
     "xpack.ml.datavisualizer.selector.dataVisualizerDescription": "L'outil de Machine Learning Data Visualizer (Visualiseur de données) vous aide à comprendre vos données en analysant les indicateurs et les champs dans un fichier log ou un index Elasticsearch existant.",
     "xpack.ml.datavisualizer.selector.dataVisualizerTitle": "Data Visualizer (Visualiseur de données)",
+    "xpack.ml.datavisualizer.selector.esqlTechnicalPreviewBadge.titleMsg": "Le visualiseur de données ES|QL est en version préliminaire technique.",
     "xpack.ml.datavisualizer.selector.importDataTitle": "Visualiser les données à partir d'un fichier",
     "xpack.ml.datavisualizer.selector.selectDataViewButtonLabel": "Sélectionner la vue de données",
     "xpack.ml.datavisualizer.selector.selectDataViewTitle": "Visualiser les données à partir d'une vue de données",
@@ -27853,6 +27856,7 @@
     "xpack.ml.deepLink.overview": "Aperçu",
     "xpack.ml.deepLink.resultExplorer": "Explorateur de résultats",
     "xpack.ml.deepLink.singleMetricViewer": "Visionneuse d’indicateur unique",
+    "xpack.ml.deepLink.suppliedConfigurations": "Configurations fournies",
     "xpack.ml.deleteSpaceAwareItemCheckModal.buttonTextCanDelete.job": "Continuer pour supprimer {length, plural, one {# tâche} other {# tâches}}",
     "xpack.ml.deleteSpaceAwareItemCheckModal.buttonTextCanDelete.model": "Continuer pour supprimer {length, plural, one {# modèle} other {# modèles}}",
     "xpack.ml.deleteSpaceAwareItemCheckModal.buttonTextCanUnTagConfirm": "Retirer de l'espace en cours",
@@ -32675,6 +32679,20 @@
     "xpack.observabilityShared.bottomBarActions.unsavedChanges": "{unsavedChangesCount, plural, =0{0 modification non enregistrée} one {1 modification non enregistrée} other {# modifications non enregistrées}}",
     "xpack.observabilityShared.breadcrumbs.observabilityLinkText": "Observabilité",
     "xpack.observabilityShared.common.constants.grouping": "Observabilité",
+    "xpack.observabilityShared.experimentalOnboardingFlow.browseDocumentationFlexItemDescription": "Guides détaillés des fonctionnalités d'Elastic",
+    "xpack.observabilityShared.experimentalOnboardingFlow.browseDocumentationFlexItemLabel": "Parcourir la documentation",
+    "xpack.observabilityShared.experimentalOnboardingFlow.browseDocumentationFlexItemLinkARIALabel": "En savoir plus sur toutes les fonctionnalités d'Elastic",
+    "xpack.observabilityShared.experimentalOnboardingFlow.browseDocumentationFlexItemLinkLabel": "En savoir plus",
+    "xpack.observabilityShared.experimentalOnboardingFlow.demoEnvironmentFlexItemDescription": "Explorer notre environnement de démonstration en direct",
+    "xpack.observabilityShared.experimentalOnboardingFlow.demoEnvironmentFlexItemLabel": "Environnement de démonstration",
+    "xpack.observabilityShared.experimentalOnboardingFlow.demoEnvironmentFlexItemLinkLabel": "Explorer la démonstration",
+    "xpack.observabilityShared.experimentalOnboardingFlow.exploreForumFlexItemDescription": "Échanger à propos d'Elastic",
+    "xpack.observabilityShared.experimentalOnboardingFlow.exploreForumFlexItemLabel": "Explorer le forum",
+    "xpack.observabilityShared.experimentalOnboardingFlow.exploreForumFlexItemLinkARIALabel": "Ouvrir le forum de discussion sur Elastic",
+    "xpack.observabilityShared.experimentalOnboardingFlow.exploreForumFlexItemLinkLabel": "Forum de discussion",
+    "xpack.observabilityShared.experimentalOnboardingFlow.supportHubFlexItemDescription": "Obtenez de l'aide dans l’ouverture d’un cas",
+    "xpack.observabilityShared.experimentalOnboardingFlow.supportHubFlexItemLabel": "Hub de support technique",
+    "xpack.observabilityShared.experimentalOnboardingFlow.supportHubFlexItemLinkLabel": "Ouvrir le Hub de support technique",
     "xpack.observabilityShared.featureFeedbackButton.tellUsWhatYouThinkLink": "Dites-nous ce que vous pensez !",
     "xpack.observabilityShared.fieldValueSelection.apply": "Appliquer",
     "xpack.observabilityShared.fieldValueSelection.apply.label": "Appliquer les filtres sélectionnés pour {label}",

x-pack/platform/plugins/private/translations/translations/ja-JP.json

@@ -10166,7 +10166,6 @@
     "xpack.aiAssistant.couldNotFindConversationTitle": "会話が見つかりません",
     "xpack.aiAssistant.disclaimer.disclaimerLabel": "この会話は、LLMプロバイダーとの統合によって提供されています。LLMは、正しくない情報を正しい情報であるかのように表示する場合があることが知られています。Elasticは、構成やLLMプロバイダーへの接続、お客様のナレッジベースへの接続はサポートしますが、LLMの応答については責任を負いません。",
     "xpack.aiAssistant.emptyConversationTitle": "新しい会話",
-    "xpack.aiAssistant.errorSettingUpInferenceEndpoint": "推論エンドポイントを作成できませんでした",
     "xpack.aiAssistant.errorUpdatingConversation": "会話を更新できませんでした",
     "xpack.aiAssistant.executedFunctionFailureEvent": "関数{functionName}の実行に失敗しました",
     "xpack.aiAssistant.flyout.confirmDeleteButtonText": "会話を削除",
@@ -18608,6 +18607,9 @@
     "xpack.fleet.agentLogs.logLevelSelectText": "ログレベル",
     "xpack.fleet.agentLogs.oldAgentWarningTitle": "ログの表示には、Elastic Agent 7.11以降が必要です。エージェントをアップグレードするには、［アクション］メニューに移動するか、新しいバージョンを{downloadLink}。",
     "xpack.fleet.agentLogs.openInDiscoverUiLinkText": "Discoverで開く",
+    "xpack.fleet.agentLogs.resetLogLevel.errorTitleText": "エージェントログレベルのリセットエラー",
+    "xpack.fleet.agentLogs.resetLogLevel.successText": "エージェントログレベルをポリシーにリセット",
+    "xpack.fleet.agentLogs.resetLogLevelLabelText": "ポリシーにリセット",
     "xpack.fleet.agentLogs.searchPlaceholderText": "ログを検索…",
     "xpack.fleet.agentLogs.selectLogLevel.errorTitleText": "エージェントログレベルの更新エラー",
     "xpack.fleet.agentLogs.selectLogLevel.successText": "エージェントログレベルを''{logLevel}''に変更しました",
@@ -27831,6 +27833,7 @@
     "xpack.ml.deepLink.overview": "概要",
     "xpack.ml.deepLink.resultExplorer": "結果エクスプローラー",
     "xpack.ml.deepLink.singleMetricViewer": "シングルメトリックビューアー",
+    "xpack.ml.deepLink.suppliedConfigurations": "提供された構成",
     "xpack.ml.deleteSpaceAwareItemCheckModal.buttonTextCanDelete.job": "続行して、{length, plural, other  {# 個のジョブ}}を削除します",
     "xpack.ml.deleteSpaceAwareItemCheckModal.buttonTextCanDelete.model": "続行して、{length, plural, other {# 個のモデル}}を削除します",
     "xpack.ml.deleteSpaceAwareItemCheckModal.buttonTextCanUnTagConfirm": "現在のスペースから削除",
@@ -32655,6 +32658,20 @@
     "xpack.observabilityShared.bottomBarActions.unsavedChanges": "{unsavedChangesCount, plural, other  {# 未保存変更}}",
     "xpack.observabilityShared.breadcrumbs.observabilityLinkText": "Observability",
     "xpack.observabilityShared.common.constants.grouping": "Observability",
+    "xpack.observabilityShared.experimentalOnboardingFlow.browseDocumentationFlexItemDescription": "すべてのElastic機能に関する詳細なガイド",
+    "xpack.observabilityShared.experimentalOnboardingFlow.browseDocumentationFlexItemLabel": "ドキュメントを参照",
+    "xpack.observabilityShared.experimentalOnboardingFlow.browseDocumentationFlexItemLinkARIALabel": "すべてのElastic機能の詳細",
+    "xpack.observabilityShared.experimentalOnboardingFlow.browseDocumentationFlexItemLinkLabel": "詳細",
+    "xpack.observabilityShared.experimentalOnboardingFlow.demoEnvironmentFlexItemDescription": "Elasticのライブデモを見る",
+    "xpack.observabilityShared.experimentalOnboardingFlow.demoEnvironmentFlexItemLabel": "デモ環境",
+    "xpack.observabilityShared.experimentalOnboardingFlow.demoEnvironmentFlexItemLinkLabel": "デモの探索",
+    "xpack.observabilityShared.experimentalOnboardingFlow.exploreForumFlexItemDescription": "Elasticに関する意見を交換",
+    "xpack.observabilityShared.experimentalOnboardingFlow.exploreForumFlexItemLabel": "フォーラムを探索",
+    "xpack.observabilityShared.experimentalOnboardingFlow.exploreForumFlexItemLinkARIALabel": "Elasticディスカッションフォーラムを開く",
+    "xpack.observabilityShared.experimentalOnboardingFlow.exploreForumFlexItemLinkLabel": "ディスカッションフォーラム",
+    "xpack.observabilityShared.experimentalOnboardingFlow.supportHubFlexItemDescription": "ケースを作成してヘルプを依頼",
+    "xpack.observabilityShared.experimentalOnboardingFlow.supportHubFlexItemLabel": "サポートハブ",
+    "xpack.observabilityShared.experimentalOnboardingFlow.supportHubFlexItemLinkLabel": "サポートハブを開く",
     "xpack.observabilityShared.featureFeedbackButton.tellUsWhatYouThinkLink": "ご意見をお聞かせください。",
     "xpack.observabilityShared.fieldValueSelection.apply": "適用",
     "xpack.observabilityShared.fieldValueSelection.apply.label": "{label}に選択したフィルターを適用",