@sogni-ai/sogni-client 0.4.0-aplha.1 → 0.4.0-aplha.11

package/src/Projects/types/RawProject.ts

@@ -0,0 +1,121 @@
+import { SupernetType } from '../../ApiClient/WebSocketClient/types';
+
+export interface RawProject {
+  id: string;
+  SID: number;
+  artist: Account;
+  model: ProjectModel;
+  imageCount: number;
+  stepCount: number;
+  previewCount: number;
+  hasGuideImage: boolean;
+  denoiseStrength: string;
+  costEstimate: CostEstimate;
+  costActual: CostActual;
+  createTime: number;
+  updateTime: number;
+  endTime: number;
+  status: RawProjectStatus;
+  reason: 'allJobsCompleted' | 'artistCanceled';
+  network: SupernetType;
+  txId: string;
+  workerJobs: RawJob[];
+  completedWorkerJobs: RawJob[];
+}
+
+type RawProjectStatus =
+  | 'pending'
+  | 'active'
+  | 'assigned'
+  | 'progress'
+  | 'errored'
+  | 'completed'
+  | 'cancelled';
+
+export interface Account {
+  id?: string;
+  clientSID?: number;
+  address?: string;
+  addressSID?: number;
+  name?: string;
+  username?: string;
+}
+
+export interface RawJob {
+  id: string;
+  SID: string;
+  imgID?: string;
+  worker: Account;
+  createTime: number;
+  startTime: number | null;
+  updateTime: number;
+  endTime: number;
+  status: WorkerJobStatus;
+  reason: WorkerJobReason;
+  performedSteps: number;
+  triggeredNSFWFilter: boolean;
+  seedUsed: number;
+  costActual: CostActual;
+  network: SupernetType;
+  txId?: string;
+}
+
+export interface CostActual {
+  costInRenderSec: string;
+  costInUSD: string;
+  costInToken: string;
+  calculatedStepCount?: number;
+}
+
+export type WorkerJobReason =
+  | 'artistCanceled'
+  | 'artistDisconnected'
+  | 'genfailure'
+  | 'imgUploadFailure'
+  | 'jobCompleted'
+  | 'jobTimedOut'
+  | 'workerDisconnected'
+  | 'workerReconnected';
+
+export type WorkerJobStatus =
+  | 'created'
+  | 'queued'
+  | 'assigned'
+  | 'initiatingModel'
+  | 'jobStarted'
+  | 'jobProgress'
+  | 'jobCompleted'
+  | 'jobError';
+
+export interface CostEstimate {
+  rate: Rate;
+  quote: Quote;
+}
+
+export interface Quote {
+  model: QuoteModel;
+  job: CostActual;
+  project: CostActual;
+}
+
+export interface QuoteModel {
+  weight: string;
+  secPerStep: string;
+  secPerPreview: string;
+  secForCN: string;
+}
+
+export interface Rate {
+  costPerBaseHQRenderInUSD: string;
+  tokenMarkePriceUSD: string;
+  costPerRenderSecUSD: string;
+  costPerRenderSecToken: string;
+  network: SupernetType;
+  networkCostMultiplier: string;
+}
+
+export interface ProjectModel {
+  id: string;
+  SID: number;
+  name: string;
+}

package/src/Projects/types/index.ts

@@ -68,6 +68,10 @@ export interface ProjectParams {
    * Guidance scale. For most Stable Diffusion models, optimal value is 7.5
    */
   guidance: number;
+  /**
+   * Override current network type. Default value can be read from `client.account.currentAccount.network`
+   */
+  network?: SupernetType;
   /**
    * Disable NSFW filter for Project. Default is false, meaning NSFW filter is enabled.
    * If image triggers NSFW filter, it will not be available for download.

package/src/lib/DataEntity.ts

@@ -11,6 +11,8 @@ export interface EntityEvents {
 abstract class DataEntity<D, E extends EntityEvents = EntityEvents> extends TypedEventEmitter<E> {
   protected data: D;
 
+  protected lastUpdated: Date = new Date();
+
   constructor(data: D) {
     super();
     this.data = data;
@@ -24,6 +26,7 @@ abstract class DataEntity<D, E extends EntityEvents = EntityEvents> extends Type
     //@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);
   }
 

package/src/lib/base64.ts

@@ -1,9 +1,13 @@
-import isNodejs from './isNodejs';
-
 export function base64Encode(str: string): string {
-  return isNodejs ? Buffer.from(str).toString('base64') : btoa(str);
+  const encoder = new TextEncoder();
+  const uint8Array = encoder.encode(str);
+  const binaryString = String.fromCharCode(...uint8Array);
+  return btoa(binaryString);
 }
 
 export function base64Decode(str: string): string {
-  return isNodejs ? Buffer.from(str, 'base64').toString() : atob(str);
+  const binaryString = atob(str);
+  const binaryArray = Uint8Array.from(binaryString, (char) => char.charCodeAt(0));
+  const decoder = new TextDecoder();
+  return decoder.decode(binaryArray);
 }

package/src/types/ErrorData.ts

@@ -1,5 +1,6 @@
 interface ErrorData {
   code: number;
+  originalCode?: string;
   message: string;
 }
 

package/src/version.ts

@@ -1 +1 @@
-export const LIB_VERSION = "0.4.0-aplha.1";
+export const LIB_VERSION = "0.4.0-aplha.11";