Skip to content

Commit

Permalink
## [Security Solution] [Elastic AI Assistant] Delete the _Retrieval A…
Browse files Browse the repository at this point in the history
…ugmented Generation (RAG) for Alerts_ Feature Flag

This PR deletes the `assistantRagOnAlerts` feature flag introduced in [[Security Solution] [Elastic AI Assistant] Retrieval Augmented Generation (RAG) for Alerts #172542](#172542).

Deleting the `assistantRagOnAlerts` feature flag makes the `Alerts` toggle available in the assistant settings, per the screenshot below:

![alerts_setting](https://github.com/elastic/kibana/assets/4459398/f8d20751-48e0-4d75-ac0b-66f63c10b7cb)

This PR should not be merged until the docs describing the feature in <elastic/security-docs#4456> have been merged.

### Desk testing

To desk test this change:

1) Delete the following `assistantRagOnAlerts` feature flag from your local `config/kibana.dev.yml`:

```
xpack.securitySolution.enableExperimental: ['assistantRagOnAlerts']
```

2) Start Kibana

3) Generate alerts with a variety of severity (e.g. `low`, `medium`, `high`, and `critical`)

4) Navigate to Security > Alerts

5) Click the `AI Assistant` button to open the assistant

6) Click the `X` button to clear the conversation

7) Click the assistant's `Settings` gear

8) Click the `Knowledge Base` category

**Expected result**

- The `Alerts` toggle shown in the screenshot below is available

![alerts_setting](https://github.com/elastic/kibana/assets/4459398/f8d20751-48e0-4d75-ac0b-66f63c10b7cb)

9) Click the `Alerts` toggle to enable the feature

10) Click the `Save` button to close settings

11) Enter the following prompt:

```
How many open alerts do I have?
```

**Expected result**

- A response with alert counts grouped by workflow status will be returned, similar to the example below:

```
You currently have 48 open alerts in your system. These are categorized by severity as following: 19 of them are low severity, 16 are high severity, 12 are of medium severity and 1 is of critical severity. There is also 1 critical severity alert which is acknowledged.
```

12) Enter the following prompt:

```
Which alerts should I look at first?
```

**Expected result**

A response with alert details, similar to the following is returned:

```
Based on the latest information, the alerts to prioritize first are those related to a mimikatz process starting on the hosts, which have a critical severity and the highest risk score of 99. There are also a series of alerts related to an EQL process sequence with a high severity and risk scores of 73. There is one alert about an Elastic Endpoint Security alert with a medium severity and risk score of 47.
```

13) Once again, click the assistant's `Settings` gear

14) Click the `Knowledge Base` category

15) Click the `Alerts` toggle to disable the feature

16) Click the `Save` button to close settings

17) Once again, enter the following prompt:

```
How many open alerts do I have?
```

**Expected result**

- The assistant does NOT respond with a breakdown of alerts by severity. Instead it replies with something like the following example response:

```
I'm sorry for any confusion, but as an AI, I don't have real-time access to your data or system to provide the number of your current open alerts. You can check your Elastic Security dashboard or use the appropriate querying commands to get the updated count of your open alerts.
```

18) One more time, enter the following prompt:

```
Which alerts should I look at first?
```

**Expected result**

- The assistant does NOT respond with alert details. Instead it replies with something like  the following example response:

```
As an AI model, I don't have the capability to access real-time data. However, when it comes to managing alerts in Elastic Security, it's generally recommended to first look at the ones with the highest severity and risk score. Alerts related to malware, unauthorized access attempts, and abnormal data transfers or process activities, for example, may need immediate attention due to their potential high impact.
```
  • Loading branch information
andrew-goldstein committed Dec 21, 2023
1 parent 3a41138 commit ada3a90
Show file tree
Hide file tree
Showing 10 changed files with 22 additions and 65 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,6 @@ const fetchConnectorArgs: FetchConnectorExecuteAction = {
http: mockHttp,
messages,
onNewReplacements: jest.fn(),
ragOnAlerts: false,
};
describe('API tests', () => {
beforeEach(() => {
Expand Down Expand Up @@ -91,7 +90,6 @@ describe('API tests', () => {
alertsIndexPattern: '.alerts-security.alerts-default',
allow: ['a', 'b', 'c'],
allowReplacement: ['b', 'c'],
ragOnAlerts: true,
replacements: { auuid: 'real.hostname' },
size: 30,
};
Expand Down
4 changes: 0 additions & 4 deletions x-pack/packages/kbn-elastic-assistant/impl/assistant/api.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,6 @@ export interface FetchConnectorExecuteAction {
http: HttpSetup;
messages: Message[];
onNewReplacements: (newReplacements: Record<string, string>) => void;
ragOnAlerts: boolean;
replacements?: Record<string, string>;
signal?: AbortSignal | undefined;
size?: number;
Expand All @@ -55,7 +54,6 @@ export const fetchConnectorExecuteAction = async ({
http,
messages,
onNewReplacements,
ragOnAlerts,
replacements,
apiConfig,
signal,
Expand Down Expand Up @@ -90,7 +88,6 @@ export const fetchConnectorExecuteAction = async ({
alertsIndexPattern,
allow,
allowReplacement,
ragOnAlerts,
replacements,
size,
});
Expand Down Expand Up @@ -192,7 +189,6 @@ export const fetchConnectorExecuteAction = async ({
response: hasParsableResponse({
alerts,
assistantLangChain,
ragOnAlerts,
})
? getFormattedMessageContent(response.data)
: response.data,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -235,29 +235,12 @@ describe('getBlockBotConversation', () => {
});

describe('getOptionalRequestParams', () => {
it('should return an empty object when ragOnAlerts is false', () => {
const params = {
alerts: true,
alertsIndexPattern: 'indexPattern',
allow: ['a', 'b', 'c'],
allowReplacement: ['b', 'c'],
ragOnAlerts: false, // <-- false
replacements: { key: 'value' },
size: 10,
};

const result = getOptionalRequestParams(params);

expect(result).toEqual({});
});

it('should return an empty object when alerts is false', () => {
const params = {
alerts: false, // <-- false
alertsIndexPattern: 'indexPattern',
allow: ['a', 'b', 'c'],
allowReplacement: ['b', 'c'],
ragOnAlerts: true,
replacements: { key: 'value' },
size: 10,
};
Expand All @@ -267,13 +250,12 @@ describe('getBlockBotConversation', () => {
expect(result).toEqual({});
});

it('should return the optional request params when ragOnAlerts is true and alerts is true', () => {
it('should return the optional request params when alerts is true', () => {
const params = {
alerts: true,
alertsIndexPattern: 'indexPattern',
allow: ['a', 'b', 'c'],
allowReplacement: ['b', 'c'],
ragOnAlerts: true,
replacements: { key: 'value' },
size: 10,
};
Expand All @@ -292,7 +274,6 @@ describe('getBlockBotConversation', () => {
it('should return (only) the optional request params that are defined when some optional params are not provided', () => {
const params = {
alerts: true,
ragOnAlerts: true,
allow: ['a', 'b', 'c'], // all the others are undefined
};

Expand All @@ -305,31 +286,37 @@ describe('getBlockBotConversation', () => {
});

describe('hasParsableResponse', () => {
it('returns true when assistantLangChain is true', () => {
it('returns true when just assistantLangChain is true', () => {
const result = hasParsableResponse({
alerts: false,
assistantLangChain: true,
ragOnAlerts: false,
});

expect(result).toBe(true);
});

it('returns true when ragOnAlerts is true and alerts is true', () => {
it('returns true when just alerts is true', () => {
const result = hasParsableResponse({
alerts: true,
assistantLangChain: false,
ragOnAlerts: true,
});

expect(result).toBe(true);
});

it('returns false when assistantLangChain, ragOnAlerts, and alerts are all false', () => {
it('returns true when both assistantLangChain and alerts are true', () => {
const result = hasParsableResponse({
alerts: true,
assistantLangChain: true,
});

expect(result).toBe(true);
});

it('returns false when both assistantLangChain and alerts are false', () => {
const result = hasParsableResponse({
alerts: false,
assistantLangChain: false,
ragOnAlerts: false,
});

expect(result).toBe(false);
Expand Down
12 changes: 3 additions & 9 deletions x-pack/packages/kbn-elastic-assistant/impl/assistant/helpers.ts
Original file line number Diff line number Diff line change
Expand Up @@ -101,15 +101,13 @@ export const getOptionalRequestParams = ({
alertsIndexPattern,
allow,
allowReplacement,
ragOnAlerts,
replacements,
size,
}: {
alerts: boolean;
alertsIndexPattern?: string;
allow?: string[];
allowReplacement?: string[];
ragOnAlerts: boolean;
replacements?: Record<string, string>;
size?: number;
}): OptionalRequestParams => {
Expand All @@ -119,10 +117,8 @@ export const getOptionalRequestParams = ({
const optionalReplacements = replacements ? { replacements } : undefined;
const optionalSize = size ? { size } : undefined;

if (
!ragOnAlerts || // the feature flag must be enabled
!alerts // the settings toggle must also be enabled
) {
// the settings toggle must be enabled:
if (!alerts) {
return {}; // don't send any optional params
}

Expand All @@ -138,9 +134,7 @@ export const getOptionalRequestParams = ({
export const hasParsableResponse = ({
alerts,
assistantLangChain,
ragOnAlerts,
}: {
alerts: boolean;
assistantLangChain: boolean;
ragOnAlerts: boolean;
}): boolean => assistantLangChain || (ragOnAlerts && alerts);
}): boolean => assistantLangChain || alerts;
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,6 @@ export const useSendMessages = (): UseSendMessages => {
assistantStreamingEnabled,
defaultAllow,
defaultAllowReplacement,
ragOnAlerts,
knowledgeBase,
} = useAssistantContext();
const [isLoading, setIsLoading] = useState(false);
Expand All @@ -56,7 +55,6 @@ export const useSendMessages = (): UseSendMessages => {
assistantLangChain: knowledgeBase.assistantLangChain,
assistantStreamingEnabled,
http,
ragOnAlerts, // feature flag
replacements,
messages,
size: knowledgeBase.latestAlerts,
Expand All @@ -74,7 +72,6 @@ export const useSendMessages = (): UseSendMessages => {
knowledgeBase.alerts,
knowledgeBase.assistantLangChain,
knowledgeBase.latestAlerts,
ragOnAlerts,
]
);

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,6 @@ export interface AssistantProviderProps {
getInitialConversations: () => Record<string, Conversation>;
modelEvaluatorEnabled?: boolean;
nameSpace?: string;
ragOnAlerts?: boolean;
setConversations: React.Dispatch<React.SetStateAction<Record<string, Conversation>>>;
setDefaultAllow: React.Dispatch<React.SetStateAction<string[]>>;
setDefaultAllowReplacement: React.Dispatch<React.SetStateAction<string[]>>;
Expand Down Expand Up @@ -140,7 +139,6 @@ export interface UseAssistantContext {
promptContexts: Record<string, PromptContext>;
modelEvaluatorEnabled: boolean;
nameSpace: string;
ragOnAlerts: boolean;
registerPromptContext: RegisterPromptContext;
selectedSettingsTab: SettingsTabs;
setAllQuickPrompts: React.Dispatch<React.SetStateAction<QuickPrompt[] | undefined>>;
Expand Down Expand Up @@ -182,7 +180,6 @@ export const AssistantProvider: React.FC<AssistantProviderProps> = ({
getInitialConversations,
modelEvaluatorEnabled = false,
nameSpace = DEFAULT_ASSISTANT_NAMESPACE,
ragOnAlerts = false,
setConversations,
setDefaultAllow,
setDefaultAllowReplacement,
Expand Down Expand Up @@ -319,7 +316,6 @@ export const AssistantProvider: React.FC<AssistantProviderProps> = ({
modelEvaluatorEnabled,
promptContexts,
nameSpace,
ragOnAlerts,
registerPromptContext,
selectedSettingsTab,
setAllQuickPrompts: setLocalStorageQuickPrompts,
Expand Down Expand Up @@ -365,7 +361,6 @@ export const AssistantProvider: React.FC<AssistantProviderProps> = ({
nameSpace,
onConversationsUpdated,
promptContexts,
ragOnAlerts,
registerPromptContext,
selectedSettingsTab,
setDefaultAllow,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,6 @@ const mockUseAssistantContext = {
prepend: jest.fn(),
},
},
ragOnAlerts: true,
setAllSystemPrompts: jest.fn(),
setConversations: jest.fn(),
};
Expand Down Expand Up @@ -210,7 +209,7 @@ describe('Knowledge base settings', () => {
expect(queryByTestId('knowledgeBaseActionButton')).not.toBeInTheDocument();
});

it('renders the alerts settings when ragOnAlerts is true', () => {
it('renders the alerts settings', () => {
const { getByTestId } = render(
<TestProviders>
<KnowledgeBaseSettings {...defaultProps} />
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ interface Props {
*/
export const KnowledgeBaseSettings: React.FC<Props> = React.memo(
({ knowledgeBase, setUpdatedKnowledgeBaseSettings }) => {
const { http, ragOnAlerts } = useAssistantContext();
const { http } = useAssistantContext();
const {
data: kbStatus,
isLoading,
Expand Down Expand Up @@ -303,12 +303,10 @@ export const KnowledgeBaseSettings: React.FC<Props> = React.memo(

<EuiSpacer size="s" />

{ragOnAlerts && (
<AlertsSettings
knowledgeBase={knowledgeBase}
setUpdatedKnowledgeBaseSettings={setUpdatedKnowledgeBaseSettings}
/>
)}
<AlertsSettings
knowledgeBase={knowledgeBase}
setUpdatedKnowledgeBaseSettings={setUpdatedKnowledgeBaseSettings}
/>
</>
);
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -94,11 +94,6 @@ export const allowedExperimentalValues = Object.freeze({
*/
assistantModelEvaluation: false,

/**
* Enables Retrieval Augmented Generation (RAG) on Alerts in the assistant
*/
assistantRagOnAlerts: false,

/*
* Enables the new user details flyout displayed on the Alerts page and timeline.
*
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,6 @@ export const AssistantProvider: React.FC = ({ children }) => {

const { signalIndexName } = useSignalIndex();
const alertsIndexPattern = signalIndexName ?? undefined;
const ragOnAlerts = useIsExperimentalFeatureEnabled('assistantRagOnAlerts');
const toasts = useAppToasts() as unknown as IToasts; // useAppToasts is the current, non-deprecated method of getting the toasts service in the Security Solution, but it doesn't return the IToasts interface (defined by core)

return (
Expand All @@ -82,7 +81,6 @@ export const AssistantProvider: React.FC = ({ children }) => {
assistantStreamingEnabled={assistantStreamingEnabled}
modelEvaluatorEnabled={isModelEvaluationEnabled}
nameSpace={nameSpace}
ragOnAlerts={ragOnAlerts}
setConversations={setConversations}
setDefaultAllow={setDefaultAllow}
setDefaultAllowReplacement={setDefaultAllowReplacement}
Expand Down

0 comments on commit ada3a90

Please sign in to comment.