@comfyorg/sdk 0.1.0 → 0.1.2

package/README.md

@@ -44,11 +44,11 @@ await job.getOutputs("13")[0].toFile("out.png");
 
 ## Auth, per surface
 
-| Surface                                                            | Auth                                       |
-| ------------------------------------------------------------------ | ------------------------------------------ |
-| Self-hosted proxy (`comfy-api-proxy` in front of your own ComfyUI) | none — do **not** pass `apiKey`            |
-| Comfy Cloud                                                        | `new Comfy(baseUrl, { apiKey: "ck_..." })` |
-| Serverless                                                         | `new Comfy(baseUrl, { apiKey: "ck_..." })` |
+| Surface                                                            | Auth                                            |
+| ------------------------------------------------------------------ | ----------------------------------------------- |
+| Self-hosted proxy (`comfy-api-proxy` in front of your own ComfyUI) | none — do **not** pass `apiKey`                 |
+| Comfy Cloud                                                        | `new Comfy(baseUrl, { apiKey: "comfyui-..." })` |
+| Serverless                                                         | `new Comfy(baseUrl, { apiKey: "comfyui-..." })` |
 
 The client only attaches the `Authorization` header to requests aimed at its
 own `baseUrl`'s origin. If the server hands back an absolute URL on a
@@ -56,6 +56,11 @@ different host (for example a job's `events`/`cancel` link, or a redirect on
 an asset download), the key is not sent there — see "Typed errors" below for
 the exception classes this surface can raise.
 
+The SDK identifies itself via `User-Agent` (for support + usage analytics);
+no other data is collected. Pass `clientInfo` to `new Comfy(baseUrl, { ... })`
+to append your own app's name to it, for example when attributing traffic
+from a Worker built on top of this SDK.
+
 ## Quickstart
 
 ```ts
@@ -79,6 +84,23 @@ await job.wait(); // poll to terminal (adaptive backoff); or call job.refresh()
 console.log(job.status, job.outputs);
 ```
 
+## Partner (API) node auth
+
+Workflows that use partner/API nodes (Gemini, etc.) need a Comfy API key to
+authenticate them. Pass it per submit with `apiKey`. This is **not** the same as
+the credential you construct `Comfy` with: the constructor key authenticates
+_you_ to the server, while this one authenticates the partner nodes _inside_ the
+workflow (it is often the same `comfyui-…` key):
+
+```ts
+const job = await client.run(wf, { apiKey: "comfyui-…" });
+// or: await client.submit(wf, { apiKey: "comfyui-…" });
+```
+
+The SDK sends it once as `extra_data.api_key_comfy_org` alongside the workflow —
+one key authenticates every partner node in the graph. It is never logged or
+persisted by the SDK. Omit `apiKey` and no `extra_data` is sent at all.
+
 ## Assets and `core/ASSET`
 
 `client.assets.fromFile(path)` / `client.assets.fromBytes(data, options)`
@@ -156,6 +178,32 @@ managing an `AbortController` yourself:
 await client.run(wf, { timeoutMs: 60_000 });
 ```
 
+## Downloading outputs
+
+A finished job exposes its results as output handles — `job.outputs`, or
+`job.getOutputs(nodeId)` to filter to one node. Each is an asset you can pull
+down whichever way suits the caller:
+
+```ts
+const out = job.getOutputs("13")[0];
+await out.toFile("result.png"); // stream to disk
+const data = await out.toBytes(); // buffer into memory
+await out.toFile("head.png", { range: [0, 1023] }); // range-aware: first 1 KiB only
+```
+
+`getDownloadUrl()` hands back a fetchable URL instead of streaming the bytes
+through your process — give it to a browser, a CDN, or another service:
+
+```ts
+const { url, expiresAt } = await out.getDownloadUrl();
+```
+
+On Comfy Cloud / serverless it's a short-lived, **self-authorizing** signed
+storage URL: whoever holds it can read the asset until `expiresAt` with no API
+key of their own. On a self-hosted proxy it's the content endpoint (normal auth
+still applies) and `expiresAt` is `null`. It works on every backend and never
+downloads the bytes first.
+
 ## Typed errors
 
 Protocol-level failures are raised as one exception class per error code, so

package/dist/low/errors.js

@@ -66,6 +66,11 @@ const BY_CODE = {
     queue_full: QueueFull,
     insufficient_credits: InsufficientCredits,
     not_found: NotFound,
+    // public-api currently returns entity-specific 404 codes even though the spec
+    // documents the generic `not_found`; map them so callers still get a typed
+    // NotFound. (Server/spec reconciliation is a separate follow-up.)
+    job_not_found: NotFound,
+    asset_not_found: NotFound,
     unauthorized: Unauthorized,
     forbidden: Forbidden,
 };

package/dist/low/generated/types.gen.d.ts

@@ -371,6 +371,15 @@ export type PostJobsData = {
         workflow: {
             [key: string]: unknown;
         };
+        /**
+         * Per-prompt ComfyUI `extra_data`, same shape as Comfy Cloud and local ComfyUI. Closed object: only the enumerated keys are accepted, keeping the contract fully typed. Forwarded to the worker per-prompt, never persisted, and excluded from idempotency comparison.
+         */
+        extra_data?: {
+            /**
+             * API key for partner (API) nodes.
+             */
+            api_key_comfy_org?: string;
+        };
     };
     headers?: {
         /**
@@ -437,6 +446,10 @@ export type GetJobErrors = {
      * `not_found`.
      */
     404: ErrorEnvelope;
+    /**
+     * `rate_limited` — the caller has exceeded the request rate limit for this account. Account/rate-scoped, not job-specific — this can be returned even for a job id the caller doesn't own or that doesn't exist, without revealing which.
+     */
+    429: ErrorEnvelope;
     /**
      * `upstream_error` — an unexpected failure reaching or processing the request in this implementation's backing services. The message is always a generic, safe-to-display string; implementation detail (the specific upstream, its error text, transport failures) is never included here — see each implementation's own error-mapping notes. Every operation in this contract can fail this way.
      */
@@ -513,6 +526,10 @@ export type CancelJobErrors = {
      * `not_found`.
      */
     404: ErrorEnvelope;
+    /**
+     * `rate_limited` — the caller has exceeded the request rate limit for this account. Account/rate-scoped, not job-specific — this can be returned even for a job id the caller doesn't own or that doesn't exist, without revealing which.
+     */
+    429: ErrorEnvelope;
     /**
      * `upstream_error` — an unexpected failure reaching or processing the request in this implementation's backing services. The message is always a generic, safe-to-display string; implementation detail (the specific upstream, its error text, transport failures) is never included here — see each implementation's own error-mapping notes. Every operation in this contract can fail this way.
      */

package/dist/low/generated/zod.gen.d.ts

@@ -253,6 +253,9 @@ export declare const zGetAssetContentPath: z.ZodObject<{
 export declare const zGetAssetContentResponse: z.ZodString;
 export declare const zPostJobsBody: z.ZodObject<{
     workflow: z.ZodRecord<z.ZodString, z.ZodUnknown>;
+    extra_data: z.ZodOptional<z.ZodObject<{
+        api_key_comfy_org: z.ZodOptional<z.ZodString>;
+    }, z.core.$strip>>;
 }, z.core.$strip>;
 export declare const zPostJobsHeaders: z.ZodObject<{
     'Idempotency-Key': z.ZodOptional<z.ZodString>;

package/dist/low/generated/zod.gen.js

@@ -172,7 +172,10 @@ export const zGetAssetContentPath = z.object({
  */
 export const zGetAssetContentResponse = z.string();
 export const zPostJobsBody = z.object({
-    workflow: z.record(z.string(), z.unknown())
+    workflow: z.record(z.string(), z.unknown()),
+    extra_data: z.object({
+        api_key_comfy_org: z.string().optional()
+    }).optional()
 });
 export const zPostJobsHeaders = z.object({
     'Idempotency-Key': z.string().optional()

package/dist/low/index.d.ts

@@ -18,4 +18,4 @@ export * as schemas from "./generated/zod.gen.js";
 export * from "./models.js";
 export * from "./errors.js";
 export { type RawEvent, iterateSse } from "./sse.js";
-export { ComfyLow, OPERATION_IDS, OPERATION_METHODS, type ComfyLowOptions, type RequestOptions, } from "./transport.js";
+export { ComfyLow, OPERATION_IDS, OPERATION_METHODS, type ComfyLowOptions, type RequestOptions, type AssetContentUrl, } from "./transport.js";

package/dist/low/sse.d.ts

@@ -14,11 +14,24 @@ export interface RawEvent {
     event: string;
     data: Record<string, unknown>;
 }
+/**
+ * Default read-idle timeout for the SSE byte stream. The server sends keepalive
+ * comments well within this window, so no *bytes at all* for this long means a
+ * silently-stalled ("zombie") connection — open but delivering nothing, not
+ * legitimately idle. We error the stream in that case so the caller's
+ * reconnect/poll fallback runs (see `Job.events`) instead of hanging forever.
+ * The timeout is at the byte level, not the parsed-message level, so keepalive
+ * comments (which `eventsource-parser` drops) still count as liveness.
+ */
+export declare const SSE_IDLE_TIMEOUT_MS = 45000;
 /**
  * Decode a fetch response body into a stream of {@link RawEvent}s.
  *
  * No cursor/replay support by design — the contract carries no `id`/
  * `Last-Event-ID`; a blank `data:` frame (no payload) is dropped, matching
- * the Python decoder.
+ * the Python decoder. A read-idle timeout (default {@link SSE_IDLE_TIMEOUT_MS})
+ * errors a silently-stalled connection instead of blocking forever.
  */
-export declare function iterateSse(body: ReadableStream<Uint8Array>): AsyncGenerator<RawEvent, void, void>;
+export declare function iterateSse(body: ReadableStream<Uint8Array>, options?: {
+    idleTimeoutMs?: number;
+}): AsyncGenerator<RawEvent, void, void>;

package/dist/low/sse.js

@@ -24,15 +24,48 @@ function parseData(raw) {
     }
     return { value: parsed };
 }
+/**
+ * Default read-idle timeout for the SSE byte stream. The server sends keepalive
+ * comments well within this window, so no *bytes at all* for this long means a
+ * silently-stalled ("zombie") connection — open but delivering nothing, not
+ * legitimately idle. We error the stream in that case so the caller's
+ * reconnect/poll fallback runs (see `Job.events`) instead of hanging forever.
+ * The timeout is at the byte level, not the parsed-message level, so keepalive
+ * comments (which `eventsource-parser` drops) still count as liveness.
+ */
+export const SSE_IDLE_TIMEOUT_MS = 45_000;
+/** A pass-through stream that errors if no chunk arrives within `ms`. */
+function idleTimeout(ms) {
+    let timer;
+    const clear = () => {
+        if (timer !== undefined)
+            clearTimeout(timer);
+    };
+    const arm = (controller) => {
+        clear();
+        timer = setTimeout(() => controller.error(new Error(`SSE idle timeout: no data for ${ms}ms`)), ms);
+    };
+    return new TransformStream({
+        start: arm,
+        transform(chunk, controller) {
+            arm(controller);
+            controller.enqueue(chunk);
+        },
+        flush: clear,
+        cancel: clear,
+    });
+}
 /**
  * Decode a fetch response body into a stream of {@link RawEvent}s.
  *
  * No cursor/replay support by design — the contract carries no `id`/
  * `Last-Event-ID`; a blank `data:` frame (no payload) is dropped, matching
- * the Python decoder.
+ * the Python decoder. A read-idle timeout (default {@link SSE_IDLE_TIMEOUT_MS})
+ * errors a silently-stalled connection instead of blocking forever.
  */
-export async function* iterateSse(body) {
+export async function* iterateSse(body, options = {}) {
     const stream = body
+        .pipeThrough(idleTimeout(options.idleTimeoutMs ?? SSE_IDLE_TIMEOUT_MS))
         .pipeThrough(new TextDecoderStream())
         .pipeThrough(new EventSourceParserStream());
     for await (const message of stream) {

package/dist/low/transport.d.ts

@@ -25,7 +25,7 @@
  * the Python SDK (its `AsyncComfyLow`; there is no sync variant here — see
  * the package README for why).
  */
-import type { Asset, Job } from "./generated/types.gen.js";
+import type { Asset, Job, PostJobsData } from "./generated/types.gen.js";
 import { type RawEvent } from "./sse.js";
 export interface RequestOptions {
     headers?: Record<string, string>;
@@ -38,10 +38,24 @@ export interface RequestOptions {
      * must not time out while idle mid-job).
      */
     timeoutMs?: number | null;
+    /** Defaults to `"follow"`; `getAssetContentUrl` passes `"manual"` so it can
+     * read a redirect's `Location` instead of following it. */
+    redirect?: "follow" | "manual" | "error";
 }
 export interface ComfyLowOptions {
     fetch?: typeof fetch;
     timeoutMs?: number;
+    /**
+     * Appended to the default `User-Agent` as `app/{clientInfo}` — lets an
+     * app built on this SDK attribute its own traffic in request logs.
+     * Opt-in; omitted by default.
+     */
+    clientInfo?: string;
+}
+/** Return shape of {@link ComfyLow.getAssetContentUrl}. */
+export interface AssetContentUrl {
+    url: string;
+    expiresAt: Date | null;
 }
 /** Synchronous protocol bindings — async throughout (JS is async-native). */
 export declare class ComfyLow {
@@ -49,6 +63,7 @@ export declare class ComfyLow {
     private readonly apiKey?;
     private readonly fetchImpl;
     private readonly defaultTimeoutMs;
+    private readonly userAgent;
     constructor(baseUrl: string, apiKey?: string, options?: ComfyLowOptions);
     private urlFor;
     /**
@@ -100,9 +115,31 @@ export declare class ComfyLow {
         range?: readonly [number, number];
         signal?: AbortSignal;
     }): Promise<Response>;
-    /** `POST /api/v2/jobs`. */
+    /**
+     * `GET /api/v2/assets/{id}/content` with the redirect **not** followed —
+     * hands back a directly-fetchable URL instead of the bytes, so a caller
+     * (e.g. a serverless/Cloudflare Worker) can pass the URL to a downstream
+     * consumer without streaming the body through itself. On an object-storage
+     * backend (Cloud/serverless) the server redirects to a short-lived signed
+     * URL; that redirect is deliberately not followed (unlike
+     * {@link getAssetContent}), both so its `Location` can be read and so this
+     * client's bearer token is never attached to the object-storage host. On a
+     * self-hosted proxy (which serves the bytes inline, no redirect) this
+     * returns the endpoint's own absolute URL instead. Works on every backend
+     * and never throws for either shape — only a genuine failure maps to the
+     * usual typed error.
+     */
+    getAssetContentUrl(assetId: string, options?: {
+        signal?: AbortSignal;
+    }): Promise<AssetContentUrl>;
+    /**
+     * `POST /api/v2/jobs`. `extraData` (e.g. the partner-node API key) is a
+     * sibling of `workflow` on the wire, per the spec's closed `extra_data`
+     * object; omitted entirely when not provided, never sent empty.
+     */
     postJobs(workflow: Record<string, unknown>, options?: {
         idempotencyKey?: string;
+        extraData?: PostJobsData["body"]["extra_data"];
         signal?: AbortSignal;
     }): Promise<Job>;
     /** `GET /api/v2/jobs/{id}` (or an absolute self link). */

package/dist/low/transport.js

@@ -29,9 +29,55 @@ import { errorFromEnvelope } from "./errors.js";
 import { iterateSse } from "./sse.js";
 const API_PREFIX = "/api/v2";
 const DEFAULT_TIMEOUT_MS = 30_000;
+// Mirrors `version` in package.json. Hand-kept in sync: this package builds
+// with plain `tsc` (no bundler/codegen step to inline it), and importing
+// `../../package.json` directly would step outside `tsc`'s configured
+// `rootDir`.
+const SDK_VERSION = "0.1.0";
 function looksLikePath(value) {
     return value.startsWith("http") || value.startsWith("/");
 }
+function buildUserAgent(clientInfo) {
+    const base = `comfy-sdk-typescript/${SDK_VERSION} (node ${process.version})`;
+    if (!clientInfo)
+        return base;
+    // A caller-set token goes verbatim into a header value; reject CR/LF so it
+    // can never split/inject headers (undici would reject it anyway, but fail
+    // fast with a clear message at construction).
+    if (/[\r\n]/.test(clientInfo)) {
+        throw new Error("clientInfo must not contain CR or LF characters");
+    }
+    return `${base} app/${clientInfo}`;
+}
+const GOOG_DATE_RE = /^(\d{4})(\d{2})(\d{2})T(\d{2})(\d{2})(\d{2})Z$/;
+/**
+ * Expiry off a GCS V4 signed URL's `X-Goog-Date` (`YYYYMMDDTHHMMSSZ`, UTC) +
+ * `X-Goog-Expires` (seconds) query params. `null` if either is missing or
+ * doesn't match — an unrecognized signed-URL flavor degrades to "unknown
+ * expiry" rather than throwing.
+ */
+function parseExpiry(url) {
+    let parsed;
+    try {
+        parsed = new URL(url);
+    }
+    catch {
+        return null;
+    }
+    const dateParam = parsed.searchParams.get("X-Goog-Date");
+    const expiresParam = parsed.searchParams.get("X-Goog-Expires");
+    if (!dateParam || !expiresParam)
+        return null;
+    const match = GOOG_DATE_RE.exec(dateParam);
+    if (!match)
+        return null;
+    const expiresSeconds = Number.parseInt(expiresParam, 10);
+    if (Number.isNaN(expiresSeconds))
+        return null;
+    const [, year, month, day, hour, minute, second] = match;
+    const epochMs = Date.UTC(Number(year), Number(month) - 1, Number(day), Number(hour), Number(minute), Number(second));
+    return new Date(epochMs + expiresSeconds * 1000);
+}
 function parseRetryAfter(response) {
     const raw = response.headers.get("Retry-After");
     if (raw === null)
@@ -45,11 +91,13 @@ export class ComfyLow {
     apiKey;
     fetchImpl;
     defaultTimeoutMs;
+    userAgent;
     constructor(baseUrl, apiKey, options = {}) {
         this.baseUrl = baseUrl.replace(/\/$/, "");
         this.apiKey = apiKey;
         this.fetchImpl = options.fetch ?? fetch;
         this.defaultTimeoutMs = options.timeoutMs ?? DEFAULT_TIMEOUT_MS;
+        this.userAgent = buildUserAgent(options.clientInfo);
     }
     urlFor(path) {
         if (path.startsWith("http"))
@@ -90,6 +138,12 @@ export class ComfyLow {
                 headers.set(key, value);
             }
         }
+        // Identifies this SDK's traffic in request logs (support + adoption
+        // metrics, no other data collected) — never overrides a caller-supplied
+        // header.
+        if (!headers.has("User-Agent")) {
+            headers.set("User-Agent", this.userAgent);
+        }
         return headers;
     }
     resolveSignal(callerSignal, timeoutMs) {
@@ -113,7 +167,13 @@ export class ComfyLow {
             body = JSON.stringify(options.json);
         }
         const signal = this.resolveSignal(options.signal, options.timeoutMs);
-        return this.fetchImpl(url, { method, headers, body, signal, redirect: "follow" });
+        return this.fetchImpl(url, {
+            method,
+            headers,
+            body,
+            signal,
+            redirect: options.redirect ?? "follow",
+        });
     }
     async parseOrRaise(response, ok) {
         if (ok.includes(response.status)) {
@@ -137,8 +197,11 @@ export class ComfyLow {
     // -- assets -------------------------------------------------------------
     /** `POST /api/v2/assets` — streaming multipart upload. */
     async postAssets(file, contentType, filePath, options = {}) {
+        // public-api streams the multipart upload and requires the metadata fields
+        // BEFORE the file part (so it can route the file stream without buffering).
+        // The `file` part MUST be appended last, or the server rejects with 422
+        // "content_type is required and must be sent before the file field".
         const form = new FormData();
-        form.append("file", file, filePath);
         form.append("content_type", contentType);
         form.append("file_path", filePath);
         if (options.expectedHash !== undefined) {
@@ -147,6 +210,7 @@ export class ComfyLow {
         for (const tag of options.tags ?? []) {
             form.append("tags", tag);
         }
+        form.append("file", file, filePath);
         const headers = {};
         if (options.idempotencyKey) {
             headers["Idempotency-Key"] = options.idempotencyKey;
@@ -209,14 +273,58 @@ export class ComfyLow {
         }
         return response;
     }
+    /**
+     * `GET /api/v2/assets/{id}/content` with the redirect **not** followed —
+     * hands back a directly-fetchable URL instead of the bytes, so a caller
+     * (e.g. a serverless/Cloudflare Worker) can pass the URL to a downstream
+     * consumer without streaming the body through itself. On an object-storage
+     * backend (Cloud/serverless) the server redirects to a short-lived signed
+     * URL; that redirect is deliberately not followed (unlike
+     * {@link getAssetContent}), both so its `Location` can be read and so this
+     * client's bearer token is never attached to the object-storage host. On a
+     * self-hosted proxy (which serves the bytes inline, no redirect) this
+     * returns the endpoint's own absolute URL instead. Works on every backend
+     * and never throws for either shape — only a genuine failure maps to the
+     * usual typed error.
+     */
+    async getAssetContentUrl(assetId, options = {}) {
+        const path = `/assets/${encodeURIComponent(assetId)}/content`;
+        const response = await this.request("GET", path, {
+            signal: options.signal,
+            redirect: "manual",
+        });
+        const location = response.headers.get("Location");
+        // Node/undici hands back the real 3xx status with a readable `Location`
+        // for a manual redirect; status 0 covers a browser's opaque redirect.
+        // We only want the URL, not the bytes — release the response body so
+        // undici can reuse the connection instead of pinning it until GC.
+        if (location && (response.status === 0 || (response.status >= 300 && response.status < 400))) {
+            await response.body?.cancel();
+            return { url: location, expiresAt: parseExpiry(location) };
+        }
+        if (response.status === 200 || response.status === 206) {
+            await response.body?.cancel();
+            return { url: this.urlFor(path), expiresAt: null };
+        }
+        return this.parseOrRaise(response, [200, 206]); // always throws here
+    }
     // -- jobs -----------------------------------------------------------------
-    /** `POST /api/v2/jobs`. */
+    /**
+     * `POST /api/v2/jobs`. `extraData` (e.g. the partner-node API key) is a
+     * sibling of `workflow` on the wire, per the spec's closed `extra_data`
+     * object; omitted entirely when not provided, never sent empty.
+     */
     async postJobs(workflow, options = {}) {
         const headers = {};
         if (options.idempotencyKey) {
             headers["Idempotency-Key"] = options.idempotencyKey;
         }
         const json = { workflow };
+        // Only attach a non-empty extra_data — never serialize an empty object onto
+        // the wire (mirrors the Python low layer's truthy guard).
+        if (options.extraData && Object.keys(options.extraData).length > 0) {
+            json.extra_data = options.extraData;
+        }
         const response = await this.request("POST", "/jobs", { headers, json, signal: options.signal });
         return this.parseOrRaise(response, [201]);
     }

package/dist/sdk/client.d.ts

@@ -37,6 +37,9 @@ export interface ComfyOptions {
     apiKey?: string;
     timeoutMs?: number;
     fetch?: ComfyLowOptions["fetch"];
+    /** Appended to the SDK's default `User-Agent` as `app/{clientInfo}` — lets
+     * an app built on this SDK attribute its own traffic in request logs. */
+    clientInfo?: string;
 }
 export declare class Comfy {
     private readonly low;
@@ -57,14 +60,22 @@ export declare class Comfy {
      * idempotent, pass an explicit `idempotencyKey` and reuse it. Note a reused
      * key is *rejected*, not replayed: on reuse, catch the error and poll/list
      * for the job the first attempt already created.
+     *
+     * Pass `apiKey` to authenticate partner (API) nodes in the workflow (for
+     * example Gemini) — it is sent once, as `extra_data.api_key_comfy_org`
+     * alongside the workflow, and is unrelated to the `Idempotency-Key`: it
+     * does not affect idempotency and is never persisted or logged by this
+     * SDK. Omit it and no `extra_data` is sent at all.
      */
     submit(workflow: Workflow, options?: {
         idempotencyKey?: string;
+        apiKey?: string;
         signal?: AbortSignal;
     }): Promise<Job>;
     /** Submit, then poll to terminal (authoritative). Throws on failure. */
     run(workflow: Workflow, options?: {
         timeoutMs?: number;
+        apiKey?: string;
         signal?: AbortSignal;
     }): Promise<Job>;
 }

package/dist/sdk/client.js

@@ -52,6 +52,7 @@ export class Comfy {
         this.low = new ComfyLow(baseUrl, options.apiKey, {
             timeoutMs: options.timeoutMs,
             fetch: options.fetch,
+            clientInfo: options.clientInfo,
         });
         this.assets = new AssetFactory(this.low);
         this.workflows = new WorkflowFactory();
@@ -77,16 +78,26 @@ export class Comfy {
      * idempotent, pass an explicit `idempotencyKey` and reuse it. Note a reused
      * key is *rejected*, not replayed: on reuse, catch the error and poll/list
      * for the job the first attempt already created.
+     *
+     * Pass `apiKey` to authenticate partner (API) nodes in the workflow (for
+     * example Gemini) — it is sent once, as `extra_data.api_key_comfy_org`
+     * alongside the workflow, and is unrelated to the `Idempotency-Key`: it
+     * does not affect idempotency and is never persisted or logged by this
+     * SDK. Omit it and no `extra_data` is sent at all.
      */
     async submit(workflow, options = {}) {
         guardUiFormat(workflow);
         const graph = await this.materialize(workflow, options.signal);
         const key = options.idempotencyKey ?? newIdempotencyKey();
+        // A falsy apiKey (undefined or "") means "no key" — send no extra_data,
+        // matching the Python SDK's behavior so the two stay in lockstep.
+        const extraData = options.apiKey ? { api_key_comfy_org: options.apiKey } : undefined;
         const deadline = Date.now() + QUEUE_RETRY_BUDGET_MS;
         for (;;) {
             try {
                 const job = await this.low.postJobs(graph, {
                     idempotencyKey: key,
+                    extraData,
                     signal: options.signal,
                 });
                 return new Job(this.low, job);
@@ -105,7 +116,7 @@ export class Comfy {
     }
     /** Submit, then poll to terminal (authoritative). Throws on failure. */
     async run(workflow, options = {}) {
-        const job = await this.submit(workflow, { signal: options.signal });
+        const job = await this.submit(workflow, { apiKey: options.apiKey, signal: options.signal });
         return options.timeoutMs === undefined
             ? job.result(options.signal)
             : runWithTimeout(job, options.timeoutMs, options.signal);

package/dist/sdk/exceptions.js

@@ -79,6 +79,11 @@ const BY_CODE = {
     idempotency_key_reuse: IdempotencyKeyReuse,
     insufficient_credits: InsufficientCredits,
     not_found: NotFound,
+    // public-api returns entity-specific 404 codes even though the spec documents
+    // the generic not_found; map them so a missing job/asset raises the typed
+    // NotFound. (Server/spec reconciliation of the code set is a separate follow-up.)
+    job_not_found: NotFound,
+    asset_not_found: NotFound,
     unauthorized: Unauthorized,
     forbidden: Forbidden,
 };

package/dist/sdk/outputs.d.ts

@@ -23,4 +23,18 @@ export declare class Output {
     toBytes(options?: {
         range?: readonly [number, number];
     }): Promise<Uint8Array>;
+    /**
+     * A directly-fetchable URL for this output's bytes — a short-lived,
+     * self-authorizing bearer credential (readable until `expiresAt`) — so a
+     * caller (e.g. a serverless/Cloudflare Worker) can hand the URL to a
+     * downstream consumer instead of streaming the bytes through itself. On an
+     * object-storage backend (Cloud/serverless) this is a signed URL and
+     * `expiresAt` is set; on a self-hosted proxy (which serves the bytes
+     * inline) it's this asset's own content URL and `expiresAt` is `null`.
+     * Works on every backend and never throws.
+     */
+    getDownloadUrl(): Promise<{
+        url: string;
+        expiresAt: Date | null;
+    }>;
 }

package/dist/sdk/outputs.js

@@ -45,4 +45,17 @@ export class Output {
         const response = await this.low.getAssetContent(this.model.id, { range: options.range });
         return new Uint8Array(await response.arrayBuffer());
     }
+    /**
+     * A directly-fetchable URL for this output's bytes — a short-lived,
+     * self-authorizing bearer credential (readable until `expiresAt`) — so a
+     * caller (e.g. a serverless/Cloudflare Worker) can hand the URL to a
+     * downstream consumer instead of streaming the bytes through itself. On an
+     * object-storage backend (Cloud/serverless) this is a signed URL and
+     * `expiresAt` is set; on a self-hosted proxy (which serves the bytes
+     * inline) it's this asset's own content URL and `expiresAt` is `null`.
+     * Works on every backend and never throws.
+     */
+    async getDownloadUrl() {
+        return this.low.getAssetContentUrl(this.model.id);
+    }
 }

package/package.json

@@ -1,8 +1,16 @@
 {
   "name": "@comfyorg/sdk",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "private": false,
   "description": "TypeScript SDK for running ComfyUI workflows via the Comfy API v2 (self-hosted, Comfy Cloud, serverless).",
+  "homepage": "https://github.com/Comfy-Org/ComfyTypeScriptSDK#readme",
+  "bugs": {
+    "url": "https://github.com/Comfy-Org/ComfyTypeScriptSDK/issues"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/Comfy-Org/ComfyTypeScriptSDK.git"
+  },
   "files": [
     "dist"
   ],
@@ -19,17 +27,6 @@
       "default": "./dist/low/index.js"
     }
   },
-  "scripts": {
-    "build": "tsc",
-    "generate": "openapi-ts",
-    "lint": "oxlint .",
-    "format": "oxfmt --write .",
-    "format:check": "oxfmt --check .",
-    "typecheck": "tsc --noEmit",
-    "test": "vitest run",
-    "test:coverage": "vitest run --coverage",
-    "check:spec-drift": "node scripts/check-spec-drift.mjs"
-  },
   "dependencies": {
     "eventsource-parser": "^3.0.0",
     "hash-wasm": "^4.11.0",
@@ -48,5 +45,15 @@
   "engines": {
     "node": ">=22"
   },
-  "packageManager": "pnpm@11.13.0"
-}
+  "scripts": {
+    "build": "tsc",
+    "generate": "openapi-ts",
+    "lint": "oxlint .",
+    "format": "oxfmt --write .",
+    "format:check": "oxfmt --check .",
+    "typecheck": "tsc --noEmit",
+    "test": "vitest run",
+    "test:coverage": "vitest run --coverage",
+    "check:spec-drift": "node scripts/check-spec-drift.mjs"
+  }
+}