From 4fc44abc0e6d7b46d405371fd012d0b9a030f000 Mon Sep 17 00:00:00 2001
From: Paul Lewis <617438+paullewis@users.noreply.github.com>
Date: Thu, 30 Jan 2025 15:35:21 +0000
Subject: [PATCH] [shared-ui] Only show quick add for context-only nodes
 (#4273)

Fixes #4271
---
 .changeset/forty-cougars-exist.md                    | 5 +++++
 packages/shared-ui/src/elements/editor/graph-node.ts | 7 ++++++-
 2 files changed, 11 insertions(+), 1 deletion(-)
 create mode 100644 .changeset/forty-cougars-exist.md

diff --git a/.changeset/forty-cougars-exist.md b/.changeset/forty-cougars-exist.md
new file mode 100644
index 00000000000..edcb3977ca5
--- /dev/null
+++ b/.changeset/forty-cougars-exist.md
@@ -0,0 +1,5 @@
+---
+"@breadboard-ai/shared-ui": patch
+---
+
+Only show quick add for context-only nodes
diff --git a/packages/shared-ui/src/elements/editor/graph-node.ts b/packages/shared-ui/src/elements/editor/graph-node.ts
index 8e4d69ed494..22a7072ef2f 100644
--- a/packages/shared-ui/src/elements/editor/graph-node.ts
+++ b/packages/shared-ui/src/elements/editor/graph-node.ts
@@ -1734,7 +1734,12 @@ export class GraphNode extends PIXI.Container {
   }
 
   #drawQuickAddIfNeeded() {
-    if (!this.selected || this.#isConnectedOut() || !this.#canShowQuickAdd) {
+    if (
+      !this.selected ||
+      this.#isConnectedOut() ||
+      !this.#canShowQuickAdd ||
+      !this.#isContextOnly()
+    ) {
       this.#quickAdd.removeFromParent();
       return;
     }