- An AI app calling by an older tool list (tool_help or use_tool not offered, use_tool for a tool offered directly, a tool the list lacks) is told so, and the task header shows how to reconnect until the app fetches tools/list again. - add_lora / set_model compare the base models on the model line (model_advice.py; the page reports the line) and add base_model_note. - get_model gives author_examples (sampler, steps, CFG, size, a LoRA's weight from the Civitai record's example pictures) and the version name. - run_workflow says ComfyUI's refusal as which node, what is wrong and the tool that fixes it (run_refusal.py; mcp_run.js passes the refusal as data). - Gates "When I don't answer": wait for me, or go with the recommendation once the waiting time runs out (choices with a recommended option only), handed to the AI in the same call and shown on the stage and under Decisions. - A tool budget test (count, size, description length, required arguments) and the exam tasks with their scorecard (docs/architecture/tool-exam.md, tools/tool_exam.mjs). The tool set's text recommends Full for Antigravity. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
67 lines
4.1 KiB
JavaScript
67 lines
4.1 KiB
JavaScript
// ComfyUI's own words when it refuses a prompt (which node, what is wrong), not just "failed".
|
|
const reason = failure => String((failure?.response && failure.toString?.()) || failure?.message || failure).slice(0, 800);
|
|
|
|
/** The same refusal as data (the frontend's PromptExecutionError.response), for the server to say what to do next (run_refusal.py). */
|
|
const refusal = failure => {
|
|
const response = failure?.response;
|
|
if (!response || typeof response !== 'object') return null;
|
|
const top = typeof response.error === 'string' ? { message: response.error } : (response.error || {});
|
|
const nodes = Object.entries(response.node_errors || {}).slice(0, 12).flatMap(([node, item]) => (item?.errors || []).slice(0, 4).map(error => ({
|
|
node, class_type: item?.class_type, type: error?.type, message: error?.message, details: String(error?.details ?? '').slice(0, 200),
|
|
input: error?.extra_info?.input_name, value: error?.extra_info?.received_value == null ? undefined : String(error.extra_info.received_value).slice(0, 120),
|
|
})));
|
|
return { type: top.type, message: top.message, details: String(top.details ?? '').slice(0, 300), nodes };
|
|
};
|
|
|
|
/** Capture submission receipts from one native ComfyUI queue request, preserving its seed/widget hooks. */
|
|
export async function queueTracked(app, api, batch = 1) {
|
|
if (app.processingQueue) throw new Error('ComfyUI is submitting another batch. Wait for submission to finish, then try again.');
|
|
if (!Number.isSafeInteger(app.nextQueueRequestId) || typeof api.queuePrompt !== 'function') {
|
|
throw new Error('This ComfyUI frontend cannot identify queue requests. Update the frontend before running from AI.');
|
|
}
|
|
const requestId = app.nextQueueRequestId;
|
|
const original = api.queuePrompt;
|
|
const ids = [];
|
|
let capturing = false, error = '', refused = null, attempts = 0;
|
|
let complete;
|
|
const ownDone = new Promise(resolve => { complete = resolve; });
|
|
const started = event => { if (event.detail?.requestId === requestId) capturing = true; };
|
|
const finished = event => { if (event.detail?.requestId === requestId) { capturing = false; complete(); } };
|
|
const wrapped = async function (...args) {
|
|
// Calls queued by the user behind this batch run after its promptQueued event.
|
|
const ours = capturing && attempts < batch;
|
|
if (ours) attempts++;
|
|
try {
|
|
const result = await original.apply(this, args);
|
|
if (ours) {
|
|
if (typeof result?.prompt_id === 'string' && result.prompt_id) ids.push(result.prompt_id);
|
|
else { error = 'ComfyUI did not return a prompt ID for this submission.'; capturing = false; complete(); }
|
|
}
|
|
return result;
|
|
} catch (failure) {
|
|
if (ours) { error = reason(failure); refused = refusal(failure); capturing = false; complete(); }
|
|
throw failure; // the native error display and partial-batch handling still run
|
|
}
|
|
};
|
|
api.addEventListener('promptQueueing', started);
|
|
api.addEventListener('promptQueued', finished);
|
|
api.queuePrompt = wrapped;
|
|
try {
|
|
// Native queuePrompt may also drain later manual requests. Return our receipt as soon as
|
|
// our batch ends, but still handle a later rejection from the native queue promise.
|
|
const nativeDone = Promise.resolve(app.queuePrompt(0, batch)).catch(failure => {
|
|
if (ids.length < batch) { error ||= reason(failure); refused ||= refusal(failure); }
|
|
});
|
|
await Promise.race([nativeDone, ownDone]);
|
|
} catch (failure) {
|
|
error ||= reason(failure);
|
|
} finally {
|
|
capturing = false;
|
|
if (api.queuePrompt === wrapped) api.queuePrompt = original;
|
|
api.removeEventListener('promptQueueing', started);
|
|
api.removeEventListener('promptQueued', finished);
|
|
}
|
|
if (ids.length < batch) error ||= `Only ${ids.length} of ${batch} prompts were accepted. Check ComfyUI before retrying.`;
|
|
return { prompt_ids: ids, queued: ids.length, requested: batch, ...(error ? { submission_error: error } : {}), ...(refused ? { refusal: refused } : {}) };
|
|
}
|