Ethanfel
2c1777ae4e
ComfyUI-Tenaciousload speeds up ComfyUI page loads for large model/LoRA collections by caching the slow /api/object_info response in memory and on disk (survives restarts) and serving it gzipped in ~milliseconds, instead of rebuilding it (and freezing the event loop) on every load. Adds a "Refresh Models / LoRAs" menu button, a graph node, and a POST /tenaciousload/refresh endpoint to rebuild the cache after adding or removing models. No external dependencies; no nginx/docker required. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
56 lines
1.9 KiB
JavaScript
56 lines
1.9 KiB
JavaScript
import { app } from "../../scripts/app.js";
|
|
import { api } from "../../scripts/api.js";
|
|
|
|
let busy = false;
|
|
|
|
function notify(severity, detail, life = 6000) {
|
|
const toast = app.extensionManager?.toast;
|
|
if (toast?.add) {
|
|
toast.add({ severity, summary: "Tenaciousload", detail, life });
|
|
} else if (severity === "error") {
|
|
alert("Tenaciousload: " + detail);
|
|
}
|
|
console[severity === "error" ? "error" : "log"]("[Tenaciousload]", detail);
|
|
}
|
|
|
|
async function doRefresh() {
|
|
if (busy) {
|
|
notify("warn", "A refresh is already running…");
|
|
return;
|
|
}
|
|
busy = true;
|
|
notify("info", "Refreshing models/LoRAs — scanning over the network, this can take a few minutes…", 10000);
|
|
try {
|
|
// 1) clear ComfyUI's folder cache + drop the object_info cache.
|
|
await api.fetchApi("/tenaciousload/refresh", { method: "POST" });
|
|
// 2) force a fresh object_info build and re-cache it (this is the slow step).
|
|
await api.fetchApi("/object_info?nocache=1").then((r) => r.arrayBuffer());
|
|
// 3) update open node dropdowns live if the frontend supports it.
|
|
try { await app.refreshComboInNodes?.(); } catch (e) { /* non-fatal */ }
|
|
notify("success", "Done — new models are now available (reload the page if a dropdown still looks stale).", 8000);
|
|
} catch (e) {
|
|
notify("error", "Refresh failed: " + (e?.message || e), 10000);
|
|
} finally {
|
|
busy = false;
|
|
}
|
|
}
|
|
|
|
app.registerExtension({
|
|
name: "Tenaciousload.Refresh",
|
|
commands: [
|
|
{
|
|
id: "Tenaciousload.refresh",
|
|
label: "🔄 Refresh Models / LoRAs",
|
|
icon: "pi pi-refresh",
|
|
function: doRefresh,
|
|
},
|
|
],
|
|
// Adds the command under the top "Extensions" menu (and the command palette).
|
|
menuCommands: [
|
|
{
|
|
path: ["Extensions"],
|
|
commands: ["Tenaciousload.refresh"],
|
|
},
|
|
],
|
|
});
|