feat(api): OpenAPI spec update via Stainless API (#62)

Author: stainless-app[bot]

.stats.yml

@@ -1,2 +1,2 @@
 configured_endpoints: 15
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/togetherai%2FTogetherAI-0e56a0b2cfbbe66858d33d487ae2bef80a5acdaf99bf16029ef2006d5804b27d.yml
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/togetherai%2FTogetherAI-4365478ddf8533c4aeced4c0be4bde57b5930a13385c7500da6e4eb493b2cbcc.yml

src/core.ts

@@ -994,11 +994,6 @@ const validatePositiveInteger = (name: string, n: unknown): number => {
 
 export const castToError = (err: any): Error => {
   if (err instanceof Error) return err;
-  if (typeof err === 'object' && err !== null) {
-    try {
-      return new Error(JSON.stringify(err));
-    } catch {}
-  }
   return new Error(err);
 };
 

src/error.ts

@@ -49,7 +49,7 @@ export class APIError extends TogetherError {
     headers: Headers | undefined,
   ) {
     if (!status) {
-      return new APIConnectionError({ message, cause: castToError(errorResponse) });
+      return new APIConnectionError({ cause: castToError(errorResponse) });
     }
 
     const error = errorResponse as Record<string, any>;
@@ -101,7 +101,7 @@ export class APIUserAbortError extends APIError {
 export class APIConnectionError extends APIError {
   override readonly status: undefined = undefined;
 
-  constructor({ message, cause }: { message?: string | undefined; cause?: Error | undefined }) {
+  constructor({ message, cause }: { message?: string; cause?: Error | undefined }) {
     super(undefined, undefined, message || 'Connection error.', undefined);
     // in some environments the 'cause' property is already declared
     // @ts-ignore

src/resources/fine-tune.ts

@@ -79,16 +79,6 @@ export interface FineTune {
 
   learning_rate?: number;
 
-  lora?: boolean;
-
-  lora_alpha?: number;
-
-  lora_dropout?: number;
-
-  lora_r?: number;
-
-  lora_trainable_modules?: string;
-
   model?: string;
 
   model_output_name?: string;
@@ -111,6 +101,8 @@ export interface FineTune {
 
   training_file?: string;
 
+  training_type?: FineTune.FullTrainingType | FineTune.LoRaTrainingType;
+
   trainingfile_numlines?: number;
 
   trainingfile_size?: number;
@@ -169,6 +161,22 @@ export namespace FineTune {
 
     wandb_url?: string;
   }
+
+  export interface FullTrainingType {
+    type: 'Full';
+  }
+
+  export interface LoRaTrainingType {
+    lora_alpha: number;
+
+    lora_r: number;
+
+    type: 'Lora';
+
+    lora_dropout?: number;
+
+    lora_trainable_modules?: string;
+  }
 }
 
 export interface FineTuneEvent {
@@ -269,32 +277,6 @@ export interface FineTuneCreateParams {
    */
   learning_rate?: number;
 
-  /**
-   * Whether to enable LoRA training. If not provided, full fine-tuning will be
-   * applied.
-   */
-  lora?: boolean;
-
-  /**
-   * The alpha value for LoRA adapter training.
-   */
-  lora_alpha?: number;
-
-  /**
-   * The dropout probability for Lora layers.
-   */
-  lora_dropout?: number;
-
-  /**
-   * Rank for LoRA adapter weights
-   */
-  lora_r?: number;
-
-  /**
-   * A list of LoRA trainable modules, separated by a comma
-   */
-  lora_trainable_modules?: string;
-
   /**
    * Number of checkpoints to save during fine-tuning
    */
@@ -315,6 +297,8 @@ export interface FineTuneCreateParams {
    */
   suffix?: string;
 
+  training_type?: FineTuneCreateParams.FullTrainingType | FineTuneCreateParams.LoRaTrainingType;
+
   /**
    * File-ID of a validation file uploaded to the Together API
    */
@@ -326,6 +310,24 @@ export interface FineTuneCreateParams {
   wandb_api_key?: string;
 }
 
+export namespace FineTuneCreateParams {
+  export interface FullTrainingType {
+    type: 'Full';
+  }
+
+  export interface LoRaTrainingType {
+    lora_alpha: number;
+
+    lora_r: number;
+
+    type: 'Lora';
+
+    lora_dropout?: number;
+
+    lora_trainable_modules?: string;
+  }
+}
+
 export interface FineTuneDownloadParams {
   /**
    * Fine-tune ID to download. A string that starts with `ft-`.

tests/api-resources/fine-tune.test.ts

@@ -26,15 +26,11 @@ describe('resource fineTune', () => {
       training_file: 'training_file',
       batch_size: 0,
       learning_rate: 0,
-      lora: true,
-      lora_alpha: 0,
-      lora_dropout: 0,
-      lora_r: 0,
-      lora_trainable_modules: 'lora_trainable_modules',
       n_checkpoints: 0,
       n_epochs: 0,
       n_evals: 0,
       suffix: 'suffix',
+      training_type: { type: 'Full' },
       validation_file: 'validation_file',
       wandb_api_key: 'wandb_api_key',
     });