@sogni-ai/sogni-client 5.1.0 → 5.3.0

package/dist-esm/lib/validation.js

@@ -9,19 +9,27 @@ const EXTENDED_IMAGE_SIZE_MODEL_IDS = new Set([
     'qwen_image_2512_fp8_lightning',
     'flux2_dev_fp8'
 ]);
-function getMaxCustomImageSize(modelId) {
+const KREA_IDENTITY_EDIT_MODEL_IDS = new Set([
+    'krea2_identity_edit_v1_2',
+    'dark_beast_krea2_identity_edit_v1_2'
+]);
+function getCustomImageSizeBounds(modelId) {
+    if (modelId && KREA_IDENTITY_EDIT_MODEL_IDS.has(modelId)) {
+        return { min: 512, max: 2048 };
+    }
     if (modelId === 'gpt-image-2') {
-        return 3840;
+        return { min: 256, max: 3840 };
     }
     if (modelId && EXTENDED_IMAGE_SIZE_MODEL_IDS.has(modelId)) {
-        return 2560;
+        return { min: 256, max: 2560 };
     }
-    return 2048;
+    return { min: 256, max: 2048 };
 }
 export function validateCustomImageSize(value, { modelId, propertyName = 'Width and height' } = {}) {
+    const bounds = getCustomImageSizeBounds(modelId);
     return validateNumber(value, {
-        min: 256,
-        max: getMaxCustomImageSize(modelId),
+        min: bounds.min,
+        max: bounds.max,
         propertyName
     });
 }
@@ -79,7 +87,16 @@ export function validateTeacacheThreshold(value) {
     return num;
 }
 export function isComfyModel(modelId) {
-    const COMFY_PREFIXES = ['z_image_', 'krea2_', 'qwen_image_', 'flux2_', 'wan_', 'ace_step'];
+    const COMFY_PREFIXES = [
+        'z_image_',
+        'dark_beast_z_image_',
+        'krea2_',
+        'dark_beast_krea2_',
+        'qwen_image_',
+        'flux2_',
+        'wan_',
+        'ace_step'
+    ];
     return COMFY_PREFIXES.some((prefix) => modelId.startsWith(prefix));
 }
 /**
@@ -87,6 +104,7 @@ export function isComfyModel(modelId) {
  * - GPT Image 2: 16 images
  * - Flux.2 Dev: 6 images
  * - Qwen Image Edit: 3 images
+ * - Krea 2 Identity Edit: 2 images
  * - Flux Kontext: 2 images
  * - Default: 3 images
  */
@@ -100,6 +118,9 @@ export function getMaxContextImages(modelId) {
     if (modelId.startsWith('qwen_image_')) {
         return 3;
     }
+    if (KREA_IDENTITY_EDIT_MODEL_IDS.has(modelId)) {
+        return 2;
+    }
     if (modelId.includes('kontext')) {
         return 2;
     }
@@ -136,4 +157,15 @@ export function validateScheduler(value, options) {
     }
     return validateOption(value, options.scheduler.allowed, `Invalid scheduler ${value}. Must be one of "${options.scheduler.allowed.join('", "')}".`);
 }
+/**
+ * Validate a model-specific VAE value against allowed options.
+ * Returns the validated value unchanged; sogni-socket passes the filename to ComfyUI.
+ */
+export function validateVae(value, options) {
+    var _a;
+    if (!('vae' in options) || !((_a = options.vae) === null || _a === void 0 ? void 0 : _a.allowed.length) || !value) {
+        return null;
+    }
+    return validateOption(value, options.vae.allowed, `Invalid VAE ${value}. Must be one of "${options.vae.allowed.join('", "')}".`);
+}
 //# sourceMappingURL=validation.js.map

package/dist-esm/lib/validation.js.map

@@ -1 +1 @@
-{"version":3,"file":"validation.js","sourceRoot":"","sources":["../../src/lib/validation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AAGjD,MAAM,6BAA6B,GAAG,IAAI,GAAG,CAAC;IAC5C,cAAc;IACd,oBAAoB;IACpB,wBAAwB;IACxB,0BAA0B;IAC1B,oCAAoC;IACpC,qBAAqB;IACrB,+BAA+B;IAC/B,eAAe;CAChB,CAAC,CAAC;AAOH,SAAS,qBAAqB,CAAC,OAAgB;IAC7C,IAAI,OAAO,KAAK,aAAa,EAAE,CAAC;QAC9B,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,OAAO,IAAI,6BAA6B,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QAC1D,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,KAAU,EACV,EAAE,OAAO,EAAE,YAAY,GAAG,kBAAkB,KAAiC,EAAE;IAE/E,OAAO,cAAc,CAAC,KAAK,EAAE;QAC3B,GAAG,EAAE,GAAG;QACR,GAAG,EAAE,qBAAqB,CAAC,OAAO,CAAC;QACnC,YAAY;KACb,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,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,MAAM,UAAU,qBAAqB,CAAC,KAAU,EAAE,GAAG,GAAG,CAAC,EAAE,GAAG,GAAG,EAAE;IACjE,OAAO,cAAc,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,YAAY,EAAE,gBAAgB,EAAE,CAAC,CAAC;AAC7E,CAAC;AASD,MAAM,UAAU,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,CAAC,YAAY,EAAE,CAAC;QAClB,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,MAAM,UAAU,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,MAAM,UAAU,YAAY,CAAC,OAAe;IAC1C,MAAM,cAAc,GAAG,CAAC,UAAU,EAAE,QAAQ,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;IAC3F,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;AACrE,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,mBAAmB,CAAC,OAAe;IACjD,IAAI,OAAO,KAAK,aAAa,EAAE,CAAC;QAC9B,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,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,QAAQ,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,MAAM,UAAU,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,MAAM,UAAU,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"}
+{"version":3,"file":"validation.js","sourceRoot":"","sources":["../../src/lib/validation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AAGjD,MAAM,6BAA6B,GAAG,IAAI,GAAG,CAAC;IAC5C,cAAc;IACd,oBAAoB;IACpB,wBAAwB;IACxB,0BAA0B;IAC1B,oCAAoC;IACpC,qBAAqB;IACrB,+BAA+B;IAC/B,eAAe;CAChB,CAAC,CAAC;AAEH,MAAM,4BAA4B,GAAG,IAAI,GAAG,CAAC;IAC3C,0BAA0B;IAC1B,qCAAqC;CACtC,CAAC,CAAC;AAOH,SAAS,wBAAwB,CAAC,OAAgB;IAChD,IAAI,OAAO,IAAI,4BAA4B,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QACzD,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;IACjC,CAAC;IACD,IAAI,OAAO,KAAK,aAAa,EAAE,CAAC;QAC9B,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;IACjC,CAAC;IACD,IAAI,OAAO,IAAI,6BAA6B,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QAC1D,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;IACjC,CAAC;IACD,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;AACjC,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,KAAU,EACV,EAAE,OAAO,EAAE,YAAY,GAAG,kBAAkB,KAAiC,EAAE;IAE/E,MAAM,MAAM,GAAG,wBAAwB,CAAC,OAAO,CAAC,CAAC;IACjD,OAAO,cAAc,CAAC,KAAK,EAAE;QAC3B,GAAG,EAAE,MAAM,CAAC,GAAG;QACf,GAAG,EAAE,MAAM,CAAC,GAAG;QACf,YAAY;KACb,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,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,MAAM,UAAU,qBAAqB,CAAC,KAAU,EAAE,GAAG,GAAG,CAAC,EAAE,GAAG,GAAG,EAAE;IACjE,OAAO,cAAc,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,YAAY,EAAE,gBAAgB,EAAE,CAAC,CAAC;AAC7E,CAAC;AASD,MAAM,UAAU,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,CAAC,YAAY,EAAE,CAAC;QAClB,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,MAAM,UAAU,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,MAAM,UAAU,YAAY,CAAC,OAAe;IAC1C,MAAM,cAAc,GAAG;QACrB,UAAU;QACV,qBAAqB;QACrB,QAAQ;QACR,mBAAmB;QACnB,aAAa;QACb,QAAQ;QACR,MAAM;QACN,UAAU;KACX,CAAC;IACF,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;AACrE,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,mBAAmB,CAAC,OAAe;IACjD,IAAI,OAAO,KAAK,aAAa,EAAE,CAAC;QAC9B,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,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,4BAA4B,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QAC9C,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,QAAQ,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,MAAM,UAAU,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,MAAM,UAAU,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;AAED;;;GAGG;AACH,MAAM,UAAU,WAAW,CAAC,KAAyB,EAAE,OAAqB;;IAC1E,IAAI,CAAC,CAAC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAA,MAAA,OAAO,CAAC,GAAG,0CAAE,OAAO,CAAC,MAAM,CAAA,IAAI,CAAC,KAAK,EAAE,CAAC;QAClE,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,cAAc,CACnB,KAAK,EACL,OAAO,CAAC,GAAG,CAAC,OAAO,EACnB,eAAe,KAAK,qBAAqB,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAC9E,CAAC;AACJ,CAAC"}

package/dist-esm/types/attribution.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=attribution.js.map

package/dist-esm/types/attribution.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"attribution.js","sourceRoot":"","sources":["../../src/types/attribution.ts"],"names":[],"mappings":""}

package/dist-esm/version.js

@@ -1,4 +1,4 @@
 // AUTO-GENERATED by scripts/generate-version.mjs — do not edit.
 // Source of truth: package.json version.
-export const LIB_VERSION = '5.1.0';
+export const LIB_VERSION = '5.3.0';
 //# sourceMappingURL=version.js.map

package/llms-full.txt

@@ -254,6 +254,33 @@ const portal = await sogni.account.createSubscriptionPortalSession();
 
 `checkAuth()` is only for cookie-auth browser flows on permitted Sogni domains. API-key auth auto-authenticates during `createInstance()`. Username/password auth uses `sogni.account.login()`. Token auth can be resumed with `sogni.setTokens()`.
 
+AGENT ATTRIBUTION:
+
+Keep `appSource` as the stable product/integration identifier. Agent wrappers may add immutable `attribution.connection` and `attribution.workload` defaults:
+
+```typescript
+const sogni = await SogniClient.createInstance({
+  appId: 'my-agent',
+  appSource: 'my-agent-integration',
+  apiKey: process.env.SOGNI_API_KEY,
+  attribution: {
+    connection: {
+      interactionKind: 'external_agent',
+      agentFramework: 'codex',
+      agentSurface: 'plugin'
+    },
+    workload: {
+      workloadKind: 'agent_mediated',
+      agentFramework: 'codex',
+      agentSurface: 'plugin',
+      executionMode: 'server'
+    }
+  }
+});
+```
+
+Project, socket chat, hosted chat/tool, durable chat run, and creative workflow calls accept an optional per-operation `attribution` override. Standalone attributed requests default to `operationScope: 'top_level'` and use the underlying project/job request ID. For child work, pass `operationScope: 'child'`, a stable `rootOperationId`, and the immediate `parentOperationId`; omit `operationId` to use the request's generated ID. The exported attribution types include `native_desktop`. Invalid metadata is dropped, the server remains authoritative, and all new wire fields are omitted when attribution is not configured.
+
 NETWORK TYPES:
 
 - 'fast': High-end GPU workers. Faster, higher cost. REQUIRED for video.
@@ -407,6 +434,7 @@ RECOMMENDED MODELS:
 | z_image_turbo_bf16 | Ultra-fast | 8 | 1 | Fastest generation |
 | z_image_bf16 | Quality | 20 | 3 | High quality Z-Image generation |
 | krea2_turbo_fp8_scaled | Fast | 8 | 1 | Distilled few-step, strong in-image text, up to 2K |
+| krea2_identity_edit_v1_2 | Edit | 10 | 1 | Identity-preserving Krea 2 edit LoRA with 1-2 context images |
 | chroma1-hd_fp8_scaled | Quality | 26 | 3.8 | Final high-res Chroma (uncensored, LoRA-capable) |
 | flux2_dev_fp8 | Quality | 20 | 4 | Pro quality + up to 6 context images (also image edit) |
 | qwen_image_edit_2511_fp8_lightning | Edit | 4 | 1 | Multi-image context |
@@ -417,6 +445,7 @@ COMMUNITY FINE-TUNES (uncensored; discover with sogni.projects.getAvailableModel
 |----------|------|-------|----------|-------|
 | dark_beast_z_image_turbo_v9_bf16 | Ultra-fast | 8 | 1 | Community Z-Image Turbo fine-tune (2K needs 24GB VRAM) |
 | dark_beast_krea2_fp8 | Fast | 16 | 1 | Community Krea 2 fine-tune (2K needs 24GB VRAM) |
+| dark_beast_krea2_identity_edit_v1_2 | Edit | 10 | 1 | Community Krea 2 identity edit LoRA with 1-2 context images |
 | one_obsession_v22_fp16 | Quality | 28 | 6 | Community Illustrious/anime checkpoint, LoRA-capable |
 
 ================================================================================ 5. VIDEO GENERATION - WAN 2.2 MODELS
@@ -880,9 +909,10 @@ sv (Swedish), ro (Romanian), el (Greek), and more. Use 'unknown' for auto-detect
 ================================================================================
 
 The Sogni SDK supports LLM text generation through the Supernet with
-socket-native chat completions, hosted REST chat completions, durable hosted
-chat runs, streaming, multi-turn conversations, thinking/reasoning controls,
-structured outputs, and tool calling.
+socket-native chat completions, hosted REST chat completions, direct hosted
+synchronous tool execution, durable hosted chat runs, streaming, multi-turn
+conversations, thinking/reasoning controls, structured outputs, and tool
+calling.
 
 DEFAULT LLM MODEL: qwen3.6-35b-a3b-gguf-iq4xs
 
@@ -892,6 +922,9 @@ PRIMARY CHAT SURFACES:
   `ChatCompletionResult` for `stream: false` and `ChatStream` for `stream: true`.
 - `sogni.chat.hosted.create(params)` - hosted REST chat completion at
   `/v1/chat/completions`; can execute hosted Sogni tools server-side.
+- `sogni.chat.hosted.executeTool(params)` - direct synchronous hosted
+  composition/planning tool execution at `/v1/creative-agent/tools/execute`;
+  use when the caller already has the exact tool and JSON arguments.
 - `sogni.chat.runs.{create,get,cancel,streamEvents}` - durable hosted runs at
   `/v1/chat/runs`; the server owns the LLM/tool loop and clients can reconnect
   through SSE event replay.
@@ -1190,7 +1223,7 @@ Sogni's generation APIs automatically:
 - Music Generation: "Compose a jazz song about the rain"
 
 CANONICAL HOSTED CREATIVE-TOOL SURFACE (`SogniTools.all`, 24 tools):
-The full hosted surface executes server-side via `sogni.chat.hosted.create()` / `sogni.chat.runs.create()`. The default `creative-tools` value of `sogni_tools` injects this surface server-side; `sogni_tools: "creative-agent"` adds workflow control and asset-manifest tools on top. The client-side `sogni.chat.tools.execute()` path directly dispatches only the core generation tools (`generate_image`, `edit_image`, `generate_video`, `sound_to_video`, `video_to_video`, `generate_music`); composition and post-production tools should be routed through hosted chat or durable runs.
+The full hosted surface executes server-side via `sogni.chat.hosted.create()` / `sogni.chat.runs.create()`. The default `creative-tools` value of `sogni_tools` injects this surface server-side; `sogni_tools: "creative-agent"` adds workflow control and asset-manifest tools on top. The client-side `sogni.chat.tools.execute()` path directly dispatches only the core generation tools (`generate_image`, `edit_image`, `generate_video`, `sound_to_video`, `video_to_video`, `generate_music`). Known synchronous composition/planning calls can use `sogni.chat.hosted.executeTool()`; post-production tools should be routed through hosted chat, durable runs, or explicit workflows.
 
 - `generate_image`, `edit_image`, `generate_video`, `generate_music`, `sound_to_video`, `video_to_video` — core media-generation tools with creative-agent naming.
 - `restore_photo`, `apply_style`, `refine_result`, `change_angle` — image-edit adapters backed by the shared image-edit workflow.
@@ -1207,6 +1240,9 @@ The full hosted surface executes server-side via `sogni.chat.hosted.create()` /
 - `compose_lyrics`, `compose_instrumental` — vocal lyrics or instrumental music structures.
 - `compose_workflow`, `compose_workflow_template` — durable workflow and workflow-template planners.
 
+DIRECT SYNCHRONOUS HOSTED TOOL EXECUTION:
+When the caller already knows the exact synchronous composition/planning tool and JSON arguments, use `sogni.chat.hosted.executeTool()` instead of forcing a `/v1/chat/completions` tool-choice round. Direct execution supports `enhance_prompt`, `compose_script`, `compose_lyrics`, `compose_instrumental`, `compose_workflow`, and `compose_workflow_template`; long-running media tools still use hosted chat/runs or `sogni.workflows`.
+
 PER-REQUEST MEDIA CONTEXT:
 Generated images, videos, and audio are addressable by stable indices across tool rounds. Every hosted Sogni tool result returns `startIndex` and `indices` in its envelope so a later tool can target the result with `sourceImageIndex`, `videoIndices`, `audioIndex`, etc., without the model having to copy URLs back into prompt text.
 
@@ -1243,6 +1279,8 @@ PLATFORM TOOL WORKFLOW:
 See `workflow_text_chat_sogni_tools.mjs` for a composition-pipeline example
 covering the core image/video/music flows. See `workflow_creative_agent_tools.mjs`
 for server-side hosted Sogni tool injection with `sogni_tools`.
+See `workflow_direct_creative_tool.mjs` for known synchronous tool calls through
+`sogni.chat.hosted.executeTool()`.
 Pass `SogniTools.all` (or individual tool defs) to the LLM via `tools` and
 execute calls through `sogni.chat.hosted.create()` / `sogni.chat.runs.create()`.
 
@@ -1275,6 +1313,9 @@ node workflow_text_chat_sogni_tools.mjs -n 4 "Create images of fantasy landscape
 # Server-side hosted creative tools
 node workflow_creative_agent_tools.mjs "Create an orbit video plan for a crystal perfume bottle"
 
+# Direct synchronous hosted tool execution
+node workflow_direct_creative_tool.mjs --tool compose_workflow "Plan a 3-shot neon bakery teaser"
+
 # Durable creative-agent workflow
 node workflow_creative_agent_workflows.mjs "A chrome monorail over neon gardens" --watch
 

package/llms.txt

@@ -140,6 +140,10 @@ const portal = await sogni.account.createSubscriptionPortalSession();
 
 Checkout and portal sessions require user authentication; API-key auth is rejected for those redirect operations.
 
+AGENT ATTRIBUTION:
+
+Keep `appSource` as the stable product/integration ID. Agent wrappers may additionally set immutable `attribution.connection` and `attribution.workload` defaults on `createInstance()`, then override `attribution` per project, socket/hosted/durable chat request, hosted tool, or creative workflow. Exported unions cover `interactionKind`, `workloadKind`, `agentSurface` (including `native_desktop`), `executionMode`, and operation lineage. Standalone attributed calls receive a top-level operation ID from the underlying job/request; child calls should provide `operationScope: 'child'`, `rootOperationId`, and `parentOperationId`. All fields are optional and omitted for legacy callers.
+
 ### Network Types
 
 - `fast` - High-end GPUs, faster, more expensive. **Required for video.**
@@ -205,22 +209,28 @@ The full catalog is dynamic — enumerate it at runtime with
 | `z_image_turbo_bf16` | Z-Image Turbo — ultra-fast 8-step (free-tier eligible) |
 | `z_image_bf16` | Z-Image — high quality 20-50 step |
 | `krea2_turbo_fp8_scaled` | Krea 2 Turbo — few-step, strong in-image text, up to 2K |
+| `krea2_identity_edit_v1_2` | Krea 2 Identity Edit LoRA v1.2 — identity-preserving edits with 1-2 reference images |
 | `chroma-v.46-flash_fp8` | Chroma v.46 Flash — fast, high quality |
 | `chroma-v48-detail-svd_fp8` | Chroma v.48 Detail — high detail |
 | `chroma1-hd_fp8_scaled` | Chroma1-HD — final high-res Chroma (uncensored, LoRA-capable) |
 | `flux1-schnell-fp8` | Flux.1 [schnell] — 1-5 step |
 | `flux1-krea-dev_fp8_scaled` | Flux.1 Krea [dev] — creative detail |
 | `flux2_dev_fp8` | Flux.2 [dev] — pro quality + context images (also image edit) |
+| `dark_beast_krea2_identity_edit_v1_2` | Dark Beast Krea 2 Identity Edit — community identity edit LoRA with 1-2 reference images |
 | `qwen_image_2512_fp8` / `_lightning` | Qwen Image 2512 — text rendering |
 | `gpt-image-2` | GPT Image 2 — OpenAI partner (premium, Spark-only) |
 
 Community fine-tunes (uncensored; pass the id directly):
 `dark_beast_z_image_turbo_v9_bf16` (fast Z-Image fine-tune, 2K needs 24GB VRAM),
 `dark_beast_krea2_fp8` (Krea 2 fine-tune, 2K needs 24GB VRAM),
+`dark_beast_krea2_identity_edit_v1_2` (Krea 2 identity edit LoRA),
 `one_obsession_v22_fp16` (Illustrious/anime, LoRA-capable).
 
 Krea 2 Turbo supports the `krea2-filter-bypass-3` style LoRA via
 `loras: ['krea2-filter-bypass-3']` + `loraStrengths: [1.0]`.
+For identity-preserving edits, use `krea2_identity_edit_v1_2` or
+`dark_beast_krea2_identity_edit_v1_2` with 1-2 `contextImages`, `steps: 10`,
+and `guidance: 1`.
 
 ---
 
@@ -448,7 +458,7 @@ const project = await sogni.projects.create({
 
 ## 7. LLM Text Generation
 
-The Sogni SDK supports LLM text generation via the Supernet. Socket-native chat uses `sogni.chat.completions.create()`. Hosted REST chat uses `sogni.chat.hosted.create()`. Durable server-side chat runs use `sogni.chat.runs`.
+The Sogni SDK supports LLM text generation via the Supernet. Socket-native chat uses `sogni.chat.completions.create()`. Hosted REST chat uses `sogni.chat.hosted.create()`. Direct synchronous hosted tool execution uses `sogni.chat.hosted.executeTool()`. Durable server-side chat runs use `sogni.chat.runs`.
 
 ### Chat Completion (Non-Streaming)
 
@@ -570,9 +580,11 @@ Canonical hosted creative-tool surface (`SogniTools.all`, 24 tools, executed ser
 
 The default `creative-tools` surface includes the tools above. `sogni_tools: "creative-agent"` adds workflow control and asset-manifest tools. Per-request media context lets generated images / videos / audio be addressed by stable indices across tool rounds (each tool result envelope returns `startIndex` and `indices`).
 
+When the caller already knows the exact synchronous composition/planning tool and JSON arguments, use `sogni.chat.hosted.executeTool()` instead of forcing a `/v1/chat/completions` tool-choice round. Direct execution supports `enhance_prompt`, `compose_script`, `compose_lyrics`, `compose_instrumental`, `compose_workflow`, and `compose_workflow_template`; long-running media tools still use hosted chat/runs or `sogni.workflows`.
+
 Media-conditioned tools accept explicit inline base64 `data:` URIs such as `source_image_url`, `reference_image_url`, `reference_audio_url`, and `reference_video_url`. Tool image inputs accept PNG/JPEG, tool audio inputs accept MP3/M4A/WAV, and tool video inputs accept MP4 or MOV/QuickTime. For uploaded or Sogni-hosted HTTP(S) assets, pass media through request-level `mediaReferences` / `mediaContext` or through `sogni.workflows` dependencies.
 
-The `workflow_text_chat_sogni_tools.mjs` example demonstrates the core image/video/music composition flows. `workflow_creative_agent_tools.mjs` demonstrates server-side hosted Sogni tool injection through `/v1/chat/completions`. Pass `SogniTools.all` (or individual tool defs) to the LLM and execute calls through `sogni.chat.hosted.create()` / `sogni.chat.runs.create()`.
+The `workflow_text_chat_sogni_tools.mjs` example demonstrates the core image/video/music composition flows. `workflow_creative_agent_tools.mjs` demonstrates server-side hosted Sogni tool injection through `/v1/chat/completions`. `workflow_direct_creative_tool.mjs` demonstrates `sogni.chat.hosted.executeTool()` for known synchronous tool calls. Pass `SogniTools.all` (or individual tool defs) to the LLM and execute calls through `sogni.chat.hosted.create()` / `sogni.chat.runs.create()`.
 
 Durable creative workflows: the SDK exposes persistent multi-step workflows through `sogni.workflows` (`start`, `list`, `get`, `events`, `streamEvents`, `resume`, `reseed`, `cancel`) backed by `/v1/creative-agent/workflows`. `start()` accepts either an inline `input` plan or a saved `workflowId` + `inputs`. Saved workflow templates are managed via `sogni.workflows.templates` (`list`, `get`, `create`, `update`, `delete`, `fork`). All endpoints use the SDK's active auth. See `workflow_creative_agent_workflows.mjs` for start/list/get/events/stream/cancel coverage.
 

package/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "5.1.0",
+  "version": "5.3.0",
   "engines": {
     "node": ">=22"
   },
@@ -61,6 +61,7 @@
     "prepublishOnly": "npm run build",
     "build": "npm run clean && npm run codegen && tsc --project tsconfig.json && tsc --project tsconfig.esm.json && node scripts/write-dist-esm-package-json.mjs",
     "test:chat-routing": "npm run build && node scripts/check-chat-model-routing.cjs",
+    "test:attribution": "npm run build && node scripts/check-attribution-transport.cjs",
     "test:subscription-api": "npm run build && node scripts/check-subscription-api.cjs",
     "test:workflow-template-envelope": "npm run build && node scripts/check-workflow-template-envelope.cjs",
     "watch": "npm run clean && npm run codegen && tsc --watch --project tsconfig.json",

package/src/ApiClient/WebSocketClient/BrowserWebSocketClient/index.ts

@@ -14,6 +14,7 @@ import type {
   SocketEventSubscriptionInput,
   SocketEventSubscriptions
 } from '../eventSubscriptions.js';
+import type { NormalizedConnectionAttribution } from '../../../lib/attribution.js';
 
 interface SocketSend<T extends MessageType = MessageType> {
   type: 'socket-send';
@@ -98,6 +99,9 @@ class BrowserWebSocketClient extends RestClient<SocketEventMap> implements IWebS
   // a long wait, since the server skips swarmModels for artists that have
   // active jobs. Cached and replayed like the balance snapshot.
   private _lastSwarmModels: SocketEventMap['swarmModels'] | null = null;
+  // Last swarmLLMModels the primary observed. ChatApi builds `models` from
+  // this live event, so replay it to late-joining tabs just like swarmModels.
+  private _lastSwarmLLMModels: SocketEventMap['swarmLLMModels'] | null = null;
   // Last subscription entitlement snapshot observed on the primary tab. Replayed
   // to late-joining secondaries so CurrentAccount reflects purchases made after
   // the shared socket originally connected.
@@ -111,7 +115,8 @@ class BrowserWebSocketClient extends RestClient<SocketEventMap> implements IWebS
     supernetType: SupernetType,
     logger: Logger,
     appSource?: string,
-    socketEventSubscriptions?: SocketEventSubscriptions
+    socketEventSubscriptions?: SocketEventSubscriptions,
+    connectionAttribution?: NormalizedConnectionAttribution
   ) {
     const socketClient = new WrappedClient(
       baseUrl,
@@ -120,7 +125,8 @@ class BrowserWebSocketClient extends RestClient<SocketEventMap> implements IWebS
       supernetType,
       logger,
       appSource,
-      socketEventSubscriptions
+      socketEventSubscriptions,
+      connectionAttribution
     );
     super(socketClient.baseUrl, auth, logger);
     this.socketClient = socketClient;
@@ -262,6 +268,12 @@ class BrowserWebSocketClient extends RestClient<SocketEventMap> implements IWebS
               payload: { type: 'swarmModels', data: this._lastSwarmModels }
             });
           }
+          if (this._lastSwarmLLMModels) {
+            this.coordinator.notify({
+              type: 'socket-event',
+              payload: { type: 'swarmLLMModels', data: this._lastSwarmLLMModels }
+            });
+          }
           if (this._lastSubscriptionEntitlement) {
             this.coordinator.notify({
               type: 'socket-event',
@@ -330,6 +342,8 @@ class BrowserWebSocketClient extends RestClient<SocketEventMap> implements IWebS
         this._lastBalanceUpdate = payload;
       } else if (eventType === 'swarmModels') {
         this._lastSwarmModels = payload;
+      } else if (eventType === 'swarmLLMModels') {
+        this._lastSwarmLLMModels = payload;
       } else if (eventType === 'subscriptionEntitlementUpdated') {
         this._lastSubscriptionEntitlement = payload;
       } else if (eventType === 'authenticated' && payload?.subscriptionEntitlement) {

package/src/ApiClient/WebSocketClient/events.ts

@@ -117,6 +117,17 @@ export type JobETAData = {
   etaSeconds: number;
 };
 
+export type JobPreparation = {
+  phase: 'downloadingAssets';
+  assetType: 'lora';
+  requested: number;
+  cached: number;
+  total: number;
+  completed: number;
+  current: number;
+  currentProgress?: number;
+};
+
 export type JobResultData = {
   jobID: string;
   imgID: string;
@@ -145,6 +156,7 @@ export type JobStateData =
       positivePrompt?: string;
       negativePrompt?: string;
       jobIndex?: number;
+      preparation?: JobPreparation;
     }
   | {
       type: 'assigned';

package/src/ApiClient/WebSocketClient/index.ts

@@ -16,6 +16,11 @@ import type {
   SocketEventSubscriptionInput,
   SocketEventSubscriptions
 } from './eventSubscriptions.js';
+import {
+  appendConnectionAttributionQuery,
+  normalizeConnectionAttribution,
+  type NormalizedConnectionAttribution
+} from '../../lib/attribution.js';
 
 const PROTOCOL_VERSION = '3.0.0';
 
@@ -24,6 +29,7 @@ const PING_INTERVAL = 15000;
 class WebSocketClient extends RestClient<SocketEventMap> implements IWebSocketClient {
   appId: string;
   appSource?: string;
+  connectionAttribution?: NormalizedConnectionAttribution;
   baseUrl: string;
   socketEventSubscriptions?: SocketEventSubscriptions;
   private socket: WebSocket | null = null;
@@ -37,7 +43,8 @@ class WebSocketClient extends RestClient<SocketEventMap> implements IWebSocketCl
     supernetType: SupernetType,
     logger: Logger,
     appSource?: string,
-    socketEventSubscriptions?: SocketEventSubscriptions
+    socketEventSubscriptions?: SocketEventSubscriptions,
+    connectionAttribution?: NormalizedConnectionAttribution
   ) {
     const _baseUrl = new URL(baseUrl);
     switch (_baseUrl.protocol) {
@@ -55,6 +62,7 @@ class WebSocketClient extends RestClient<SocketEventMap> implements IWebSocketCl
     super(_baseUrl.toString(), auth, logger);
     this.appId = appId;
     this.appSource = appSource?.trim() || undefined;
+    this.connectionAttribution = normalizeConnectionAttribution(connectionAttribution);
     this.socketEventSubscriptions = socketEventSubscriptions;
     this.baseUrl = _baseUrl.toString();
     this._supernetType = supernetType;
@@ -87,6 +95,7 @@ class WebSocketClient extends RestClient<SocketEventMap> implements IWebSocketCl
     if (this.appSource) {
       url.searchParams.set('appSource', this.appSource);
     }
+    appendConnectionAttributionQuery(url, this.connectionAttribution);
     const socketEventSubscriptions = serializeSocketEventSubscriptions(
       this.socketEventSubscriptions
     );

package/src/ApiClient/index.ts

@@ -16,6 +16,17 @@ import CookieAuthManager from '../lib/AuthManager/CookieAuthManager.js';
 import { AuthManager, TokenAuthManager } from '../lib/AuthManager/index.js';
 import isNodejs from '../lib/isNodejs.js';
 import BrowserWebSocketClient from './WebSocketClient/BrowserWebSocketClient/index.js';
+import {
+  buildSogniAttributionHeaders,
+  normalizeConnectionAttribution,
+  resolveWorkloadAttribution,
+  type NormalizedConnectionAttribution
+} from '../lib/attribution.js';
+import type {
+  SogniAttributionConfig,
+  WorkloadAttributionDefaults,
+  WorkloadAttributionInput
+} from '../types/attribution.js';
 
 const WS_RECONNECT_ATTEMPTS = 5;
 
@@ -46,6 +57,7 @@ export interface ApiClientOptions {
   socketUrl: string;
   appId: string;
   appSource?: string;
+  attribution?: SogniAttributionConfig;
   socketEventSubscriptions?: SocketEventSubscriptions;
   networkType: SupernetType;
   logger: Logger;
@@ -57,6 +69,10 @@ export interface ApiClientOptions {
 class ApiClient extends TypedEventEmitter<ApiClientEvents> {
   readonly appId: string;
   readonly appSource?: string;
+  readonly attribution: Readonly<{
+    connection?: Readonly<NormalizedConnectionAttribution>;
+    workload?: Readonly<WorkloadAttributionDefaults>;
+  }>;
   readonly logger: Logger;
   private _rest: RestClient;
   private _socket: IWebSocketClient;
@@ -69,6 +85,7 @@ class ApiClient extends TypedEventEmitter<ApiClientEvents> {
     socketUrl,
     appId,
     appSource,
+    attribution,
     socketEventSubscriptions,
     networkType,
     authType,
@@ -79,6 +96,11 @@ class ApiClient extends TypedEventEmitter<ApiClientEvents> {
     super();
     this.appId = appId;
     this.appSource = appSource?.trim() || undefined;
+    const connectionAttribution = normalizeConnectionAttribution(attribution?.connection);
+    this.attribution = Object.freeze({
+      ...(connectionAttribution ? { connection: Object.freeze(connectionAttribution) } : {}),
+      ...(attribution?.workload ? { workload: Object.freeze({ ...attribution.workload }) } : {})
+    });
     this.logger = logger;
     if (authType === 'apiKey') {
       this._auth = new ApiKeyAuthManager(logger);
@@ -98,7 +120,8 @@ class ApiClient extends TypedEventEmitter<ApiClientEvents> {
         networkType,
         logger,
         this.appSource,
-        socketEventSubscriptions
+        socketEventSubscriptions,
+        this.attribution.connection
       );
     } else {
       this._socket = new WebSocketClient(
@@ -108,7 +131,8 @@ class ApiClient extends TypedEventEmitter<ApiClientEvents> {
         networkType,
         logger,
         this.appSource,
-        socketEventSubscriptions
+        socketEventSubscriptions,
+        this.attribution.connection
       );
     }
     this._disableSocket = disableSocket;
@@ -137,6 +161,22 @@ class ApiClient extends TypedEventEmitter<ApiClientEvents> {
     return !this._disableSocket;
   }
 
+  resolveWorkloadAttribution(override?: WorkloadAttributionInput, fallbackOperationId?: string) {
+    return resolveWorkloadAttribution(this.attribution.workload, override, fallbackOperationId);
+  }
+
+  attributionHeaders(
+    appSource: string | undefined,
+    override?: WorkloadAttributionInput,
+    fallbackOperationId?: string
+  ): Record<string, string> {
+    return buildSogniAttributionHeaders({
+      appSource,
+      connection: this.attribution.connection,
+      workload: this.resolveWorkloadAttribution(override, fallbackOperationId)
+    });
+  }
+
   setSocketEventSubscriptions(update: SocketEventSubscriptionInput): Promise<void> {
     return this.socket.setSocketEventSubscriptions(update);
   }

package/src/ApiGroup.ts

@@ -1,6 +1,8 @@
 import ApiClient from './ApiClient/index.js';
 import EIP712Helper from './lib/EIP712Helper.js';
 import TypedEventEmitter, { EventMap } from './lib/TypedEventEmitter.js';
+import { buildSogniAttributionHeaders, resolveWorkloadAttribution } from './lib/attribution.js';
+import type { WorkloadAttributionInput } from './types/attribution.js';
 
 export interface ApiConfig {
   client: ApiClient;
@@ -16,6 +18,29 @@ abstract class ApiGroup<E extends EventMap = {}> extends TypedEventEmitter<E> {
     this.client = config.client;
     this.eip712 = config.eip712;
   }
+
+  protected resolveWorkloadAttribution(
+    override?: WorkloadAttributionInput,
+    fallbackOperationId?: string
+  ) {
+    return resolveWorkloadAttribution(
+      this.client.attribution?.workload,
+      override,
+      fallbackOperationId
+    );
+  }
+
+  protected attributionHeaders(
+    appSource: string | undefined,
+    override?: WorkloadAttributionInput,
+    fallbackOperationId?: string
+  ): Record<string, string> {
+    return buildSogniAttributionHeaders({
+      appSource,
+      connection: this.client.attribution?.connection,
+      workload: this.resolveWorkloadAttribution(override, fallbackOperationId)
+    });
+  }
 }
 
 export default ApiGroup;

package/src/Chat/ChatTools.ts

@@ -155,6 +155,7 @@ class ChatToolsApi {
           tokenType: options?.tokenType,
           network: options?.network,
           numberOfMedia: options?.numberOfMedia,
+          attribution: options?.attribution,
           timeout: options?.timeout,
           onProgress: options?.onToolProgress
             ? (progress: ToolExecutionProgress) => options.onToolProgress!(toolCall, progress)
@@ -213,7 +214,10 @@ class ChatToolsApi {
   ): Promise<ToolExecutionResult> {
     options?.onProgress?.({ status: 'creating', percent: 0 });
 
-    const project = await this.projects.create(projectParams as any);
+    const project = await this.projects.create({
+      ...projectParams,
+      ...(options?.attribution ? { attribution: options.attribution } : {})
+    } as any);
     const timeout = options?.timeout ?? DEFAULT_TIMEOUT;
     let timeoutId: ReturnType<typeof setTimeout> | null = null;
     let jobsCompleted = 0;