diff --git a/packages/compass-data-modeling/src/services/export-diagram.tsx b/packages/compass-data-modeling/src/services/export-diagram.tsx index 5edf1aa8c93..9cae1352347 100644 --- a/packages/compass-data-modeling/src/services/export-diagram.tsx +++ b/packages/compass-data-modeling/src/services/export-diagram.tsx @@ -115,6 +115,9 @@ export function getExportPngDataUri(diagram: DiagramInstance): Promise { height: `${bounds.height}px`, transform: `translate(${transform.x}px, ${transform.y}px) scale(${transform.zoom})`, }, + filter: (node) => { + return !node.classList?.contains('node-add-field-button'); + }, }) .then(resolve) .catch(reject) diff --git a/packages/compass-data-modeling/src/utils/nodes-and-edges.tsx b/packages/compass-data-modeling/src/utils/nodes-and-edges.tsx index 17b527a02ff..66fa4fdbc1f 100644 --- a/packages/compass-data-modeling/src/utils/nodes-and-edges.tsx +++ b/packages/compass-data-modeling/src/utils/nodes-and-edges.tsx @@ -5,6 +5,7 @@ import { IconButton, InlineDefinition, css, + cx, } from '@mongodb-js/compass-components'; import type { NodeProps, EdgeProps, BaseNode } from '@mongodb-js/diagramming'; import type { MongoDBJSONSchema } from 'mongodb-schema'; @@ -187,7 +188,7 @@ export function collectionToDiagramNode({ actions: onClickAddNewFieldToCollection ? ( ) => { event.stopPropagation(); onClickAddNewFieldToCollection();