last type fixes

elastic · Oct 30, 2024 · 7387e47 · 7387e47
1 parent ac2aedc
commit 7387e47
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 12 deletions.
diff --git a/...vability_solution/observability_ai_assistant_app/common/convert_messages_for_inference.ts b/...vability_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/...ins/observability_solution/observability_ai_assistant_app/server/functions/query/index.ts b/...ins/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
@@ -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,
         })
       );
     };