@sogni-ai/sogni-client 4.0.0-alpha.9 → 4.0.0

package/src/index.ts

@@ -12,7 +12,27 @@ 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,
+  Sampler,
+  ForgeSampler,
+  ComfySampler,
+  Scheduler,
+  ForgeScheduler,
+  ComfyScheduler,
+  SupportedForgeSamplers,
+  SupportedForgeSchedulers,
+  SupportedComfySamplers,
+  SupportedComfySchedulers,
+  VideoProjectParams,
+  AudioFormat,
+  VideoFormat,
+  VideoOutputFormat,
+  VideoWorkflowType
+} from './Projects/types';
 // Stats API
 import StatsApi from './Stats';
 // Base Types
@@ -22,21 +42,40 @@ 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,
+  ForgeSampler,
+  ComfySampler,
   SupernetType,
   Scheduler,
-  TokenType
+  ForgeScheduler,
+  ComfyScheduler,
+  TokenType,
+  VideoFormat,
+  VideoOutputFormat,
+  VideoProjectParams,
+  VideoWorkflowType
 };
 
-export { ApiError, CurrentAccount, Job, Project };
+export {
+  ApiError,
+  CurrentAccount,
+  Job,
+  Project,
+  SupportedComfySchedulers,
+  SupportedComfySamplers,
+  SupportedForgeSchedulers,
+  SupportedForgeSamplers
+};
 
 export interface SogniClientConfig {
   /**
@@ -55,7 +94,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 +126,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 +180,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 +212,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,32 @@
-import { isSampler, SamplerMap } from '../Projects/types/SamplerParams';
-import { isScheduler, SchedulerMap } from '../Projects/types/SchedulerParams';
+import {
+  isRawForgeSampler,
+  isForgeSampler,
+  SupportedForgeSamplers
+} from '../Projects/types/ForgeSamplerParams';
+import {
+  isForgeScheduler,
+  isRawForgeScheduler,
+  SupportedForgeSchedulers
+} from '../Projects/types/ForgeSchedulerParams';
+import { isComfySampler, SupportedComfySamplers } from '../Projects/types/ComfySamplerParams';
+import { isComfyScheduler, SupportedComfySchedulers } from '../Projects/types/ComfySchedulerParams';
 
 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 +64,101 @@ export function validateNumber(
   return number;
 }
 
-export function validateSampler(value?: string) {
+export function validateForgeSampler(value?: string) {
   if (!value) {
     return null;
   }
-  if (!isSampler(value)) {
-    throw new Error(
-      `Invalid sampler: ${value}. Supported options: ${Object.keys(SamplerMap).join(', ')}`
-    );
+  if (isRawForgeSampler(value)) {
+    return value;
   }
-  return SamplerMap[value];
+  if (isForgeSampler(value)) {
+    return SupportedForgeSamplers[value];
+  }
+  throw new Error(
+    `Invalid sampler: ${value}. Supported options: ${Object.keys(SupportedForgeSamplers).join(', ')}`
+  );
 }
 
-export function validateScheduler(value?: string) {
+export function validateForgeScheduler(value?: string) {
   if (!value) {
     return null;
   }
-  if (!isScheduler(value)) {
-    throw new Error(
-      `Invalid scheduler: ${value}. Supported options: ${Object.keys(SchedulerMap).join(', ')}`
-    );
+  if (isRawForgeScheduler(value)) {
+    return value;
+  }
+  if (isForgeScheduler(value)) {
+    return SupportedForgeSchedulers[value];
+  }
+  throw new Error(
+    `Invalid scheduler: ${value}. Supported options: ${Object.keys(SupportedForgeSchedulers).join(', ')}`
+  );
+}
+
+/**
+ * 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;
+  }
+  const num = Number(value);
+  if (isNaN(num)) {
+    throw new Error(`teacacheThreshold must be a number, got ${value}`);
+  }
+  if (num < 0.0 || num > 1.0) {
+    throw new Error(`teacacheThreshold must be between 0.0 and 1.0 (got ${num})`);
+  }
+  return num;
+}
+
+/**
+ * Validate ComfyUI sampler for video models.
+ * Returns the sampler string directly (no mapping needed).
+ */
+export function validateComfySampler(value?: string): string | undefined {
+  if (!value) {
+    return undefined;
+  }
+  if (isComfySampler(value)) {
+    return SupportedComfySamplers[value];
+  }
+  throw new Error(
+    `Invalid comfySampler: ${value}. Supported options: ${Object.keys(SupportedComfySamplers).join(', ')}`
+  );
+}
+
+/**
+ * Validate ComfyUI scheduler for video models.
+ * Returns the scheduler string directly (no mapping needed).
+ */
+export function validateComfyScheduler(value?: string): string | undefined {
+  if (!value) {
+    return undefined;
+  }
+  if (isComfyScheduler(value)) {
+    return SupportedComfySchedulers[value];
+  }
+  throw new Error(
+    `Invalid comfyScheduler: ${value}. Supported options: ${Object.keys(SupportedComfySchedulers).join(', ')}`
+  );
+}
+
+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 validateSampler(modelId: string, sampler: string) {
+  if (isComfyModel(modelId)) {
+    return validateComfySampler(sampler);
+  }
+  return validateForgeSampler(sampler);
+}
+
+export function validateScheduler(modelId: string, scheduler: string) {
+  if (isComfyModel(modelId)) {
+    return validateComfyScheduler(scheduler);
   }
-  return SchedulerMap[value];
+  return validateForgeScheduler(scheduler);
 }

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

@@ -1,12 +0,0 @@
-export declare const SamplerMap: {
-    dfs_sd3: string;
-    dpm_pp: string;
-    dpm_pp_sde: string;
-    dpm_pp_2m: string;
-    euler: string;
-    euler_a: string;
-    lcm: string;
-    pndm_plms: 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,gCAAgC;IAChC,KAAK,EAAE,OAAO;IACd,OAAO,EAAE,SAAS;IAClB,eAAe;IACf,GAAG,EAAE,gCAAgC;IACrC,SAAS,EAAE,iCAAiC;IAC5C,iBAAiB;CAClB,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: 'Leading',
-    linear: 'Linear',
-    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,SAAS;IAClB,MAAM,EAAE,QAAQ;IAChB,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/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: 'Leading',
-  linear: 'Linear',
-  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;