llama-stack-client 0.7.0 → 0.7.2

package/src/resources/alpha/post-training/post-training.ts

@@ -1,385 +0,0 @@
-// Copyright (c) Meta Platforms, Inc. and affiliates.
-// All rights reserved.
-//
-// This source code is licensed under the terms described in the LICENSE file in
-// the root directory of this source tree.
-//
-// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-import { APIResource } from '../../../resource';
-import * as Core from '../../../core';
-import * as JobAPI from './job';
-import { Job, JobArtifactsResponse, JobListResponse, JobStatusResponse } from './job';
-
-export class PostTraining extends APIResource {
-  job: JobAPI.Job = new JobAPI.Job(this._client);
-
-  /**
-   * Run preference optimization of a model.
-   */
-  preferenceOptimize(
-    body: PostTrainingPreferenceOptimizeParams,
-    options?: Core.RequestOptions,
-  ): Core.APIPromise<PostTrainingJob> {
-    return this._client.post('/v1alpha/post-training/preference-optimize', { body, ...options });
-  }
-
-  /**
-   * Run supervised fine-tuning of a model.
-   */
-  supervisedFineTune(
-    body: PostTrainingSupervisedFineTuneParams,
-    options?: Core.RequestOptions,
-  ): Core.APIPromise<PostTrainingJob> {
-    return this._client.post('/v1alpha/post-training/supervised-fine-tune', { body, ...options });
-  }
-}
-
-/**
- * Configuration for Low-Rank Adaptation (LoRA) fine-tuning.
- */
-export type AlgorithmConfig = AlgorithmConfig.LoraFinetuningConfig | AlgorithmConfig.QatFinetuningConfig;
-
-export namespace AlgorithmConfig {
-  /**
-   * Configuration for Low-Rank Adaptation (LoRA) fine-tuning.
-   */
-  export interface LoraFinetuningConfig {
-    alpha: number;
-
-    apply_lora_to_mlp: boolean;
-
-    apply_lora_to_output: boolean;
-
-    lora_attn_modules: Array<string>;
-
-    rank: number;
-
-    quantize_base?: boolean | null;
-
-    type?: 'LoRA';
-
-    use_dora?: boolean | null;
-  }
-
-  /**
-   * Configuration for Quantization-Aware Training (QAT) fine-tuning.
-   */
-  export interface QatFinetuningConfig {
-    group_size: number;
-
-    quantizer_name: string;
-
-    type?: 'QAT';
-  }
-}
-
-export interface ListPostTrainingJobsResponse {
-  data: JobAPI.JobListResponse;
-}
-
-export interface PostTrainingJob {
-  job_uuid: string;
-}
-
-export interface PostTrainingPreferenceOptimizeParams {
-  /**
-   * The algorithm configuration.
-   */
-  algorithm_config: PostTrainingPreferenceOptimizeParams.AlgorithmConfig;
-
-  /**
-   * The model to fine-tune.
-   */
-  finetuned_model: string;
-
-  /**
-   * The hyperparam search configuration.
-   */
-  hyperparam_search_config: { [key: string]: unknown };
-
-  /**
-   * The UUID of the job to create.
-   */
-  job_uuid: string;
-
-  /**
-   * The logger configuration.
-   */
-  logger_config: { [key: string]: unknown };
-
-  /**
-   * The training configuration.
-   */
-  training_config: PostTrainingPreferenceOptimizeParams.TrainingConfig;
-}
-
-export namespace PostTrainingPreferenceOptimizeParams {
-  /**
-   * The algorithm configuration.
-   */
-  export interface AlgorithmConfig {
-    beta: number;
-
-    loss_type?: 'sigmoid' | 'hinge' | 'ipo' | 'kto_pair';
-  }
-
-  /**
-   * The training configuration.
-   */
-  export interface TrainingConfig {
-    n_epochs: number;
-
-    /**
-     * Configuration for training data and data loading.
-     */
-    data_config?: TrainingConfig.DataConfig | null;
-
-    dtype?: string | null;
-
-    /**
-     * Configuration for memory and compute efficiency optimizations.
-     */
-    efficiency_config?: TrainingConfig.EfficiencyConfig | null;
-
-    gradient_accumulation_steps?: number;
-
-    max_steps_per_epoch?: number;
-
-    max_validation_steps?: number | null;
-
-    /**
-     * Configuration parameters for the optimization algorithm.
-     */
-    optimizer_config?: TrainingConfig.OptimizerConfig | null;
-  }
-
-  export namespace TrainingConfig {
-    /**
-     * Configuration for training data and data loading.
-     */
-    export interface DataConfig {
-      batch_size: number;
-
-      /**
-       * Format of the training dataset.
-       */
-      data_format: 'instruct' | 'dialog';
-
-      dataset_id: string;
-
-      shuffle: boolean;
-
-      packed?: boolean | null;
-
-      train_on_input?: boolean | null;
-
-      validation_dataset_id?: string | null;
-    }
-
-    /**
-     * Configuration for memory and compute efficiency optimizations.
-     */
-    export interface EfficiencyConfig {
-      enable_activation_checkpointing?: boolean | null;
-
-      enable_activation_offloading?: boolean | null;
-
-      fsdp_cpu_offload?: boolean | null;
-
-      memory_efficient_fsdp_wrap?: boolean | null;
-    }
-
-    /**
-     * Configuration parameters for the optimization algorithm.
-     */
-    export interface OptimizerConfig {
-      lr: number;
-
-      num_warmup_steps: number;
-
-      /**
-       * Available optimizer algorithms for training.
-       */
-      optimizer_type: 'adam' | 'adamw' | 'sgd';
-
-      weight_decay: number;
-    }
-  }
-}
-
-export interface PostTrainingSupervisedFineTuneParams {
-  /**
-   * The hyperparam search configuration.
-   */
-  hyperparam_search_config: { [key: string]: unknown };
-
-  /**
-   * The UUID of the job to create.
-   */
-  job_uuid: string;
-
-  /**
-   * The logger configuration.
-   */
-  logger_config: { [key: string]: unknown };
-
-  /**
-   * The training configuration.
-   */
-  training_config: PostTrainingSupervisedFineTuneParams.TrainingConfig;
-
-  /**
-   * The algorithm configuration.
-   */
-  algorithm_config?:
-    | PostTrainingSupervisedFineTuneParams.LoraFinetuningConfig
-    | PostTrainingSupervisedFineTuneParams.QatFinetuningConfig
-    | null;
-
-  /**
-   * The directory to save checkpoint(s) to.
-   */
-  checkpoint_dir?: string | null;
-
-  /**
-   * Model descriptor for training if not in provider config
-   */
-  model?: string | null;
-}
-
-export namespace PostTrainingSupervisedFineTuneParams {
-  /**
-   * The training configuration.
-   */
-  export interface TrainingConfig {
-    n_epochs: number;
-
-    /**
-     * Configuration for training data and data loading.
-     */
-    data_config?: TrainingConfig.DataConfig | null;
-
-    dtype?: string | null;
-
-    /**
-     * Configuration for memory and compute efficiency optimizations.
-     */
-    efficiency_config?: TrainingConfig.EfficiencyConfig | null;
-
-    gradient_accumulation_steps?: number;
-
-    max_steps_per_epoch?: number;
-
-    max_validation_steps?: number | null;
-
-    /**
-     * Configuration parameters for the optimization algorithm.
-     */
-    optimizer_config?: TrainingConfig.OptimizerConfig | null;
-  }
-
-  export namespace TrainingConfig {
-    /**
-     * Configuration for training data and data loading.
-     */
-    export interface DataConfig {
-      batch_size: number;
-
-      /**
-       * Format of the training dataset.
-       */
-      data_format: 'instruct' | 'dialog';
-
-      dataset_id: string;
-
-      shuffle: boolean;
-
-      packed?: boolean | null;
-
-      train_on_input?: boolean | null;
-
-      validation_dataset_id?: string | null;
-    }
-
-    /**
-     * Configuration for memory and compute efficiency optimizations.
-     */
-    export interface EfficiencyConfig {
-      enable_activation_checkpointing?: boolean | null;
-
-      enable_activation_offloading?: boolean | null;
-
-      fsdp_cpu_offload?: boolean | null;
-
-      memory_efficient_fsdp_wrap?: boolean | null;
-    }
-
-    /**
-     * Configuration parameters for the optimization algorithm.
-     */
-    export interface OptimizerConfig {
-      lr: number;
-
-      num_warmup_steps: number;
-
-      /**
-       * Available optimizer algorithms for training.
-       */
-      optimizer_type: 'adam' | 'adamw' | 'sgd';
-
-      weight_decay: number;
-    }
-  }
-
-  /**
-   * Configuration for Low-Rank Adaptation (LoRA) fine-tuning.
-   */
-  export interface LoraFinetuningConfig {
-    alpha: number;
-
-    apply_lora_to_mlp: boolean;
-
-    apply_lora_to_output: boolean;
-
-    lora_attn_modules: Array<string>;
-
-    rank: number;
-
-    quantize_base?: boolean | null;
-
-    type?: 'LoRA';
-
-    use_dora?: boolean | null;
-  }
-
-  /**
-   * Configuration for Quantization-Aware Training (QAT) fine-tuning.
-   */
-  export interface QatFinetuningConfig {
-    group_size: number;
-
-    quantizer_name: string;
-
-    type?: 'QAT';
-  }
-}
-
-PostTraining.Job = Job;
-
-export declare namespace PostTraining {
-  export {
-    type AlgorithmConfig as AlgorithmConfig,
-    type ListPostTrainingJobsResponse as ListPostTrainingJobsResponse,
-    type PostTrainingJob as PostTrainingJob,
-    type PostTrainingPreferenceOptimizeParams as PostTrainingPreferenceOptimizeParams,
-    type PostTrainingSupervisedFineTuneParams as PostTrainingSupervisedFineTuneParams,
-  };
-
-  export {
-    Job as Job,
-    type JobListResponse as JobListResponse,
-    type JobArtifactsResponse as JobArtifactsResponse,
-    type JobStatusResponse as JobStatusResponse,
-  };
-}

package/src/resources/tool-runtime.ts

@@ -1,217 +0,0 @@
-// Copyright (c) Meta Platforms, Inc. and affiliates.
-// All rights reserved.
-//
-// This source code is licensed under the terms described in the LICENSE file in
-// the root directory of this source tree.
-//
-// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-import { APIResource } from '../resource';
-import { isRequestOptions } from '../core';
-import * as Core from '../core';
-
-export class ToolRuntime extends APIResource {
-  /**
-   * Run a tool with the given arguments.
-   *
-   * @deprecated
-   */
-  invokeTool(
-    body: ToolRuntimeInvokeToolParams,
-    options?: Core.RequestOptions,
-  ): Core.APIPromise<ToolInvocationResult> {
-    return this._client.post('/v1/tool-runtime/invoke', { body, ...options });
-  }
-
-  /**
-   * List all tools in the runtime.
-   *
-   * @deprecated
-   */
-  listTools(
-    query?: ToolRuntimeListToolsParams,
-    options?: Core.RequestOptions,
-  ): Core.APIPromise<ToolRuntimeListToolsResponse>;
-  listTools(options?: Core.RequestOptions): Core.APIPromise<ToolRuntimeListToolsResponse>;
-  listTools(
-    query: ToolRuntimeListToolsParams | Core.RequestOptions = {},
-    options?: Core.RequestOptions,
-  ): Core.APIPromise<ToolRuntimeListToolsResponse> {
-    if (isRequestOptions(query)) {
-      return this.listTools({}, query);
-    }
-    return (
-      this._client.get('/v1/tool-runtime/list-tools', { query, ...options }) as Core.APIPromise<{
-        data: ToolRuntimeListToolsResponse;
-      }>
-    )._thenUnwrap((obj) => obj.data);
-  }
-}
-
-/**
- * Tool definition used in runtime contexts.
- */
-export interface ToolDef {
-  name: string;
-
-  description?: string | null;
-
-  input_schema?: { [key: string]: unknown } | null;
-
-  metadata?: { [key: string]: unknown } | null;
-
-  output_schema?: { [key: string]: unknown } | null;
-
-  toolgroup_id?: string | null;
-}
-
-/**
- * Result of a tool invocation.
- */
-export interface ToolInvocationResult {
-  /**
-   * A image content item
-   */
-  content?:
-    | string
-    | ToolInvocationResult.ImageContentItemOutput
-    | ToolInvocationResult.TextContentItem
-    | Array<ToolInvocationResult.ImageContentItemOutput | ToolInvocationResult.TextContentItem>
-    | null;
-
-  error_code?: number | null;
-
-  error_message?: string | null;
-
-  metadata?: { [key: string]: unknown } | null;
-}
-
-export namespace ToolInvocationResult {
-  /**
-   * A image content item
-   */
-  export interface ImageContentItemOutput {
-    /**
-     * A URL or a base64 encoded string
-     */
-    image: ImageContentItemOutput.Image;
-
-    type?: 'image';
-  }
-
-  export namespace ImageContentItemOutput {
-    /**
-     * A URL or a base64 encoded string
-     */
-    export interface Image {
-      data?: string | null;
-
-      /**
-       * A URL reference to external content.
-       */
-      url?: Image.URL | null;
-    }
-
-    export namespace Image {
-      /**
-       * A URL reference to external content.
-       */
-      export interface URL {
-        uri: string;
-      }
-    }
-  }
-
-  /**
-   * A text content item
-   */
-  export interface TextContentItem {
-    text: string;
-
-    type?: 'text';
-  }
-
-  /**
-   * A image content item
-   */
-  export interface ImageContentItemOutput {
-    /**
-     * A URL or a base64 encoded string
-     */
-    image: ImageContentItemOutput.Image;
-
-    type?: 'image';
-  }
-
-  export namespace ImageContentItemOutput {
-    /**
-     * A URL or a base64 encoded string
-     */
-    export interface Image {
-      data?: string | null;
-
-      /**
-       * A URL reference to external content.
-       */
-      url?: Image.URL | null;
-    }
-
-    export namespace Image {
-      /**
-       * A URL reference to external content.
-       */
-      export interface URL {
-        uri: string;
-      }
-    }
-  }
-
-  /**
-   * A text content item
-   */
-  export interface TextContentItem {
-    text: string;
-
-    type?: 'text';
-  }
-}
-
-export type ToolRuntimeListToolsResponse = Array<ToolDef>;
-
-export interface ToolRuntimeInvokeToolParams {
-  kwargs: { [key: string]: unknown };
-
-  tool_name: string;
-
-  authorization?: string | null;
-}
-
-export interface ToolRuntimeListToolsParams {
-  authorization?: string | null;
-
-  /**
-   * A URL reference to external content.
-   */
-  mcp_endpoint?: ToolRuntimeListToolsParams.McpEndpoint | null;
-
-  tool_group_id?: string | null;
-}
-
-export namespace ToolRuntimeListToolsParams {
-  /**
-   * A URL reference to external content.
-   */
-  export interface McpEndpoint {
-    uri: string;
-  }
-}
-
-export declare namespace ToolRuntime {
-  export {
-    type ToolDef as ToolDef,
-    type ToolInvocationResult as ToolInvocationResult,
-    type ToolRuntimeListToolsResponse as ToolRuntimeListToolsResponse,
-    type ToolRuntimeInvokeToolParams as ToolRuntimeInvokeToolParams,
-    type ToolRuntimeListToolsParams as ToolRuntimeListToolsParams,
-  };
-}