@sogni-ai/sogni-client 4.0.0-alpha.45 → 4.0.0-alpha.47

package/dist/lib/validation.js

@@ -4,18 +4,12 @@ exports.validateCustomImageSize = validateCustomImageSize;
 exports.validateVideoSize = validateVideoSize;
 exports.validateVideoDuration = validateVideoDuration;
 exports.validateNumber = validateNumber;
-exports.validateForgeSampler = validateForgeSampler;
-exports.validateForgeScheduler = validateForgeScheduler;
 exports.validateTeacacheThreshold = validateTeacacheThreshold;
-exports.validateComfySampler = validateComfySampler;
-exports.validateComfyScheduler = validateComfyScheduler;
 exports.isComfyModel = isComfyModel;
+exports.getMaxContextImages = getMaxContextImages;
 exports.validateSampler = validateSampler;
 exports.validateScheduler = validateScheduler;
-const ForgeSamplerParams_1 = require("../Projects/types/ForgeSamplerParams");
-const ForgeSchedulerParams_1 = require("../Projects/types/ForgeSchedulerParams");
-const ComfySamplerParams_1 = require("../Projects/types/ComfySamplerParams");
-const ComfySchedulerParams_1 = require("../Projects/types/ComfySchedulerParams");
+const ApiClient_1 = require("../ApiClient");
 function validateCustomImageSize(value) {
     return validateNumber(value, { min: 256, max: 2048, propertyName: 'Width and height' });
 }
@@ -55,30 +49,6 @@ function validateNumber(value, { min, max, propertyName, defaultValue } = {}) {
     }
     return number;
 }
-function validateForgeSampler(value) {
-    if (!value) {
-        return null;
-    }
-    if ((0, ForgeSamplerParams_1.isRawForgeSampler)(value)) {
-        return value;
-    }
-    if ((0, ForgeSamplerParams_1.isForgeSampler)(value)) {
-        return ForgeSamplerParams_1.SupportedForgeSamplers[value];
-    }
-    throw new Error(`Invalid sampler: ${value}. Supported options: ${Object.keys(ForgeSamplerParams_1.SupportedForgeSamplers).join(', ')}`);
-}
-function validateForgeScheduler(value) {
-    if (!value) {
-        return null;
-    }
-    if ((0, ForgeSchedulerParams_1.isRawForgeScheduler)(value)) {
-        return value;
-    }
-    if ((0, ForgeSchedulerParams_1.isForgeScheduler)(value)) {
-        return ForgeSchedulerParams_1.SupportedForgeSchedulers[value];
-    }
-    throw new Error(`Invalid scheduler: ${value}. Supported options: ${Object.keys(ForgeSchedulerParams_1.SupportedForgeSchedulers).join(', ')}`);
-}
 /**
  * Validate teacacheThreshold for T2V and I2V models.
  * Range: 0.0-1.0. 0.0 = disabled.
@@ -96,46 +66,58 @@ function validateTeacacheThreshold(value) {
     }
     return num;
 }
-/**
- * Validate ComfyUI sampler for video models.
- * Returns the sampler string directly (no mapping needed).
- */
-function validateComfySampler(value) {
-    if (!value) {
-        return undefined;
-    }
-    if ((0, ComfySamplerParams_1.isComfySampler)(value)) {
-        return ComfySamplerParams_1.SupportedComfySamplers[value];
-    }
-    throw new Error(`Invalid comfySampler: ${value}. Supported options: ${Object.keys(ComfySamplerParams_1.SupportedComfySamplers).join(', ')}`);
+function isComfyModel(modelId) {
+    const COMFY_PREFIXES = ['z_image_', 'qwen_image_', 'flux2_', 'wan_'];
+    return COMFY_PREFIXES.some((prefix) => modelId.startsWith(prefix));
 }
 /**
- * Validate ComfyUI scheduler for video models.
- * Returns the scheduler string directly (no mapping needed).
+ * Get the maximum number of context images supported by a model.
+ * - Flux.2 Dev: 6 images
+ * - Qwen Image Edit: 3 images
+ * - Flux Kontext: 2 images
+ * - Default: 3 images
  */
-function validateComfyScheduler(value) {
-    if (!value) {
-        return undefined;
+function getMaxContextImages(modelId) {
+    if (modelId.startsWith('flux2_')) {
+        return 6;
     }
-    if ((0, ComfySchedulerParams_1.isComfyScheduler)(value)) {
-        return ComfySchedulerParams_1.SupportedComfySchedulers[value];
+    if (modelId.startsWith('qwen_image_')) {
+        return 3;
     }
-    throw new Error(`Invalid comfyScheduler: ${value}. Supported options: ${Object.keys(ComfySchedulerParams_1.SupportedComfySchedulers).join(', ')}`);
+    if (modelId.includes('kontext')) {
+        return 2;
+    }
+    // Default fallback for other models that might support context images
+    return 3;
 }
-function isComfyModel(modelId) {
-    const COMFY_PREFIXES = ['z_image_', 'qwen_image_', 'flux2_', 'wan_'];
-    return COMFY_PREFIXES.some((prefix) => modelId.startsWith(prefix));
+function validateOption(value, options, errorMessage) {
+    if (!options.includes(value)) {
+        throw new ApiClient_1.ApiError(400, {
+            status: 'error',
+            errorCode: 0,
+            message: errorMessage
+        });
+    }
+    return value;
 }
-function validateSampler(modelId, sampler) {
-    if (isComfyModel(modelId)) {
-        return validateComfySampler(sampler);
+/**
+ * Validate sampler value against allowed options.
+ * Returns the validated value unchanged - sogni-socket handles normalization.
+ */
+function validateSampler(value, options) {
+    if (!options.sampler.allowed.length || !value) {
+        return null;
     }
-    return validateForgeSampler(sampler);
+    return validateOption(value, options.sampler.allowed, `Invalid sampler ${value}. Must be one of "${options.sampler.allowed.join('", "')}".`);
 }
-function validateScheduler(modelId, scheduler) {
-    if (isComfyModel(modelId)) {
-        return validateComfyScheduler(scheduler);
+/**
+ * Validate scheduler value against allowed options.
+ * Returns the validated value unchanged - sogni-socket handles normalization.
+ */
+function validateScheduler(value, options) {
+    if (!options.scheduler.allowed.length || !value) {
+        return null;
     }
-    return validateForgeScheduler(scheduler);
+    return validateOption(value, options.scheduler.allowed, `Invalid scheduler ${value}. Must be one of "${options.scheduler.allowed.join('", "')}".`);
 }
 //# sourceMappingURL=validation.js.map

package/dist/lib/validation.js.map

@@ -1 +1 @@
-{"version":3,"file":"validation.js","sourceRoot":"","sources":["../../src/lib/validation.ts"],"names":[],"mappings":";;AAaA,0DAEC;AAMD,8CAEC;AAED,sDAEC;AASD,wCA4BC;AAED,oDAaC;AAED,wDAaC;AAMD,8DAYC;AAMD,oDAUC;AAMD,wDAUC;AAED,oCAGC;AAED,0CAKC;AAED,8CAKC;AAnKD,6EAI8C;AAC9C,iFAIgD;AAChD,6EAA8F;AAC9F,iFAAoG;AAEpG,SAAgB,uBAAuB,CAAC,KAAU;IAChD,OAAO,cAAc,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,YAAY,EAAE,kBAAkB,EAAE,CAAC,CAAC;AAC1F,CAAC;AAED;;;GAGG;AACH,SAAgB,iBAAiB,CAAC,KAAU,EAAE,YAAgC;IAC5E,OAAO,cAAc,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,YAAY,EAAE,SAAS,YAAY,EAAE,EAAE,CAAC,CAAC;AACpF,CAAC;AAED,SAAgB,qBAAqB,CAAC,KAAU;IAC9C,OAAO,cAAc,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,YAAY,EAAE,gBAAgB,EAAE,CAAC,CAAC;AACpF,CAAC;AASD,SAAgB,cAAc,CAC5B,KAAU,EACV,EAAE,GAAG,EAAE,GAAG,EAAE,YAAY,EAAE,YAAY,KAA8B,EAAE;IAEtE,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC7B,MAAM,eAAe,GAAG,YAAY,KAAK,SAAS,CAAC;IACnD,IAAI,YAAY,EAAE,CAAC;QACjB,YAAY,GAAG,OAAO,CAAC;IACzB,CAAC;IACD,IAAI,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;QAClB,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,YAAY,CAAC;QACtB,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,GAAG,YAAY,0BAA0B,KAAK,EAAE,CAAC,CAAC;IACpE,CAAC;IACD,IAAI,GAAG,KAAK,SAAS,IAAI,MAAM,GAAG,GAAG,EAAE,CAAC;QACtC,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,YAAY,CAAC;QACtB,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,GAAG,YAAY,0BAA0B,GAAG,SAAS,MAAM,EAAE,CAAC,CAAC;IACjF,CAAC;IACD,IAAI,GAAG,KAAK,SAAS,IAAI,MAAM,GAAG,GAAG,EAAE,CAAC;QACtC,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,YAAY,CAAC;QACtB,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,GAAG,YAAY,0BAA0B,GAAG,SAAS,MAAM,EAAE,CAAC,CAAC;IACjF,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAgB,oBAAoB,CAAC,KAAc;IACjD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,IAAA,sCAAiB,EAAC,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,IAAA,mCAAc,EAAC,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,2CAAsB,CAAC,KAAK,CAAC,CAAC;IACvC,CAAC;IACD,MAAM,IAAI,KAAK,CACb,oBAAoB,KAAK,wBAAwB,MAAM,CAAC,IAAI,CAAC,2CAAsB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAClG,CAAC;AACJ,CAAC;AAED,SAAgB,sBAAsB,CAAC,KAAc;IACnD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,IAAA,0CAAmB,EAAC,KAAK,CAAC,EAAE,CAAC;QAC/B,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,IAAA,uCAAgB,EAAC,KAAK,CAAC,EAAE,CAAC;QAC5B,OAAO,+CAAwB,CAAC,KAAK,CAAC,CAAC;IACzC,CAAC;IACD,MAAM,IAAI,KAAK,CACb,sBAAsB,KAAK,wBAAwB,MAAM,CAAC,IAAI,CAAC,+CAAwB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACtG,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,SAAgB,yBAAyB,CAAC,KAAc;IACtD,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QAC1C,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC1B,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,2CAA2C,KAAK,EAAE,CAAC,CAAC;IACtE,CAAC;IACD,IAAI,GAAG,GAAG,GAAG,IAAI,GAAG,GAAG,GAAG,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,sDAAsD,GAAG,GAAG,CAAC,CAAC;IAChF,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED;;;GAGG;AACH,SAAgB,oBAAoB,CAAC,KAAc;IACjD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,IAAA,mCAAc,EAAC,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,2CAAsB,CAAC,KAAK,CAAC,CAAC;IACvC,CAAC;IACD,MAAM,IAAI,KAAK,CACb,yBAAyB,KAAK,wBAAwB,MAAM,CAAC,IAAI,CAAC,2CAAsB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACvG,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,SAAgB,sBAAsB,CAAC,KAAc;IACnD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,IAAA,uCAAgB,EAAC,KAAK,CAAC,EAAE,CAAC;QAC5B,OAAO,+CAAwB,CAAC,KAAK,CAAC,CAAC;IACzC,CAAC;IACD,MAAM,IAAI,KAAK,CACb,2BAA2B,KAAK,wBAAwB,MAAM,CAAC,IAAI,CAAC,+CAAwB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC3G,CAAC;AACJ,CAAC;AAED,SAAgB,YAAY,CAAC,OAAe;IAC1C,MAAM,cAAc,GAAG,CAAC,UAAU,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;IACrE,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;AACrE,CAAC;AAED,SAAgB,eAAe,CAAC,OAAe,EAAE,OAAe;IAC9D,IAAI,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC;QAC1B,OAAO,oBAAoB,CAAC,OAAO,CAAC,CAAC;IACvC,CAAC;IACD,OAAO,oBAAoB,CAAC,OAAO,CAAC,CAAC;AACvC,CAAC;AAED,SAAgB,iBAAiB,CAAC,OAAe,EAAE,SAAiB;IAClE,IAAI,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC;QAC1B,OAAO,sBAAsB,CAAC,SAAS,CAAC,CAAC;IAC3C,CAAC;IACD,OAAO,sBAAsB,CAAC,SAAS,CAAC,CAAC;AAC3C,CAAC"}
+{"version":3,"file":"validation.js","sourceRoot":"","sources":["../../src/lib/validation.ts"],"names":[],"mappings":";;AAGA,0DAEC;AAMD,8CAEC;AAED,sDAEC;AASD,wCA4BC;AAMD,8DAYC;AAED,oCAGC;AASD,kDAYC;AAiBD,0CASC;AAMD,8CASC;AA3ID,4CAAwC;AAGxC,SAAgB,uBAAuB,CAAC,KAAU;IAChD,OAAO,cAAc,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,YAAY,EAAE,kBAAkB,EAAE,CAAC,CAAC;AAC1F,CAAC;AAED;;;GAGG;AACH,SAAgB,iBAAiB,CAAC,KAAU,EAAE,YAAgC;IAC5E,OAAO,cAAc,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,YAAY,EAAE,SAAS,YAAY,EAAE,EAAE,CAAC,CAAC;AACpF,CAAC;AAED,SAAgB,qBAAqB,CAAC,KAAU;IAC9C,OAAO,cAAc,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,YAAY,EAAE,gBAAgB,EAAE,CAAC,CAAC;AACpF,CAAC;AASD,SAAgB,cAAc,CAC5B,KAAU,EACV,EAAE,GAAG,EAAE,GAAG,EAAE,YAAY,EAAE,YAAY,KAA8B,EAAE;IAEtE,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC7B,MAAM,eAAe,GAAG,YAAY,KAAK,SAAS,CAAC;IACnD,IAAI,YAAY,EAAE,CAAC;QACjB,YAAY,GAAG,OAAO,CAAC;IACzB,CAAC;IACD,IAAI,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;QAClB,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,YAAY,CAAC;QACtB,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,GAAG,YAAY,0BAA0B,KAAK,EAAE,CAAC,CAAC;IACpE,CAAC;IACD,IAAI,GAAG,KAAK,SAAS,IAAI,MAAM,GAAG,GAAG,EAAE,CAAC;QACtC,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,YAAY,CAAC;QACtB,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,GAAG,YAAY,0BAA0B,GAAG,SAAS,MAAM,EAAE,CAAC,CAAC;IACjF,CAAC;IACD,IAAI,GAAG,KAAK,SAAS,IAAI,MAAM,GAAG,GAAG,EAAE,CAAC;QACtC,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,YAAY,CAAC;QACtB,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,GAAG,YAAY,0BAA0B,GAAG,SAAS,MAAM,EAAE,CAAC,CAAC;IACjF,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,SAAgB,yBAAyB,CAAC,KAAc;IACtD,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QAC1C,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC1B,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,2CAA2C,KAAK,EAAE,CAAC,CAAC;IACtE,CAAC;IACD,IAAI,GAAG,GAAG,GAAG,IAAI,GAAG,GAAG,GAAG,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,sDAAsD,GAAG,GAAG,CAAC,CAAC;IAChF,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAgB,YAAY,CAAC,OAAe;IAC1C,MAAM,cAAc,GAAG,CAAC,UAAU,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;IACrE,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;AACrE,CAAC;AAED;;;;;;GAMG;AACH,SAAgB,mBAAmB,CAAC,OAAe;IACjD,IAAI,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACjC,OAAO,CAAC,CAAC;IACX,CAAC;IACD,IAAI,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;QACtC,OAAO,CAAC,CAAC;IACX,CAAC;IACD,IAAI,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QAChC,OAAO,CAAC,CAAC;IACX,CAAC;IACD,sEAAsE;IACtE,OAAO,CAAC,CAAC;AACX,CAAC;AAED,SAAS,cAAc,CAAc,KAAQ,EAAE,OAAY,EAAE,YAAoB;IAC/E,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QAC7B,MAAM,IAAI,oBAAQ,CAAC,GAAG,EAAE;YACtB,MAAM,EAAE,OAAO;YACf,SAAS,EAAE,CAAC;YACZ,OAAO,EAAE,YAAY;SACtB,CAAC,CAAC;IACL,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;GAGG;AACH,SAAgB,eAAe,CAAC,KAAyB,EAAE,OAAqB;IAC9E,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QAC9C,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,cAAc,CACnB,KAAK,EACL,OAAO,CAAC,OAAO,CAAC,OAAO,EACvB,mBAAmB,KAAK,qBAAqB,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CACtF,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,SAAgB,iBAAiB,CAAC,KAAyB,EAAE,OAAqB;IAChF,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QAChD,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,cAAc,CACnB,KAAK,EACL,OAAO,CAAC,SAAS,CAAC,OAAO,EACzB,qBAAqB,KAAK,qBAAqB,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAC1F,CAAC;AACJ,CAAC"}

package/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "4.0.0-alpha.45",
+  "version": "4.0.0-alpha.47",
   "description": "Sogni Supernet Client",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/Projects/createJobRequestMessage.ts

@@ -9,17 +9,16 @@ import { ControlNetParams, ControlNetParamsRaw } from './types/ControlNetParams'
 import {
   validateNumber,
   validateCustomImageSize,
-  validateForgeSampler,
-  validateForgeScheduler,
   validateVideoSize,
   validateTeacacheThreshold,
-  validateComfySampler,
-  validateComfyScheduler,
   isComfyModel,
-  validateVideoDuration
+  validateVideoDuration,
+  validateSampler,
+  validateScheduler
 } from '../lib/validation';
 import { getVideoWorkflowType, isVideoModel, VIDEO_WORKFLOW_ASSETS } from './utils';
 import { ApiError } from '../ApiClient';
+import { ImageModelOptions, ModelOptions, VideoModelOptions } from './types/ModelOptions';
 
 /**
  * Validate that the provided assets match the workflow requirements.
@@ -178,7 +177,11 @@ function getControlNet(params: ControlNetParams): ControlNetParamsRaw[] {
   return [cn];
 }
 
-function applyImageParams(inputKeyframe: Record<string, any>, params: ImageProjectParams) {
+function applyImageParams(
+  inputKeyframe: Record<string, any>,
+  params: ImageProjectParams,
+  options: ImageModelOptions
+) {
   const keyFrame: Record<string, any> = {
     ...inputKeyframe,
     sizePreset: params.sizePreset,
@@ -189,23 +192,16 @@ function applyImageParams(inputKeyframe: Record<string, any>, params: ImageProje
     hasContextImage5: !!params.contextImages?.[4],
     hasContextImage6: !!params.contextImages?.[5]
   };
-
-  // Handle sampler/scheduler: ComfyUI models use comfySampler/comfyScheduler,
-  // legacy models use sampler/scheduler (mapped to scheduler/timeStepSpacing)
+  // Sampler/scheduler handling: SDK validates and passes through as-is.
+  // sogni-socket normalizes values for both ComfyUI and Forge workers.
   if (isComfyModel(params.modelId)) {
-    if (params.sampler !== undefined) {
-      keyFrame.comfySampler = validateComfySampler(params.sampler);
-    }
-    if (params.scheduler !== undefined) {
-      keyFrame.comfyScheduler = validateComfyScheduler(params.scheduler);
-    }
+    // ComfyUI models use comfySampler/comfyScheduler fields
+    keyFrame.comfySampler = validateSampler(params.sampler, options);
+    keyFrame.comfyScheduler = validateScheduler(params.scheduler, options);
   } else {
-    if (params.sampler !== undefined) {
-      keyFrame.scheduler = validateForgeSampler(params.sampler);
-    }
-    if (params.scheduler !== undefined) {
-      keyFrame.timeStepSpacing = validateForgeScheduler(params.scheduler);
-    }
+    // Legacy Forge models use scheduler/timeStepSpacing fields
+    keyFrame.scheduler = validateSampler(params.sampler, options);
+    keyFrame.timeStepSpacing = validateScheduler(params.scheduler, options);
   }
 
   if (params.startingImage) {
@@ -232,7 +228,11 @@ function applyImageParams(inputKeyframe: Record<string, any>, params: ImageProje
   return keyFrame;
 }
 
-function applyVideoParams(inputKeyframe: Record<string, any>, params: VideoProjectParams) {
+function applyVideoParams(
+  inputKeyframe: Record<string, any>,
+  params: VideoProjectParams,
+  options: VideoModelOptions
+) {
   if (!isVideoModel(params.modelId)) {
     throw new ApiError(400, {
       status: 'error',
@@ -295,19 +295,13 @@ function applyVideoParams(inputKeyframe: Record<string, any>, params: VideoProje
     keyFrame.height = validateVideoSize(params.height, 'height');
   }
 
-  // Video models are ComfyUI models - only accept comfySampler/comfyScheduler
-  // Legacy sampler/scheduler fields are NOT supported for video models
-  if (params.sampler !== undefined) {
-    keyFrame.comfySampler = validateComfySampler(params.sampler);
-  }
-  if (params.scheduler !== undefined) {
-    keyFrame.comfyScheduler = validateComfyScheduler(params.scheduler);
-  }
+  keyFrame.comfySampler = validateSampler(params.sampler, options);
+  keyFrame.comfyScheduler = validateScheduler(params.scheduler, options);
 
   return keyFrame;
 }
 
-function createJobRequestMessage(id: string, params: ProjectParams) {
+function createJobRequestMessage(id: string, params: ProjectParams, options: ModelOptions) {
   const template = getTemplate();
   // Base keyFrame with common params
   let keyFrame: Record<string, any> = {
@@ -323,10 +317,26 @@ function createJobRequestMessage(id: string, params: ProjectParams) {
 
   switch (params.type) {
     case 'image':
-      keyFrame = applyImageParams(keyFrame, params);
+      if (options.type !== 'image') {
+        throw new ApiError(400, {
+          status: 'error',
+          errorCode: 0,
+          message:
+            'Invalid model type. Model does not support image generation. Please use a different model.'
+        });
+      }
+      keyFrame = applyImageParams(keyFrame, params, options);
       break;
     case 'video':
-      keyFrame = applyVideoParams(keyFrame, params);
+      if (options.type !== 'video') {
+        throw new ApiError(400, {
+          status: 'error',
+          errorCode: 0,
+          message:
+            'Invalid model type. Model does not support video generation. Please use a different model.'
+        });
+      }
+      keyFrame = applyVideoParams(keyFrame, params, options);
       break;
     default:
       throw new ApiError(400, {

package/src/Projects/index.ts

@@ -11,15 +11,7 @@ import {
   SupportedModel,
   ImageProjectParams,
   VideoProjectParams,
-  VideoEstimateRequest,
-  SupportedComfySamplers,
-  SupportedForgeSamplers,
-  SupportedComfySchedulers,
-  SupportedForgeSchedulers,
-  ComfyScheduler,
-  ForgeScheduler,
-  ComfySampler,
-  ForgeSampler
+  VideoEstimateRequest
 } from './types';
 import {
   JobErrorData,
@@ -47,7 +39,9 @@ import {
   VIDEO_WORKFLOW_ASSETS
 } from './utils';
 import { TokenType } from '../types/token';
-import { isComfyModel, validateSampler } from '../lib/validation';
+import { getMaxContextImages, validateSampler } from '../lib/validation';
+import ModelTiersRaw, { isComfyImageTier, isImageTier, isVideoTier } from './types/ModelTiersRaw';
+import { mapComfyImageTier, mapImageTier, mapVideoTier, ModelOptions } from './types/ModelOptions';
 
 const sizePresetCache = new Cache<SizePreset[]>(10 * 60 * 1000);
 const GARBAGE_COLLECT_TIMEOUT = 30000;
@@ -106,6 +100,13 @@ class ProjectsApi extends ApiGroup<ProjectApiEvents> {
     data: null,
     updatedAt: new Date(0)
   };
+  private _modelTiers: {
+    data: ModelTiersRaw;
+    updatedAt: Date;
+  } = {
+    data: {},
+    updatedAt: new Date(0)
+  };
 
   get availableModels() {
     return this._availableModels;
@@ -524,7 +525,8 @@ class ProjectsApi extends ApiGroup<ProjectApiEvents> {
    */
   async create(data: ProjectParams): Promise<Project> {
     const project = new Project({ ...data }, { api: this, logger: this.client.logger });
-    const request = createJobRequestMessage(project.id, data);
+    const modelOptions = await this.getModelOptions(data.modelId);
+    const request = createJobRequestMessage(project.id, data, modelOptions);
 
     switch (data.type) {
       case 'image':
@@ -550,19 +552,20 @@ class ProjectsApi extends ApiGroup<ProjectApiEvents> {
       await this.uploadCNImage(project.id, data.controlNet.image);
     }
 
-    // Context images (Flux.2 Dev, Qwen Image Edit Plus support up to 3; Flux Kontext supports up to 2)
+    // Context images (Flux.2 Dev supports up to 6; Qwen Image Edit Plus supports up to 3; Flux Kontext supports up to 2)
     if (data.contextImages?.length) {
-      if (data.contextImages.length > 3) {
+      const maxContextImages = getMaxContextImages(data.modelId);
+      if (data.contextImages.length > maxContextImages) {
         throw new ApiError(500, {
           status: 'error',
           errorCode: 0,
-          message: `Up to 3 context images are supported`
+          message: `Up to ${maxContextImages} context images are supported for this model`
         });
       }
       await Promise.all(
         data.contextImages.map((image, index) => {
           if (image && image !== true) {
-            return this.uploadContextImage(project.id, index as 0 | 1 | 2, image);
+            return this.uploadContextImage(project.id, index as 0 | 1 | 2 | 3 | 4 | 5, image);
           }
         })
       );
@@ -672,11 +675,11 @@ class ProjectsApi extends ApiGroup<ProjectApiEvents> {
 
   private async uploadContextImage(
     projectId: string,
-    index: 0 | 1 | 2,
+    index: 0 | 1 | 2 | 3 | 4 | 5,
     file: File | Buffer | Blob
   ) {
     const imageId = getUUID();
-    const imageIndex = (index + 1) as 1 | 2 | 3;
+    const imageIndex = (index + 1) as 1 | 2 | 3 | 4 | 5 | 6;
     const presignedUrl = await this.uploadUrl({
       imageId,
       jobId: projectId,
@@ -833,6 +836,7 @@ class ProjectsApi extends ApiGroup<ProjectApiEvents> {
     contextImages
   }: EstimateRequest): Promise<CostEstimation> {
     let apiVersion = 2;
+    const modelOptions = await this.getModelOptions(model);
     const pathParams = [
       tokenType || 'spark',
       network,
@@ -858,7 +862,7 @@ class ProjectsApi extends ApiGroup<ProjectApiEvents> {
     if (sampler) {
       apiVersion = 3;
       pathParams.push(guidance || 0);
-      pathParams.push(validateSampler(model, sampler)!);
+      pathParams.push(validateSampler(sampler, modelOptions)!);
       pathParams.push(contextImages || 0);
     }
     const r = await this.client.socket.get<EstimationResponse>(
@@ -1005,6 +1009,19 @@ class ProjectsApi extends ApiGroup<ProjectApiEvents> {
     return models;
   }
 
+  private async _getModelTiers(forceRefresh = false) {
+    if (
+      this._modelTiers.data &&
+      !forceRefresh &&
+      Date.now() - this._modelTiers.updatedAt.getTime() < MODELS_REFRESH_INTERVAL
+    ) {
+      return this._modelTiers.data;
+    }
+    const tiers = await this.client.socket.get<ModelTiersRaw>(`/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.
    *
@@ -1104,18 +1121,27 @@ class ProjectsApi extends ApiGroup<ProjectApiEvents> {
     });
   }
 
-  async getSamplers(modelId: string) {
-    if (isComfyModel(modelId)) {
-      return Object.keys(SupportedComfySamplers) as ComfySampler[];
+  async getModelOptions(modelId: string): Promise<ModelOptions> {
+    const models = await this.getSupportedModels();
+    const tiers = await this._getModelTiers();
+    const model = models.find((m) => m.id === modelId);
+    if (!model) {
+      throw new Error(`Model ${modelId} not supported`);
     }
-    return Object.keys(SupportedForgeSamplers) as ForgeSampler[];
-  }
-
-  async getSchedulers(modelId: string) {
-    if (isComfyModel(modelId)) {
-      return Object.keys(SupportedComfySchedulers) as ComfyScheduler[];
+    const tier = tiers[model.tier];
+    if (!tier) {
+      throw new Error(`Unable to find model tier "${model.tier}" please contact support`);
+    }
+    if (isImageTier(tier)) {
+      return mapImageTier(tier);
+    }
+    if (isVideoTier(tier)) {
+      return mapVideoTier(tier);
+    }
+    if (isComfyImageTier(tier)) {
+      return mapComfyImageTier(tier);
     }
-    return Object.keys(SupportedForgeSchedulers) as ForgeScheduler[];
+    throw new Error(`Unsupported model tier "${model.tier}"`);
   }
 }
 

package/src/Projects/types/ComfySamplerParams.ts

@@ -1,34 +0,0 @@
-/**
- * ComfyUI sampler parameters for video models and ComfyUI-based image models.
- * These use lowercase underscore format directly matching ComfyUI's internal names.
- */
-export const SupportedComfySamplers = {
-  euler: 'euler',
-  euler_ancestral: 'euler_ancestral',
-  heun: 'heun',
-  dpmpp_2m: 'dpmpp_2m',
-  dpmpp_2m_sde: 'dpmpp_2m_sde',
-  dpmpp_sde: 'dpmpp_sde',
-  dpmpp_3m_sde: 'dpmpp_3m_sde',
-  uni_pc: 'uni_pc',
-  lcm: 'lcm',
-  // Additional ComfyUI samplers
-  lms: 'lms',
-  dpm_2: 'dpm_2',
-  dpm_2_ancestral: 'dpm_2_ancestral',
-  dpm_fast: 'dpm_fast',
-  dpm_adaptive: 'dpm_adaptive',
-  dpmpp_2s_ancestral: 'dpmpp_2s_ancestral',
-  ddpm: 'ddpm',
-  ddim: 'ddim',
-  uni_pc_bh2: 'uni_pc_bh2',
-  // Z-Image specific samplers
-  res_multistep: 'res_multistep',
-  res_multistep_cfg_pp: 'res_multistep_cfg_pp'
-};
-
-export function isComfySampler(sampler: string): sampler is ComfySampler {
-  return sampler in SupportedComfySamplers;
-}
-
-export type ComfySampler = keyof typeof SupportedComfySamplers;

package/src/Projects/types/ModelOptions.ts

@@ -0,0 +1,92 @@
+import { ComfyImageTier, ImageTier, NumericDefaults, VideoTier } from './ModelTiersRaw';
+import { samplerValueToAlias } from '../utils/samplers';
+import { schedulerValueToAlias } from '../utils/scheduler';
+
+interface NumRange {
+  min: number;
+  max: number;
+  step: number;
+  default: number;
+}
+
+interface Options<T> {
+  allowed: T[];
+  default: T | null;
+}
+
+interface NumOptions {
+  options: number[];
+  default: number;
+}
+
+export interface ImageModelOptions {
+  type: 'image';
+  steps: NumRange;
+  guidance: NumRange;
+  scheduler: Options<string>;
+  sampler: Options<string>;
+}
+
+export interface VideoModelOptions {
+  type: 'video';
+  steps: NumRange;
+  guidance: NumRange;
+  fps: Options<number>;
+  sampler: Options<string>;
+  scheduler: Options<string>;
+}
+
+export type ModelOptions = ImageModelOptions | VideoModelOptions;
+
+function mapRange(data: NumericDefaults): NumRange {
+  return {
+    min: data.min,
+    max: data.max,
+    step: data.decimals ? Math.pow(10, 0 - data.decimals) : data.step || 1,
+    default: data.default
+  };
+}
+
+function mapOptions<T>(data: Options<T> | undefined, mapper = (value: T) => value): Options<T> {
+  if (!data) {
+    return {
+      allowed: [],
+      default: null
+    };
+  }
+  return {
+    allowed: data.allowed.map(mapper),
+    default: data.default !== null ? mapper(data.default) : null
+  };
+}
+
+export function mapImageTier(tier: ImageTier): ImageModelOptions {
+  return {
+    type: 'image',
+    steps: mapRange(tier.steps),
+    guidance: mapRange(tier.guidance),
+    scheduler: mapOptions(tier.scheduler, schedulerValueToAlias),
+    sampler: mapOptions(tier.sampler, samplerValueToAlias)
+  };
+}
+
+export function mapComfyImageTier(tier: ComfyImageTier): ImageModelOptions {
+  return {
+    type: 'image',
+    steps: mapRange(tier.steps),
+    guidance: mapRange(tier.guidance),
+    scheduler: mapOptions(tier.comfyScheduler, schedulerValueToAlias),
+    sampler: mapOptions(tier.comfySampler, samplerValueToAlias)
+  };
+}
+
+export function mapVideoTier(tier: VideoTier): VideoModelOptions {
+  return {
+    type: 'video',
+    steps: mapRange(tier.steps),
+    guidance: mapRange(tier.guidance),
+    scheduler: mapOptions(tier.comfyScheduler, schedulerValueToAlias),
+    sampler: mapOptions(tier.comfySampler, samplerValueToAlias),
+    fps: tier.fps
+  };
+}