111 lines
3.8 KiB
JavaScript
111 lines
3.8 KiB
JavaScript
import { app } from "../../../scripts/app.js";
|
|
import { ComfyButtonGroup } from "../../scripts/ui/components/buttonGroup.js";
|
|
import { ComfyButton } from "../../scripts/ui/components/button.js";
|
|
|
|
import { api } from "../../../scripts/api.js";
|
|
import { nodeFixes } from "./nodeFixes.js";
|
|
import { setupConfigurationNodes } from "./ConfigurationNodes.js";
|
|
import { setupNodeColors } from "./ColorNodes.js";
|
|
import { setupDisplayNodes } from "./DisplayNodes.js";
|
|
import { setupCombineNodes } from "./CombineNodes.js";
|
|
import { setupExtractionNodes } from "./ExtractionNodes.js";
|
|
import { setupTextLoaderModuleNodes } from "./TextLoaderModuleNodes.js";
|
|
import { setupFlowNodes } from "./FlowNodes.js";
|
|
import {
|
|
gtUIAddUploadWidget,
|
|
gtUIAddUrlButtonWidget,
|
|
gtUIAddButtonWidget,
|
|
} from "./gtUIUtils.js";
|
|
import { setupMenuSeparator } from "./gtUIMenuSeparator.js";
|
|
import { keys_organized } from "./griptape_api_keys.js";
|
|
import { setupVisibilityToggles } from "./NodesWithVisibilityToggles.js";
|
|
import { setupCodeExecutionNode } from "./CodeExecutionNode.js";
|
|
import { setupApiKeyButtons } from "./apiKeyButtons.js";
|
|
import { createSettings } from "./settings.js";
|
|
const createUrlCallback = function (url) {
|
|
return function () {
|
|
window.open(url, "_blank"); // Opens the provided URL in a new tab
|
|
};
|
|
};
|
|
|
|
// function addGriptapeTopBarButtons() {
|
|
// const buttons = [];
|
|
// const griptapeButton = new ComfyButton({
|
|
// tooltip: "Griptape",
|
|
// app,
|
|
// enabled: true,
|
|
// classList: "comfyui-button comfyui-menu-mobile-collapse primary",
|
|
// });
|
|
// console.log(griptapeButton);
|
|
// }
|
|
app.registerExtension({
|
|
name: "comfy.gtUI",
|
|
// addGriptapeTopBarButtons,
|
|
beforeConfigureGraph: (graphData, missingNodeTypes) => {
|
|
for (let node of graphData.nodes) {
|
|
if (nodeFixes.fixes[node.type]) {
|
|
node.type = nodeFixes.fixes[node.type];
|
|
}
|
|
}
|
|
},
|
|
setup: (app) => {
|
|
setupMenuSeparator(app);
|
|
function messageHandler(event) {
|
|
// console.log(event.detail.message);
|
|
}
|
|
api.addEventListener("comfy.gtUI.runagent", messageHandler);
|
|
|
|
createSettings(app);
|
|
},
|
|
|
|
init() {},
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
setupNodeColors(nodeType, nodeData, app);
|
|
setupFlowNodes(nodeType, nodeData, app);
|
|
setupConfigurationNodes(nodeType, nodeData, app);
|
|
setupDisplayNodes(nodeType, nodeData, app);
|
|
setupCombineNodes(nodeType, nodeData, app);
|
|
setupExtractionNodes(nodeType, nodeData, app);
|
|
setupVisibilityToggles(nodeType, nodeData, app);
|
|
setupCodeExecutionNode(nodeType, nodeData, app);
|
|
setupApiKeyButtons(nodeType, nodeData, app);
|
|
// Create Audio Node
|
|
if (nodeData.name === "Griptape Load: Audio") {
|
|
gtUIAddUploadWidget(nodeType, nodeData, "audio", "audio");
|
|
}
|
|
// Load Text Node
|
|
if (nodeData.name === "Griptape Load: Text") {
|
|
gtUIAddUploadWidget(nodeType, nodeData, "text", "text");
|
|
}
|
|
if (nodeData.name === "Griptape RAG Retrieve: Text Loader Module") {
|
|
gtUIAddUploadWidget(nodeType, nodeData, "file_path", "text");
|
|
}
|
|
setupTextLoaderModuleNodes(nodeType, nodeData, app);
|
|
|
|
if (nodeData.name === "Griptape Code: Run Griptape Cloud Structure") {
|
|
gtUIAddUrlButtonWidget(
|
|
nodeType,
|
|
"Open Griptape Cloud Structure Dashboard",
|
|
"https://cloud.griptape.ai/structures",
|
|
"structure_id"
|
|
);
|
|
}
|
|
if (nodeData.name === "Griptape Run: Cloud Assistant") {
|
|
gtUIAddUrlButtonWidget(
|
|
nodeType,
|
|
"Open Griptape Cloud Assistant Dashboard",
|
|
"https://cloud.griptape.ai/assistants",
|
|
"assistant_id"
|
|
);
|
|
}
|
|
if (nodeData.name === "Griptape Retrieve: Cloud Ruleset") {
|
|
gtUIAddUrlButtonWidget(
|
|
nodeType,
|
|
"Open Griptape Cloud Ruleset Dashboard",
|
|
"https://cloud.griptape.ai/rulesets",
|
|
"ruleset_id"
|
|
);
|
|
}
|
|
},
|
|
});
|