fix circular dep

stephmilovic · stephmilovic · commit 62d0fba916ae · 2024-09-17T09:18:03.000-06:00
diff --git a/x-pack/packages/kbn-langchain/server/language_models/chat_vertex/chat_vertex.test.ts b/x-pack/packages/kbn-langchain/server/language_models/chat_vertex/chat_vertex.test.ts
@@ -9,10 +9,9 @@ import { PassThrough } from 'stream';
 import { loggerMock } from '@kbn/logging-mocks';
 import { actionsClientMock } from '@kbn/actions-plugin/server/actions_client/actions_client.mock';
 
+import { BaseMessage, HumanMessage, SystemMessage } from '@langchain/core/messages';
 import { ActionsClientChatVertexAI } from './chat_vertex';
-import { BaseMessage } from '@langchain/core/messages';
 import { CallbackManagerForLLMRun } from '@langchain/core/callbacks/manager';
-import { getLangChainMessage } from '@kbn/elastic-assistant-plugin/server/lib/langchain/helpers';
 
 const connectorId = 'mock-connector-id';
 
@@ -56,11 +55,8 @@ const mockStreamExecute = jest.fn().mockImplementation(() => {
 });
 
 const callMessages = [
-  getLangChainMessage({
-    role: 'system',
-    content: 'Answer the following questions truthfully and as best you can.',
-  }),
-  getLangChainMessage({ role: 'user', content: 'Question: Do you know my name?\n\n' }),
+  new SystemMessage('Answer the following questions truthfully and as best you can.'),
+  new HumanMessage('Question: Do you know my name?\n\n'),
 ] as unknown as BaseMessage[];
 
 const callOptions = {
diff --git a/x-pack/packages/kbn-langchain/tsconfig.json b/x-pack/packages/kbn-langchain/tsconfig.json
@@ -19,7 +19,6 @@
     "@kbn/actions-plugin",
     "@kbn/logging-mocks",
     "@kbn/utility-types",
-    "@kbn/tooling-log",
-    "@kbn/elastic-assistant-plugin"
+    "@kbn/tooling-log"
   ]
 }

Original file line number	Diff line number	Diff line change
`@@ -19,7 +19,6 @@`
`19`	`19`	`"@kbn/actions-plugin",`
`20`	`20`	`"@kbn/logging-mocks",`
`21`	`21`	`"@kbn/utility-types",`
`22`		`- "@kbn/tooling-log",`
`23`		`- "@kbn/elastic-assistant-plugin"`
	`22`	`+ "@kbn/tooling-log"`
`24`	`23`	`]`
`25`	`24`	`}`