Fix KB recall for inference endpoint connectors

Author: yuliia-fryshko

x-pack/platform/plugins/shared/observability_ai_assistant/server/functions/context/context.ts

@@ -35,9 +35,6 @@ export function registerContextFunction({
     async ({ connectorId, messages, screenContexts, chat }, signal) => {
       const { request, plugins } = resources;
       const { analytics } = await plugins.core.start();
-      const actionsClient = await (
-        await plugins.actions.start()
-      ).getActionsClientWithRequest(request);
 
       async function getContext() {
         const screenDescription = compact(
@@ -64,10 +61,8 @@ export function registerContextFunction({
           return { content };
         }
 
-        const connector = await actionsClient.get({
-          id: connectorId,
-          throwIfSystemAction: true,
-        });
+        const inferenceStart = await plugins.inference.start();
+        const connector = await inferenceStart.getConnectorById(connectorId, request);
 
         const { llmScores, relevantDocuments, suggestions } = await recallAndScore({
           recall: client.recall,

x-pack/platform/plugins/shared/observability_ai_assistant/server/functions/context/utils/recall_and_score.ts

@@ -8,7 +8,7 @@
 import type { Logger } from '@kbn/logging';
 import type { AnalyticsServiceStart } from '@kbn/core/server';
 import type { AssistantScope } from '@kbn/ai-assistant-common';
-import type { Connector } from '@kbn/actions-plugin/server';
+import type { InferenceConnector } from '@kbn/inference-common';
 import { scoreSuggestions } from './score_suggestions';
 import type { Message } from '../../../../common';
 import { getInferenceConnectorInfo } from '../../../../common/utils/get_inference_connector';
@@ -36,7 +36,7 @@ export async function recallAndScore({
   chat: FunctionCallChatFunction;
   analytics: AnalyticsServiceStart;
   scopes: AssistantScope[];
-  connector?: Connector;
+  connector?: InferenceConnector;
   screenDescription: string;
   messages: Message[];
   logger: Logger;

x-pack/platform/plugins/shared/observability_ai_assistant/server/routes/chat/route.ts

@@ -203,16 +203,10 @@ const chatRecallRoute = createObservabilityAIAssistantServerRoute({
 
     const { request, plugins } = resources;
 
-    const actionsClient = await (
-      await plugins.actions.start()
-    ).getActionsClientWithRequest(request);
-
     const { connectorId, screenDescription, messages, scopes } = resources.params.body;
 
-    const connector = await actionsClient.get({
-      id: connectorId,
-      throwIfSystemAction: true,
-    });
+    const inferenceStart = await plugins.inference.start();
+    const connector = await inferenceStart.getConnectorById(connectorId, request);
     const response$ = from(
       recallAndScore({
         analytics: (await resources.plugins.core.start()).analytics,