jaidaken/ComfyUI
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Add custom nodes, Civitai loras (LFS), and vast.ai setup script
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Details
Python Linting / Run Pylint (push) Has been cancelled
Details
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Has been cancelled
Details
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Has been cancelled
Details
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Has been cancelled
Details
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Has been cancelled
Details
Execution Tests / test (macos-latest) (push) Has been cancelled
Details
Execution Tests / test (ubuntu-latest) (push) Has been cancelled
Details
Execution Tests / test (windows-latest) (push) Has been cancelled
Details
Test server launches without errors / test (push) Has been cancelled
Details
Unit Tests / test (macos-latest) (push) Has been cancelled
Details
Unit Tests / test (ubuntu-latest) (push) Has been cancelled
Details
Unit Tests / test (windows-2022) (push) Has been cancelled
Details

Browse Source 
Includes 30 custom nodes committed directly, 7 Civitai-exclusive
loras stored via Git LFS, and a setup script that installs all
dependencies and downloads HuggingFace-hosted models on vast.ai.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
jaidaken
2026-02-09 00:55:26 +00:00
parent 2b70ab9ad0
commit f09734b0ee
2274 changed files with 748556 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

188
custom_nodes/rgthree-comfy/web/common/prompt_service.js Normal file
View File

@@ -0,0 +1,188 @@
import { api } from "../../scripts/api.js";
import { getResolver } from "./shared_utils.js";
export class PromptExecution {
constructor(id) {
this.promptApi = null;
this.executedNodeIds = [];
this.totalNodes = 0;
this.currentlyExecuting = null;
this.errorDetails = null;
this.apiPrompt = getResolver();
this.id = id;
}
setPrompt(prompt) {
this.promptApi = prompt.output;
this.totalNodes = Object.keys(this.promptApi).length;
this.apiPrompt.resolve(null);
}
getApiNode(nodeId) {
var _a;
return ((_a = this.promptApi) === null || _a === void 0 ? void 0 : _a[String(nodeId)]) || null;
}
getNodeLabel(nodeId) {
var _a, _b;
const apiNode = this.getApiNode(nodeId);
let label = ((_a = apiNode === null || apiNode === void 0 ? void 0 : apiNode._meta) === null || _a === void 0 ? void 0 : _a.title) || (apiNode === null || apiNode === void 0 ? void 0 : apiNode.class_type) || undefined;
if (!label) {
const graphNode = (_b = this.maybeGetComfyGraph()) === null || _b === void 0 ? void 0 : _b.getNodeById(Number(nodeId));
label = (graphNode === null || graphNode === void 0 ? void 0 : graphNode.title) || (graphNode === null || graphNode === void 0 ? void 0 : graphNode.type) || undefined;
}
return label;
}
executing(nodeId, step, maxSteps) {
var _a;
if (nodeId == null) {
this.currentlyExecuting = null;
return;
}
if (((_a = this.currentlyExecuting) === null || _a === void 0 ? void 0 : _a.nodeId) !== nodeId) {
if (this.currentlyExecuting != null) {
this.executedNodeIds.push(nodeId);
}
this.currentlyExecuting = { nodeId, nodeLabel: this.getNodeLabel(nodeId), pass: 0 };
this.apiPrompt.promise.then(() => {
var _a;
if (this.currentlyExecuting == null) {
return;
}
const apiNode = this.getApiNode(nodeId);
if (!this.currentlyExecuting.nodeLabel) {
this.currentlyExecuting.nodeLabel = this.getNodeLabel(nodeId);
}
if ((apiNode === null || apiNode === void 0 ? void 0 : apiNode.class_type) === "UltimateSDUpscale") {
this.currentlyExecuting.pass--;
this.currentlyExecuting.maxPasses = -1;
}
else if ((apiNode === null || apiNode === void 0 ? void 0 : apiNode.class_type) === "IterativeImageUpscale") {
this.currentlyExecuting.maxPasses = (_a = apiNode === null || apiNode === void 0 ? void 0 : apiNode.inputs["steps"]) !== null && _a !== void 0 ? _a : -1;
}
});
}
if (step != null) {
if (!this.currentlyExecuting.step || step < this.currentlyExecuting.step) {
this.currentlyExecuting.pass++;
}
this.currentlyExecuting.step = step;
this.currentlyExecuting.maxSteps = maxSteps;
}
}
error(details) {
this.errorDetails = details;
}
maybeGetComfyGraph() {
var _a;
return ((_a = window === null || window === void 0 ? void 0 : window.app) === null || _a === void 0 ? void 0 : _a.graph) || null;
}
}
class PromptService extends EventTarget {
constructor(api) {
super();
this.promptsMap = new Map();
this.currentExecution = null;
this.lastQueueRemaining = 0;
const that = this;
const queuePrompt = api.queuePrompt;
api.queuePrompt = async function (num, prompt, ...args) {
let response;
try {
response = await queuePrompt.apply(api, [...arguments]);
}
catch (e) {
const promptExecution = that.getOrMakePrompt("error");
promptExecution.error({ exception_type: "Unknown." });
throw e;
}
const promptExecution = that.getOrMakePrompt(response.prompt_id);
promptExecution.setPrompt(prompt);
if (!that.currentExecution) {
that.currentExecution = promptExecution;
}
that.promptsMap.set(response.prompt_id, promptExecution);
that.dispatchEvent(new CustomEvent("queue-prompt", {
detail: {
prompt: promptExecution,
},
}));
return response;
};
api.addEventListener("status", (e) => {
var _a;
if (!((_a = e.detail) === null || _a === void 0 ? void 0 : _a.exec_info))
return;
this.lastQueueRemaining = e.detail.exec_info.queue_remaining;
this.dispatchProgressUpdate();
});
api.addEventListener("execution_start", (e) => {
if (!this.promptsMap.has(e.detail.prompt_id)) {
console.warn("'execution_start' fired before prompt was made.");
}
const prompt = this.getOrMakePrompt(e.detail.prompt_id);
this.currentExecution = prompt;
this.dispatchProgressUpdate();
});
api.addEventListener("executing", (e) => {
if (!this.currentExecution) {
this.currentExecution = this.getOrMakePrompt("unknown");
console.warn("'executing' fired before prompt was made.");
}
this.currentExecution.executing(e.detail);
this.dispatchProgressUpdate();
if (e.detail == null) {
this.currentExecution = null;
}
});
api.addEventListener("progress", (e) => {
if (!this.currentExecution) {
this.currentExecution = this.getOrMakePrompt(e.detail.prompt_id);
console.warn("'progress' fired before prompt was made.");
}
this.currentExecution.executing(e.detail.node, e.detail.value, e.detail.max);
this.dispatchProgressUpdate();
});
api.addEventListener("execution_cached", (e) => {
if (!this.currentExecution) {
this.currentExecution = this.getOrMakePrompt(e.detail.prompt_id);
console.warn("'execution_cached' fired before prompt was made.");
}
for (const cached of e.detail.nodes) {
this.currentExecution.executing(cached);
}
this.dispatchProgressUpdate();
});
api.addEventListener("executed", (e) => {
if (!this.currentExecution) {
this.currentExecution = this.getOrMakePrompt(e.detail.prompt_id);
console.warn("'executed' fired before prompt was made.");
}
});
api.addEventListener("execution_error", (e) => {
var _a;
if (!this.currentExecution) {
this.currentExecution = this.getOrMakePrompt(e.detail.prompt_id);
console.warn("'execution_error' fired before prompt was made.");
}
(_a = this.currentExecution) === null || _a === void 0 ? void 0 : _a.error(e.detail);
this.dispatchProgressUpdate();
});
}
async queuePrompt(prompt) {
return await api.queuePrompt(-1, prompt);
}
dispatchProgressUpdate() {
this.dispatchEvent(new CustomEvent("progress-update", {
detail: {
queue: this.lastQueueRemaining,
prompt: this.currentExecution,
},
}));
}
getOrMakePrompt(id) {
let prompt = this.promptsMap.get(id);
if (!prompt) {
prompt = new PromptExecution(id);
this.promptsMap.set(id, prompt);
}
return prompt;
}
}
export const SERVICE = new PromptService(api);