@openclaw/comfy-provider 2026.9.2 → 2026.9.4

package/dist/workflow-runtime.js

@@ -6,8 +6,7 @@ import { extensionForMime } from "openclaw/plugin-sdk/media-mime";
 import { resolvePositiveTimerTimeoutMs } from "openclaw/plugin-sdk/number-runtime";
 import { isProviderApiKeyConfigured } from "openclaw/plugin-sdk/provider-auth";
 import { resolveApiKeyForProvider } from "openclaw/plugin-sdk/provider-auth-runtime";
-import { assertOkOrThrowHttpError, normalizeBaseUrl, readProviderJsonResponse, redactProviderResponseErrorText, resolveProviderHttpRequestConfig } from "openclaw/plugin-sdk/provider-http";
-import { readResponseWithLimit } from "openclaw/plugin-sdk/response-limit-runtime";
+import { assertOkOrThrowHttpError, normalizeBaseUrl, readProviderBinaryResponse, readProviderJsonResponse, redactProviderResponseErrorText, resolveProviderHttpRequestConfig } from "openclaw/plugin-sdk/provider-http";
 import { normalizeSecretInputString, resolveConfiguredSecretInputString, resolveSecretInputString } from "openclaw/plugin-sdk/secret-input-runtime";
 import { canResolveEnvSecretRefInReadOnlyPath } from "openclaw/plugin-sdk/secret-ref-readonly";
 import { fetchWithSsrFGuard, isPrivateOrLoopbackHost, mergeSsrFPolicies, ssrfPolicyFromHttpBaseUrlAllowedOrigin } from "openclaw/plugin-sdk/ssrf-runtime";
@@ -218,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) => {
@@ -323,11 +305,13 @@ async function downloadOutputFile(params) {
 	try {
 		await assertOkOrThrowHttpError(firstResponse.response, "Comfy output download failed", { requestHeaders: params.headers });
 		const mimeType = normalizeOptionalString(firstResponse.response.headers.get("content-type")) || "application/octet-stream";
+		const downloadLabel = `Comfy ${params.capability} output download`;
 		return {
-			buffer: await readResponseWithLimit(firstResponse.response, params.maxBytes, {
+			buffer: await readProviderBinaryResponse(firstResponse.response, downloadLabel, params.capability, {
+				maxBytes: params.maxBytes,
 				chunkTimeoutMs: params.timeoutMs,
-				onOverflow: ({ maxBytes }) => /* @__PURE__ */ new Error(`Comfy ${params.capability} output download exceeds ${maxBytes} bytes`),
-				onIdleTimeout: ({ chunkTimeoutMs }) => /* @__PURE__ */ new Error(`Comfy ${params.capability} output download stalled after ${chunkTimeoutMs}ms`)
+				onOverflow: ({ maxBytes }) => /* @__PURE__ */ new Error(`${downloadLabel} exceeds ${maxBytes} bytes`),
+				onIdleTimeout: ({ chunkTimeoutMs }) => /* @__PURE__ */ new Error(`${downloadLabel} stalled after ${chunkTimeoutMs}ms`)
 			}),
 			mimeType
 		};
@@ -465,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.2",
+  "version": "2026.9.4",
   "description": "OpenClaw ComfyUI provider plugin.",
   "repository": {
     "type": "git",
@@ -18,10 +18,10 @@
       "minHostVersion": ">=2026.7.2"
     },
     "compat": {
-      "pluginApi": ">=2026.9.2"
+      "pluginApi": ">=2026.9.4"
     },
     "build": {
-      "openclawVersion": "2026.9.2",
+      "openclawVersion": "2026.9.4",
       "bundledDist": false
     },
     "release": {
@@ -38,7 +38,7 @@
     "README.md"
   ],
   "peerDependencies": {
-    "openclaw": ">=2026.9.2"
+    "openclaw": ">=2026.9.4"
   },
   "peerDependenciesMeta": {
     "openclaw": {