hide ai assitant button when in preview mode

Author: alexreal1314

x-pack/solutions/security/plugins/security_solution/public/flyout/entity_details/generic_right/footer.tsx

@@ -42,6 +42,7 @@ export const GenericEntityFlyoutFooter = ({
   const { showAssistant, showAssistantOverlay } = useAssetInventoryAssistant({
     entityId,
     entityFields,
+    isPreviewMode,
   });
 
   const openDocumentFlyout = useCallback(() => {

x-pack/solutions/security/plugins/security_solution/public/flyout/entity_details/generic_right/hooks/use_asset_inventory_assistant.test.tsx

@@ -30,7 +30,7 @@ const entityId = 'test-entity-id';
 
 const renderUseAssetInventoryAssistant = () =>
   renderHook((props: UseAssetInventoryAssistantParams) => useAssetInventoryAssistant(props), {
-    initialProps: { entityId, entityFields: mockEntityFields },
+    initialProps: { entityId, entityFields: mockEntityFields, isPreviewMode: false },
   });
 
 const useAssistantOverlayMock = useAssistantOverlay as jest.Mock;
@@ -151,6 +151,17 @@ describe('useAssetInventoryAssistant', () => {
     expect(hookResult.result.current.showAssistant).toEqual(false);
   });
 
+  it('should return showAssistant false when isPreviewMode is true', () => {
+    const hookResultWithPreviewMode = renderHook(
+      (props: UseAssetInventoryAssistantParams) => useAssetInventoryAssistant(props),
+      {
+        initialProps: { entityId, entityFields: mockEntityFields, isPreviewMode: true },
+      }
+    );
+
+    expect(hookResultWithPreviewMode.result.current.showAssistant).toEqual(false);
+  });
+
   it('returns entity fields as prompt context data', async () => {
     hookResult = renderUseAssetInventoryAssistant();
 
@@ -169,15 +180,15 @@ describe('useAssetInventoryAssistant', () => {
 
   it('falls back to default conversation ID when entityId is not provided', () => {
     renderHook((props: UseAssetInventoryAssistantParams) => useAssetInventoryAssistant(props), {
-      initialProps: { entityId: '', entityFields: mockEntityFields },
+      initialProps: { entityId: '', entityFields: mockEntityFields, isPreviewMode: false },
     });
 
     expect(useAssistantOverlayMock.mock.calls[0][1]).toEqual('Entity Summary');
   });
 
   it('returns empty prompt context when entityFields is empty', async () => {
     renderHook((props: UseAssetInventoryAssistantParams) => useAssetInventoryAssistant(props), {
-      initialProps: { entityId, entityFields: {} },
+      initialProps: { entityId, entityFields: {}, isPreviewMode: false },
     });
 
     const getPromptContext = (useAssistantOverlay as jest.Mock).mock.calls[0][3];

x-pack/solutions/security/plugins/security_solution/public/flyout/entity_details/generic_right/hooks/use_asset_inventory_assistant.ts

@@ -26,6 +26,7 @@ const useAssistantNoop = () => ({
 export interface UseAssetInventoryAssistantParams {
   entityId: EntityEcs['id'];
   entityFields: Record<string, string[]>;
+  isPreviewMode: boolean;
 }
 
 export interface UseAssetInventoryAssistantResult {
@@ -49,6 +50,7 @@ export interface UseAssetInventoryAssistantResult {
 export const useAssetInventoryAssistant = ({
   entityId,
   entityFields,
+  isPreviewMode,
 }: UseAssetInventoryAssistantParams): UseAssetInventoryAssistantResult => {
   const { hasAssistantPrivilege, isAssistantEnabled, isAssistantVisible } =
     useAssistantAvailability();
@@ -81,7 +83,11 @@ export const useAssetInventoryAssistant = ({
 
   return {
     showAssistant:
-      isAssistantEnabled && hasAssistantPrivilege && promptContextId !== null && isAssistantVisible,
+      isAssistantEnabled &&
+      hasAssistantPrivilege &&
+      promptContextId !== null &&
+      isAssistantVisible &&
+      !isPreviewMode,
     showAssistantOverlay,
     promptContextId: promptContextId || '',
   };