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

package/src/Projects/utils/scheduler.ts

@@ -0,0 +1,27 @@
+/**
+ * Maps display name formats (from API) to lowercase aliases.
+ * Used to normalize the allowed scheduler list so users can use consistent aliases.
+ */
+const SchedulerDisplayToAlias: Record<string, string> = {
+  Simple: 'simple',
+  Normal: 'normal',
+  Karras: 'karras',
+  Exponential: 'exponential',
+  'SGM Uniform': 'sgm_uniform',
+  'DDIM Uniform': 'ddim_uniform',
+  Beta: 'beta',
+  'Linear Quadratic': 'linear_quadratic',
+  'KL Optimal': 'kl_optimal',
+  DDIM: 'ddim',
+  Leading: 'leading',
+  Linear: 'linear'
+};
+
+/**
+ * Convert a scheduler display name to its alias.
+ * If already an alias or unknown, returns unchanged.
+ * Used to normalize API tier data for consistent validation.
+ */
+export function schedulerValueToAlias(value: string): string {
+  return SchedulerDisplayToAlias[value] || value;
+}

package/src/index.ts

@@ -12,7 +12,17 @@ import EIP712Helper from './lib/EIP712Helper';
 import ProjectsApi from './Projects';
 import Job, { JobStatus } from './Projects/Job';
 import Project, { ProjectStatus } from './Projects/Project';
-import { AvailableModel, OutputFormat, ProjectParams, Sampler, Scheduler } from './Projects/types';
+import {
+  AvailableModel,
+  ImageProjectParams,
+  ImageOutputFormat,
+  ProjectParams,
+  VideoProjectParams,
+  AudioFormat,
+  VideoFormat,
+  VideoOutputFormat,
+  VideoWorkflowType
+} from './Projects/types';
 // Stats API
 import StatsApi from './Stats';
 // Base Types
@@ -22,18 +32,22 @@ import { CookieAuthManager, TokenAuthData, TokenAuthManager } from './lib/AuthMa
 import { MeData } from './Account/types';
 
 export type {
+  AudioFormat,
   AvailableModel,
   ErrorData,
+  ImageProjectParams,
+  ImageOutputFormat,
   JobStatus,
   Logger,
   LogLevel,
-  OutputFormat,
   ProjectParams,
   ProjectStatus,
-  Sampler,
   SupernetType,
-  Scheduler,
-  TokenType
+  TokenType,
+  VideoFormat,
+  VideoOutputFormat,
+  VideoProjectParams,
+  VideoWorkflowType
 };
 
 export { ApiError, CurrentAccount, Job, Project };
@@ -55,7 +69,7 @@ export interface SogniClientConfig {
   socketEndpoint?: string;
   /**
    * Disable WebSocket connection. Useful for testing or when WebSocket is not needed.
-   * Note that many may not work without WebSocket connection.
+   * Note that many APIs may not work without WebSocket connection.
    * @experimental
    * @internal
    */
@@ -87,6 +101,14 @@ export interface SogniClientConfig {
    * @experimental
    */
   authType?: 'token' | 'cookies';
+  /**
+   * Browser only. If true, the client will use a single WebSocket connection shared across multiple
+   * tabs. This is useful for browser apps that need to process multiple projects at the same time.
+   * Only works in browser environment and with cookie authentication.
+   * @default false
+   * @experimental
+   */
+  multiInstance?: boolean;
 }
 
 export class SogniClient {
@@ -133,9 +155,13 @@ export class SogniClient {
       throw Error('This method should only be called when using cookie auth');
     }
     try {
-      await this.apiClient.rest.get<ApiResponse<MeData>>('/v1/account/me');
+      const res = await this.apiClient.rest.get<ApiResponse<MeData>>('/v1/account/me');
       await auth.authenticate();
-      await this.account.me();
+      this.currentAccount._update({
+        username: res.data.username,
+        email: res.data.currentEmail,
+        walletAddress: res.data.walletAddress
+      });
       return true;
     } catch (e) {
       this.apiClient.logger.info('Client is not authenticated');
@@ -161,7 +187,8 @@ export class SogniClient {
       networkType: network,
       logger,
       authType: config.authType || 'token',
-      disableSocket: config.disableSocket
+      disableSocket: config.disableSocket,
+      multiInstance: config.multiInstance
     });
     const eip712 = new EIP712Helper({
       name: isTestnet ? 'Sogni-testnet' : 'Sogni AI',

package/src/lib/AuthManager/TokenAuthManager.ts

@@ -46,7 +46,6 @@ class TokenAuthManager extends AuthManagerBase<TokenAuthData | null> {
       const { expiresAt } = decodeToken(token);
       if (expiresAt > new Date()) {
         this._updateTokens({ token, refreshToken });
-        this.emit('updated', true);
         return;
       }
     }
@@ -55,7 +54,6 @@ class TokenAuthManager extends AuthManagerBase<TokenAuthData | null> {
     const { expiresAt: refreshExpiresAt } = decodeRefreshToken(refreshToken);
     this._refreshTokenExpiresAt = refreshExpiresAt;
     await this._renewTokenSafe();
-    this.emit('updated', true);
   }
 
   clear() {

package/src/lib/DataEntity.ts

@@ -25,9 +25,11 @@ abstract class DataEntity<D, E extends EntityEvents = EntityEvents> extends Type
   _update(delta: Partial<D>) {
     //@ts-ignore
     const changedKeys = Object.keys(delta).filter((key) => this.data[key] !== delta[key]);
-    this.data = { ...this.data, ...delta };
     this.lastUpdated = new Date();
-    this.emit('updated', changedKeys);
+    if (changedKeys.length > 0) {
+      this.data = { ...this.data, ...delta };
+      this.emit('updated', changedKeys);
+    }
   }
 
   /**

package/src/lib/RestClient.ts

@@ -30,7 +30,21 @@ class RestClient<E extends EventMap = never> extends TypedEventEmitter<E> {
 
   private async request<T = JSONValue>(url: string, options: RequestInit = {}): Promise<T> {
     const init = await this.auth.authenticateRequest(options);
-    return fetch(url, init).then((r) => this.processResponse(r) as T);
+
+    // Add a timeout to detect hanging requests
+    const controller = new AbortController();
+    const timeoutId = setTimeout(() => {
+      controller.abort();
+    }, 30000);
+
+    try {
+      const response = await fetch(url, { ...init, signal: controller.signal });
+      clearTimeout(timeoutId);
+      return this.processResponse(response) as T;
+    } catch (fetchError: any) {
+      clearTimeout(timeoutId);
+      throw fetchError;
+    }
   }
 
   private async processResponse(response: Response): Promise<JSONValue> {
@@ -52,7 +66,7 @@ class RestClient<E extends EventMap = never> extends TypedEventEmitter<E> {
   }
 
   get<T = JSONValue>(path: string, query: Record<string, any> = {}): Promise<T> {
-    return this.request<T>(this.formatUrl(path, query), query);
+    return this.request<T>(this.formatUrl(path, query));
   }
 
   post<T = JSONValue>(path: string, body: Record<string, unknown> = {}): Promise<T> {

package/src/lib/validation.ts

@@ -1,10 +1,22 @@
-import { isSampler, SamplerMap } from '../Projects/types/SamplerParams';
-import { isScheduler, SchedulerMap } from '../Projects/types/SchedulerParams';
+import { ApiError } from '../ApiClient';
+import { ModelOptions } from '../Projects/types/ModelOptions';
 
 export function validateCustomImageSize(value: any): number {
   return validateNumber(value, { min: 256, max: 2048, propertyName: 'Width and height' });
 }
 
+/**
+ * Validate video dimensions for Wan 2.2 models.
+ * Minimum dimension is 480px for both width and height.
+ */
+export function validateVideoSize(value: any, propertyName: 'width' | 'height'): number {
+  return validateNumber(value, { min: 480, propertyName: `Video ${propertyName}` });
+}
+
+export function validateVideoDuration(value: any): number {
+  return validateNumber(value, { min: 1, max: 10, propertyName: 'Video duration' });
+}
+
 interface NumberValidationOptions {
   min?: number;
   max?: number;
@@ -42,26 +54,87 @@ export function validateNumber(
   return number;
 }
 
-export function validateSampler(value?: string) {
-  if (!value) {
-    return null;
+/**
+ * Validate teacacheThreshold for T2V and I2V models.
+ * Range: 0.0-1.0. 0.0 = disabled.
+ */
+export function validateTeacacheThreshold(value?: number): number | undefined {
+  if (value === undefined || value === null) {
+    return undefined;
   }
-  if (!isSampler(value)) {
-    throw new Error(
-      `Invalid sampler: ${value}. Supported options: ${Object.keys(SamplerMap).join(', ')}`
-    );
+  const num = Number(value);
+  if (isNaN(num)) {
+    throw new Error(`teacacheThreshold must be a number, got ${value}`);
   }
-  return SamplerMap[value];
+  if (num < 0.0 || num > 1.0) {
+    throw new Error(`teacacheThreshold must be between 0.0 and 1.0 (got ${num})`);
+  }
+  return num;
+}
+
+export function isComfyModel(modelId: string): boolean {
+  const COMFY_PREFIXES = ['z_image_', 'qwen_image_', 'flux2_', 'wan_'];
+  return COMFY_PREFIXES.some((prefix) => modelId.startsWith(prefix));
 }
 
-export function validateScheduler(value?: string) {
-  if (!value) {
+/**
+ * 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
+ */
+export function getMaxContextImages(modelId: string): number {
+  if (modelId.startsWith('flux2_')) {
+    return 6;
+  }
+  if (modelId.startsWith('qwen_image_')) {
+    return 3;
+  }
+  if (modelId.includes('kontext')) {
+    return 2;
+  }
+  // Default fallback for other models that might support context images
+  return 3;
+}
+
+function validateOption<T = unknown>(value: T, options: T[], errorMessage: string): T {
+  if (!options.includes(value)) {
+    throw new ApiError(400, {
+      status: 'error',
+      errorCode: 0,
+      message: errorMessage
+    });
+  }
+  return value;
+}
+
+/**
+ * Validate sampler value against allowed options.
+ * Returns the validated value unchanged - sogni-socket handles normalization.
+ */
+export function validateSampler(value: string | undefined, options: ModelOptions) {
+  if (!options.sampler.allowed.length || !value) {
     return null;
   }
-  if (!isScheduler(value)) {
-    throw new Error(
-      `Invalid scheduler: ${value}. Supported options: ${Object.keys(SchedulerMap).join(', ')}`
-    );
+  return validateOption(
+    value,
+    options.sampler.allowed,
+    `Invalid sampler ${value}. Must be one of "${options.sampler.allowed.join('", "')}".`
+  );
+}
+
+/**
+ * Validate scheduler value against allowed options.
+ * Returns the validated value unchanged - sogni-socket handles normalization.
+ */
+export function validateScheduler(value: string | undefined, options: ModelOptions) {
+  if (!options.scheduler.allowed.length || !value) {
+    return null;
   }
-  return SchedulerMap[value];
+  return validateOption(
+    value,
+    options.scheduler.allowed,
+    `Invalid scheduler ${value}. Must be one of "${options.scheduler.allowed.join('", "')}".`
+  );
 }

package/dist/Projects/types/SamplerParams.d.ts

@@ -1,15 +0,0 @@
-export declare const SamplerMap: {
-    dfs_sd3: string;
-    dpm_pp: string;
-    dpm_pp_sde: string;
-    dpm_pp_2m: string;
-    dpm_pp_2m_sde: string;
-    euler: string;
-    euler_a: string;
-    heun: string;
-    lcm: string;
-    pndm_plms: string;
-    uni_pc: string;
-};
-export declare function isSampler(sampler: string): sampler is Sampler;
-export type Sampler = keyof typeof SamplerMap;

package/dist/Projects/types/SamplerParams.js

@@ -1,21 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.SamplerMap = void 0;
-exports.isSampler = isSampler;
-exports.SamplerMap = {
-    dfs_sd3: 'Discrete Flow Scheduler (SD3)',
-    dpm_pp: 'DPM Solver Multistep (DPM-Solver++)',
-    dpm_pp_sde: 'DPM++ SDE',
-    dpm_pp_2m: 'DPM++ 2M',
-    dpm_pp_2m_sde: 'DPM++ 2M SDE',
-    euler: 'Euler',
-    euler_a: 'Euler a',
-    heun: 'Heun',
-    lcm: 'LCM (Latent Consistency Model)',
-    pndm_plms: 'PNDM (Pseudo-linear multi-step)',
-    uni_pc: 'UniPC'
-};
-function isSampler(sampler) {
-    return sampler in exports.SamplerMap;
-}
-//# sourceMappingURL=SamplerParams.js.map

package/dist/Projects/types/SamplerParams.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"SamplerParams.js","sourceRoot":"","sources":["../../../src/Projects/types/SamplerParams.ts"],"names":[],"mappings":";;;AAcA,8BAEC;AAhBY,QAAA,UAAU,GAAG;IACxB,OAAO,EAAE,+BAA+B;IACxC,MAAM,EAAE,qCAAqC;IAC7C,UAAU,EAAE,WAAW;IACvB,SAAS,EAAE,UAAU;IACrB,aAAa,EAAE,cAAc;IAC7B,KAAK,EAAE,OAAO;IACd,OAAO,EAAE,SAAS;IAClB,IAAI,EAAE,MAAM;IACZ,GAAG,EAAE,gCAAgC;IACrC,SAAS,EAAE,iCAAiC;IAC5C,MAAM,EAAE,OAAO;CAChB,CAAC;AAEF,SAAgB,SAAS,CAAC,OAAe;IACvC,OAAO,OAAO,IAAI,kBAAU,CAAC;AAC/B,CAAC"}

package/dist/Projects/types/SchedulerParams.d.ts

@@ -1,13 +0,0 @@
-export declare const SchedulerMap: {
-    beta: string;
-    ddim: string;
-    karras: string;
-    kl_optimal: string;
-    leading: string;
-    linear: string;
-    normal: string;
-    sgm_uniform: string;
-    simple: string;
-};
-export declare function isScheduler(scheduler: string): scheduler is Scheduler;
-export type Scheduler = keyof typeof SchedulerMap;

package/dist/Projects/types/SchedulerParams.js

@@ -1,19 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.SchedulerMap = void 0;
-exports.isScheduler = isScheduler;
-exports.SchedulerMap = {
-    beta: 'Beta',
-    ddim: 'DDIM',
-    karras: 'Karras',
-    kl_optimal: 'KL Optimal',
-    leading: 'Automatic',
-    linear: 'Automatic',
-    normal: 'Normal',
-    sgm_uniform: 'SGM Uniform',
-    simple: 'Simple'
-};
-function isScheduler(scheduler) {
-    return scheduler in exports.SchedulerMap;
-}
-//# sourceMappingURL=SchedulerParams.js.map

package/dist/Projects/types/SchedulerParams.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"SchedulerParams.js","sourceRoot":"","sources":["../../../src/Projects/types/SchedulerParams.ts"],"names":[],"mappings":";;;AAYA,kCAEC;AAdY,QAAA,YAAY,GAAG;IAC1B,IAAI,EAAE,MAAM;IACZ,IAAI,EAAE,MAAM;IACZ,MAAM,EAAE,QAAQ;IAChB,UAAU,EAAE,YAAY;IACxB,OAAO,EAAE,WAAW;IACpB,MAAM,EAAE,WAAW;IACnB,MAAM,EAAE,QAAQ;IAChB,WAAW,EAAE,aAAa;IAC1B,MAAM,EAAE,QAAQ;CACjB,CAAC;AAEF,SAAgB,WAAW,CAAC,SAAiB;IAC3C,OAAO,SAAS,IAAI,oBAAY,CAAC;AACnC,CAAC"}

package/dist/Projects/utils.d.ts

@@ -1,2 +0,0 @@
-import { EnhancementStrength } from './types';
-export declare function getEnhacementStrength(strength: EnhancementStrength): number;

package/dist/Projects/utils.js

@@ -1,14 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.getEnhacementStrength = getEnhacementStrength;
-function getEnhacementStrength(strength) {
-    switch (strength) {
-        case 'light':
-            return 0.15;
-        case 'heavy':
-            return 0.49;
-        default:
-            return 0.35;
-    }
-}
-//# sourceMappingURL=utils.js.map

package/dist/Projects/utils.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/Projects/utils.ts"],"names":[],"mappings":";;AAEA,sDASC;AATD,SAAgB,qBAAqB,CAAC,QAA6B;IACjE,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,OAAO;YACV,OAAO,IAAI,CAAC;QACd,KAAK,OAAO;YACV,OAAO,IAAI,CAAC;QACd;YACE,OAAO,IAAI,CAAC;IAChB,CAAC;AACH,CAAC"}

package/dist/lib/utils.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/lib/utils.ts"],"names":[],"mappings":";;;;;;;;;;;AAEA,kCAMC;AAED,gDAMC;AAED,sBAEC;AApBD,2CAAuC;AAEvC,SAAgB,WAAW,CAAC,KAAa;IACvC,MAAM,IAAI,GAAG,IAAA,sBAAS,EAA0D,KAAK,CAAC,CAAC;IACvF,OAAO;QACL,aAAa,EAAE,IAAI,CAAC,IAAI;QACxB,SAAS,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC;KACrC,CAAC;AACJ,CAAC;AAED,SAAgB,kBAAkB,CAAC,KAAa;IAC9C,MAAM,IAAI,GAAG,IAAA,sBAAS,EAA0D,KAAK,CAAC,CAAC;IACvF,OAAO;QACL,GAAG,EAAE,IAAI,CAAC,GAAG;QACb,SAAS,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC;KACrC,CAAC;AACJ,CAAC;AAED,SAAsB,KAAK,CAAC,EAAU;;QACpC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC;CAAA"}

package/src/Projects/types/SamplerParams.ts

@@ -1,19 +0,0 @@
-export const SamplerMap = {
-  dfs_sd3: 'Discrete Flow Scheduler (SD3)',
-  dpm_pp: 'DPM Solver Multistep (DPM-Solver++)',
-  dpm_pp_sde: 'DPM++ SDE',
-  dpm_pp_2m: 'DPM++ 2M',
-  dpm_pp_2m_sde: 'DPM++ 2M SDE',
-  euler: 'Euler',
-  euler_a: 'Euler a',
-  heun: 'Heun',
-  lcm: 'LCM (Latent Consistency Model)',
-  pndm_plms: 'PNDM (Pseudo-linear multi-step)',
-  uni_pc: 'UniPC'
-};
-
-export function isSampler(sampler: string): sampler is Sampler {
-  return sampler in SamplerMap;
-}
-
-export type Sampler = keyof typeof SamplerMap;

package/src/Projects/types/SchedulerParams.ts

@@ -1,17 +0,0 @@
-export const SchedulerMap = {
-  beta: 'Beta',
-  ddim: 'DDIM',
-  karras: 'Karras',
-  kl_optimal: 'KL Optimal',
-  leading: 'Automatic',
-  linear: 'Automatic',
-  normal: 'Normal',
-  sgm_uniform: 'SGM Uniform',
-  simple: 'Simple'
-};
-
-export function isScheduler(scheduler: string): scheduler is Scheduler {
-  return scheduler in SchedulerMap;
-}
-
-export type Scheduler = keyof typeof SchedulerMap;

package/src/Projects/utils.ts

@@ -1,12 +0,0 @@
-import { EnhancementStrength } from './types';
-
-export function getEnhacementStrength(strength: EnhancementStrength): number {
-  switch (strength) {
-    case 'light':
-      return 0.15;
-    case 'heavy':
-      return 0.49;
-    default:
-      return 0.35;
-  }
-}