github-mirrors/kibana
1
0
Fork 0
mirror of https://github.com/elastic/kibana.git synced 2025-04-23 09:19:04 -04:00
Code Issues Projects Releases Packages Wiki Activity

[Rule Migration] Add inference connector as supported LLM type (#208032)

Browse source 
## Summary

Summarize your PR. If it involves visual changes include a screenshot or
gif.

Adds .inference as a supported type, so it can be tested with EIS both
with custom providers and the default EIS provider.
This commit is contained in:
Marius Iversen 2025-01-23 14:44:26 +01:00 • committed by GitHub
parent cc38fbea29
commit 811c539fff
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
x-pack/solutions/security/plugins/security_solution/public/onboarding/components/onboarding_body/cards/common/connectors/constants.ts
View file

@ -5,4 +5,4 @@
* 2.0.
*/
export const AIActionTypeIds = ['.bedrock', '.gen-ai', '.gemini'];
export const AIActionTypeIds = ['.bedrock', '.gen-ai', '.gemini', '.inference'];

8
x-pack/solutions/security/plugins/security_solution/server/lib/siem_migrations/rules/task/util/actions_client_chat.ts
View file

@ -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<string, string> = {
[`.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;
}