@openclaw/comfy-provider 2026.9.3 → 2026.9.4

package/dist/workflow-runtime.js

@@ -217,49 +217,32 @@ function extractHistoryEntry(history, promptId) {
 	if (isRecord(nested)) return nested;
 	return null;
 }
-async function waitForLocalHistory(params) {
-	const deadline = Date.now() + params.timeoutMs;
-	for (;;) {
-		const requestTimeoutMs = resolveComfyRemainingMs(deadline, params.timeoutMs);
-		const entry = extractHistoryEntry(await readJsonResponse({
-			url: `${params.baseUrl}/history/${params.promptId}`,
-			init: {
-				method: "GET",
-				headers: params.headers
-			},
-			timeoutMs: requestTimeoutMs,
-			policy: params.policy,
-			dispatcherPolicy: params.dispatcherPolicy,
-			auditContext: "comfy-history",
-			errorPrefix: "Comfy history lookup failed"
-		}), params.promptId);
-		if (entry?.outputs && Object.keys(entry.outputs).length > 0) return entry;
-		const pollDelayMs = resolveComfyRemainingMs(deadline, params.timeoutMs, params.pollIntervalMs);
-		await new Promise((resolve) => {
-			setTimeout(resolve, pollDelayMs);
-		});
-	}
-}
-async function waitForCloudCompletion(params) {
+async function waitForComfyHistory(params) {
 	const deadline = Date.now() + params.timeoutMs;
+	const read = (path, kind, timeoutMs) => readJsonResponse({
+		url: `${params.baseUrl}${path}`,
+		init: {
+			method: "GET",
+			headers: params.headers
+		},
+		timeoutMs,
+		policy: params.policy,
+		dispatcherPolicy: params.dispatcherPolicy,
+		auditContext: `comfy-${kind}`,
+		errorPrefix: `Comfy ${kind} lookup failed`
+	});
 	for (;;) {
 		const requestTimeoutMs = resolveComfyRemainingMs(deadline, params.timeoutMs);
-		const status = await readJsonResponse({
-			url: `${params.baseUrl}/api/job/${params.promptId}/status`,
-			init: {
-				method: "GET",
-				headers: params.headers
-			},
-			timeoutMs: requestTimeoutMs,
-			policy: params.policy,
-			dispatcherPolicy: params.dispatcherPolicy,
-			auditContext: "comfy-status",
-			errorPrefix: "Comfy status lookup failed"
-		});
-		if (status.status === "completed") return;
-		if (status.status === "failed" || status.status === "cancelled") {
-			const detail = redactProviderResponseErrorText(status.error ?? status.message ?? params.promptId, params.headers);
-			throw new Error(`Comfy workflow ${status.status}: ${detail}`);
+		if (params.mode === "cloud") {
+			const status = await read(`/api/job/${params.promptId}/status`, "status", requestTimeoutMs);
+			if (status.status === "completed") return await read(`/api/history_v2/${params.promptId}`, "history", params.timeoutMs);
+			if (status.status === "failed" || status.status === "cancelled") {
+				const detail = redactProviderResponseErrorText(status.error ?? status.message ?? params.promptId, params.headers);
+				throw new Error(`Comfy workflow ${status.status}: ${detail}`);
+			}
+		} else {
+			const entry = extractHistoryEntry(await read(`/history/${params.promptId}`, "history", requestTimeoutMs), params.promptId);
+			if (entry?.outputs && Object.keys(entry.outputs).length > 0) return entry;
 		}
 		const pollDelayMs = resolveComfyRemainingMs(deadline, params.timeoutMs, params.pollIntervalMs);
 		await new Promise((resolve) => {
@@ -466,36 +449,15 @@ async function runComfyWorkflow(params) {
 	});
 	const promptId = normalizeOptionalString(promptResponse.prompt_id);
 	if (!promptId) throw new Error("Comfy workflow submit response missing prompt_id");
-	const historyEntry = extractHistoryEntry(mode === "cloud" ? await (async () => {
-		await waitForCloudCompletion({
-			baseUrl: normalizedBaseUrl,
-			promptId,
-			headers: new Headers(headers),
-			timeoutMs,
-			pollIntervalMs,
-			policy: networkPolicy.apiPolicy,
-			dispatcherPolicy
-		});
-		return await readJsonResponse({
-			url: `${normalizedBaseUrl}/api/history_v2/${promptId}`,
-			init: {
-				method: "GET",
-				headers: new Headers(headers)
-			},
-			timeoutMs,
-			policy: networkPolicy.apiPolicy,
-			dispatcherPolicy,
-			auditContext: "comfy-history",
-			errorPrefix: "Comfy history lookup failed"
-		});
-	})() : await waitForLocalHistory({
+	const historyEntry = extractHistoryEntry(await waitForComfyHistory({
 		baseUrl: normalizedBaseUrl,
 		promptId,
 		headers: new Headers(headers),
 		timeoutMs,
 		pollIntervalMs,
 		policy: networkPolicy.apiPolicy,
-		dispatcherPolicy
+		dispatcherPolicy,
+		mode
 	}), promptId);
 	if (!historyEntry) throw new Error(`Comfy history response missing outputs for prompt ${promptId}`);
 	const outputFiles = collectOutputFiles({

package/openclaw.plugin.json

@@ -1,5 +1,6 @@
 {
   "id": "comfy",
+  "categories": ["models", "media"],
   "activation": {
     "onStartup": false
   },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@openclaw/comfy-provider",
-  "version": "2026.9.3",
+  "version": "2026.9.4",
   "description": "OpenClaw ComfyUI provider plugin.",
   "repository": {
     "type": "git",
@@ -18,10 +18,10 @@
       "minHostVersion": ">=2026.7.2"
     },
     "compat": {
-      "pluginApi": ">=2026.9.3"
+      "pluginApi": ">=2026.9.4"
     },
     "build": {
-      "openclawVersion": "2026.9.3",
+      "openclawVersion": "2026.9.4",
       "bundledDist": false
     },
     "release": {
@@ -38,7 +38,7 @@
     "README.md"
   ],
   "peerDependencies": {
-    "openclaw": ">=2026.9.3"
+    "openclaw": ">=2026.9.4"
   },
   "peerDependenciesMeta": {
     "openclaw": {