diff --git a/openmetadata-ui/src/main/resources/ui/src/components/WorkflowDefinitions/WorkflowBuilder/WorkflowHeader.test.tsx b/openmetadata-ui/src/main/resources/ui/src/components/WorkflowDefinitions/WorkflowBuilder/WorkflowHeader.test.tsx index 7064fe8a1272..0e39735d0fb6 100644 --- a/openmetadata-ui/src/main/resources/ui/src/components/WorkflowDefinitions/WorkflowBuilder/WorkflowHeader.test.tsx +++ b/openmetadata-ui/src/main/resources/ui/src/components/WorkflowDefinitions/WorkflowBuilder/WorkflowHeader.test.tsx @@ -275,4 +275,21 @@ describe('WorkflowHeader — System badge', () => { screen.queryByTestId('edit-workflow-title-button') ).not.toBeInTheDocument(); }); + + it('renders the AI-mode header with the breadcrumb when isAiMode is true', () => { + mockUseWorkflowModeContext.mockReturnValue(buildContextMock()); + + render( + crumb} + /> + ); + + expect(screen.getByTestId('ai-breadcrumb')).toBeInTheDocument(); + expect(screen.getByTestId('workflow-title')).toHaveTextContent( + 'Glossary Approval Workflow' + ); + }); }); diff --git a/openmetadata-ui/src/main/resources/ui/src/components/WorkflowDefinitions/WorkflowBuilder/WorkflowHeader.tsx b/openmetadata-ui/src/main/resources/ui/src/components/WorkflowDefinitions/WorkflowBuilder/WorkflowHeader.tsx index 271990970d5e..e91544b5074a 100644 --- a/openmetadata-ui/src/main/resources/ui/src/components/WorkflowDefinitions/WorkflowBuilder/WorkflowHeader.tsx +++ b/openmetadata-ui/src/main/resources/ui/src/components/WorkflowDefinitions/WorkflowBuilder/WorkflowHeader.tsx @@ -14,6 +14,7 @@ import { Badge, Button, + Card, Dialog, Input, Modal, @@ -35,6 +36,7 @@ import { WorkflowControls } from './WorkflowControls'; export const WorkflowHeader: React.FC = ({ breadcrumb, + isAiMode = false, title, workflowName, handleTestWorkflow, @@ -89,97 +91,143 @@ export const WorkflowHeader: React.FC = ({ setIsEditModalOpen(false); }; + const workflowControls = ( + + ); + + const editWorkflowButton = showEditButton && ( + + ); + + const systemBadge = isNoOp && ( + + + + {t('label.system')} + + + + ); + + const editTitleButton = !isViewMode && !isNoOp && ( + - )} - - } - badge={ - <> - {isNoOp && ( - - - - {t('label.system')} - - - - )} - {!isViewMode && !isNoOp && ( -