fix bug related to tool mode

langflow-ai · Nov 19, 2024 · 9106b7c · 9106b7c
1 parent ebe4f34
commit 9106b7c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/frontend/src/pages/FlowPage/components/nodeToolbarComponent/index.tsx b/src/frontend/src/pages/FlowPage/components/nodeToolbarComponent/index.tsx
@@ -783,7 +783,10 @@ export default function NodeToolbarComponent({
                 open={openModal}
                 setOpen={setOpenModal}
                 dynamic={true}
-                setNodeClass={handleNodeClass}
+                setNodeClass={(apiClassType, type) => {
+                  handleNodeClass(apiClassType, type);
+                  setToolMode(false);
+                }}
                 nodeClass={data.node}
                 value={data.node?.template[name].value ?? ""}
               >