fix(registry): force re-registration on WS refresh to prevent timeout, demote empty-registry log to debug

- workflow_registry.js: add force param to refreshRegistry(), bypass fingerprint
  dedup when responding to lora_registry_refresh WS message. Without this, the
  backend's wait_for_all() times out after 0.5s because the frontend skips the
  register-nodes POST when the workflow fingerprint hasn't changed (common after
  ComfyUI restart with an empty or unchanged workflow).
- misc_handlers.py: demote 'No nodes registered after refresh' from WARNING to
  DEBUG — empty workflows are a normal operational state, not a warning-worthy
  condition.
This commit is contained in:
Will Miao
2026-07-13 19:10:48 +08:00
parent 2018722cc8
commit 966024e534
2 changed files with 10 additions and 5 deletions

View File

@@ -3314,7 +3314,10 @@ class NodeRegistryHandler:
self._last_slow_path_ts = time.monotonic()
if registry_info["node_count"] == 0:
logger.warning("No nodes registered after refresh")
logger.debug(
"[LM:Registry] refresh OK — %s connected tab(s) but 0 compatible nodes found",
registry_info["tab_count"],
)
return web.json_response(
{
"success": False,

View File

@@ -84,7 +84,7 @@ app.registerExtension({
this._log("extension initialized, clientId=%s", api.clientId ?? api.initialClientId ?? "(pending)");
api.addEventListener("lora_registry_refresh", () => {
this.refreshRegistry();
this.refreshRegistry(true);
});
api.addEventListener("lm_widget_update", (event) => {
@@ -151,7 +151,7 @@ app.registerExtension({
console.debug(`[LM:Registry ${ts}] ${msg}`);
},
async refreshRegistry() {
async refreshRegistry(force = false) {
try {
const workflowNodes = [];
const nodeEntries = getAllGraphNodes(app.graph);
@@ -204,11 +204,13 @@ app.registerExtension({
const clientId = api.clientId ?? api.initialClientId ?? "";
// Content-based dedup: skip POST if identical to last sent payload.
// Content-based dedup: skip POST if identical to last sent payload,
// unless forced (e.g. responding to a lora_registry_refresh WS message
// where the backend explicitly requests a re-registration).
const fingerprint = JSON.stringify(
workflowNodes.map(n => `${n.graph_id}:${n.node_id}|${n.marker_role ?? ""}|${n.mode ?? 0}`).sort()
);
if (fingerprint === this._lastFingerprint) {
if (!force && fingerprint === this._lastFingerprint) {
return;
}
this._lastFingerprint = fingerprint;