From 7387e47eb3ab27941a0b6c34672debddc6f64aad Mon Sep 17 00:00:00 2001
From: pgayvallet <pierre.gayvallet@elastic.co>
Date: Wed, 30 Oct 2024 13:15:16 +0100
Subject: [PATCH] last type fixes

---
 .../common/convert_messages_for_inference.ts               | 4 ++--
 .../server/functions/query/index.ts                        | 7 ++-----
 .../server/assistant/tools/esql/nl_to_esql_tool.ts         | 6 +-----
 3 files changed, 5 insertions(+), 12 deletions(-)

diff --git a/x-pack/plugins/observability_solution/observability_ai_assistant_app/common/convert_messages_for_inference.ts b/x-pack/plugins/observability_solution/observability_ai_assistant_app/common/convert_messages_for_inference.ts
index 1dc8638626d0b..7ab9516440988 100644
--- a/x-pack/plugins/observability_solution/observability_ai_assistant_app/common/convert_messages_for_inference.ts
+++ b/x-pack/plugins/observability_solution/observability_ai_assistant_app/common/convert_messages_for_inference.ts
@@ -10,8 +10,8 @@ import {
   AssistantMessage,
   Message as InferenceMessage,
   MessageRole as InferenceMessageRole,
-  generateFakeToolCallId,
-} from '@kbn/inference-plugin/common';
+} from '@kbn/inference-common';
+import { generateFakeToolCallId } from '@kbn/inference-plugin/common';
 
 export function convertMessagesForInference(messages: Message[]): InferenceMessage[] {
   const inferenceMessages: InferenceMessage[] = [];
diff --git a/x-pack/plugins/observability_solution/observability_ai_assistant_app/server/functions/query/index.ts b/x-pack/plugins/observability_solution/observability_ai_assistant_app/server/functions/query/index.ts
index 3643c54365248..210dee20339af 100644
--- a/x-pack/plugins/observability_solution/observability_ai_assistant_app/server/functions/query/index.ts
+++ b/x-pack/plugins/observability_solution/observability_ai_assistant_app/server/functions/query/index.ts
@@ -5,11 +5,8 @@
  * 2.0.
  */
 
-import {
-  correctCommonEsqlMistakes,
-  isChatCompletionChunkEvent,
-  isOutputEvent,
-} from '@kbn/inference-plugin/common';
+import { isChatCompletionChunkEvent, isOutputEvent } from '@kbn/inference-common';
+import { correctCommonEsqlMistakes } from '@kbn/inference-plugin/common';
 import { naturalLanguageToEsql } from '@kbn/inference-plugin/server';
 import {
   FunctionVisibility,
diff --git a/x-pack/plugins/security_solution/server/assistant/tools/esql/nl_to_esql_tool.ts b/x-pack/plugins/security_solution/server/assistant/tools/esql/nl_to_esql_tool.ts
index 1205fb03b0458..11ca8ffd94edd 100644
--- a/x-pack/plugins/security_solution/server/assistant/tools/esql/nl_to_esql_tool.ts
+++ b/x-pack/plugins/security_solution/server/assistant/tools/esql/nl_to_esql_tool.ts
@@ -49,11 +49,7 @@ export const NL_TO_ESQL_TOOL: AssistantTool = {
           connectorId,
           input: question,
           ...(isOssModel ? { functionCalling: 'simulated' } : {}),
-          logger: {
-            debug: (source) => {
-              logger.debug(typeof source === 'function' ? source() : source);
-            },
-          },
+          logger,
         })
       );
     };