From 710104d709d4d85d7fc95cfc8adebce47e5f1e6f Mon Sep 17 00:00:00 2001
From: shanexi <shane@myshell.ai>
Date: Sun, 27 Oct 2024 16:02:48 +0800
Subject: [PATCH] Image input

---
 web/shellagent.js | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/web/shellagent.js b/web/shellagent.js
index e113910..d59ef83 100644
--- a/web/shellagent.js
+++ b/web/shellagent.js
@@ -313,6 +313,12 @@ app.registerExtension({
       })
     }
   },
+
+  afterConfigureGraph(missingNodeTypes, app) {
+    const type = 'IMAGE'
+    const nodeId = 'ShellAgentPluginInputImage'
+    LiteGraph.slot_types_default_in[type].unshift(nodeId)
+  }
 });
 
 function addMenuHandler(nodeType, cb) {