27 lines
1.0 KiB
JavaScript
27 lines
1.0 KiB
JavaScript
import {app} from "../../scripts/app.js";
|
|
import {createTextWidget} from "./utils.js"
|
|
|
|
app.registerExtension({
|
|
name: "Searge_LLM_Node.Searge_Output_Node",
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (nodeData.name === "Searge_Output_Node") {
|
|
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
|
|
|
nodeType.prototype.onNodeCreated = function () {
|
|
const result = onNodeCreated?.apply(this, arguments);
|
|
const text = createTextWidget(app, this, "text");
|
|
const nodeWidth = this.size[0];
|
|
const nodeHeight = this.size[1];
|
|
this.setSize([nodeWidth, nodeHeight * 3]);
|
|
return result;
|
|
};
|
|
|
|
// Update widgets
|
|
const onExecuted = nodeType.prototype.onExecuted;
|
|
nodeType.prototype.onExecuted = function (message) {
|
|
onExecuted?.apply(this, arguments);
|
|
this.widgets.find(obj => obj.name === "text").value = message.text;
|
|
};
|
|
}
|
|
},
|
|
}); |