From b266c660a6cbb8940dd2bcecf7c5d6e7517f2b11 Mon Sep 17 00:00:00 2001 From: Marius Iversen Date: Thu, 23 Jan 2025 12:47:41 +0100 Subject: [PATCH 1/2] add inference as supported LLM type --- .../rules/task/util/actions_client_chat.ts | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/x-pack/solutions/security/plugins/security_solution/server/lib/siem_migrations/rules/task/util/actions_client_chat.ts b/x-pack/solutions/security/plugins/security_solution/server/lib/siem_migrations/rules/task/util/actions_client_chat.ts index 1659862543078..555662c8312c9 100644 --- a/x-pack/solutions/security/plugins/security_solution/server/lib/siem_migrations/rules/task/util/actions_client_chat.ts +++ b/x-pack/solutions/security/plugins/security_solution/server/lib/siem_migrations/rules/task/util/actions_client_chat.ts @@ -5,16 +5,16 @@ * 2.0. */ +import type { ActionsClient } from '@kbn/actions-plugin/server'; +import type { Logger } from '@kbn/core/server'; import type { ActionsClientSimpleChatModel } from '@kbn/langchain/server'; import { ActionsClientBedrockChatModel, ActionsClientChatOpenAI, ActionsClientChatVertexAI, } from '@kbn/langchain/server'; -import type { Logger } from '@kbn/core/server'; -import type { ActionsClient } from '@kbn/actions-plugin/server'; -import type { ActionsClientChatOpenAIParams } from '@kbn/langchain/server/language_models/chat_openai'; import type { CustomChatModelInput as ActionsClientBedrockChatModelParams } from '@kbn/langchain/server/language_models/bedrock_chat'; +import type { ActionsClientChatOpenAIParams } from '@kbn/langchain/server/language_models/chat_openai'; import type { CustomChatModelInput as ActionsClientChatVertexAIParams } from '@kbn/langchain/server/language_models/gemini_chat'; import type { CustomChatModelInput as ActionsClientSimpleChatModelParams } from '@kbn/langchain/server/language_models/simple_chat_model'; @@ -39,6 +39,7 @@ const llmTypeDictionary: Record = { [`.gen-ai`]: `openai`, [`.bedrock`]: `bedrock`, [`.gemini`]: `gemini`, + [`.inference`]: `inference`, }; export class ActionsClientChat { @@ -83,6 +84,7 @@ export class ActionsClientChat { case 'gemini': return ActionsClientChatVertexAI; case 'openai': + case 'inference': default: return ActionsClientChatOpenAI; } From 2ab030e4e7fb3b3b5ecb812ebb68291bc751cadf Mon Sep 17 00:00:00 2001 From: Marius Iversen Date: Thu, 23 Jan 2025 12:50:30 +0100 Subject: [PATCH 2/2] add support for public UI side as well --- .../onboarding_body/cards/common/connectors/constants.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/x-pack/solutions/security/plugins/security_solution/public/onboarding/components/onboarding_body/cards/common/connectors/constants.ts b/x-pack/solutions/security/plugins/security_solution/public/onboarding/components/onboarding_body/cards/common/connectors/constants.ts index 5c9c94e369854..1a41d3e5562dc 100644 --- a/x-pack/solutions/security/plugins/security_solution/public/onboarding/components/onboarding_body/cards/common/connectors/constants.ts +++ b/x-pack/solutions/security/plugins/security_solution/public/onboarding/components/onboarding_body/cards/common/connectors/constants.ts @@ -5,4 +5,4 @@ * 2.0. */ -export const AIActionTypeIds = ['.bedrock', '.gen-ai', '.gemini']; +export const AIActionTypeIds = ['.bedrock', '.gen-ai', '.gemini', '.inference'];