@sogni-ai/sogni-client 3.0.0-alpha.2 → 3.0.0-alpha.21

package/src/Projects/types/index.ts

@@ -1,5 +1,6 @@
 import { SupernetType } from '../../ApiClient/WebSocketClient/types';
 import { ControlNetParams } from './ControlNetParams';
+import { TokenType } from '../../types/token';
 
 export interface SupportedModel {
   id: string;
@@ -141,6 +142,11 @@ export interface ProjectParams {
    * ControlNet model parameters
    */
   controlNet?: ControlNetParams;
+  /**
+   * Select which tokens to use for the project.
+   * If not specified, the Sogni token will be used.
+   */
+  tokenType?: TokenType;
 }
 
 export type ImageUrlParams = {
@@ -195,3 +201,5 @@ export interface EstimateRequest {
    */
   height?: number;
 }
+
+export type EnhancementStrength = 'light' | 'medium' | 'heavy';

package/src/Projects/utils.ts

@@ -0,0 +1,12 @@
+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;
+  }
+}

package/src/Stats/types.ts

@@ -4,7 +4,9 @@ export type LeaderboardType =
   | 'renderSecCompleteWorker'
   | 'renderSecCompleteArtist'
   | 'renderTokenCompleteWorker'
+  | 'renderTokenCompleteWorker2'
   | 'renderTokenCompleteArtist'
+  | 'renderTokenCompleteArtist2'
   | 'jobCompleteWorker'
   | 'jobCompleteArtist'
   | 'projectCompleteArtist'

package/src/index.ts

@@ -18,6 +18,7 @@ import { AvailableModel, ProjectParams, Scheduler, TimeStepSpacing } from './Pro
 import StatsApi from './Stats';
 // Base Types
 import ErrorData from './types/ErrorData';
+import { TokenType } from './types/token';
 
 export type {
   AvailableModel,
@@ -29,7 +30,8 @@ export type {
   ProjectStatus,
   Scheduler,
   SupernetType,
-  TimeStepSpacing
+  TimeStepSpacing,
+  TokenType
 };
 
 export { ApiError, CurrentAccount, Job, Project };

package/src/types/token.ts

@@ -0,0 +1 @@
+export type TokenType = 'sogni' | 'spark';

package/src/version.ts

@@ -1 +1,2 @@
-export const LIB_VERSION = "1.0.0-alpha.4";
+import { version } from '../package.json';
+export const LIB_VERSION = version;