From ae8ed607674294b8fab5ff096c9c909723d80c57 Mon Sep 17 00:00:00 2001
From: shanexi <shane@myshell.ai>
Date: Mon, 28 Oct 2024 09:52:28 +0800
Subject: [PATCH] Save Image(s) on output connect pop menu

---
 web/shellagent.js | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/web/shellagent.js b/web/shellagent.js
index d59ef83..38df820 100644
--- a/web/shellagent.js
+++ b/web/shellagent.js
@@ -302,7 +302,7 @@ app.registerExtension({
 
           if (toOutput.length) {
             options.unshift({
-              content: "Convert to ShellAgent (Output)",
+              content: "Connect to ShellAgent (Output)",
               submenu: {
                 options: toOutput
               }
@@ -315,9 +315,9 @@ app.registerExtension({
   },
 
   afterConfigureGraph(missingNodeTypes, app) {
-    const type = 'IMAGE'
-    const nodeId = 'ShellAgentPluginInputImage'
-    LiteGraph.slot_types_default_in[type].unshift(nodeId)
+    LiteGraph.slot_types_default_in['IMAGE'].unshift('ShellAgentPluginInputImage')
+    LiteGraph.slot_types_default_out['IMAGE'].unshift('ShellAgentPluginSaveImage')
+    LiteGraph.slot_types_default_out['IMAGE'].unshift('ShellAgentPluginSaveImages')
   }
 });