@sogni-ai/sogni-client 4.0.0-alpha.5 → 4.0.0-alpha.51

package/dist/Projects/index.js

@@ -20,9 +20,39 @@ const getUUID_1 = __importDefault(require("../lib/getUUID"));
 const Cache_1 = __importDefault(require("../lib/Cache"));
 const Job_1 = require("./Job");
 const utils_1 = require("./utils");
+const validation_1 = require("../lib/validation");
+const ModelTiersRaw_1 = require("./types/ModelTiersRaw");
+const ModelOptions_1 = require("./types/ModelOptions");
 const sizePresetCache = new Cache_1.default(10 * 60 * 1000);
 const GARBAGE_COLLECT_TIMEOUT = 30000;
 const MODELS_REFRESH_INTERVAL = 1000 * 60 * 60 * 24; // 24 hours
+/**
+ * Detect content type from a file object.
+ * For File objects in browser, uses the type property.
+ * Returns undefined if content type cannot be detected.
+ */
+function getFileContentType(file) {
+    if (file instanceof Blob && 'type' in file && file.type) {
+        return file.type;
+    }
+    return undefined;
+}
+/**
+ * Convert file to a format compatible with fetch body.
+ * Converts Node.js Buffer to Blob for cross-platform compatibility.
+ */
+function toFetchBody(file) {
+    // Node.js Buffer is not supported in browsers, so we can skip this conversion
+    if (typeof Buffer === 'undefined') {
+        return file;
+    }
+    if (Buffer.isBuffer(file)) {
+        // Copy Buffer data to a new ArrayBuffer to ensure type compatibility
+        const arrayBuffer = file.buffer.slice(file.byteOffset, file.byteOffset + file.byteLength);
+        return new Blob([arrayBuffer]);
+    }
+    return file;
+}
 function mapErrorCodes(code) {
     switch (code) {
         case 'serverRestarting':
@@ -43,6 +73,20 @@ class ProjectsApi extends ApiGroup_1.default {
     get availableModels() {
         return this._availableModels;
     }
+    /**
+     * Check if a model produces video output using the cached models list.
+     * Uses the `media` property from the models API when available,
+     * falls back to model ID prefix check if models aren't loaded yet.
+     */
+    isVideoModelId(modelId) {
+        var _a;
+        const model = (_a = this._supportedModels.data) === null || _a === void 0 ? void 0 : _a.find((m) => m.id === modelId);
+        if (model) {
+            return model.media === 'video';
+        }
+        // Fallback to prefix check if models not loaded
+        return (0, utils_1.isVideoModel)(modelId);
+    }
     constructor(config) {
         super(config);
         this._availableModels = [];
@@ -51,19 +95,38 @@ class ProjectsApi extends ApiGroup_1.default {
             data: null,
             updatedAt: new Date(0)
         };
+        this._modelTiers = {
+            data: {},
+            updatedAt: new Date(0)
+        };
         // Listen to server events and emit them as project and job events
         this.client.socket.on('changeNetwork', this.handleChangeNetwork.bind(this));
         this.client.socket.on('swarmModels', this.handleSwarmModels.bind(this));
         this.client.socket.on('jobState', this.handleJobState.bind(this));
         this.client.socket.on('jobProgress', this.handleJobProgress.bind(this));
+        this.client.socket.on('jobETA', this.handleJobETA.bind(this));
         this.client.socket.on('jobError', this.handleJobError.bind(this));
-        this.client.socket.on('jobResult', this.handleJobResult.bind(this));
+        this.client.socket.on('jobResult', (data) => {
+            this.handleJobResult(data).catch((err) => {
+                this.client.logger.error('Error in handleJobResult:', err);
+            });
+        });
         // Listen to the server disconnect event
         this.client.on('disconnected', this.handleServerDisconnected.bind(this));
         // Listen to project and job events and update project and job instances
         this.on('project', this.handleProjectEvent.bind(this));
         this.on('job', this.handleJobEvent.bind(this));
     }
+    /**
+     * Retrieves a list of projects created and tracked by this SogniClient instance.
+     *
+     * Note: When a project is finished, it will be removed from this list after 30 seconds
+     *
+     * @return {Array} A copy of the array containing the tracked projects.
+     */
+    get trackedProjects() {
+        return this.projects.slice(0);
+    }
     handleChangeNetwork() {
         this._availableModels = [];
         this.emit('availableModels', this._availableModels);
@@ -82,11 +145,12 @@ class ProjectsApi extends ApiGroup_1.default {
                 return acc;
             }, {});
             this._availableModels = Object.entries(data).map(([id, workerCount]) => {
-                var _a;
+                var _a, _b;
                 return ({
                     id,
                     name: ((_a = modelIndex[id]) === null || _a === void 0 ? void 0 : _a.name) || id.replace(/-/g, ' '),
-                    workerCount
+                    workerCount,
+                    media: ((_b = modelIndex[id]) === null || _b === void 0 ? void 0 : _b.media) || 'image'
                 });
             });
             this.emit('availableModels', this._availableModels);
@@ -154,20 +218,61 @@ class ProjectsApi extends ApiGroup_1.default {
             }
         });
     }
+    handleJobETA(data) {
+        return __awaiter(this, void 0, void 0, function* () {
+            this.emit('job', {
+                type: 'jobETA',
+                projectId: data.jobID,
+                jobId: data.imgID || '',
+                etaSeconds: data.etaSeconds
+            });
+        });
+    }
     handleJobResult(data) {
         return __awaiter(this, void 0, void 0, function* () {
             const project = this.projects.find((p) => p.id === data.jobID);
             const passNSFWCheck = !data.triggeredNSFWFilter || !project || project.params.disableNSFWFilter;
-            let downloadUrl = null;
-            // If NSFW filter is triggered, image will be only available for download if user explicitly
-            // disabled the filter for this project
-            if (passNSFWCheck && !data.userCanceled) {
-                downloadUrl = yield this.downloadUrl({
-                    jobId: data.jobID,
-                    imageId: data.imgID,
-                    type: 'complete'
-                });
+            let downloadUrl = data.resultUrl || null; // Use resultUrl from event if provided
+            // If no resultUrl provided and NSFW check passes, generate download URL
+            if (!downloadUrl && passNSFWCheck && !data.userCanceled) {
+                // Use media endpoint for video models, image endpoint for image models
+                const isVideo = project && this.isVideoModelId(project.params.modelId);
+                try {
+                    if (isVideo) {
+                        downloadUrl = yield this.mediaDownloadUrl({
+                            jobId: data.jobID,
+                            id: data.imgID,
+                            type: 'complete'
+                        });
+                    }
+                    else {
+                        downloadUrl = yield this.downloadUrl({
+                            jobId: data.jobID,
+                            imageId: data.imgID,
+                            type: 'complete'
+                        });
+                    }
+                }
+                catch (error) {
+                    this.client.logger.error('Failed to generate download URL for job result');
+                    this.client.logger.error(error);
+                }
             }
+            // Update the job directly with the result URL to prevent duplicate API calls
+            if (project) {
+                const job = project.job(data.imgID);
+                if (job) {
+                    job._update({
+                        status: data.userCanceled ? 'canceled' : 'completed',
+                        step: data.performedStepCount,
+                        seed: Number(data.lastSeed),
+                        resultUrl: downloadUrl,
+                        isNSFW: data.triggeredNSFWFilter,
+                        userCanceled: data.userCanceled
+                    });
+                }
+            }
+            // Emit job completion event with the generated download URL
             this.emit('job', {
                 type: 'completed',
                 projectId: data.jobID,
@@ -238,7 +343,11 @@ class ProjectsApi extends ApiGroup_1.default {
         if (project.finished) {
             // Sync project data with the server and remove it from the list after some time
             project._syncToServer().catch((e) => {
-                this.client.logger.error(e);
+                // 404 errors are expected when project is still initializing
+                // Only log non-404 errors to avoid confusing users
+                if (e.status !== 404) {
+                    this.client.logger.error(e);
+                }
             });
             setTimeout(() => {
                 this.projects = this.projects.filter((p) => !p.finished);
@@ -246,6 +355,7 @@ class ProjectsApi extends ApiGroup_1.default {
         }
     }
     handleJobEvent(event) {
+        var _a, _b, _c;
         let project = this.projects.find((p) => p.id === event.projectId);
         if (!project) {
             return;
@@ -257,7 +367,7 @@ class ProjectsApi extends ApiGroup_1.default {
                 projectId: event.projectId,
                 status: 'pending',
                 step: 0,
-                stepCount: project.params.steps
+                stepCount: (_a = project.params.steps) !== null && _a !== void 0 ? _a : 0
             });
         }
         switch (event.type) {
@@ -284,7 +394,7 @@ class ProjectsApi extends ApiGroup_1.default {
             case 'progress':
                 job._update({
                     status: 'processing',
-                    // Jus in case event comes out of order
+                    // Just in case event comes out of order
                     step: Math.max(event.step, job.step),
                     stepCount: event.stepCount
                 });
@@ -292,6 +402,23 @@ class ProjectsApi extends ApiGroup_1.default {
                     project._update({ status: 'processing' });
                 }
                 break;
+            case 'jobETA': {
+                // ETA updates keep the project alive (refreshes lastUpdated) and store the ETA value.
+                // This is critical for long-running jobs like video generation that can take several
+                // minutes and may not send frequent progress updates.
+                // We always call _keepAlive() to ensure lastUpdated is refreshed, preventing premature timeouts.
+                project._keepAlive();
+                const newEta = new Date(Date.now() + event.etaSeconds * 1000);
+                if (((_b = job.eta) === null || _b === void 0 ? void 0 : _b.getTime()) !== (newEta === null || newEta === void 0 ? void 0 : newEta.getTime())) {
+                    job._update({ eta: newEta });
+                    const maxEta = project.jobs.reduce((max, j) => { var _a; return Math.max(max, ((_a = j.eta) === null || _a === void 0 ? void 0 : _a.getTime()) || 0); }, 0);
+                    const projectETA = maxEta ? new Date(maxEta) : undefined;
+                    if (((_c = project.eta) === null || _c === void 0 ? void 0 : _c.getTime()) !== (projectETA === null || projectETA === void 0 ? void 0 : projectETA.getTime())) {
+                        project._update({ eta: projectETA });
+                    }
+                }
+                break;
+            }
             case 'preview':
                 job._update({ previewUrl: event.url });
                 break;
@@ -308,6 +435,17 @@ class ProjectsApi extends ApiGroup_1.default {
             }
             case 'error':
                 job._update({ status: 'failed', error: event.error });
+                // Check if project should also fail when a job fails
+                // For video jobs (single image) or when all jobs have failed, propagate to project
+                const allJobsStarted = project.jobs.length >= project.params.numberOfMedia;
+                const allJobsFailed = allJobsStarted && project.jobs.every((j) => j.status === 'failed');
+                const isSingleJobProject = project.params.numberOfMedia === 1;
+                if (isSingleJobProject || allJobsFailed) {
+                    project._update({
+                        status: 'failed',
+                        error: event.error
+                    });
+                }
                 break;
         }
     }
@@ -328,41 +466,69 @@ class ProjectsApi extends ApiGroup_1.default {
             return Promise.resolve(this._availableModels);
         }
         return new Promise((resolve, reject) => {
+            let settled = false;
             const timeoutId = setTimeout(() => {
-                reject(new Error('Timeout waiting for models'));
+                if (!settled) {
+                    settled = true;
+                    this.off('availableModels', handler);
+                    reject(new Error('Timeout waiting for models'));
+                }
             }, timeout);
-            this.once('availableModels', (models) => {
-                clearTimeout(timeoutId);
-                if (models.length) {
+            const handler = (models) => {
+                // Only resolve when we get a non-empty models list
+                // Empty arrays may be emitted during disconnects/reconnects
+                if (models.length && !settled) {
+                    settled = true;
+                    clearTimeout(timeoutId);
+                    this.off('availableModels', handler);
                     resolve(models);
                 }
-                else {
-                    reject(new Error('No models available'));
-                }
-            });
+            };
+            this.on('availableModels', handler);
         });
     }
     /**
      * Send new project request to the network. Returns project instance which can be used to track
-     * progress and get resulting images.
+     * progress and get resulting images or videos.
      * @param data
      */
     create(data) {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b;
             const project = new Project_1.default(Object.assign({}, data), { api: this, logger: this.client.logger });
+            const modelOptions = yield this.getModelOptions(data.modelId);
+            const request = (0, createJobRequestMessage_1.default)(project.id, data, modelOptions);
+            switch (data.type) {
+                case 'image':
+                    yield this._processImageAssets(project, data);
+                    break;
+                case 'video':
+                    yield this._processVideoAssets(project, data);
+                    break;
+            }
+            yield this.client.socket.send('jobRequest', request);
+            this.projects.push(project);
+            return project;
+        });
+    }
+    _processImageAssets(project, data) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b;
+            //Guide image
             if (data.startingImage && data.startingImage !== true) {
                 yield this.uploadGuideImage(project.id, data.startingImage);
             }
+            // ControlNet image
             if (((_a = data.controlNet) === null || _a === void 0 ? void 0 : _a.image) && data.controlNet.image !== true) {
                 yield this.uploadCNImage(project.id, data.controlNet.image);
             }
+            // Context images (Flux.2 Dev supports up to 6; Qwen Image Edit Plus supports up to 3; Flux Kontext supports up to 2)
             if ((_b = data.contextImages) === null || _b === void 0 ? void 0 : _b.length) {
-                if (data.contextImages.length > 2) {
+                const maxContextImages = (0, validation_1.getMaxContextImages)(data.modelId);
+                if (data.contextImages.length > maxContextImages) {
                     throw new ApiClient_1.ApiError(500, {
                         status: 'error',
                         errorCode: 0,
-                        message: `Up to 2 context images are supported`
+                        message: `Up to ${maxContextImages} context images are supported for this model`
                     });
                 }
                 yield Promise.all(data.contextImages.map((image, index) => {
@@ -371,10 +537,22 @@ class ProjectsApi extends ApiGroup_1.default {
                     }
                 }));
             }
-            const request = (0, createJobRequestMessage_1.default)(project.id, data);
-            yield this.client.socket.send('jobRequest', request);
-            this.projects.push(project);
-            return project;
+        });
+    }
+    _processVideoAssets(project, data) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if ((data === null || data === void 0 ? void 0 : data.referenceImage) && data.referenceImage !== true) {
+                yield this.uploadReferenceImage(project.id, data.referenceImage);
+            }
+            if ((data === null || data === void 0 ? void 0 : data.referenceImageEnd) && data.referenceImageEnd !== true) {
+                yield this.uploadReferenceImageEnd(project.id, data.referenceImageEnd);
+            }
+            if ((data === null || data === void 0 ? void 0 : data.referenceAudio) && data.referenceAudio !== true) {
+                yield this.uploadReferenceAudio(project.id, data.referenceAudio);
+            }
+            if ((data === null || data === void 0 ? void 0 : data.referenceVideo) && data.referenceVideo !== true) {
+                yield this.uploadReferenceVideo(project.id, data.referenceVideo);
+            }
         });
     }
     /**
@@ -425,13 +603,13 @@ class ProjectsApi extends ApiGroup_1.default {
         return __awaiter(this, void 0, void 0, function* () {
             const imageId = (0, getUUID_1.default)();
             const presignedUrl = yield this.uploadUrl({
-                imageId: imageId,
+                imageId,
                 jobId: projectId,
                 type: 'startingImage'
             });
             const res = yield fetch(presignedUrl, {
                 method: 'PUT',
-                body: file
+                body: toFetchBody(file)
             });
             if (!res.ok) {
                 throw new ApiClient_1.ApiError(res.status, {
@@ -447,13 +625,13 @@ class ProjectsApi extends ApiGroup_1.default {
         return __awaiter(this, void 0, void 0, function* () {
             const imageId = (0, getUUID_1.default)();
             const presignedUrl = yield this.uploadUrl({
-                imageId: imageId,
+                imageId,
                 jobId: projectId,
                 type: 'cnImage'
             });
             const res = yield fetch(presignedUrl, {
                 method: 'PUT',
-                body: file
+                body: toFetchBody(file)
             });
             if (!res.ok) {
                 throw new ApiClient_1.ApiError(res.status, {
@@ -474,9 +652,10 @@ class ProjectsApi extends ApiGroup_1.default {
                 jobId: projectId,
                 type: `contextImage${imageIndex}`
             });
+            const body = toFetchBody(file);
             const res = yield fetch(presignedUrl, {
                 method: 'PUT',
-                body: file
+                body
             });
             if (!res.ok) {
                 throw new ApiClient_1.ApiError(res.status, {
@@ -488,14 +667,133 @@ class ProjectsApi extends ApiGroup_1.default {
             return imageId;
         });
     }
+    // ============================================
+    // VIDEO WORKFLOW UPLOADS (WAN 2.2)
+    // ============================================
     /**
-     * Estimate project cost
+     * Upload reference image for WAN video workflows
+     * @internal
+     */
+    uploadReferenceImage(projectId, file) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const imageId = (0, getUUID_1.default)();
+            const presignedUrl = yield this.uploadUrl({
+                imageId,
+                jobId: projectId,
+                type: 'referenceImage'
+            });
+            const res = yield fetch(presignedUrl, {
+                method: 'PUT',
+                body: toFetchBody(file)
+            });
+            if (!res.ok) {
+                throw new ApiClient_1.ApiError(res.status, {
+                    status: 'error',
+                    errorCode: 0,
+                    message: 'Failed to upload reference image'
+                });
+            }
+            return imageId;
+        });
+    }
+    /**
+     * Upload reference image end for i2v interpolation
+     * @internal
+     */
+    uploadReferenceImageEnd(projectId, file) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const imageId = (0, getUUID_1.default)();
+            const presignedUrl = yield this.uploadUrl({
+                imageId,
+                jobId: projectId,
+                type: 'referenceImageEnd'
+            });
+            const res = yield fetch(presignedUrl, {
+                method: 'PUT',
+                body: toFetchBody(file)
+            });
+            if (!res.ok) {
+                throw new ApiClient_1.ApiError(res.status, {
+                    status: 'error',
+                    errorCode: 0,
+                    message: 'Failed to upload reference image end'
+                });
+            }
+            return imageId;
+        });
+    }
+    /**
+     * Upload reference audio for s2v workflows
+     * Supported formats: mp3, m4a, wav
+     * @internal
+     */
+    uploadReferenceAudio(projectId, file) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const contentType = getFileContentType(file);
+            const presignedUrl = yield this.mediaUploadUrl({
+                jobId: projectId,
+                type: 'referenceAudio'
+            });
+            const headers = {};
+            if (contentType) {
+                headers['Content-Type'] = contentType;
+            }
+            const res = yield fetch(presignedUrl, {
+                method: 'PUT',
+                body: toFetchBody(file),
+                headers
+            });
+            if (!res.ok) {
+                throw new ApiClient_1.ApiError(res.status, {
+                    status: 'error',
+                    errorCode: 0,
+                    message: 'Failed to upload reference audio'
+                });
+            }
+        });
+    }
+    /**
+     * Upload reference video for animate workflows
+     * Supported formats: mp4, mov
+     * @internal
+     */
+    uploadReferenceVideo(projectId, file) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const contentType = getFileContentType(file);
+            const presignedUrl = yield this.mediaUploadUrl({
+                jobId: projectId,
+                type: 'referenceVideo'
+            });
+            const headers = {};
+            if (contentType) {
+                headers['Content-Type'] = contentType;
+            }
+            const res = yield fetch(presignedUrl, {
+                method: 'PUT',
+                body: toFetchBody(file),
+                headers
+            });
+            if (!res.ok) {
+                throw new ApiClient_1.ApiError(res.status, {
+                    status: 'error',
+                    errorCode: 0,
+                    message: 'Failed to upload reference video'
+                });
+            }
+        });
+    }
+    // ============================================
+    // COST ESTIMATION
+    // ============================================
+    /**
+     * Estimate image project cost
      */
     estimateCost(_a) {
-        return __awaiter(this, arguments, void 0, function* ({ network, tokenType, model, imageCount, stepCount, previewCount, cnEnabled, startingImageStrength, width, height, sizePreset, guidance, scheduler, contextImages }) {
+        return __awaiter(this, arguments, void 0, function* ({ network, tokenType, model, imageCount, stepCount, previewCount, cnEnabled, startingImageStrength, width, height, sizePreset, guidance, sampler, contextImages }) {
             let apiVersion = 2;
+            const modelOptions = yield this.getModelOptions(model);
             const pathParams = [
-                tokenType || 'sogni',
+                tokenType || 'spark',
                 network,
                 model,
                 imageCount,
@@ -518,21 +816,28 @@ class ProjectsApi extends ApiGroup_1.default {
             else {
                 pathParams.push(0, 0);
             }
-            if (scheduler) {
+            if (sampler) {
                 apiVersion = 3;
                 pathParams.push(guidance || 0);
-                pathParams.push(scheduler || '');
+                pathParams.push((0, validation_1.validateSampler)(sampler, modelOptions));
                 pathParams.push(contextImages || 0);
             }
             const r = yield this.client.socket.get(`/api/v${apiVersion}/job/estimate/${pathParams.join('/')}`);
             return {
                 token: r.quote.project.costInToken,
-                usd: r.quote.project.costInUSD
+                usd: r.quote.project.costInUSD,
+                spark: r.quote.project.costInSpark,
+                sogni: r.quote.project.costInSogni
             };
         });
     }
+    /**
+     * Estimate image enhancement cost
+     * @param strength
+     * @param tokenType
+     */
     estimateEnhancementCost(strength_1) {
-        return __awaiter(this, arguments, void 0, function* (strength, tokenType = 'sogni') {
+        return __awaiter(this, arguments, void 0, function* (strength, tokenType = 'spark') {
             return this.estimateCost({
                 network: Job_1.enhancementDefaults.network,
                 tokenType,
@@ -545,10 +850,51 @@ class ProjectsApi extends ApiGroup_1.default {
             });
         });
     }
+    /**
+     * Estimates the cost of generating a video based on the provided parameters.
+     *
+     * @param {VideoEstimateRequest} params - The parameters required for video cost estimation. This includes:
+     *   - tokenType: The type of token to be used for generation.
+     *   - model: The model to be used for video generation.
+     *   - width: The width of the video in pixels.
+     *   - height: The height of the video in pixels.
+     *   - frames: The total number of frames in the video.
+     *   - fps: The frames per second for the video.
+     *   - steps: Number of steps.
+     * @return {Promise<Object>} Returns an object containing the estimated costs for the video in different units:
+     *   - token: Cost in tokens.
+     *   - usd: Cost in USD.
+     *   - spark: Cost in Spark.
+     *   - sogni: Cost in Sogni.
+     */
+    estimateVideoCost(params) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const pathParams = [
+                params.tokenType,
+                params.model,
+                params.width,
+                params.height,
+                params.frames ? params.frames : params.duration * 16 + 1,
+                params.fps,
+                params.steps,
+                params.numberOfMedia
+            ];
+            const path = pathParams.map((p) => encodeURIComponent(p)).join('/');
+            const r = yield this.client.socket.get(`/api/v1/job-video/estimate/${path}`);
+            return {
+                token: r.quote.project.costInToken,
+                usd: r.quote.project.costInUSD,
+                spark: r.quote.project.costInSpark,
+                sogni: r.quote.project.costInSogni
+            };
+        });
+    }
+    // ============================================
+    // URL HELPERS
+    // ============================================
     /**
      * Get upload URL for image
      * @internal
-     * @param params
      */
     uploadUrl(params) {
         return __awaiter(this, void 0, void 0, function* () {
@@ -559,14 +905,44 @@ class ProjectsApi extends ApiGroup_1.default {
     /**
      * Get download URL for image
      * @internal
-     * @param params
      */
     downloadUrl(params) {
         return __awaiter(this, void 0, void 0, function* () {
+            var _a;
             const r = yield this.client.rest.get(`/v1/image/downloadUrl`, params);
+            if (!((_a = r === null || r === void 0 ? void 0 : r.data) === null || _a === void 0 ? void 0 : _a.downloadUrl)) {
+                throw new Error(`API returned no downloadUrl: ${JSON.stringify(r)}`);
+            }
+            return r.data.downloadUrl;
+        });
+    }
+    /**
+     * Get upload URL for media (video/audio)
+     * @internal
+     */
+    mediaUploadUrl(params) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const r = yield this.client.rest.get(`/v1/media/uploadUrl`, params);
+            return r.data.uploadUrl;
+        });
+    }
+    /**
+     * Get download URL for media (video/audio)
+     * @internal
+     */
+    mediaDownloadUrl(params) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            const r = yield this.client.rest.get(`/v1/media/downloadUrl`, params);
+            if (!((_a = r === null || r === void 0 ? void 0 : r.data) === null || _a === void 0 ? void 0 : _a.downloadUrl)) {
+                throw new Error(`API returned no downloadUrl: ${JSON.stringify(r)}`);
+            }
             return r.data.downloadUrl;
         });
     }
+    // ============================================
+    // MODEL/PRESET HELPERS
+    // ============================================
     getSupportedModels() {
         return __awaiter(this, arguments, void 0, function* (forceRefresh = false) {
             if (this._supportedModels.data &&
@@ -579,12 +955,24 @@ class ProjectsApi extends ApiGroup_1.default {
             return models;
         });
     }
+    _getModelTiers() {
+        return __awaiter(this, arguments, void 0, function* (forceRefresh = false) {
+            if (this._modelTiers.data &&
+                !forceRefresh &&
+                Date.now() - this._modelTiers.updatedAt.getTime() < MODELS_REFRESH_INTERVAL) {
+                return this._modelTiers.data;
+            }
+            const tiers = yield this.client.socket.get(`/api/v2/models/tiers`);
+            this._modelTiers = { data: tiers, updatedAt: new Date() };
+            return tiers;
+        });
+    }
     /**
      * Get supported size presets for the model and network. Size presets are cached for 10 minutes.
      *
      * @example
      * ```ts
-     * const presets = await client.projects.getSizePresets('fast', 'flux1-schnell-fp8');
+     * const presets = await sogni.projects.getSizePresets('fast', 'flux1-schnell-fp8');
      * console.log(presets);
      * ```
      *
@@ -612,6 +1000,50 @@ class ProjectsApi extends ApiGroup_1.default {
             return data;
         });
     }
+    /**
+     * Retrieves the video asset configuration for a given video model identifier.
+     * Validates whether the provided model ID corresponds to a video model. If it does,
+     * returns the appropriate video asset configuration based on the workflow type.
+     *
+     * @example Returned object for a model that implements image to video workflow:
+     * ```json
+     * {
+     *   "workflowType": "i2v",
+     *   "assets": {
+     *     "referenceImage": "required",
+     *     "referenceImageEnd": "optional",
+     *     "referenceAudio": "forbidden",
+     *     "referenceVideo": "forbidden"
+     *   }
+     * }
+     * ```
+     *
+     * @param {string} modelId - The identifier of the video model to retrieve the configuration for.
+     * @return {Object} The video asset configuration object where key is asset field and value is
+     * either `required`, `forbidden` or `optional`. Returns `null` if no rules defined for the model.
+     * @throws {ApiError} Throws an error if the provided model ID is not a video model.
+     */
+    getVideoAssetConfig(modelId) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!this.isVideoModelId(modelId)) {
+                throw new ApiClient_1.ApiError(400, {
+                    status: 'error',
+                    errorCode: 0,
+                    message: `Model ${modelId} is not a video model`
+                });
+            }
+            const workflow = (0, utils_1.getVideoWorkflowType)(modelId);
+            if (!workflow) {
+                return {
+                    workflowType: null
+                };
+            }
+            return {
+                workflowType: workflow,
+                assets: utils_1.VIDEO_WORKFLOW_ASSETS[workflow]
+            };
+        });
+    }
     /**
      * Get available models and their worker counts. Normally, you would get list once you connect
      * to the server, but you can also call this method to get the list of available models manually.
@@ -627,11 +1059,36 @@ class ProjectsApi extends ApiGroup_1.default {
                 return {
                     id: (model === null || model === void 0 ? void 0 : model.id) || sid,
                     name: (model === null || model === void 0 ? void 0 : model.name) || sid.replace(/-/g, ' '),
-                    workerCount
+                    workerCount,
+                    media: (model === null || model === void 0 ? void 0 : model.media) || 'image'
                 };
             });
         });
     }
+    getModelOptions(modelId) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const models = yield this.getSupportedModels();
+            const tiers = yield this._getModelTiers();
+            const model = models.find((m) => m.id === modelId);
+            if (!model) {
+                throw new Error(`Model ${modelId} not supported`);
+            }
+            const tier = tiers[model.tier];
+            if (!tier) {
+                throw new Error(`Unable to find model tier "${model.tier}" please contact support`);
+            }
+            if ((0, ModelTiersRaw_1.isImageTier)(tier)) {
+                return (0, ModelOptions_1.mapImageTier)(tier);
+            }
+            if ((0, ModelTiersRaw_1.isVideoTier)(tier)) {
+                return (0, ModelOptions_1.mapVideoTier)(tier);
+            }
+            if ((0, ModelTiersRaw_1.isComfyImageTier)(tier)) {
+                return (0, ModelOptions_1.mapComfyImageTier)(tier);
+            }
+            throw new Error(`Unsupported model tier "${model.tier}"`);
+        });
+    }
 }
 exports.default = ProjectsApi;
 //# sourceMappingURL=index.js.map