@civitai/client 0.2.0-beta.10 → 0.2.0-beta.12

package/dist/generated/sdk.gen.d.ts

@@ -5,6 +5,9 @@ import type {
   HeadBlobData,
   HeadBlobResponses,
   HeadBlobErrors,
+  GetBlobContentData,
+  GetBlobContentResponses,
+  GetBlobContentErrors,
   InvokeAgeClassificationStepTemplateData,
   InvokeAgeClassificationStepTemplateResponses,
   InvokeAgeClassificationStepTemplateErrors,
@@ -20,6 +23,9 @@ import type {
   InvokeImageResourceTrainingStepTemplateData,
   InvokeImageResourceTrainingStepTemplateResponses,
   InvokeImageResourceTrainingStepTemplateErrors,
+  InvokeTrainingStepTemplateData,
+  InvokeTrainingStepTemplateResponses,
+  InvokeTrainingStepTemplateErrors,
   InvokeImageUploadStepTemplateData,
   InvokeImageUploadStepTemplateResponses,
   InvokeImageUploadStepTemplateErrors,
@@ -92,7 +98,7 @@ export type Options<
   meta?: Record<string, unknown>;
 };
 /**
- * Get blob by ID. This will return the blob as a binary stream.
+ * Get blob by ID. This will redirect to a cacheable content URL.
  */
 export declare const getBlob: <ThrowOnError extends boolean = false>(
   options: Options<GetBlobData, ThrowOnError>
@@ -103,6 +109,17 @@ export declare const getBlob: <ThrowOnError extends boolean = false>(
 export declare const headBlob: <ThrowOnError extends boolean = false>(
   options: Options<HeadBlobData, ThrowOnError>
 ) => import('./client').RequestResult<HeadBlobResponses, HeadBlobErrors, ThrowOnError, 'fields'>;
+/**
+ * Serves cacheable blob content using a deterministic encrypted token
+ */
+export declare const getBlobContent: <ThrowOnError extends boolean = false>(
+  options: Options<GetBlobContentData, ThrowOnError>
+) => import('./client').RequestResult<
+  GetBlobContentResponses,
+  GetBlobContentErrors,
+  ThrowOnError,
+  'fields'
+>;
 /**
  * Age classification
  * Detects minors in media content. Returns a boolean value indicating whether the content contains minors as well as details on where minors are detected.
@@ -166,6 +183,18 @@ export declare const invokeImageResourceTrainingStepTemplate: <
   ThrowOnError,
   'fields'
 >;
+/**
+ * AI Toolkit Training
+ * Train models using AI Toolkit engine
+ */
+export declare const invokeTrainingStepTemplate: <ThrowOnError extends boolean = false>(
+  options?: Options<InvokeTrainingStepTemplateData, ThrowOnError>
+) => import('./client').RequestResult<
+  InvokeTrainingStepTemplateResponses,
+  InvokeTrainingStepTemplateErrors,
+  ThrowOnError,
+  'fields'
+>;
 /**
  * Image upload
  * Uploads an image to be used in a workflow

package/dist/generated/sdk.gen.js

@@ -1,7 +1,7 @@
 // This file is auto-generated by @hey-api/openapi-ts
 import { client as _heyApiClient } from './client.gen';
 /**
- * Get blob by ID. This will return the blob as a binary stream.
+ * Get blob by ID. This will redirect to a cacheable content URL.
  */
 export const getBlob = (options) => {
     var _a;
@@ -32,6 +32,22 @@ export const headBlob = (options) => {
         ...options,
     });
 };
+/**
+ * Serves cacheable blob content using a deterministic encrypted token
+ */
+export const getBlobContent = (options) => {
+    var _a;
+    return ((_a = options.client) !== null && _a !== void 0 ? _a : _heyApiClient).get({
+        security: [
+            {
+                scheme: 'bearer',
+                type: 'http',
+            },
+        ],
+        url: '/v2/consumer/blobs/content/{encryptedToken}',
+        ...options,
+    });
+};
 /**
  * Age classification
  * Detects minors in media content. Returns a boolean value indicating whether the content contains minors as well as details on where minors are detected.
@@ -138,6 +154,27 @@ export const invokeImageResourceTrainingStepTemplate = (options) => {
         },
     });
 };
+/**
+ * AI Toolkit Training
+ * Train models using AI Toolkit engine
+ */
+export const invokeTrainingStepTemplate = (options) => {
+    var _a;
+    return ((_a = options === null || options === void 0 ? void 0 : options.client) !== null && _a !== void 0 ? _a : _heyApiClient).post({
+        security: [
+            {
+                scheme: 'bearer',
+                type: 'http',
+            },
+        ],
+        url: '/v2/consumer/recipes/training',
+        ...options,
+        headers: {
+            'Content-Type': 'application/json',
+            ...options === null || options === void 0 ? void 0 : options.headers,
+        },
+    });
+};
 /**
  * Image upload
  * Uploads an image to be used in a workflow

package/dist/generated/types.gen.d.ts

@@ -628,6 +628,160 @@ export type ImageResourceTrainingStepTemplate = WorkflowStepTemplate & {
 } & {
   $type: 'imageResourceTraining';
 };
+/**
+ * Input for a training step.
+ */
+export type TrainingInput = {
+  /**
+   * The training engine to use
+   */
+  engine: 'ai-toolkit';
+  /**
+   * The model ecosystem (sd1, sdxl, flux1, sd3, wan)
+   */
+  ecosystem: string;
+  /**
+   * The model URN to train upon
+   */
+  model: string;
+  /**
+   * Model variant (e.g., 'dev', 'large', '2.1')
+   */
+  modelVariant?: string | null;
+  /**
+   * Training data specification
+   */
+  trainingData: {
+    /**
+     * Type of training data (e.g., 'zip')
+     */
+    type: string;
+    /**
+     * URN/URL to training data
+     */
+    sourceUrl: string;
+    /**
+     * Number of training items
+     */
+    count: number;
+  };
+  /**
+   * Sample generation configuration
+   */
+  samples: {
+    /**
+     * Sample generation prompts
+     */
+    prompts: Array<string>;
+  };
+  /**
+   * Number of training epochs
+   */
+  epochs?: number;
+  /**
+   * Number of repeats per image
+   */
+  numRepeats?: number;
+  /**
+   * Training batch size
+   */
+  trainBatchSize?: number | null;
+  /**
+   * Training resolution (512, 1024, etc.)
+   */
+  resolution?: number | null;
+  /**
+   * Learning rate
+   */
+  lr?: number;
+  /**
+   * Text encoder learning rate
+   */
+  textEncoderLr?: number | null;
+  /**
+   * Whether to train text encoder
+   */
+  trainTextEncoder?: boolean;
+  /**
+   * Learning rate scheduler type
+   */
+  lrScheduler?: string;
+  /**
+   * Optimizer type
+   */
+  optimizerType?: string;
+  /**
+   * LoRA network dimension
+   */
+  networkDim?: number;
+  /**
+   * LoRA network alpha
+   */
+  networkAlpha?: number;
+  /**
+   * Noise offset for training
+   */
+  noiseOffset?: number;
+  /**
+   * Min SNR gamma value
+   */
+  minSnrGamma?: number | null;
+  /**
+   * Enable horizontal flip augmentation
+   */
+  flipAugmentation?: boolean;
+  /**
+   * Shuffle tokens during training
+   */
+  shuffleTokens?: boolean;
+  /**
+   * Number of tokens to keep at start
+   */
+  keepTokens?: number;
+};
+export type TrainingOutput = {
+  /**
+   * Training status
+   */
+  status: string;
+  /**
+   * Training epoch results
+   */
+  epochs?: Array<EpochResult> | null;
+  /**
+   * Generated sample images
+   */
+  samples?: Array<{
+    prompt?: string;
+    blob?: Blob;
+  }> | null;
+  modelBlob?: Blob;
+  /**
+   * Estimated time remaining (minutes)
+   */
+  eta?: number | null;
+};
+/**
+ * AI Toolkit Training Step
+ */
+export type TrainingStep = WorkflowStep & {
+  $type: 'training';
+} & {
+  input: TrainingInput;
+  output?: TrainingOutput;
+} & {
+  $type: 'training';
+};
+/**
+ * AI Toolkit Training Step Template
+ */
+export type TrainingStepTemplate = WorkflowStepTemplate & {
+  $type: 'training';
+} & {
+  input: TrainingInput;
+} & {
+  $type: 'training';
+};
 /**
  * Available image transformers.
  */
@@ -1308,7 +1462,7 @@ export type SoraVideoGenInput = VideoGenInput & {
   duration?: number;
   seed?: number | null;
   resolution?: '720p' | '1080p';
-  aspectRatio?: '16:9' | '9:16';
+  aspectRatio?: 'auto' | '16:9' | '9:16';
   usePro?: boolean;
 } & {
   engine: 'sora';
@@ -1563,6 +1717,11 @@ export declare const Veo3AspectRatio: {
   readonly '1:1': '1:1';
 };
 export type Veo3AspectRatio = (typeof Veo3AspectRatio)[keyof typeof Veo3AspectRatio];
+export declare const Veo3Version: {
+  readonly '3_0': '3.0';
+  readonly '3_1': '3.1';
+};
+export type Veo3Version = (typeof Veo3Version)[keyof typeof Veo3Version];
 export type Veo3VideoGenInput = VideoGenInput & {
   engine: 'veo3';
 } & {
@@ -1574,6 +1733,7 @@ export type Veo3VideoGenInput = VideoGenInput & {
   seed?: number | null;
   fastMode?: boolean;
   images?: Array<string>;
+  version?: Veo3Version;
 } & {
   engine: 'veo3';
 };
@@ -2263,6 +2423,14 @@ export type GetBlobErrors = {
    * Unauthorized
    */
   401: ProblemDetails;
+  /**
+   * Forbidden
+   */
+  403: ProblemDetails;
+  /**
+   * Not Found
+   */
+  404: ProblemDetails;
 };
 export type GetBlobError = GetBlobErrors[keyof GetBlobErrors];
 export type HeadBlobData = {
@@ -2294,6 +2462,42 @@ export type HeadBlobResponses = {
   204: void;
 };
 export type HeadBlobResponse = HeadBlobResponses[keyof HeadBlobResponses];
+export type GetBlobContentData = {
+  body?: never;
+  path: {
+    /**
+     * The encrypted token containing blob access parameters
+     */
+    encryptedToken: string;
+  };
+  query?: never;
+  url: '/v2/consumer/blobs/content/{encryptedToken}';
+};
+export type GetBlobContentErrors = {
+  /**
+   * Bad Request
+   */
+  400: ProblemDetails;
+  /**
+   * Unauthorized
+   */
+  401: ProblemDetails;
+  /**
+   * Not Found
+   */
+  404: ProblemDetails;
+  /**
+   * Internal Server Error
+   */
+  500: unknown;
+};
+export type GetBlobContentError = GetBlobContentErrors[keyof GetBlobContentErrors];
+export type GetBlobContentResponses = {
+  /**
+   * OK
+   */
+  200: unknown;
+};
 export type InvokeAgeClassificationStepTemplateData = {
   body?: AgeClassificationInput;
   path?: never;
@@ -2439,6 +2643,35 @@ export type InvokeImageResourceTrainingStepTemplateResponses = {
 };
 export type InvokeImageResourceTrainingStepTemplateResponse =
   InvokeImageResourceTrainingStepTemplateResponses[keyof InvokeImageResourceTrainingStepTemplateResponses];
+export type InvokeTrainingStepTemplateData = {
+  body?: TrainingInput;
+  path?: never;
+  query?: {
+    experimental?: boolean;
+    allowMatureContent?: boolean;
+  };
+  url: '/v2/consumer/recipes/training';
+};
+export type InvokeTrainingStepTemplateErrors = {
+  /**
+   * Bad Request
+   */
+  400: ProblemDetails;
+  /**
+   * Unauthorized
+   */
+  401: ProblemDetails;
+};
+export type InvokeTrainingStepTemplateError =
+  InvokeTrainingStepTemplateErrors[keyof InvokeTrainingStepTemplateErrors];
+export type InvokeTrainingStepTemplateResponses = {
+  /**
+   * OK
+   */
+  200: TrainingOutput;
+};
+export type InvokeTrainingStepTemplateResponse =
+  InvokeTrainingStepTemplateResponses[keyof InvokeTrainingStepTemplateResponses];
 export type InvokeImageUploadStepTemplateData = {
   body?: string;
   path?: never;

package/dist/generated/types.gen.js

@@ -169,6 +169,10 @@ export const Veo3AspectRatio = {
     '9:16': '9:16',
     '1:1': '1:1',
 };
+export const Veo3Version = {
+    '3_0': '3.0',
+    '3_1': '3.1',
+};
 export const ViduVideoGenModel = {
     DEFAULT: 'default',
     Q1: 'q1',

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@civitai/client",
-  "version": "0.2.0-beta.10",
+  "version": "0.2.0-beta.12",
   "description": "Civitai's javascript client for generating ai content",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -54,4 +54,4 @@
   "publishConfig": {
     "access": "public"
   }
-}
+}