@scenario-labs/sdk 0.3.0 → 1.1.0

package/lib/models.mjs

@@ -1,6 +1,34 @@
 import { Models } from "../resources/models/models.mjs";
 import { Train } from "../resources/models/train.mjs";
 import { enhanceJob } from "./job.mjs";
+/**
+ * Enhanced Model entity with helper methods.
+ * Has all original model fields plus `.run()`.
+ *
+ * @example
+ * ```ts
+ * const { model } = await client.models.retrieve(modelId);
+ * const gen = await model.run({ body: { prompt: 'a red car' } });
+ * const completed = await gen.job.wait();
+ * ```
+ */
+export class ModelEntity {
+    /**
+     * Generate with this model. Shortcut for
+     * `client.generate.runModel(this.id, params)`.
+     *
+     * Returns an enhanced response where `response.job` has `.wait()`.
+     */
+    run(params, options) {
+        return this._client.generate.runModel(this.id, params, options);
+    }
+    /** @internal Create a ModelEntity wrapping raw model data. */
+    static from(client, data) {
+        const entity = Object.assign(Object.create(ModelEntity.prototype), data);
+        Object.defineProperty(entity, '_client', { value: client, enumerable: false });
+        return entity;
+    }
+}
 /**
  * Enhanced Train resource.
  * `trigger()` returns a response where `job` has `.wait()`.
@@ -12,12 +40,20 @@ export class EnhancedTrain extends Train {
 }
 /**
  * Enhanced Models resource.
- * All original methods inherited. `train` is an {@link EnhancedTrain}.
+ * All original methods inherited.
+ * - `train` is an {@link EnhancedTrain}
+ * - `retrieve()` returns a response where `model` has `.run()`
  */
 export class EnhancedModels extends Models {
     constructor() {
         super(...arguments);
         this.train = new EnhancedTrain(this._client);
     }
+    retrieve(modelID, query = {}, options) {
+        return super.retrieve(modelID, query, options)._thenUnwrap((data) => ({
+            ...data,
+            model: ModelEntity.from(this._client, data.model),
+        }));
+    }
 }
 //# sourceMappingURL=models.mjs.map

package/lib/models.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"models.mjs","sourceRoot":"","sources":["../src/lib/models.ts"],"names":[],"mappings":"OAEO,EAAE,MAAM,EAAE;OACV,EAAE,KAAK,EAAsD;OAC7D,EAAgB,UAAU,EAAE;AAEnC;;;GAGG;AACH,MAAM,OAAO,aAAc,SAAQ,KAAK;IAC7B,OAAO,CACd,OAAe,EACf,MAA0B,EAC1B,OAAwB;QAExB,OAAO,UAAU,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IAC3E,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,cAAe,SAAQ,MAAM;IAA1C;;QACW,UAAK,GAAkB,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAClE,CAAC;CAAA"}
+{"version":3,"file":"models.mjs","sourceRoot":"","sources":["../src/lib/models.ts"],"names":[],"mappings":"OAGO,EAAE,MAAM,EAAwD;OAChE,EAAE,KAAK,EAAsD;OAC7D,EAAgB,UAAU,EAAE;AAInC;;;;;;;;;;GAUG;AACH,MAAM,OAAO,WAAW;IAGtB;;;;;OAKG;IACH,GAAG,CAED,MAA8B,EAC9B,OAAwB;QAExB,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IAClE,CAAC;IAED,8DAA8D;IAC9D,MAAM,CAAC,IAAI,CACT,MAAgB,EAChB,IAAiC;QAEjC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,CAAC;QACzE,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/E,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAOD;;;GAGG;AACH,MAAM,OAAO,aAAc,SAAQ,KAAK;IAC7B,OAAO,CACd,OAAe,EACf,MAA0B,EAC1B,OAAwB;QAExB,OAAO,UAAU,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IAC3E,CAAC;CACF;AAED;;;;;GAKG;AACH,MAAM,OAAO,cAAe,SAAQ,MAAM;IAA1C;;QACW,UAAK,GAAkB,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAYlE,CAAC;IAVU,QAAQ,CACf,OAAe,EACf,QAAgD,EAAE,EAClD,OAAwB;QAExB,OAAO,KAAK,CAAC,QAAQ,CAAC,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACpE,GAAG,IAAI;YACP,KAAK,EAAE,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,OAAmB,EAAE,IAAI,CAAC,KAAK,CAAC;SAC9D,CAAC,CAAC,CAAC;IACN,CAAC;CACF"}

package/lib/scenario.d.mts

@@ -1,6 +1,9 @@
 import { Scenario as BaseScenario } from "../client.mjs";
+import { EnhancedAssets } from "./asset.mjs";
 import { EnhancedGenerate } from "./generate.mjs";
+import { EnhancedJobs } from "./job.mjs";
 import { EnhancedModels } from "./models.mjs";
+import { EnhancedUploads } from "./upload.mjs";
 import { EnhancedWorkflows } from "./workflow.mjs";
 /**
  * Enhanced Scenario client.
@@ -29,11 +32,22 @@ import { EnhancedWorkflows } from "./workflow.mjs";
  * // Enhanced: generate job.wait()
  * const gen = await client.generate.runModel(modelId, { body });
  * const done = await gen.job.wait();
+ *
+ * // Enhanced: upload.wait()
+ * const res = await client.uploads.retrieve(uploadId);
+ * const imported = await res.upload.wait();
+ *
+ * // Enhanced: asset.download()
+ * const a = await client.assets.retrieve(assetId);
+ * const bytes = await a.asset.download();
  * ```
  */
 export declare class Scenario extends BaseScenario {
     workflows: EnhancedWorkflows;
     generate: EnhancedGenerate;
     models: EnhancedModels;
+    uploads: EnhancedUploads;
+    assets: EnhancedAssets;
+    jobs: EnhancedJobs;
 }
 //# sourceMappingURL=scenario.d.mts.map

package/lib/scenario.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"scenario.d.mts","sourceRoot":"","sources":["../src/lib/scenario.ts"],"names":[],"mappings":"OAAO,EAAE,QAAQ,IAAI,YAAY,EAAE;OAC5B,EAAE,gBAAgB,EAAE;OACpB,EAAE,cAAc,EAAE;OAClB,EAAE,iBAAiB,EAAE;AAE5B;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,qBAAa,QAAS,SAAQ,YAAY;IAC/B,SAAS,EAAE,iBAAiB,CAA+B;IAC3D,QAAQ,EAAE,gBAAgB,CAA8B;IACxD,MAAM,EAAE,cAAc,CAA4B;CAC5D"}
+{"version":3,"file":"scenario.d.mts","sourceRoot":"","sources":["../src/lib/scenario.ts"],"names":[],"mappings":"OAAO,EAAE,QAAQ,IAAI,YAAY,EAAE;OAC5B,EAAE,cAAc,EAAE;OAClB,EAAE,gBAAgB,EAAE;OACpB,EAAE,YAAY,EAAE;OAChB,EAAE,cAAc,EAAE;OAClB,EAAE,eAAe,EAAE;OACnB,EAAE,iBAAiB,EAAE;AAE5B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAoCG;AACH,qBAAa,QAAS,SAAQ,YAAY;IAC/B,SAAS,EAAE,iBAAiB,CAA+B;IAC3D,QAAQ,EAAE,gBAAgB,CAA8B;IACxD,MAAM,EAAE,cAAc,CAA4B;IAClD,OAAO,EAAE,eAAe,CAA6B;IACrD,MAAM,EAAE,cAAc,CAA4B;IAClD,IAAI,EAAE,YAAY,CAA0B;CACtD"}

package/lib/scenario.d.ts

@@ -1,6 +1,9 @@
 import { Scenario as BaseScenario } from "../client.js";
+import { EnhancedAssets } from "./asset.js";
 import { EnhancedGenerate } from "./generate.js";
+import { EnhancedJobs } from "./job.js";
 import { EnhancedModels } from "./models.js";
+import { EnhancedUploads } from "./upload.js";
 import { EnhancedWorkflows } from "./workflow.js";
 /**
  * Enhanced Scenario client.
@@ -29,11 +32,22 @@ import { EnhancedWorkflows } from "./workflow.js";
  * // Enhanced: generate job.wait()
  * const gen = await client.generate.runModel(modelId, { body });
  * const done = await gen.job.wait();
+ *
+ * // Enhanced: upload.wait()
+ * const res = await client.uploads.retrieve(uploadId);
+ * const imported = await res.upload.wait();
+ *
+ * // Enhanced: asset.download()
+ * const a = await client.assets.retrieve(assetId);
+ * const bytes = await a.asset.download();
  * ```
  */
 export declare class Scenario extends BaseScenario {
     workflows: EnhancedWorkflows;
     generate: EnhancedGenerate;
     models: EnhancedModels;
+    uploads: EnhancedUploads;
+    assets: EnhancedAssets;
+    jobs: EnhancedJobs;
 }
 //# sourceMappingURL=scenario.d.ts.map

package/lib/scenario.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"scenario.d.ts","sourceRoot":"","sources":["../src/lib/scenario.ts"],"names":[],"mappings":"OAAO,EAAE,QAAQ,IAAI,YAAY,EAAE;OAC5B,EAAE,gBAAgB,EAAE;OACpB,EAAE,cAAc,EAAE;OAClB,EAAE,iBAAiB,EAAE;AAE5B;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,qBAAa,QAAS,SAAQ,YAAY;IAC/B,SAAS,EAAE,iBAAiB,CAA+B;IAC3D,QAAQ,EAAE,gBAAgB,CAA8B;IACxD,MAAM,EAAE,cAAc,CAA4B;CAC5D"}
+{"version":3,"file":"scenario.d.ts","sourceRoot":"","sources":["../src/lib/scenario.ts"],"names":[],"mappings":"OAAO,EAAE,QAAQ,IAAI,YAAY,EAAE;OAC5B,EAAE,cAAc,EAAE;OAClB,EAAE,gBAAgB,EAAE;OACpB,EAAE,YAAY,EAAE;OAChB,EAAE,cAAc,EAAE;OAClB,EAAE,eAAe,EAAE;OACnB,EAAE,iBAAiB,EAAE;AAE5B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAoCG;AACH,qBAAa,QAAS,SAAQ,YAAY;IAC/B,SAAS,EAAE,iBAAiB,CAA+B;IAC3D,QAAQ,EAAE,gBAAgB,CAA8B;IACxD,MAAM,EAAE,cAAc,CAA4B;IAClD,OAAO,EAAE,eAAe,CAA6B;IACrD,MAAM,EAAE,cAAc,CAA4B;IAClD,IAAI,EAAE,YAAY,CAA0B;CACtD"}

package/lib/scenario.js

@@ -2,8 +2,11 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Scenario = void 0;
 const client_1 = require("../client.js");
+const asset_1 = require("./asset.js");
 const generate_1 = require("./generate.js");
+const job_1 = require("./job.js");
 const models_1 = require("./models.js");
+const upload_1 = require("./upload.js");
 const workflow_1 = require("./workflow.js");
 /**
  * Enhanced Scenario client.
@@ -32,6 +35,14 @@ const workflow_1 = require("./workflow.js");
  * // Enhanced: generate job.wait()
  * const gen = await client.generate.runModel(modelId, { body });
  * const done = await gen.job.wait();
+ *
+ * // Enhanced: upload.wait()
+ * const res = await client.uploads.retrieve(uploadId);
+ * const imported = await res.upload.wait();
+ *
+ * // Enhanced: asset.download()
+ * const a = await client.assets.retrieve(assetId);
+ * const bytes = await a.asset.download();
  * ```
  */
 class Scenario extends client_1.Scenario {
@@ -40,6 +51,9 @@ class Scenario extends client_1.Scenario {
         this.workflows = new workflow_1.EnhancedWorkflows(this);
         this.generate = new generate_1.EnhancedGenerate(this);
         this.models = new models_1.EnhancedModels(this);
+        this.uploads = new upload_1.EnhancedUploads(this);
+        this.assets = new asset_1.EnhancedAssets(this);
+        this.jobs = new job_1.EnhancedJobs(this);
     }
 }
 exports.Scenario = Scenario;

package/lib/scenario.js.map

@@ -1 +1 @@
-{"version":3,"file":"scenario.js","sourceRoot":"","sources":["../src/lib/scenario.ts"],"names":[],"mappings":";;;AAAA,yCAAqD;AACrD,4CAA8C;AAC9C,wCAA0C;AAC1C,4CAA+C;AAE/C;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,MAAa,QAAS,SAAQ,iBAAY;IAA1C;;QACW,cAAS,GAAsB,IAAI,4BAAiB,CAAC,IAAI,CAAC,CAAC;QAC3D,aAAQ,GAAqB,IAAI,2BAAgB,CAAC,IAAI,CAAC,CAAC;QACxD,WAAM,GAAmB,IAAI,uBAAc,CAAC,IAAI,CAAC,CAAC;IAC7D,CAAC;CAAA;AAJD,4BAIC"}
+{"version":3,"file":"scenario.js","sourceRoot":"","sources":["../src/lib/scenario.ts"],"names":[],"mappings":";;;AAAA,yCAAqD;AACrD,sCAAyC;AACzC,4CAA8C;AAC9C,kCAAqC;AACrC,wCAA0C;AAC1C,wCAA2C;AAC3C,4CAA+C;AAE/C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAoCG;AACH,MAAa,QAAS,SAAQ,iBAAY;IAA1C;;QACW,cAAS,GAAsB,IAAI,4BAAiB,CAAC,IAAI,CAAC,CAAC;QAC3D,aAAQ,GAAqB,IAAI,2BAAgB,CAAC,IAAI,CAAC,CAAC;QACxD,WAAM,GAAmB,IAAI,uBAAc,CAAC,IAAI,CAAC,CAAC;QAClD,YAAO,GAAoB,IAAI,wBAAe,CAAC,IAAI,CAAC,CAAC;QACrD,WAAM,GAAmB,IAAI,sBAAc,CAAC,IAAI,CAAC,CAAC;QAClD,SAAI,GAAiB,IAAI,kBAAY,CAAC,IAAI,CAAC,CAAC;IACvD,CAAC;CAAA;AAPD,4BAOC"}

package/lib/scenario.mjs

@@ -1,6 +1,9 @@
 import { Scenario as BaseScenario } from "../client.mjs";
+import { EnhancedAssets } from "./asset.mjs";
 import { EnhancedGenerate } from "./generate.mjs";
+import { EnhancedJobs } from "./job.mjs";
 import { EnhancedModels } from "./models.mjs";
+import { EnhancedUploads } from "./upload.mjs";
 import { EnhancedWorkflows } from "./workflow.mjs";
 /**
  * Enhanced Scenario client.
@@ -29,6 +32,14 @@ import { EnhancedWorkflows } from "./workflow.mjs";
  * // Enhanced: generate job.wait()
  * const gen = await client.generate.runModel(modelId, { body });
  * const done = await gen.job.wait();
+ *
+ * // Enhanced: upload.wait()
+ * const res = await client.uploads.retrieve(uploadId);
+ * const imported = await res.upload.wait();
+ *
+ * // Enhanced: asset.download()
+ * const a = await client.assets.retrieve(assetId);
+ * const bytes = await a.asset.download();
  * ```
  */
 export class Scenario extends BaseScenario {
@@ -37,6 +48,9 @@ export class Scenario extends BaseScenario {
         this.workflows = new EnhancedWorkflows(this);
         this.generate = new EnhancedGenerate(this);
         this.models = new EnhancedModels(this);
+        this.uploads = new EnhancedUploads(this);
+        this.assets = new EnhancedAssets(this);
+        this.jobs = new EnhancedJobs(this);
     }
 }
 //# sourceMappingURL=scenario.mjs.map

package/lib/scenario.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"scenario.mjs","sourceRoot":"","sources":["../src/lib/scenario.ts"],"names":[],"mappings":"OAAO,EAAE,QAAQ,IAAI,YAAY,EAAE;OAC5B,EAAE,gBAAgB,EAAE;OACpB,EAAE,cAAc,EAAE;OAClB,EAAE,iBAAiB,EAAE;AAE5B;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,MAAM,OAAO,QAAS,SAAQ,YAAY;IAA1C;;QACW,cAAS,GAAsB,IAAI,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAC3D,aAAQ,GAAqB,IAAI,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACxD,WAAM,GAAmB,IAAI,cAAc,CAAC,IAAI,CAAC,CAAC;IAC7D,CAAC;CAAA"}
+{"version":3,"file":"scenario.mjs","sourceRoot":"","sources":["../src/lib/scenario.ts"],"names":[],"mappings":"OAAO,EAAE,QAAQ,IAAI,YAAY,EAAE;OAC5B,EAAE,cAAc,EAAE;OAClB,EAAE,gBAAgB,EAAE;OACpB,EAAE,YAAY,EAAE;OAChB,EAAE,cAAc,EAAE;OAClB,EAAE,eAAe,EAAE;OACnB,EAAE,iBAAiB,EAAE;AAE5B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAoCG;AACH,MAAM,OAAO,QAAS,SAAQ,YAAY;IAA1C;;QACW,cAAS,GAAsB,IAAI,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAC3D,aAAQ,GAAqB,IAAI,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACxD,WAAM,GAAmB,IAAI,cAAc,CAAC,IAAI,CAAC,CAAC;QAClD,YAAO,GAAoB,IAAI,eAAe,CAAC,IAAI,CAAC,CAAC;QACrD,WAAM,GAAmB,IAAI,cAAc,CAAC,IAAI,CAAC,CAAC;QAClD,SAAI,GAAiB,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;IACvD,CAAC;CAAA"}

package/lib/upload.d.mts

@@ -0,0 +1,143 @@
+import type { Scenario } from "../client.mjs";
+import type { APIPromise } from "../core/api-promise.mjs";
+import type { RequestOptions } from "../internal/request-options.mjs";
+import { Uploads, type UploadCreateParams, type UploadCreateResponse, type UploadRetrieveResponse, type UploadTriggerActionParams, type UploadTriggerActionResponse } from "../resources/uploads.mjs";
+import type { AssetRetrieveResponse } from "../resources/assets/assets.mjs";
+import type { ModelRetrieveResponse } from "../resources/models/models.mjs";
+type UploadKind = NonNullable<UploadCreateParams['kind']>;
+/** The resolved entity returned after a successful upload, narrowed by `kind`. */
+export type UploadResult<K extends UploadKind> = K extends 'model' ? ModelRetrieveResponse : AssetRetrieveResponse;
+/**
+ * Accepted upload inputs.
+ * - `string` — a local file path, read from disk (Node only; throws in browsers).
+ * - `Uint8Array` — raw bytes. `Buffer` is a `Uint8Array` and works too.
+ * - `Blob` / `File` — browser-native, or Node ≥ 18 via `fs.openAsBlob()` /
+ *   constructors.
+ */
+export type UploadFileInput = string | Uint8Array | Blob;
+export interface UploadFileParams<K extends UploadKind> {
+    /**
+     * The file to upload. Accepts a local path (Node), a `Uint8Array` / `Buffer`,
+     * or a `Blob` / `File`.
+     */
+    file: UploadFileInput;
+    /** Original filename (e.g. `"my-model.safetensors"`). */
+    fileName: string;
+    /** MIME type (e.g. `"application/octet-stream"`, `"image/png"`). */
+    contentType: string;
+    /** The kind of upload. Determines the returned entity type. */
+    kind: K;
+    /**
+     * Extra asset options. Ignored when `kind === 'model'` — models don't
+     * produce an asset, so these options are not applicable server-side.
+     */
+    assetOptions?: UploadCreateParams['assetOptions'];
+    /**
+     * Maximum number of parts uploaded to S3 in parallel.
+     * @default 4
+     */
+    partConcurrency?: number;
+    /**
+     * Polling interval for the final "upload imported" check, in ms.
+     * @default 2000
+     */
+    pollIntervalMs?: number;
+    /**
+     * Maximum time to wait for the upload to reach a terminal state, in ms.
+     * Throws if exceeded.
+     * @default 300000
+     */
+    pollTimeoutMs?: number;
+}
+export interface UploadWaitOptions {
+    /** Polling interval in milliseconds. Default: 2000 */
+    intervalMs?: number;
+    /** Maximum wait time in milliseconds. Default: 300000 */
+    timeoutMs?: number;
+}
+/** Upload methods added on top of the original upload fields. */
+export declare class UploadMethods {
+    /** @internal */ readonly _client: Scenario;
+    /**
+     * Poll until the upload has been processed into an entity (asset or model),
+     * or reaches a terminal state (`imported`, `complete`, `failed`). Resolves
+     * with the latest upload data so you can read `entityId`, `status`, or
+     * `errorMessage` without a follow-up retrieve.
+     *
+     * @example
+     * ```ts
+     * const res = await client.uploads.retrieve(uploadId);
+     * const done = await res.upload.wait();
+     * if (done.status === 'failed') throw new Error(done.errorMessage);
+     * console.log(done.entityId);
+     * ```
+     */
+    wait(this: Upload, options?: UploadWaitOptions): Promise<Upload>;
+    /** @internal Create an Upload from raw upload data. */
+    static from(client: Scenario, data: UploadRetrieveResponse['upload']): Upload;
+}
+/** An upload with all original fields plus `.wait()`. */
+export type Upload = UploadRetrieveResponse.Upload & UploadMethods;
+export declare const Upload: typeof UploadMethods;
+/**
+ * @internal Helper type: adds `.wait()` to the `upload` field of a response.
+ * Uses intersection so the original response type is preserved — a `WithUpload<T>` is still assignable to `T`.
+ */
+export type WithUpload<T extends {
+    upload: unknown;
+}> = T & {
+    upload: UploadMethods;
+};
+/**
+ * @internal Wrap `_thenUnwrap` to replace `response.upload` with an enhanced Upload.
+ */
+export declare function enhanceUpload<T extends {
+    upload: {
+        id: string;
+        status: string;
+    };
+}>(client: Scenario, promise: APIPromise<T>): APIPromise<WithUpload<T>>;
+/**
+ * Enhanced Uploads resource.
+ * All original methods (`create`, `retrieve`, `triggerAction`) are inherited,
+ * with each response's `upload` field enriched with a `.wait()` helper.
+ * Adds `uploadFile()` — a one-call wrapper around the 4-step upload flow:
+ * init → PUT parts to S3 → trigger complete → poll until imported.
+ */
+export declare class EnhancedUploads extends Uploads {
+    create(body: UploadCreateParams, options?: RequestOptions): APIPromise<WithUpload<UploadCreateResponse>>;
+    retrieve(uploadID: string, options?: RequestOptions): APIPromise<WithUpload<UploadRetrieveResponse>>;
+    triggerAction(uploadID: string, body: UploadTriggerActionParams, options?: RequestOptions): APIPromise<WithUpload<UploadTriggerActionResponse>>;
+    /**
+     * Upload a file end-to-end. Computes an optimal part size (within AWS's
+     * 5 MiB / 5 GiB / 10,000-part bounds), uploads all parts in parallel,
+     * triggers completion, polls until the server has validated and imported
+     * the file, and returns the resolved entity.
+     *
+     * The returned type narrows on `kind`: `'model'` returns a Model,
+     * everything else returns an Asset.
+     *
+     * @example
+     * ```ts
+     * // From a file path (Node)
+     * const { asset } = await client.uploads.uploadFile({
+     *   file: './photo.jpg',
+     *   fileName: 'photo.jpg',
+     *   contentType: 'image/jpeg',
+     *   kind: 'image',
+     * });
+     * console.log(asset.id, asset.url);
+     *
+     * // Or from bytes / Blob — anything you already have in memory
+     * await client.uploads.uploadFile({
+     *   file: new Blob([pngBytes], { type: 'image/png' }),
+     *   fileName: 'generated.png',
+     *   contentType: 'image/png',
+     *   kind: 'image',
+     * });
+     * ```
+     */
+    uploadFile<K extends UploadKind>(params: UploadFileParams<K>, options?: RequestOptions): Promise<UploadResult<K>>;
+}
+export {};
+//# sourceMappingURL=upload.d.mts.map

package/lib/upload.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"upload.d.mts","sourceRoot":"","sources":["../src/lib/upload.ts"],"names":[],"mappings":"OAAO,KAAK,EAAE,QAAQ,EAAE;OACjB,KAAK,EAAE,UAAU,EAAE;OACnB,KAAK,EAAE,cAAc,EAAE;OACvB,EACL,OAAO,EACP,KAAK,kBAAkB,EACvB,KAAK,oBAAoB,EACzB,KAAK,sBAAsB,EAC3B,KAAK,yBAAyB,EAC9B,KAAK,2BAA2B,EACjC;OACM,KAAK,EAAE,qBAAqB,EAAE;OAC9B,KAAK,EAAE,qBAAqB,EAAE;AAErC,KAAK,UAAU,GAAG,WAAW,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC;AAE1D,kFAAkF;AAClF,MAAM,MAAM,YAAY,CAAC,CAAC,SAAS,UAAU,IAC3C,CAAC,SAAS,OAAO,GAAG,qBAAqB,GAAG,qBAAqB,CAAC;AAEpE;;;;;;GAMG;AACH,MAAM,MAAM,eAAe,GAAG,MAAM,GAAG,UAAU,GAAG,IAAI,CAAC;AAEzD,MAAM,WAAW,gBAAgB,CAAC,CAAC,SAAS,UAAU;IACpD;;;OAGG;IACH,IAAI,EAAE,eAAe,CAAC;IACtB,yDAAyD;IACzD,QAAQ,EAAE,MAAM,CAAC;IACjB,oEAAoE;IACpE,WAAW,EAAE,MAAM,CAAC;IACpB,+DAA+D;IAC/D,IAAI,EAAE,CAAC,CAAC;IACR;;;OAGG;IACH,YAAY,CAAC,EAAE,kBAAkB,CAAC,cAAc,CAAC,CAAC;IAClD;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;;;OAIG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAgBD,MAAM,WAAW,iBAAiB;IAChC,sDAAsD;IACtD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,yDAAyD;IACzD,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,iEAAiE;AACjE,qBAAa,aAAa;IACxB,gBAAgB,CAAC,SAAiB,OAAO,EAAE,QAAQ,CAAC;IAEpD;;;;;;;;;;;;;OAaG;IACG,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,MAAM,CAAC;IAStE,uDAAuD;IACvD,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,sBAAsB,CAAC,QAAQ,CAAC,GAAG,MAAM;CAK9E;AAED,yDAAyD;AACzD,MAAM,MAAM,MAAM,GAAG,sBAAsB,CAAC,MAAM,GAAG,aAAa,CAAC;AACnE,eAAO,MAAM,MAAM,sBAAgB,CAAC;AAEpC;;;GAGG;AACH,MAAM,MAAM,UAAU,CAAC,CAAC,SAAS;IAAE,MAAM,EAAE,OAAO,CAAA;CAAE,IAAI,CAAC,GAAG;IAAE,MAAM,EAAE,aAAa,CAAA;CAAE,CAAC;AAEtF;;GAEG;AACH,wBAAgB,aAAa,CAAC,CAAC,SAAS;IAAE,MAAM,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,EAChF,MAAM,EAAE,QAAQ,EAChB,OAAO,EAAE,UAAU,CAAC,CAAC,CAAC,GACrB,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAK3B;AAED;;;;;;GAMG;AACH,qBAAa,eAAgB,SAAQ,OAAO;IACjC,MAAM,CACb,IAAI,EAAE,kBAAkB,EACxB,OAAO,CAAC,EAAE,cAAc,GACvB,UAAU,CAAC,UAAU,CAAC,oBAAoB,CAAC,CAAC;IAItC,QAAQ,CACf,QAAQ,EAAE,MAAM,EAChB,OAAO,CAAC,EAAE,cAAc,GACvB,UAAU,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC;IAIxC,aAAa,CACpB,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,yBAAyB,EAC/B,OAAO,CAAC,EAAE,cAAc,GACvB,UAAU,CAAC,UAAU,CAAC,2BAA2B,CAAC,CAAC;IAItD;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA4BG;IACG,UAAU,CAAC,CAAC,SAAS,UAAU,EACnC,MAAM,EAAE,gBAAgB,CAAC,CAAC,CAAC,EAC3B,OAAO,CAAC,EAAE,cAAc,GACvB,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;CAuE5B"}

package/lib/upload.d.ts

@@ -0,0 +1,143 @@
+import type { Scenario } from "../client.js";
+import type { APIPromise } from "../core/api-promise.js";
+import type { RequestOptions } from "../internal/request-options.js";
+import { Uploads, type UploadCreateParams, type UploadCreateResponse, type UploadRetrieveResponse, type UploadTriggerActionParams, type UploadTriggerActionResponse } from "../resources/uploads.js";
+import type { AssetRetrieveResponse } from "../resources/assets/assets.js";
+import type { ModelRetrieveResponse } from "../resources/models/models.js";
+type UploadKind = NonNullable<UploadCreateParams['kind']>;
+/** The resolved entity returned after a successful upload, narrowed by `kind`. */
+export type UploadResult<K extends UploadKind> = K extends 'model' ? ModelRetrieveResponse : AssetRetrieveResponse;
+/**
+ * Accepted upload inputs.
+ * - `string` — a local file path, read from disk (Node only; throws in browsers).
+ * - `Uint8Array` — raw bytes. `Buffer` is a `Uint8Array` and works too.
+ * - `Blob` / `File` — browser-native, or Node ≥ 18 via `fs.openAsBlob()` /
+ *   constructors.
+ */
+export type UploadFileInput = string | Uint8Array | Blob;
+export interface UploadFileParams<K extends UploadKind> {
+    /**
+     * The file to upload. Accepts a local path (Node), a `Uint8Array` / `Buffer`,
+     * or a `Blob` / `File`.
+     */
+    file: UploadFileInput;
+    /** Original filename (e.g. `"my-model.safetensors"`). */
+    fileName: string;
+    /** MIME type (e.g. `"application/octet-stream"`, `"image/png"`). */
+    contentType: string;
+    /** The kind of upload. Determines the returned entity type. */
+    kind: K;
+    /**
+     * Extra asset options. Ignored when `kind === 'model'` — models don't
+     * produce an asset, so these options are not applicable server-side.
+     */
+    assetOptions?: UploadCreateParams['assetOptions'];
+    /**
+     * Maximum number of parts uploaded to S3 in parallel.
+     * @default 4
+     */
+    partConcurrency?: number;
+    /**
+     * Polling interval for the final "upload imported" check, in ms.
+     * @default 2000
+     */
+    pollIntervalMs?: number;
+    /**
+     * Maximum time to wait for the upload to reach a terminal state, in ms.
+     * Throws if exceeded.
+     * @default 300000
+     */
+    pollTimeoutMs?: number;
+}
+export interface UploadWaitOptions {
+    /** Polling interval in milliseconds. Default: 2000 */
+    intervalMs?: number;
+    /** Maximum wait time in milliseconds. Default: 300000 */
+    timeoutMs?: number;
+}
+/** Upload methods added on top of the original upload fields. */
+export declare class UploadMethods {
+    /** @internal */ readonly _client: Scenario;
+    /**
+     * Poll until the upload has been processed into an entity (asset or model),
+     * or reaches a terminal state (`imported`, `complete`, `failed`). Resolves
+     * with the latest upload data so you can read `entityId`, `status`, or
+     * `errorMessage` without a follow-up retrieve.
+     *
+     * @example
+     * ```ts
+     * const res = await client.uploads.retrieve(uploadId);
+     * const done = await res.upload.wait();
+     * if (done.status === 'failed') throw new Error(done.errorMessage);
+     * console.log(done.entityId);
+     * ```
+     */
+    wait(this: Upload, options?: UploadWaitOptions): Promise<Upload>;
+    /** @internal Create an Upload from raw upload data. */
+    static from(client: Scenario, data: UploadRetrieveResponse['upload']): Upload;
+}
+/** An upload with all original fields plus `.wait()`. */
+export type Upload = UploadRetrieveResponse.Upload & UploadMethods;
+export declare const Upload: typeof UploadMethods;
+/**
+ * @internal Helper type: adds `.wait()` to the `upload` field of a response.
+ * Uses intersection so the original response type is preserved — a `WithUpload<T>` is still assignable to `T`.
+ */
+export type WithUpload<T extends {
+    upload: unknown;
+}> = T & {
+    upload: UploadMethods;
+};
+/**
+ * @internal Wrap `_thenUnwrap` to replace `response.upload` with an enhanced Upload.
+ */
+export declare function enhanceUpload<T extends {
+    upload: {
+        id: string;
+        status: string;
+    };
+}>(client: Scenario, promise: APIPromise<T>): APIPromise<WithUpload<T>>;
+/**
+ * Enhanced Uploads resource.
+ * All original methods (`create`, `retrieve`, `triggerAction`) are inherited,
+ * with each response's `upload` field enriched with a `.wait()` helper.
+ * Adds `uploadFile()` — a one-call wrapper around the 4-step upload flow:
+ * init → PUT parts to S3 → trigger complete → poll until imported.
+ */
+export declare class EnhancedUploads extends Uploads {
+    create(body: UploadCreateParams, options?: RequestOptions): APIPromise<WithUpload<UploadCreateResponse>>;
+    retrieve(uploadID: string, options?: RequestOptions): APIPromise<WithUpload<UploadRetrieveResponse>>;
+    triggerAction(uploadID: string, body: UploadTriggerActionParams, options?: RequestOptions): APIPromise<WithUpload<UploadTriggerActionResponse>>;
+    /**
+     * Upload a file end-to-end. Computes an optimal part size (within AWS's
+     * 5 MiB / 5 GiB / 10,000-part bounds), uploads all parts in parallel,
+     * triggers completion, polls until the server has validated and imported
+     * the file, and returns the resolved entity.
+     *
+     * The returned type narrows on `kind`: `'model'` returns a Model,
+     * everything else returns an Asset.
+     *
+     * @example
+     * ```ts
+     * // From a file path (Node)
+     * const { asset } = await client.uploads.uploadFile({
+     *   file: './photo.jpg',
+     *   fileName: 'photo.jpg',
+     *   contentType: 'image/jpeg',
+     *   kind: 'image',
+     * });
+     * console.log(asset.id, asset.url);
+     *
+     * // Or from bytes / Blob — anything you already have in memory
+     * await client.uploads.uploadFile({
+     *   file: new Blob([pngBytes], { type: 'image/png' }),
+     *   fileName: 'generated.png',
+     *   contentType: 'image/png',
+     *   kind: 'image',
+     * });
+     * ```
+     */
+    uploadFile<K extends UploadKind>(params: UploadFileParams<K>, options?: RequestOptions): Promise<UploadResult<K>>;
+}
+export {};
+//# sourceMappingURL=upload.d.ts.map

package/lib/upload.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"upload.d.ts","sourceRoot":"","sources":["../src/lib/upload.ts"],"names":[],"mappings":"OAAO,KAAK,EAAE,QAAQ,EAAE;OACjB,KAAK,EAAE,UAAU,EAAE;OACnB,KAAK,EAAE,cAAc,EAAE;OACvB,EACL,OAAO,EACP,KAAK,kBAAkB,EACvB,KAAK,oBAAoB,EACzB,KAAK,sBAAsB,EAC3B,KAAK,yBAAyB,EAC9B,KAAK,2BAA2B,EACjC;OACM,KAAK,EAAE,qBAAqB,EAAE;OAC9B,KAAK,EAAE,qBAAqB,EAAE;AAErC,KAAK,UAAU,GAAG,WAAW,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC;AAE1D,kFAAkF;AAClF,MAAM,MAAM,YAAY,CAAC,CAAC,SAAS,UAAU,IAC3C,CAAC,SAAS,OAAO,GAAG,qBAAqB,GAAG,qBAAqB,CAAC;AAEpE;;;;;;GAMG;AACH,MAAM,MAAM,eAAe,GAAG,MAAM,GAAG,UAAU,GAAG,IAAI,CAAC;AAEzD,MAAM,WAAW,gBAAgB,CAAC,CAAC,SAAS,UAAU;IACpD;;;OAGG;IACH,IAAI,EAAE,eAAe,CAAC;IACtB,yDAAyD;IACzD,QAAQ,EAAE,MAAM,CAAC;IACjB,oEAAoE;IACpE,WAAW,EAAE,MAAM,CAAC;IACpB,+DAA+D;IAC/D,IAAI,EAAE,CAAC,CAAC;IACR;;;OAGG;IACH,YAAY,CAAC,EAAE,kBAAkB,CAAC,cAAc,CAAC,CAAC;IAClD;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;;;OAIG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAgBD,MAAM,WAAW,iBAAiB;IAChC,sDAAsD;IACtD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,yDAAyD;IACzD,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,iEAAiE;AACjE,qBAAa,aAAa;IACxB,gBAAgB,CAAC,SAAiB,OAAO,EAAE,QAAQ,CAAC;IAEpD;;;;;;;;;;;;;OAaG;IACG,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,MAAM,CAAC;IAStE,uDAAuD;IACvD,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,sBAAsB,CAAC,QAAQ,CAAC,GAAG,MAAM;CAK9E;AAED,yDAAyD;AACzD,MAAM,MAAM,MAAM,GAAG,sBAAsB,CAAC,MAAM,GAAG,aAAa,CAAC;AACnE,eAAO,MAAM,MAAM,sBAAgB,CAAC;AAEpC;;;GAGG;AACH,MAAM,MAAM,UAAU,CAAC,CAAC,SAAS;IAAE,MAAM,EAAE,OAAO,CAAA;CAAE,IAAI,CAAC,GAAG;IAAE,MAAM,EAAE,aAAa,CAAA;CAAE,CAAC;AAEtF;;GAEG;AACH,wBAAgB,aAAa,CAAC,CAAC,SAAS;IAAE,MAAM,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,EAChF,MAAM,EAAE,QAAQ,EAChB,OAAO,EAAE,UAAU,CAAC,CAAC,CAAC,GACrB,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAK3B;AAED;;;;;;GAMG;AACH,qBAAa,eAAgB,SAAQ,OAAO;IACjC,MAAM,CACb,IAAI,EAAE,kBAAkB,EACxB,OAAO,CAAC,EAAE,cAAc,GACvB,UAAU,CAAC,UAAU,CAAC,oBAAoB,CAAC,CAAC;IAItC,QAAQ,CACf,QAAQ,EAAE,MAAM,EAChB,OAAO,CAAC,EAAE,cAAc,GACvB,UAAU,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC;IAIxC,aAAa,CACpB,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,yBAAyB,EAC/B,OAAO,CAAC,EAAE,cAAc,GACvB,UAAU,CAAC,UAAU,CAAC,2BAA2B,CAAC,CAAC;IAItD;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA4BG;IACG,UAAU,CAAC,CAAC,SAAS,UAAU,EACnC,MAAM,EAAE,gBAAgB,CAAC,CAAC,CAAC,EAC3B,OAAO,CAAC,EAAE,cAAc,GACvB,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;CAuE5B"}