api test

x-pack/packages/kbn-ai-assistant/src/chat/welcome_message_knowledge_base_setup_error_panel.tsx

@@ -31,7 +31,7 @@ export function WelcomeMessageKnowledgeBaseSetupErrorPanel({
             onClick={onRetryInstall}
           >
             {i18n.translate(
-              'xpack.observabilityAiAssistant.welcomeMessageKnowledgeBaseSetupErrorPanel.retryInstallingLinkLabel',
+              'xpack.aiAssistant.welcomeMessageKnowledgeBaseSetupErrorPanel.retryInstallingLinkLabel',
               { defaultMessage: 'Retry install knowledge base' }
             )}
           </EuiLink>

x-pack/plugins/observability_solution/observability_ai_assistant/server/plugin.ts

@@ -36,6 +36,7 @@ import { registerFunctions } from './functions';
 import { recallRankingEvent } from './analytics/recall_ranking';
 import { initLangtrace } from './service/client/instrumentation/init_langtrace';
 import { aiAssistantCapabilities } from '../common/capabilities';
+import { registerMigrateKnowledgeBaseEntriesTask } from './service/task_manager_definitions/register_migrate_knowledge_base_entries_task';
 
 export class ObservabilityAIAssistantPlugin
   implements
@@ -156,11 +157,16 @@ export class ObservabilityAIAssistantPlugin
     const service = (this.service = new ObservabilityAIAssistantService({
       logger: this.logger.get('service'),
       core,
-      taskManager: plugins.taskManager,
       getSearchConnectorModelId,
       enableKnowledgeBase: this.config.enableKnowledgeBase,
     }));
 
+    registerMigrateKnowledgeBaseEntriesTask({
+      core,
+      taskManager: plugins.taskManager,
+      logger: this.logger,
+    });
+
     service.register(registerFunctions);
 
     registerServerRoutes({

x-pack/plugins/observability_solution/observability_ai_assistant/server/routes/knowledge_base/route.ts

@@ -22,9 +22,7 @@ const getKnowledgeBaseStatus = createObservabilityAIAssistantServerRoute({
     service,
     request,
   }): Promise<
-    Partial<InferenceEndpointResponse['endpoints'][0]> & {
-      ready: boolean;
-    }
+    { ready: boolean; enabled: boolean } & Partial<InferenceEndpointResponse['endpoints'][0]>
   > => {
     const client = await service.getClient({ request });
 
@@ -44,7 +42,7 @@ const setupKnowledgeBase = createObservabilityAIAssistantServerRoute({
       idleSocket: 20 * 60 * 1000, // 20 minutes
     },
   },
-  handler: async (resources): Promise<unknown> => {
+  handler: async (resources): Promise<any> => {
     const client = await resources.service.getClient({ request: resources.request });
 
     if (!client) {
@@ -252,8 +250,8 @@ const importKnowledgeBaseEntries = createObservabilityAIAssistantServerRoute({
       throw notImplemented();
     }
 
-    const status = await client.getKnowledgeBaseStatus();
-    if (!status.ready) {
+    const { ready } = await client.getKnowledgeBaseStatus();
+    if (!ready) {
       throw new Error('Knowledge base is not ready');
     }
 

x-pack/plugins/observability_solution/observability_ai_assistant/server/routes/register_routes.ts

@@ -9,13 +9,14 @@ import type { Logger } from '@kbn/logging';
 import { registerRoutes } from '@kbn/server-route-repository';
 import { getGlobalObservabilityAIAssistantServerRouteRepository } from './get_global_observability_ai_assistant_route_repository';
 import type { ObservabilityAIAssistantRouteHandlerResources } from './types';
+import { ObservabilityAIAssistantPluginStartDependencies } from '../types';
 
 export function registerServerRoutes({
   core,
   logger,
   dependencies,
 }: {
-  core: CoreSetup;
+  core: CoreSetup<ObservabilityAIAssistantPluginStartDependencies>;
   logger: Logger;
   dependencies: Omit<
     ObservabilityAIAssistantRouteHandlerResources,

x-pack/plugins/observability_solution/observability_ai_assistant/server/service/client/index.ts

@@ -722,7 +722,7 @@ export class ObservabilityAIAssistantClient {
   };
 
   getKnowledgeBaseStatus = () => {
-    return this.dependencies.knowledgeBaseService.getElserModelStatus();
+    return this.dependencies.knowledgeBaseService.getStatus();
   };
 
   setupKnowledgeBase = () => {
@@ -737,9 +737,8 @@ export class ObservabilityAIAssistantClient {
 
   migrateKnowledgeBaseToSemanticText = () => {
     return runSemanticTextKnowledgeBaseMigration({
-      esClient: this.dependencies.esClient.asInternalUser,
+      esClient: this.dependencies.esClient,
       logger: this.dependencies.logger,
-      kbService: this.dependencies.knowledgeBaseService,
     });
   };
 

x-pack/plugins/observability_solution/observability_ai_assistant/server/service/create_inference_endpoint.ts

@@ -5,6 +5,7 @@
  * 2.0.
  */
 
+import { errors } from '@elastic/elasticsearch';
 import { ElasticsearchClient } from '@kbn/core-elasticsearch-server';
 import { Logger } from '@kbn/logging';
 
@@ -102,3 +103,30 @@ export async function getInferenceEndpoint({
     throw e;
   }
 }
+
+export async function getInferenceEndpointStatus({
+  esClient,
+  logger,
+}: {
+  esClient: { asInternalUser: ElasticsearchClient };
+  logger: Logger;
+}) {
+  try {
+    const endpoint = await getInferenceEndpoint({ esClient, logger });
+    const ready = endpoint !== undefined;
+    return { ...endpoint, ready };
+  } catch (error) {
+    if (isInferenceEndpointMissingOrUnavailable(error)) {
+      return { ready: false };
+    }
+    throw error;
+  }
+}
+
+export function isInferenceEndpointMissingOrUnavailable(error: Error) {
+  return (
+    error instanceof errors.ResponseError &&
+    (error.body?.error?.type === 'resource_not_found_exception' ||
+      error.body?.error?.type === 'status_exception')
+  );
+}

x-pack/plugins/observability_solution/observability_ai_assistant/server/service/index.ts

@@ -10,7 +10,6 @@ import { createConcreteWriteIndex, getDataStreamAdapter } from '@kbn/alerting-pl
 import type { CoreSetup, CoreStart, KibanaRequest, Logger } from '@kbn/core/server';
 import type { SecurityPluginStart } from '@kbn/security-plugin/server';
 import { getSpaceIdFromPath } from '@kbn/spaces-plugin/common';
-import type { TaskManagerSetupContract } from '@kbn/task-manager-plugin/server';
 import { once } from 'lodash';
 import type { AssistantScope } from '@kbn/ai-assistant-common';
 import { ObservabilityAIAssistantScreenContextRequest } from '../../common/types';
@@ -57,32 +56,20 @@ export class ObservabilityAIAssistantService {
   constructor({
     logger,
     core,
-    taskManager,
     getSearchConnectorModelId,
     enableKnowledgeBase,
   }: {
     logger: Logger;
     core: CoreSetup<ObservabilityAIAssistantPluginStartDependencies>;
-    taskManager: TaskManagerSetupContract;
     getSearchConnectorModelId: () => Promise<string>;
+    enableKnowledgeBase: boolean;
   }) {
     this.core = core;
     this.logger = logger;
     this.getSearchConnectorModelId = getSearchConnectorModelId;
     this.enableKnowledgeBase = enableKnowledgeBase;
 
     this.registerInit();
-
-    registerMigrateKnowledgeBaseEntriesTask({
-      taskManager,
-      logger,
-      getKbService: () => this.kbService,
-      getTaskManagerStart,
-      getEsClient: async () => {
-        const [coreStart] = await core.getStartServices();
-        return coreStart.elasticsearch.client.asInternalUser;
-      },
-    });
   }
 
   init = async () => {};
@@ -99,7 +86,7 @@ export class ObservabilityAIAssistantService {
   private doInit = async () => {
     try {
       this.logger.info('Setting up index assets');
-      const [coreStart, pluginsStart] = await this.core.getStartServices();
+      const [coreStart] = await this.core.getStartServices();
 
       const esClient = {
         asInternalUser: coreStart.elasticsearch.client.asInternalUser,
@@ -183,7 +170,6 @@ export class ObservabilityAIAssistantService {
       this.kbService = new KnowledgeBaseService({
         logger: this.logger.get('kb'),
         esClient,
-        taskManagerStart: pluginsStart.taskManager,
         getSearchConnectorModelId: this.getSearchConnectorModelId,
         enabled: this.enableKnowledgeBase,
       });

x-pack/plugins/observability_solution/observability_ai_assistant/server/service/kb_component_template.ts

@@ -60,9 +60,9 @@ export const kbComponentTemplate: ClusterComponentTemplate['component_template']
       namespace: keyword,
       text,
       semantic_text: {
-        // @ts-expect-error: @elastic/elasticsearch does not have this type yet
         type: 'semantic_text',
         inference_id: AI_ASSISTANT_KB_INFERENCE_ID,
+        // @ts-expect-error: @elastic/elasticsearch does not have this type yet
         model_settings: {
           task_type: 'sparse_embedding',
         },