diff --git a/src/frontend/src/CustomNodes/GenericNode/index.tsx b/src/frontend/src/CustomNodes/GenericNode/index.tsx index 5b8c16ea3..36ed0312a 100644 --- a/src/frontend/src/CustomNodes/GenericNode/index.tsx +++ b/src/frontend/src/CustomNodes/GenericNode/index.tsx @@ -259,8 +259,12 @@ function GenericNode({ return data.node?.description && data.node?.description !== ""; }, [data.node?.description]); + const selectedNodes = useFlowStore((state) => + state.nodes.filter((node) => node.selected), + ); + const memoizedNodeToolbarComponent = useMemo(() => { - return selected ? ( + return selected && selectedNodes.length === 1 ? ( <>