fix(ui): pending node connection stuck

invoke-ai · Aug 28, 2024 · 9692a5f · 9692a5f
1 parent 5db7b48
commit 9692a5f
Showing 1 changed file with 8 additions and 10 deletions.
diff --git a/invokeai/frontend/web/src/features/nodes/components/flow/AddNodeCmdk/AddNodeCmdk.tsx b/invokeai/frontend/web/src/features/nodes/components/flow/AddNodeCmdk/AddNodeCmdk.tsx
@@ -18,7 +18,6 @@ import { IAINoContentFallback } from 'common/components/IAIImageFallback';
 import ScrollableContent from 'common/components/OverlayScrollbars/ScrollableContent';
 import { useBuildNode } from 'features/nodes/hooks/useBuildNode';
 import {
-  $addNodeCmdk,
   $cursorPos,
   $edgePendingUpdate,
   $pendingConnection,
@@ -175,21 +174,20 @@ export const AddNodeCmdk = memo(() => {
     setSearchTerm(e.target.value);
   }, []);
 
-  const onSelect = useCallback(
-    (value: string) => {
-      addNode(value);
-      $addNodeCmdk.set(false);
-      setSearchTerm('');
-    },
-    [addNode]
-  );
-
   const onClose = useCallback(() => {
     addNodeCmdk.setFalse();
     setSearchTerm('');
     $pendingConnection.set(null);
   }, [addNodeCmdk]);
 
+  const onSelect = useCallback(
+    (value: string) => {
+      addNode(value);
+      onClose();
+    },
+    [addNode, onClose]
+  );
+
   return (
     <Modal
       isOpen={addNodeCmdkIsOpen}