[Automatic Import] Use InferenceChatModel (#233365)

## Summary

Use `InferenceChatModel` as default ChatModel.

## Testing
- Run Automatic Import flow with Bedrock , OpenAI , Gemini Connectors
and verify the flow goes through.
- Verify in Langsmith Traces that InferenceChatModel is used.

<img width="353" height="184" alt="image"
src="https://github.com/user-attachments/assets/181c7bf2-4125-4d4d-9f8c-6d53252f9706"
/>


<img width="379" height="147" alt="image"
src="https://github.com/user-attachments/assets/35c6137c-e64f-498c-bbbc-67b8cd07c684"
/>


<img width="361" height="187" alt="image"
src="https://github.com/user-attachments/assets/e380837a-bc54-41dc-8290-013fa00a0fae"
/>


<img width="386" height="197" alt="image"
src="https://github.com/user-attachments/assets/cf7b3301-b706-44a7-a0b8-eaca26b2ef06"
/>

<img width="274" height="186" alt="image"
src="https://github.com/user-attachments/assets/9c543dad-5ae5-4b63-836c-9e03f3badc7d"
/>

<img width="333" height="140" alt="image"
src="https://github.com/user-attachments/assets/b4ce3d09-7b5b-4e43-b0c8-562393e528a2"
/>

---------

Co-authored-by: kibanamachine <[email protected]>

Author: bhapas

x-pack/platform/plugins/shared/automatic_import/kibana.jsonc

@@ -19,6 +19,7 @@
       "triggersActionsUi",
       "actions",
       "stackConnectors",
+      "inference",
     ],
     "requiredBundles": [
       "kibanaUtils"

x-pack/platform/plugins/shared/automatic_import/scripts/draw_graphs_script.ts

@@ -6,15 +6,12 @@
  */
 
 import type { IScopedClusterClient } from '@kbn/core-elasticsearch-server';
-import type {
-  ActionsClientChatOpenAI,
-  ActionsClientSimpleChatModel,
-} from '@kbn/langchain/server/language_models';
 import { ToolingLog } from '@kbn/tooling-log';
 import type { Graph as RunnableGraph } from '@langchain/core/runnables/graph';
 import { FakeLLM } from '@langchain/core/utils/testing';
 import fs from 'fs/promises';
 import path from 'path';
+import type { InferenceChatModel } from '@kbn/inference-langchain';
 import { getCategorizationGraph } from '../server/graphs/categorization/graph';
 import { getEcsGraph, getEcsSubGraph } from '../server/graphs/ecs/graph';
 import { getLogFormatDetectionGraph } from '../server/graphs/log_type_detection/graph';
@@ -27,7 +24,7 @@ import { getApiAnalysisGraph } from '../server/graphs/api_analysis';
 // Some mock elements just to get the graph to compile
 const model = new FakeLLM({
   response: JSON.stringify({}, null, 2),
-}) as unknown as ActionsClientChatOpenAI | ActionsClientSimpleChatModel;
+}) as unknown as InferenceChatModel;
 const client = 'test' as unknown as IScopedClusterClient;
 
 const logger = new ToolingLog({

x-pack/platform/plugins/shared/automatic_import/server/__mocks__/request_context.ts

@@ -5,13 +5,9 @@
  * 2.0.
  */
 import { coreMock } from '@kbn/core/server/mocks';
-import type { PluginStartContract as ActionsPluginStart } from '@kbn/actions-plugin/server';
 import { loggerMock } from '@kbn/logging-mocks';
 import { FakeLLM } from '@langchain/core/utils/testing';
-import type {
-  ActionsClientChatOpenAI,
-  ActionsClientSimpleChatModel,
-} from '@kbn/langchain/server/language_models';
+import type { InferenceChatModel } from '@kbn/inference-langchain';
 
 export const createMockClients = () => {
   const core = coreMock.createRequestHandlerContext();
@@ -31,32 +27,7 @@ const convertRequestContextMock = <T extends Record<string, unknown>>(context: T
 
 const mockLlm = new FakeLLM({
   response: JSON.stringify({}, null, 2),
-}) as unknown as ActionsClientChatOpenAI | ActionsClientSimpleChatModel;
-
-jest.mock('@kbn/langchain/server/language_models', () => {
-  return {
-    ActionsClientSimpleChatModel: jest.fn().mockImplementation(() => {
-      return mockLlm;
-    }),
-    ActionsClientChatOpenAI: jest.fn().mockImplementation(() => {
-      return mockLlm;
-    }),
-  };
-});
-
-const actions = {
-  getActionsClientWithRequest: jest.fn().mockResolvedValue({
-    get: jest.fn().mockResolvedValue({
-      mockLlm,
-    }),
-    execute: jest.fn().mockResolvedValue({
-      status: 'ok',
-      data: {
-        message: '{"Answer": "testAction"}',
-      },
-    }),
-  }),
-} as unknown as ActionsPluginStart;
+}) as unknown as InferenceChatModel;
 
 const coreSetupMock = coreMock.createSetup();
 const createRequestContextMock = (clients: MockClients = createMockClients()) => {
@@ -67,7 +38,11 @@ const createRequestContextMock = (clients: MockClients = createMockClients()) =>
           return [
             {},
             {
-              actions,
+              inference: {
+                getChatModel: jest.fn().mockResolvedValue({
+                  mockLlm,
+                }),
+              },
             },
           ];
         }

x-pack/platform/plugins/shared/automatic_import/server/graphs/api_analysis/graph.test.ts

@@ -10,16 +10,13 @@ import {
   apiAnalysisPathSuggestionsMockedResponse,
   apiAnalysisExpectedResults,
 } from '../../../__jest__/fixtures/api_analysis';
-import type {
-  ActionsClientChatOpenAI,
-  ActionsClientSimpleChatModel,
-} from '@kbn/langchain/server/language_models';
 import { mockedApiAnalysisRequest } from '../../../__jest__/fixtures';
 import { handleGetSuggestedPaths } from './paths';
+import type { InferenceChatModel } from '@kbn/inference-langchain';
 
 const model = new FakeLLM({
   response: "I'll callback later.",
-}) as unknown as ActionsClientChatOpenAI | ActionsClientSimpleChatModel;
+}) as unknown as InferenceChatModel;
 
 jest.mock('./paths');
 

x-pack/platform/plugins/shared/automatic_import/server/graphs/api_analysis/paths.test.ts

@@ -4,18 +4,15 @@
  * 2.0; you may not use this file except in compliance with the Elastic License
  * 2.0.
  */
-import type {
-  ActionsClientChatOpenAI,
-  ActionsClientSimpleChatModel,
-} from '@kbn/langchain/server/language_models';
 import { FakeLLM } from '@langchain/core/utils/testing';
 import { apiAnalysisTestState } from '../../../__jest__/fixtures/api_analysis';
 import type { ApiAnalysisState } from '../../types';
 import { handleGetSuggestedPaths } from './paths';
+import type { InferenceChatModel } from '@kbn/inference-langchain';
 
 const model = new FakeLLM({
   response: '[ "/path1", "/path2" ]',
-}) as unknown as ActionsClientChatOpenAI | ActionsClientSimpleChatModel;
+}) as unknown as InferenceChatModel;
 
 const state: ApiAnalysisState = apiAnalysisTestState;
 

x-pack/platform/plugins/shared/automatic_import/server/graphs/categorization/categorization.test.ts

@@ -13,14 +13,11 @@ import {
   categorizationMockProcessors,
   categorizationExpectedHandlerResponse,
 } from '../../../__jest__/fixtures/categorization';
-import type {
-  ActionsClientChatOpenAI,
-  ActionsClientSimpleChatModel,
-} from '@kbn/langchain/server/language_models';
+import type { InferenceChatModel } from '@kbn/inference-langchain';
 
 const model = new FakeLLM({
   response: JSON.stringify(categorizationMockProcessors, null, 2),
-}) as unknown as ActionsClientChatOpenAI | ActionsClientSimpleChatModel;
+}) as unknown as InferenceChatModel;
 
 const state: CategorizationState = categorizationTestState;
 

x-pack/platform/plugins/shared/automatic_import/server/graphs/categorization/errors.test.ts

@@ -13,14 +13,11 @@ import {
   categorizationMockProcessors,
   categorizationExpectedHandlerResponse,
 } from '../../../__jest__/fixtures/categorization';
-import type {
-  ActionsClientChatOpenAI,
-  ActionsClientSimpleChatModel,
-} from '@kbn/langchain/server/language_models';
+import type { InferenceChatModel } from '@kbn/inference-langchain';
 
 const model = new FakeLLM({
   response: JSON.stringify(categorizationMockProcessors, null, 2),
-}) as unknown as ActionsClientChatOpenAI | ActionsClientSimpleChatModel;
+}) as unknown as InferenceChatModel;
 
 const state: CategorizationState = categorizationTestState;
 

x-pack/platform/plugins/shared/automatic_import/server/graphs/categorization/graph.test.ts

@@ -27,15 +27,12 @@ import { handleErrors } from './errors';
 import { handleInvalidCategorization } from './invalid';
 import { handleUpdateStableSamples } from './stable';
 import { testPipeline, combineProcessors } from '../../util';
-import type {
-  ActionsClientChatOpenAI,
-  ActionsClientSimpleChatModel,
-} from '@kbn/langchain/server/language_models';
 import { handleValidatePipeline } from '../../util/graph';
+import type { InferenceChatModel } from '@kbn/inference-langchain';
 
 const model = new FakeLLM({
   response: "I'll callback later.",
-}) as unknown as ActionsClientChatOpenAI | ActionsClientSimpleChatModel;
+}) as unknown as InferenceChatModel;
 
 jest.mock('./errors');
 jest.mock('./review');

x-pack/platform/plugins/shared/automatic_import/server/graphs/categorization/invalid.test.ts

@@ -13,14 +13,11 @@ import {
   categorizationMockProcessors,
   categorizationExpectedHandlerResponse,
 } from '../../../__jest__/fixtures/categorization';
-import type {
-  ActionsClientChatOpenAI,
-  ActionsClientSimpleChatModel,
-} from '@kbn/langchain/server/language_models';
+import type { InferenceChatModel } from '@kbn/inference-langchain';
 
 const model = new FakeLLM({
   response: JSON.stringify(categorizationMockProcessors, null, 2),
-}) as unknown as ActionsClientChatOpenAI | ActionsClientSimpleChatModel;
+}) as unknown as InferenceChatModel;
 
 const state: CategorizationState = categorizationTestState;
 

x-pack/platform/plugins/shared/automatic_import/server/graphs/categorization/review.test.ts

@@ -13,14 +13,11 @@ import {
   categorizationMockProcessors,
   categorizationExpectedHandlerResponse,
 } from '../../../__jest__/fixtures/categorization';
-import type {
-  ActionsClientChatOpenAI,
-  ActionsClientSimpleChatModel,
-} from '@kbn/langchain/server/language_models';
+import type { InferenceChatModel } from '@kbn/inference-langchain';
 
 const model = new FakeLLM({
   response: JSON.stringify(categorizationMockProcessors, null, 2),
-}) as unknown as ActionsClientChatOpenAI | ActionsClientSimpleChatModel;
+}) as unknown as InferenceChatModel;
 
 const state: CategorizationState = categorizationTestState;