nodes/myshell-ai-ComfyUI-ShellAgent-Plugin
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Drag to connect output text float integer

Browse Source
This commit is contained in:
shanexi
2024-10-28 10:09:38 +08:00
parent ae8ed60767
commit b825b62a96
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
web/shellagent.js
+9 -1
View File
@@ -37,11 +37,15 @@ app.registerExtension({
});
},
async beforeRegisterNodeDef(nodeType, nodeData, app) {
if (["ShellAgentPluginOutputText", "ShellAgentPluginOutputFloat", "ShellAgentPluginOutputInteger"].indexOf(nodeData.name) > -1) {
chainCallback(nodeType.prototype, "onNodeCreated", function () {
this.convertWidgetToInput(this.widgets[0])
})
}
if (["ShellAgentPluginInputText", "ShellAgentPluginInputFloat", "ShellAgentPluginInputInteger"].indexOf(nodeData.name) > -1) {
chainCallback(nodeType.prototype, "onNodeCreated", function () {
const widget = this.widgets.find(w => w.name === 'choices')
this.addWidget('button', 'manage choices', null, () => {
const container = document.createElement("div");
Object.assign(container.style, {
@@ -318,6 +322,10 @@ app.registerExtension({
LiteGraph.slot_types_default_in['IMAGE'].unshift('ShellAgentPluginInputImage')
LiteGraph.slot_types_default_out['IMAGE'].unshift('ShellAgentPluginSaveImage')
LiteGraph.slot_types_default_out['IMAGE'].unshift('ShellAgentPluginSaveImages')
LiteGraph.slot_types_default_out['STRING'].unshift('ShellAgentPluginOutputFloat')
LiteGraph.slot_types_default_out['STRING'].unshift('ShellAgentPluginOutputInteger')
LiteGraph.slot_types_default_out['STRING'].unshift('ShellAgentPluginOutputText')
}
});