@retab/node 1.0.98 → 1.0.99

package/dist/api/workflows/runs/steps/client.d.ts

@@ -1,34 +1,25 @@
 import { CompositionClient, RequestOptions } from "../../../../client.js";
-import { StepOutputResponse, WorkflowRunStep } from "../../../../types.js";
+import { StepOutputResponse, StepOutputsBatchResponse, WorkflowRunStep, WorkflowRun } from "../../../../types.js";
 /**
  * Workflow Run Steps API client for accessing step-level outputs.
  *
- * Usage: `client.workflows.runs.steps.get(runId, nodeId)` or `client.workflows.runs.steps.list(runId)`
+ * Usage: `client.workflows.runs.steps.get(runId, nodeId)` or `client.workflows.runs.steps.getMany(runId, nodeIds)`
  */
 export default class APIWorkflowRunSteps extends CompositionClient {
     constructor(client: CompositionClient);
     /**
-     * Get the full output data for a specific step in a workflow run.
-     *
-     * @param runId - The ID of the workflow run
-     * @param nodeId - The ID of the node/step to get output for
-     * @param options - Optional request options
-     * @returns The step output with handle outputs and inputs
+     * Get step status and handle data for a specific step in a workflow run.
      *
      * @example
      * ```typescript
      * const step = await client.workflows.runs.steps.get("run_abc123", "extract-node-1");
-     * console.log(`Step status: ${step.status}, output:`, step.output);
+     * const data = step.handle_outputs?.["output-json-0"]?.data;
      * ```
      */
     get(runId: string, nodeId: string, options?: RequestOptions): Promise<StepOutputResponse>;
     /**
      * List all persisted step documents for a workflow run.
      *
-     * @param runId - The ID of the workflow run
-     * @param options - Optional request options
-     * @returns All step documents for the run
-     *
      * @example
      * ```typescript
      * const steps = await client.workflows.runs.steps.list("run_abc123");
@@ -38,5 +29,36 @@ export default class APIWorkflowRunSteps extends CompositionClient {
      * ```
      */
     list(runId: string, options?: RequestOptions): Promise<WorkflowRunStep[]>;
+    /**
+     * Batch-get step outputs for multiple nodes in a single request.
+     *
+     * @param runId - The workflow run ID
+     * @param nodeIds - List of node IDs to fetch outputs for
+     * @returns Step outputs keyed by node ID
+     *
+     * @example
+     * ```typescript
+     * const batch = await client.workflows.runs.steps.getMany("run_abc123", ["extract-1", "classifier-1"]);
+     * console.log(batch.outputs["extract-1"]?.handle_outputs);
+     * ```
+     */
+    getMany(runId: string, nodeIds: string[], options?: RequestOptions): Promise<StepOutputsBatchResponse>;
+    get_many(runId: string, nodeIds: string[], options?: RequestOptions): Promise<StepOutputsBatchResponse>;
+    /**
+     * Fetch outputs for all steps in a workflow run in one call.
+     *
+     * Internally fetches the run to discover step node IDs, then batch-fetches all outputs.
+     *
+     * @param runId - The workflow run ID
+     * @returns Step outputs keyed by node ID
+     *
+     * @example
+     * ```typescript
+     * const allOutputs = await client.workflows.runs.steps.getAll("run_abc123");
+     * console.log(allOutputs.outputs["extract-1"]?.handle_outputs);
+     * ```
+     */
+    getAll(run: WorkflowRun | string, options?: RequestOptions): Promise<StepOutputsBatchResponse>;
+    get_all(run: WorkflowRun | string, options?: RequestOptions): Promise<StepOutputsBatchResponse>;
 }
 //# sourceMappingURL=client.d.ts.map

package/dist/api/workflows/runs/steps/client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../../../../src/api/workflows/runs/steps/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAC1E,OAAO,EACH,kBAAkB,EAElB,eAAe,EAElB,MAAM,sBAAsB,CAAC;AAG9B;;;;GAIG;AACH,MAAM,CAAC,OAAO,OAAO,mBAAoB,SAAQ,iBAAiB;gBAClD,MAAM,EAAE,iBAAiB;IAIrC;;;;;;;;;;;;;OAaG;IACG,GAAG,CACL,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,cAAc,GACzB,OAAO,CAAC,kBAAkB,CAAC;IAS9B;;;;;;;;;;;;;;OAcG;IACG,IAAI,CACN,KAAK,EAAE,MAAM,EACb,OAAO,CAAC,EAAE,cAAc,GACzB,OAAO,CAAC,eAAe,EAAE,CAAC;CAQhC"}
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../../../../src/api/workflows/runs/steps/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAC1E,OAAO,EACH,kBAAkB,EAElB,wBAAwB,EAExB,eAAe,EAEf,WAAW,EAEd,MAAM,sBAAsB,CAAC;AAG9B;;;;GAIG;AACH,MAAM,CAAC,OAAO,OAAO,mBAAoB,SAAQ,iBAAiB;gBAClD,MAAM,EAAE,iBAAiB;IAIrC;;;;;;;;OAQG;IACG,GAAG,CACL,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,cAAc,GACzB,OAAO,CAAC,kBAAkB,CAAC;IAS9B;;;;;;;;;;OAUG;IACG,IAAI,CACN,KAAK,EAAE,MAAM,EACb,OAAO,CAAC,EAAE,cAAc,GACzB,OAAO,CAAC,eAAe,EAAE,CAAC;IAS7B;;;;;;;;;;;;OAYG;IACG,OAAO,CACT,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,MAAM,EAAE,EACjB,OAAO,CAAC,EAAE,cAAc,GACzB,OAAO,CAAC,wBAAwB,CAAC;IAU9B,QAAQ,CACV,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,MAAM,EAAE,EACjB,OAAO,CAAC,EAAE,cAAc,GACzB,OAAO,CAAC,wBAAwB,CAAC;IAIpC;;;;;;;;;;;;;OAaG;IACG,MAAM,CACR,GAAG,EAAE,WAAW,GAAG,MAAM,EACzB,OAAO,CAAC,EAAE,cAAc,GACzB,OAAO,CAAC,wBAAwB,CAAC;IAkB9B,OAAO,CACT,GAAG,EAAE,WAAW,GAAG,MAAM,EACzB,OAAO,CAAC,EAAE,cAAc,GACzB,OAAO,CAAC,wBAAwB,CAAC;CAGvC"}

package/dist/api/workflows/runs/steps/client.js

@@ -1,27 +1,22 @@
 import { CompositionClient } from "../../../../client.js";
-import { ZStepOutputResponse, ZWorkflowRunStep, } from "../../../../types.js";
+import { ZStepOutputResponse, ZStepOutputsBatchResponse, ZWorkflowRunStep, ZWorkflowRun, } from "../../../../types.js";
 import * as z from "zod";
 /**
  * Workflow Run Steps API client for accessing step-level outputs.
  *
- * Usage: `client.workflows.runs.steps.get(runId, nodeId)` or `client.workflows.runs.steps.list(runId)`
+ * Usage: `client.workflows.runs.steps.get(runId, nodeId)` or `client.workflows.runs.steps.getMany(runId, nodeIds)`
  */
 export default class APIWorkflowRunSteps extends CompositionClient {
     constructor(client) {
         super(client);
     }
     /**
-     * Get the full output data for a specific step in a workflow run.
-     *
-     * @param runId - The ID of the workflow run
-     * @param nodeId - The ID of the node/step to get output for
-     * @param options - Optional request options
-     * @returns The step output with handle outputs and inputs
+     * Get step status and handle data for a specific step in a workflow run.
      *
      * @example
      * ```typescript
      * const step = await client.workflows.runs.steps.get("run_abc123", "extract-node-1");
-     * console.log(`Step status: ${step.status}, output:`, step.output);
+     * const data = step.handle_outputs?.["output-json-0"]?.data;
      * ```
      */
     async get(runId, nodeId, options) {
@@ -35,10 +30,6 @@ export default class APIWorkflowRunSteps extends CompositionClient {
     /**
      * List all persisted step documents for a workflow run.
      *
-     * @param runId - The ID of the workflow run
-     * @param options - Optional request options
-     * @returns All step documents for the run
-     *
      * @example
      * ```typescript
      * const steps = await client.workflows.runs.steps.list("run_abc123");
@@ -55,4 +46,61 @@ export default class APIWorkflowRunSteps extends CompositionClient {
             headers: options?.headers,
         });
     }
+    /**
+     * Batch-get step outputs for multiple nodes in a single request.
+     *
+     * @param runId - The workflow run ID
+     * @param nodeIds - List of node IDs to fetch outputs for
+     * @returns Step outputs keyed by node ID
+     *
+     * @example
+     * ```typescript
+     * const batch = await client.workflows.runs.steps.getMany("run_abc123", ["extract-1", "classifier-1"]);
+     * console.log(batch.outputs["extract-1"]?.handle_outputs);
+     * ```
+     */
+    async getMany(runId, nodeIds, options) {
+        return this._fetchJson(ZStepOutputsBatchResponse, {
+            url: `/workflows/runs/${runId}/steps/batch`,
+            method: "POST",
+            body: { node_ids: nodeIds, ...(options?.body || {}) },
+            params: options?.params,
+            headers: options?.headers,
+        });
+    }
+    async get_many(runId, nodeIds, options) {
+        return this.getMany(runId, nodeIds, options);
+    }
+    /**
+     * Fetch outputs for all steps in a workflow run in one call.
+     *
+     * Internally fetches the run to discover step node IDs, then batch-fetches all outputs.
+     *
+     * @param runId - The workflow run ID
+     * @returns Step outputs keyed by node ID
+     *
+     * @example
+     * ```typescript
+     * const allOutputs = await client.workflows.runs.steps.getAll("run_abc123");
+     * console.log(allOutputs.outputs["extract-1"]?.handle_outputs);
+     * ```
+     */
+    async getAll(run, options) {
+        const workflowRun = typeof run === "string"
+            ? await this._fetchJson(ZWorkflowRun, {
+                url: `/workflows/runs/${run}`,
+                method: "GET",
+                params: options?.params,
+                headers: options?.headers,
+            })
+            : run;
+        const nodeIds = workflowRun.steps.map((s) => s.node_id);
+        if (nodeIds.length === 0) {
+            return { outputs: {} };
+        }
+        return this.getMany(workflowRun.id, nodeIds, options);
+    }
+    async get_all(run, options) {
+        return this.getAll(run, options);
+    }
 }

package/dist/client.d.ts

@@ -5,7 +5,7 @@ type FetchParams = {
     params?: Record<string, any>;
     headers?: Record<string, any>;
     bodyMime?: "application/json" | "multipart/form-data";
-    body?: Record<string, any>;
+    body?: Record<string, any> | unknown[];
     auth?: string[];
 };
 export declare class AbstractClient {
@@ -18,6 +18,8 @@ export declare class CompositionClient extends AbstractClient {
     protected _client: AbstractClient;
     constructor(client: AbstractClient);
     protected _fetch(params: FetchParams): Promise<Response>;
+    private _installPrepareMethods;
+    protected _capturePreparedRequest(methodName: string, args: unknown[]): Promise<FetchParams>;
 }
 export declare class APIError extends Error {
     status: number;
@@ -50,7 +52,7 @@ export declare class FetcherClient extends AbstractClient {
         params?: Record<string, any>;
         headers?: Record<string, any>;
         bodyMime?: "application/json" | "multipart/form-data";
-        body?: Record<string, any>;
+        body?: Record<string, any> | unknown[];
     }): Promise<Response>;
     /**
      * Verify the signature of a webhook event.

package/dist/client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,CAAC,MAAM,KAAK,CAAC;AAGzB,KAAK,WAAW,GAAG;IACjB,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC7B,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC9B,QAAQ,CAAC,EAAE,kBAAkB,GAAG,qBAAqB,CAAC;IACtD,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC3B,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;CACjB,CAAC;AAyCF,qBAAa,cAAc;IACzB,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC;cAGnC,UAAU,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;cAC9C,UAAU,CAAC,SAAS,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;cAc9H,YAAY,CAAC,SAAS,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;CAQ/J;AAED,qBAAa,iBAAkB,SAAQ,cAAc;IACnD,SAAS,CAAC,OAAO,EAAE,cAAc,CAAC;gBACtB,MAAM,EAAE,cAAc;IAIlC,SAAS,CAAC,MAAM,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC;CAGzD;AAED,qBAAa,QAAS,SAAQ,KAAK;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;gBACD,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM;CAKzC;AAED,qBAAa,0BAA2B,SAAQ,KAAK;gBACvC,OAAO,EAAE,MAAM;CAI5B;AAED,eAAO,MAAM,SAAS,gGAKpB,CAAC;AAEH,KAAK,SAAS,GAAG;IAAE,MAAM,EAAE,MAAM,CAAA;CAAE,GAAG,EAAE,CAAC;AACzC,MAAM,MAAM,aAAa,GAAG;IAC1B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,GAAG,SAAS,CAAC;AAEd,MAAM,MAAM,cAAc,GAAG;IAC3B,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC7B,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC9B,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC5B,CAAC;AAEF,qBAAa,aAAc,SAAQ,cAAc;IAC/C,OAAO,EAAE,aAAa,CAAC;IACvB,OAAO,EAAE,MAAM,CAAC;gBACJ,OAAO,CAAC,EAAE,aAAa;IAgB7B,MAAM,CAAC,MAAM,EAAE;QACnB,GAAG,EAAE,MAAM,CAAC;QACZ,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC7B,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC9B,QAAQ,CAAC,EAAE,kBAAkB,GAAG,qBAAqB,CAAC;QACtD,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAC5B,GAAG,OAAO,CAAC,QAAQ,CAAC;IA8CrB;;;;;;;;;;;;;;;;;;;;;;;;;OAyBG;IACH,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,GAAG;CAc5F"}
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,CAAC,MAAM,KAAK,CAAC;AAGzB,KAAK,WAAW,GAAG;IACjB,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC7B,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC9B,QAAQ,CAAC,EAAE,kBAAkB,GAAG,qBAAqB,CAAC;IACtD,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,OAAO,EAAE,CAAC;IACvC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;CACjB,CAAC;AA2DF,qBAAa,cAAc;IACzB,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC;cAGnC,UAAU,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;cAC9C,UAAU,CAAC,SAAS,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;cAc9H,YAAY,CAAC,SAAS,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;CAQ/J;AAED,qBAAa,iBAAkB,SAAQ,cAAc;IACnD,SAAS,CAAC,OAAO,EAAE,cAAc,CAAC;gBACtB,MAAM,EAAE,cAAc;IAKlC,SAAS,CAAC,MAAM,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC;IAIxD,OAAO,CAAC,sBAAsB;cAsBd,uBAAuB,CAAC,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC;CAyCnG;AAED,qBAAa,QAAS,SAAQ,KAAK;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;gBACD,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM;CAKzC;AAED,qBAAa,0BAA2B,SAAQ,KAAK;gBACvC,OAAO,EAAE,MAAM;CAI5B;AAED,eAAO,MAAM,SAAS,gGAKpB,CAAC;AAEH,KAAK,SAAS,GAAG;IAAE,MAAM,EAAE,MAAM,CAAA;CAAE,GAAG,EAAE,CAAC;AACzC,MAAM,MAAM,aAAa,GAAG;IAC1B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,GAAG,SAAS,CAAC;AAEd,MAAM,MAAM,cAAc,GAAG;IAC3B,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC7B,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC9B,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC5B,CAAC;AAEF,qBAAa,aAAc,SAAQ,cAAc;IAC/C,OAAO,EAAE,aAAa,CAAC;IACvB,OAAO,EAAE,MAAM,CAAC;gBACJ,OAAO,CAAC,EAAE,aAAa;IAgB7B,MAAM,CAAC,MAAM,EAAE;QACnB,GAAG,EAAE,MAAM,CAAC;QACZ,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC7B,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC9B,QAAQ,CAAC,EAAE,kBAAkB,GAAG,qBAAqB,CAAC;QACtD,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,OAAO,EAAE,CAAC;KACxC,GAAG,OAAO,CAAC,QAAQ,CAAC;IA+CrB;;;;;;;;;;;;;;;;;;;;;;;;;OAyBG;IACH,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,GAAG;CAc5F"}

package/dist/client.js

@@ -1,5 +1,22 @@
 import * as z from "zod";
 import * as crypto from "crypto";
+const PYTHON_PUBLIC_PREPARE_METHODS = {
+    APIModels: ["list"],
+    APISchemas: ["generate"],
+    APIExtractions: ["list", "download", "update", "get", "sources", "delete"],
+    APIFiles: ["upload", "list", "get", "get_download_link"],
+    APIProjects: ["create", "get", "list", "delete", "publish", "extract", "split"],
+    ProjectDatasets: ["create", "get", "list", "update", "delete", "duplicate", "add_document", "get_document", "list_documents", "update_document", "delete_document"],
+    ProjectIterations: ["create", "get", "list", "update_draft", "delete", "finalize", "get_schema", "process_documents", "get_document", "list_documents", "update_document", "delete_document", "get_metrics"],
+    APIWorkflows: ["get", "list", "create", "update", "delete", "publish", "duplicate", "get_entities"],
+    APIWorkflowBlocks: ["list", "get", "create", "create_batch", "update", "delete"],
+    APIWorkflowEdges: ["list", "get", "create", "create_batch", "delete", "delete_all"],
+    APIWorkflowRuns: ["create", "get", "list", "delete", "cancel", "restart", "resume"],
+    APIWorkflowRunSteps: ["get", "list", "get_many"],
+    APIEvalsExtract: ["process", "process_stream"],
+    APIEvalsSplit: ["process"],
+    APIEvalsClassify: ["process"],
+};
 async function* streamResponse(schema, response) {
     let body = "";
     let depth = 0;
@@ -57,7 +74,7 @@ export class AbstractClient {
         if (!bodyType) {
             return;
         }
-        if (response.headers.get("Content-Type") !== "application/json")
+        if (!response.headers.get("Content-Type")?.startsWith("application/json"))
             throw new APIError(response.status, "Response is not JSON");
         return bodyType.parseAsync(await response.json());
     }
@@ -66,7 +83,7 @@ export class AbstractClient {
         if (!response.ok) {
             throw new APIError(response.status, await response.text());
         }
-        if (response.headers.get("Content-Type") !== "application/stream+json")
+        if (!response.headers.get("Content-Type")?.startsWith("application/stream+json"))
             throw new APIError(response.status, "Response is not stream JSON");
         return streamResponse(schema, response);
     }
@@ -75,10 +92,66 @@ export class CompositionClient extends AbstractClient {
     constructor(client) {
         super();
         this._client = client;
+        this._installPrepareMethods();
     }
     _fetch(params) {
         return this._client["_fetch"](params);
     }
+    _installPrepareMethods() {
+        const allowedMethods = PYTHON_PUBLIC_PREPARE_METHODS[this.constructor.name] ?? [];
+        if (allowedMethods.length === 0) {
+            return;
+        }
+        for (const methodName of allowedMethods) {
+            const method = this[methodName];
+            const prepareMethodName = `prepare_${methodName}`;
+            if (typeof method !== "function" || prepareMethodName in this) {
+                continue;
+            }
+            Object.defineProperty(this, prepareMethodName, {
+                configurable: true,
+                enumerable: false,
+                writable: false,
+                value: (...args) => this._capturePreparedRequest(methodName, args),
+            });
+        }
+    }
+    async _capturePreparedRequest(methodName, args) {
+        let capturedRequest;
+        const self = this;
+        const originalFetch = self._fetch;
+        const originalFetchJson = self._fetchJson;
+        const originalFetchStream = self._fetchStream;
+        self._fetch = async (params) => {
+            capturedRequest = params;
+            return new Response("{}", {
+                status: 200,
+                headers: {
+                    "Content-Type": "application/json",
+                },
+            });
+        };
+        self._fetchJson = async (...fetchArgs) => {
+            capturedRequest = (fetchArgs.length === 2 ? fetchArgs[1] : fetchArgs[0]);
+            return {};
+        };
+        self._fetchStream = async (...fetchArgs) => {
+            capturedRequest = fetchArgs[1];
+            return (async function* emptyStream() { })();
+        };
+        try {
+            await self[methodName](...args);
+        }
+        finally {
+            self._fetch = originalFetch;
+            self._fetchJson = originalFetchJson;
+            self._fetchStream = originalFetchStream;
+        }
+        if (!capturedRequest) {
+            throw new Error(`Unable to capture prepared request for ${methodName}`);
+        }
+        return capturedRequest;
+    }
 }
 export class APIError extends Error {
     constructor(status, info) {
@@ -124,8 +197,9 @@ export class FetcherClient extends AbstractClient {
         if (params.method !== "GET") {
             if (params.bodyMime === "multipart/form-data") {
                 let formData = new FormData();
-                for (const key of Object.keys(params.body || {})) {
-                    let value = params.body[key];
+                const multipartBody = (params.body || {});
+                for (const key of Object.keys(multipartBody)) {
+                    let value = multipartBody[key];
                     if (Array.isArray(value)) {
                         for (const item of value) {
                             formData.append(key, item);