@google/genai 0.10.0 → 0.11.0

package/dist/genai.d.ts

@@ -23,6 +23,17 @@ export declare enum ActivityHandling {
 export declare interface ActivityStart {
 }
 
+/** Optional. Adapter size for tuning. */
+export declare enum AdapterSize {
+    ADAPTER_SIZE_UNSPECIFIED = "ADAPTER_SIZE_UNSPECIFIED",
+    ADAPTER_SIZE_ONE = "ADAPTER_SIZE_ONE",
+    ADAPTER_SIZE_TWO = "ADAPTER_SIZE_TWO",
+    ADAPTER_SIZE_FOUR = "ADAPTER_SIZE_FOUR",
+    ADAPTER_SIZE_EIGHT = "ADAPTER_SIZE_EIGHT",
+    ADAPTER_SIZE_SIXTEEN = "ADAPTER_SIZE_SIXTEEN",
+    ADAPTER_SIZE_THIRTY_TWO = "ADAPTER_SIZE_THIRTY_TWO"
+}
+
 /**
  * The ApiClient class is used to send requests to the Gemini API or Vertex AI
  * endpoints.
@@ -258,7 +269,7 @@ export declare class Caches extends BaseModule {
      * ```ts
      * const contents = ...; // Initialize the content to cache.
      * const response = await ai.caches.create({
-     *   model: 'gemini-1.5-flash',
+     *   model: 'gemini-2.0-flash-001',
      *   config: {
      *    'contents': contents,
      *    'displayName': 'test cache',
@@ -277,7 +288,7 @@ export declare class Caches extends BaseModule {
      *
      * @example
      * ```ts
-     * await ai.caches.get({name: 'gemini-1.5-flash'});
+     * await ai.caches.get({name: '...'}); // The server-generated resource name.
      * ```
      */
     get(params: types.GetCachedContentParameters): Promise<types.CachedContent>;
@@ -289,7 +300,7 @@ export declare class Caches extends BaseModule {
      *
      * @example
      * ```ts
-     * await ai.caches.delete({name: 'gemini-1.5-flash'});
+     * await ai.caches.delete({name: '...'}); // The server-generated resource name.
      * ```
      */
     delete(params: types.DeleteCachedContentParameters): Promise<types.DeleteCachedContentResponse>;
@@ -302,7 +313,7 @@ export declare class Caches extends BaseModule {
      * @example
      * ```ts
      * const response = await ai.caches.update({
-     *   name: 'gemini-1.5-flash',
+     *   name: '...',  // The server-generated resource name.
      *   config: {'ttl': '7600s'}
      * });
      * ```
@@ -685,7 +696,7 @@ export declare interface CreateCachedContentConfig {
 
 /** Parameters for caches.create method. */
 export declare interface CreateCachedContentParameters {
-    /** ID of the model to use. Example: gemini-1.5-flash */
+    /** ID of the model to use. Example: gemini-2.0-flash */
     model: string;
     /** Configuration that contains optional parameters.
      */
@@ -700,7 +711,7 @@ export declare interface CreateCachedContentParameters {
 export declare interface CreateChatParameters {
     /** The name of the model to use for the chat session.
 
-     For example: 'gemini-2.0-flash', 'gemini-1.5-pro', etc. See gemini API
+     For example: 'gemini-2.0-flash', 'gemini-2.0-flash-lite', etc. See Gemini API
      docs to find the available models.
      */
     model: string;
@@ -791,11 +802,100 @@ export declare function createPartFromText(text: string): Part;
  */
 export declare function createPartFromUri(uri: string, mimeType: string): Part;
 
+/** Supervised fine-tuning job creation request - optional fields. */
+export declare interface CreateTuningJobConfig {
+    /** Used to override HTTP request options. */
+    httpOptions?: HttpOptions;
+    /** Abort signal which can be used to cancel the request.
+
+     NOTE: AbortSignal is a client-only operation. Using it to cancel an
+     operation will not cancel the request in the service. You will still
+     be charged usage for any applicable operations.
+     */
+    abortSignal?: AbortSignal;
+    /** Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file. */
+    validationDataset?: TuningValidationDataset;
+    /** The display name of the tuned Model. The name can be up to 128 characters long and can consist of any UTF-8 characters. */
+    tunedModelDisplayName?: string;
+    /** The description of the TuningJob */
+    description?: string;
+    /** Number of complete passes the model makes over the entire training dataset during training. */
+    epochCount?: number;
+    /** Multiplier for adjusting the default learning rate. */
+    learningRateMultiplier?: number;
+    /** Adapter size for tuning. */
+    adapterSize?: AdapterSize;
+    /** The batch size hyperparameter for tuning. If not set, a default of 4 or 16 will be used based on the number of training examples. */
+    batchSize?: number;
+    /** The learning rate hyperparameter for tuning. If not set, a default of 0.001 or 0.0002 will be calculated based on the number of training examples. */
+    learningRate?: number;
+}
+
+/** Supervised fine-tuning job creation parameters - optional fields. */
+export declare interface CreateTuningJobParameters {
+    /** The base model that is being tuned, e.g., "gemini-1.0-pro-002". */
+    baseModel: string;
+    /** Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file. */
+    trainingDataset: TuningDataset;
+    /** Configuration for the tuning job. */
+    config?: CreateTuningJobConfig;
+}
+
 /**
  * Creates a `Content` object with a user role from a `PartListUnion` object or `string`.
  */
 export declare function createUserContent(partOrString: PartListUnion | string): Content;
 
+/** Distribution computed over a tuning dataset. */
+export declare interface DatasetDistribution {
+    /** Output only. Defines the histogram bucket. */
+    buckets?: DatasetDistributionDistributionBucket[];
+    /** Output only. The maximum of the population values. */
+    max?: number;
+    /** Output only. The arithmetic mean of the values in the population. */
+    mean?: number;
+    /** Output only. The median of the values in the population. */
+    median?: number;
+    /** Output only. The minimum of the population values. */
+    min?: number;
+    /** Output only. The 5th percentile of the values in the population. */
+    p5?: number;
+    /** Output only. The 95th percentile of the values in the population. */
+    p95?: number;
+    /** Output only. Sum of a given population of values. */
+    sum?: number;
+}
+
+/** Dataset bucket used to create a histogram for the distribution given a population of values. */
+export declare interface DatasetDistributionDistributionBucket {
+    /** Output only. Number of values in the bucket. */
+    count?: string;
+    /** Output only. Left bound of the bucket. */
+    left?: number;
+    /** Output only. Right bound of the bucket. */
+    right?: number;
+}
+
+/** Statistics computed over a tuning dataset. */
+export declare interface DatasetStats {
+    /** Output only. Number of billable characters in the tuning dataset. */
+    totalBillableCharacterCount?: string;
+    /** Output only. Number of tuning characters in the tuning dataset. */
+    totalTuningCharacterCount?: string;
+    /** Output only. Number of examples in the tuning dataset. */
+    tuningDatasetExampleCount?: string;
+    /** Output only. Number of tuning steps for this Tuning Job. */
+    tuningStepCount?: string;
+    /** Output only. Sample user messages in the training dataset uri. */
+    userDatasetExamples?: Content[];
+    /** Output only. Dataset distributions for the user input tokens. */
+    userInputTokenDistribution?: DatasetDistribution;
+    /** Output only. Dataset distributions for the messages per example. */
+    userMessagePerExampleDistribution?: DatasetDistribution;
+    /** Output only. Dataset distributions for the user output tokens. */
+    userOutputTokenDistribution?: DatasetDistribution;
+}
+
 /** Optional parameters for caches.delete method. */
 export declare interface DeleteCachedContentConfig {
     /** Used to override HTTP request options. */
@@ -848,6 +948,63 @@ export declare interface DeleteFileParameters {
 export declare class DeleteFileResponse {
 }
 
+/** Configuration for deleting a tuned model. */
+export declare interface DeleteModelConfig {
+    /** Used to override HTTP request options. */
+    httpOptions?: HttpOptions;
+    /** Abort signal which can be used to cancel the request.
+
+     NOTE: AbortSignal is a client-only operation. Using it to cancel an
+     operation will not cancel the request in the service. You will still
+     be charged usage for any applicable operations.
+     */
+    abortSignal?: AbortSignal;
+}
+
+/** Parameters for deleting a tuned model. */
+export declare interface DeleteModelParameters {
+    model: string;
+    /** Optional parameters for the request. */
+    config?: DeleteModelConfig;
+}
+
+export declare class DeleteModelResponse {
+}
+
+/** Statistics computed for datasets used for distillation. */
+export declare interface DistillationDataStats {
+    /** Output only. Statistics computed for the training dataset. */
+    trainingDatasetStats?: DatasetStats;
+}
+
+/** Hyperparameters for Distillation. */
+export declare interface DistillationHyperParameters {
+    /** Optional. Adapter size for distillation. */
+    adapterSize?: AdapterSize;
+    /** Optional. Number of complete passes the model makes over the entire training dataset during training. */
+    epochCount?: string;
+    /** Optional. Multiplier for adjusting the default learning rate. */
+    learningRateMultiplier?: number;
+}
+
+/** Tuning Spec for Distillation. */
+export declare interface DistillationSpec {
+    /** The base teacher model that is being distilled, e.g., "gemini-1.0-pro-002". */
+    baseTeacherModel?: string;
+    /** Optional. Hyperparameters for Distillation. */
+    hyperParameters?: DistillationHyperParameters;
+    /** Required. A path in a Cloud Storage bucket, which will be treated as the root output directory of the distillation pipeline. It is used by the system to generate the paths of output artifacts. */
+    pipelineRootDirectory?: string;
+    /** The student model that is being tuned, e.g., "google/gemma-2b-1.1-it". */
+    studentModel?: string;
+    /** Required. Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file. */
+    trainingDatasetUri?: string;
+    /** The resource name of the Tuned teacher model. Format: `projects/{project}/locations/{location}/models/{model}`. */
+    tunedTeacherModelSource?: string;
+    /** Optional. Cloud Storage path to file containing validation dataset for tuning. The dataset must be formatted as a JSONL file. */
+    validationDatasetUri?: string;
+}
+
 /** Used to override the default configuration. */
 export declare interface DownloadFileConfig {
     /** Used to override HTTP request options. */
@@ -940,6 +1097,12 @@ export declare class EmbedContentResponse {
     metadata?: EmbedContentMetadata;
 }
 
+/** Represents a customer-managed encryption key spec that can be applied to a top-level resource. */
+export declare interface EncryptionSpec {
+    /** Required. The Cloud KMS resource identifier of the customer managed encryption key used to protect a resource. Has the form: `projects/my-project/locations/my-region/keyRings/my-kr/cryptoKeys/my-key`. The key needs to be in the same region as where the compute resource is created. */
+    kmsKeyName?: string;
+}
+
 /** An endpoint where you deploy models. */
 export declare interface Endpoint {
     /** Resource name of the endpoint. */
@@ -1220,21 +1383,16 @@ export declare enum FunctionCallingConfigMode {
     NONE = "NONE"
 }
 
-/** Defines a function that the model can generate JSON inputs for.
-
- The inputs are based on `OpenAPI 3.0 specifications
- <https://spec.openapis.org/oas/v3.0.3>`_.
- */
+/** Structured representation of a function declaration as defined by the [OpenAPI 3.0 specification](https://spec.openapis.org/oas/v3.0.3). Included in this declaration are the function name, description, parameters and response type. This FunctionDeclaration is a representation of a block of code that can be used as a `Tool` by the model and executed by the client. */
 export declare interface FunctionDeclaration {
-    /** Describes the output from the function in the OpenAPI JSON Schema
-     Object format. */
-    response?: Schema;
     /** Optional. Description and purpose of the function. Model uses it to decide how and whether to call the function. */
     description?: string;
     /** Required. The name of the function to call. Must start with a letter or an underscore. Must be a-z, A-Z, 0-9, or contain underscores, dots and dashes, with a maximum length of 64. */
     name?: string;
     /** Optional. Describes the parameters to this function in JSON Schema Object format. Reflects the Open API 3.03 Parameter Object. string Key: the name of the parameter. Parameter names are case sensitive. Schema Value: the Schema defining the type used for the parameter. For function with no parameters, this can be left unset. Parameter names must start with a letter or an underscore and must only contain chars a-z, A-Z, 0-9, or underscores with a maximum length of 64. Example with 1 required and 1 optional parameter: type: OBJECT properties: param1: type: STRING param2: type: INTEGER required: - param1 */
     parameters?: Schema;
+    /** Optional. Describes the output from this function in JSON Schema format. Reflects the Open API 3.03 Response Object. The Schema defines the type used for the response value of the function. */
+    response?: Schema;
 }
 
 /** A function response. */
@@ -1887,6 +2045,26 @@ export declare interface GetOperationParameters {
     config?: GetOperationConfig;
 }
 
+/** Optional parameters for tunings.get method. */
+export declare interface GetTuningJobConfig {
+    /** Used to override HTTP request options. */
+    httpOptions?: HttpOptions;
+    /** Abort signal which can be used to cancel the request.
+
+     NOTE: AbortSignal is a client-only operation. Using it to cancel an
+     operation will not cancel the request in the service. You will still
+     be charged usage for any applicable operations.
+     */
+    abortSignal?: AbortSignal;
+}
+
+/** Parameters for the get method. */
+export declare interface GetTuningJobParameters {
+    name: string;
+    /** Optional parameters for the request. */
+    config?: GetTuningJobConfig;
+}
+
 /**
  * The Google GenAI SDK.
  *
@@ -1929,6 +2107,7 @@ export declare class GoogleGenAI {
     readonly caches: Caches;
     readonly files: Files;
     readonly operations: Operations;
+    readonly tunings: Tunings;
     constructor(options: GoogleGenAIOptions);
 }
 
@@ -1943,7 +2122,7 @@ export declare interface GoogleGenAIOptions {
      *
      * @remarks
      * When true, the {@link https://cloud.google.com/vertex-ai/docs/reference/rest | Vertex AI API} will used.
-     * When false, the {@link https://cloud.google.com/vertex-ai/docs/reference/rest | Gemini API} will be used.
+     * When false, the {@link https://ai.google.dev/api | Gemini API} will be used.
      *
      * If unset, default SDK behavior is to use the Gemini API service.
      */
@@ -1951,12 +2130,14 @@ export declare interface GoogleGenAIOptions {
     /**
      * Optional. The Google Cloud project ID for Vertex AI clients.
      *
+     * Find your project ID: https://cloud.google.com/resource-manager/docs/creating-managing-projects#identifying_projects
+     *
      * @remarks
      * Only supported on Node runtimes, ignored on browser runtimes.
      */
     project?: string;
     /**
-     * Optional. The Google Cloud project region for Vertex AI clients.
+     * Optional. The Google Cloud project {@link https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations | location} for Vertex AI clients.
      *
      * @remarks
      * Only supported on Node runtimes, ignored on browser runtimes.
@@ -1993,6 +2174,16 @@ export declare interface GoogleGenAIOptions {
     httpOptions?: HttpOptions;
 }
 
+/** The `Status` type defines a logical error model that is suitable for different programming environments, including REST APIs and RPC APIs. It is used by [gRPC](https://github.com/grpc). Each `Status` message contains three pieces of data: error code, error message, and error details. You can find out more about this error model and how to work with it in the [API Design Guide](https://cloud.google.com/apis/design/errors). */
+export declare interface GoogleRpcStatus {
+    /** The status code, which should be an enum value of google.rpc.Code. */
+    code?: number;
+    /** A list of messages that carry the error details. There is a common set of message types for APIs to use. */
+    details?: Record<string, unknown>[];
+    /** A developer-facing error message, which should be in English. Any user-facing error message should be localized and sent in the google.rpc.Status.details field, or localized by the client. */
+    message?: string;
+}
+
 /** Tool to support Google Search in Model. Powered by Google. */
 export declare interface GoogleSearch {
 }
@@ -2209,6 +2400,22 @@ export declare enum ImagePromptLanguage {
     hi = "hi"
 }
 
+/** Output only. The detailed state of the job. */
+export declare enum JobState {
+    JOB_STATE_UNSPECIFIED = "JOB_STATE_UNSPECIFIED",
+    JOB_STATE_QUEUED = "JOB_STATE_QUEUED",
+    JOB_STATE_PENDING = "JOB_STATE_PENDING",
+    JOB_STATE_RUNNING = "JOB_STATE_RUNNING",
+    JOB_STATE_SUCCEEDED = "JOB_STATE_SUCCEEDED",
+    JOB_STATE_FAILED = "JOB_STATE_FAILED",
+    JOB_STATE_CANCELLING = "JOB_STATE_CANCELLING",
+    JOB_STATE_CANCELLED = "JOB_STATE_CANCELLED",
+    JOB_STATE_PAUSED = "JOB_STATE_PAUSED",
+    JOB_STATE_EXPIRED = "JOB_STATE_EXPIRED",
+    JOB_STATE_UPDATING = "JOB_STATE_UPDATING",
+    JOB_STATE_PARTIALLY_SUCCEEDED = "JOB_STATE_PARTIALLY_SUCCEEDED"
+}
+
 /** Required. Programming language of the `code`. */
 export declare enum Language {
     LANGUAGE_UNSPECIFIED = "LANGUAGE_UNSPECIFIED",
@@ -2273,6 +2480,35 @@ export declare class ListFilesResponse {
     files?: File_2[];
 }
 
+/** Configuration for the list tuning jobs method. */
+export declare interface ListTuningJobsConfig {
+    /** Used to override HTTP request options. */
+    httpOptions?: HttpOptions;
+    /** Abort signal which can be used to cancel the request.
+
+     NOTE: AbortSignal is a client-only operation. Using it to cancel an
+     operation will not cancel the request in the service. You will still
+     be charged usage for any applicable operations.
+     */
+    abortSignal?: AbortSignal;
+    pageSize?: number;
+    pageToken?: string;
+    filter?: string;
+}
+
+/** Parameters for the list tuning jobs method. */
+export declare interface ListTuningJobsParameters {
+    config?: ListTuningJobsConfig;
+}
+
+/** Response for the list tuning jobs method. */
+export declare class ListTuningJobsResponse {
+    /** A token to retrieve the next page of results. Pass to ListTuningJobsRequest.page_token to obtain that page. */
+    nextPageToken?: string;
+    /** List of TuningJobs in the requested page. */
+    tuningJobs?: TuningJob[];
+}
+
 /**
  Live class encapsulates the configuration for live interaction with the
  Generative Language API. It embeds ApiClient for general API settings.
@@ -2987,6 +3223,36 @@ export declare class Models extends BaseModule {
      * ```
      */
     get(params: types.GetModelParameters): Promise<types.Model>;
+    /**
+     * Updates a tuned model by its name.
+     *
+     * @param params - The parameters for updating the model.
+     * @return The response from the API.
+     *
+     * @example
+     * ```ts
+     * const response = await ai.models.update({
+     *   model: 'tuned-model-name',
+     *   config: {
+     *     displayName: 'New display name',
+     *     description: 'New description',
+     *   },
+     * });
+     * ```
+     */
+    update(params: types.UpdateModelParameters): Promise<types.Model>;
+    /**
+     * Deletes a tuned model by its name.
+     *
+     * @param params - The parameters for deleting the model.
+     * @return The response from the API.
+     *
+     * @example
+     * ```ts
+     * const response = await ai.models.delete({model: 'tuned-model-name'});
+     * ```
+     */
+    delete(params: types.DeleteModelParameters): Promise<types.DeleteModelResponse>;
     /**
      * Counts the number of tokens in the given contents. Multimodal input is
      * supported for Gemini models.
@@ -3055,6 +3321,18 @@ export declare interface ModelSelectionConfig {
     featureSelectionPreference?: FeatureSelectionPreference;
 }
 
+/** A long-running operation. */
+export declare interface Operation {
+    /** The server-assigned name, which is only unique within the same service that originally returns it. If you use the default HTTP mapping, the `name` should be a resource name ending with `operations/{unique_id}`. */
+    name?: string;
+    /** Service-specific metadata associated with the operation. It typically contains progress information and common metadata such as create time. Some services might not provide such metadata.  Any method that returns a long-running operation should document the metadata type, if any. */
+    metadata?: Record<string, unknown>;
+    /** If the value is `false`, it means the operation is still in progress. If `true`, the operation is completed, and either `error` or `response` is available. */
+    done?: boolean;
+    /** The error result of the operation in case of failure or cancellation. */
+    error?: Record<string, unknown>;
+}
+
 /** Parameters for the get method of the operations module. */
 export declare interface OperationGetParameters {
     /** The operation to be retrieved. */
@@ -3246,6 +3524,16 @@ export declare interface Part {
 
 export declare type PartListUnion = PartUnion[] | PartUnion;
 
+/** Tuning spec for Partner models. */
+export declare interface PartnerModelTuningSpec {
+    /** Hyperparameters for tuning. The accepted hyper_parameters and their valid range of values will differ depending on the base model. */
+    hyperParameters?: Record<string, unknown>;
+    /** Required. Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file. */
+    trainingDatasetUri?: string;
+    /** Optional. Cloud Storage path to file containing validation dataset for tuning. The dataset must be formatted as a JSONL file. */
+    validationDatasetUri?: string;
+}
+
 export declare type PartUnion = Part | string;
 
 /** Enum that controls the generation of people. */
@@ -3431,45 +3719,42 @@ export declare interface SafetySetting {
     threshold?: HarmBlockThreshold;
 }
 
-/** Schema that defines the format of input and output data.
-
- Represents a select subset of an OpenAPI 3.0 schema object.
- */
+/** Schema is used to define the format of input/output data. Represents a select subset of an [OpenAPI 3.0 schema object](https://spec.openapis.org/oas/v3.0.3#schema-object). More fields may be added in the future as needed. */
 export declare interface Schema {
-    /** Optional. Example of the object. Will only populated when the object is the root. */
-    example?: unknown;
-    /** Optional. Pattern of the Type.STRING to restrict a string to a regular expression. */
-    pattern?: string;
-    /** Optional. Default value of the data. */
-    default?: unknown;
-    /** Optional. Maximum length of the Type.STRING */
-    maxLength?: string;
-    /** Optional. SCHEMA FIELDS FOR TYPE STRING Minimum length of the Type.STRING */
-    minLength?: string;
-    /** Optional. Minimum number of the properties for Type.OBJECT. */
-    minProperties?: string;
-    /** Optional. Maximum number of the properties for Type.OBJECT. */
-    maxProperties?: string;
     /** Optional. The value should be validated against any (one or more) of the subschemas in the list. */
     anyOf?: Schema[];
+    /** Optional. Default value of the data. */
+    default?: unknown;
     /** Optional. The description of the data. */
     description?: string;
     /** Optional. Possible values of the element of primitive type with enum format. Examples: 1. We can define direction as : {type:STRING, format:enum, enum:["EAST", NORTH", "SOUTH", "WEST"]} 2. We can define apartment number as : {type:INTEGER, format:enum, enum:["101", "201", "301"]} */
     enum?: string[];
+    /** Optional. Example of the object. Will only populated when the object is the root. */
+    example?: unknown;
     /** Optional. The format of the data. Supported formats: for NUMBER type: "float", "double" for INTEGER type: "int32", "int64" for STRING type: "email", "byte", etc */
     format?: string;
     /** Optional. SCHEMA FIELDS FOR TYPE ARRAY Schema of the elements of Type.ARRAY. */
     items?: Schema;
     /** Optional. Maximum number of the elements for Type.ARRAY. */
     maxItems?: string;
+    /** Optional. Maximum length of the Type.STRING */
+    maxLength?: string;
+    /** Optional. Maximum number of the properties for Type.OBJECT. */
+    maxProperties?: string;
     /** Optional. Maximum value of the Type.INTEGER and Type.NUMBER */
     maximum?: number;
     /** Optional. Minimum number of the elements for Type.ARRAY. */
     minItems?: string;
+    /** Optional. SCHEMA FIELDS FOR TYPE STRING Minimum length of the Type.STRING */
+    minLength?: string;
+    /** Optional. Minimum number of the properties for Type.OBJECT. */
+    minProperties?: string;
     /** Optional. SCHEMA FIELDS FOR TYPE INTEGER and NUMBER Minimum value of the Type.INTEGER and Type.NUMBER */
     minimum?: number;
     /** Optional. Indicates if the value may be null. */
     nullable?: boolean;
+    /** Optional. Pattern of the Type.STRING to restrict a string to a regular expression. */
+    pattern?: string;
     /** Optional. SCHEMA FIELDS FOR TYPE OBJECT Properties of Type.OBJECT. */
     properties?: Record<string, Schema>;
     /** Optional. The order of the properties. Not a standard field in open api spec. Only used to support the order of the properties. */
@@ -3778,6 +4063,84 @@ export declare enum SubjectReferenceType {
     SUBJECT_TYPE_PRODUCT = "SUBJECT_TYPE_PRODUCT"
 }
 
+/** Hyperparameters for SFT. */
+export declare interface SupervisedHyperParameters {
+    /** Optional. Adapter size for tuning. */
+    adapterSize?: AdapterSize;
+    /** Optional. Number of complete passes the model makes over the entire training dataset during training. */
+    epochCount?: string;
+    /** Optional. Multiplier for adjusting the default learning rate. */
+    learningRateMultiplier?: number;
+}
+
+/** Dataset distribution for Supervised Tuning. */
+export declare interface SupervisedTuningDatasetDistribution {
+    /** Output only. Sum of a given population of values that are billable. */
+    billableSum?: string;
+    /** Output only. Defines the histogram bucket. */
+    buckets?: SupervisedTuningDatasetDistributionDatasetBucket[];
+    /** Output only. The maximum of the population values. */
+    max?: number;
+    /** Output only. The arithmetic mean of the values in the population. */
+    mean?: number;
+    /** Output only. The median of the values in the population. */
+    median?: number;
+    /** Output only. The minimum of the population values. */
+    min?: number;
+    /** Output only. The 5th percentile of the values in the population. */
+    p5?: number;
+    /** Output only. The 95th percentile of the values in the population. */
+    p95?: number;
+    /** Output only. Sum of a given population of values. */
+    sum?: string;
+}
+
+/** Dataset bucket used to create a histogram for the distribution given a population of values. */
+export declare interface SupervisedTuningDatasetDistributionDatasetBucket {
+    /** Output only. Number of values in the bucket. */
+    count?: number;
+    /** Output only. Left bound of the bucket. */
+    left?: number;
+    /** Output only. Right bound of the bucket. */
+    right?: number;
+}
+
+/** Tuning data statistics for Supervised Tuning. */
+export declare interface SupervisedTuningDataStats {
+    /** Output only. Number of billable characters in the tuning dataset. */
+    totalBillableCharacterCount?: string;
+    /** Output only. Number of billable tokens in the tuning dataset. */
+    totalBillableTokenCount?: string;
+    /** The number of examples in the dataset that have been truncated by any amount. */
+    totalTruncatedExampleCount?: string;
+    /** Output only. Number of tuning characters in the tuning dataset. */
+    totalTuningCharacterCount?: string;
+    /** A partial sample of the indices (starting from 1) of the truncated examples. */
+    truncatedExampleIndices?: string[];
+    /** Output only. Number of examples in the tuning dataset. */
+    tuningDatasetExampleCount?: string;
+    /** Output only. Number of tuning steps for this Tuning Job. */
+    tuningStepCount?: string;
+    /** Output only. Sample user messages in the training dataset uri. */
+    userDatasetExamples?: Content[];
+    /** Output only. Dataset distributions for the user input tokens. */
+    userInputTokenDistribution?: SupervisedTuningDatasetDistribution;
+    /** Output only. Dataset distributions for the messages per example. */
+    userMessagePerExampleDistribution?: SupervisedTuningDatasetDistribution;
+    /** Output only. Dataset distributions for the user output tokens. */
+    userOutputTokenDistribution?: SupervisedTuningDatasetDistribution;
+}
+
+/** Tuning Spec for Supervised Tuning for first party models. */
+export declare interface SupervisedTuningSpec {
+    /** Optional. Hyperparameters for SFT. */
+    hyperParameters?: SupervisedHyperParameters;
+    /** Required. Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file. */
+    trainingDatasetUri?: string;
+    /** Optional. Cloud Storage path to file containing validation dataset for tuning. The dataset must be formatted as a JSONL file. */
+    validationDatasetUri?: string;
+}
+
 export declare interface TestTableFile {
     comment?: string;
     testMethod?: string;
@@ -3826,8 +4189,6 @@ export declare interface TokensInfo {
 
 /** Tool details of a tool that the model may use to generate a response. */
 export declare interface Tool {
-    /** List of function declarations that the tool supports. */
-    functionDeclarations?: FunctionDeclaration[];
     /** Optional. Retrieval tool type. System will always execute the provided retrieval tool(s) to get external knowledge to answer the prompt. Retrieval results are presented to the model for generation. */
     retrieval?: Retrieval;
     /** Optional. Google Search tool type. Specialized retrieval tool
@@ -3837,6 +4198,8 @@ export declare interface Tool {
     googleSearchRetrieval?: GoogleSearchRetrieval;
     /** Optional. CodeExecution tool type. Enables the model to execute code as part of generation. This field is only used by the Gemini Developer API services. */
     codeExecution?: ToolCodeExecution;
+    /** Optional. Function tool type. One or more function declarations to be passed to the model along with the current user query. Model may decide to call a subset of these functions by populating FunctionCall in the response. User should provide a FunctionResponse for each function call in the next turn. Based on the function responses, Model will generate the final response back to the user. Maximum 128 function declarations can be provided. */
+    functionDeclarations?: FunctionDeclaration[];
 }
 
 /** Tool that executes code generated by the model, and automatically returns the result to the model. See also [ExecutableCode]and [CodeExecutionResult] which are input and output to this tool. */
@@ -3871,6 +4234,13 @@ export declare interface Transcription {
     finished?: boolean;
 }
 
+export declare interface TunedModel {
+    /** Output only. The resource name of the TunedModel. Format: `projects/{project}/locations/{location}/models/{model}`. */
+    model?: string;
+    /** Output only. A resource name of an Endpoint. Format: `projects/{project}/locations/{location}/endpoints/{endpoint}`. */
+    endpoint?: string;
+}
+
 /** A tuned machine learning model. */
 export declare interface TunedModelInfo {
     /** ID of the base model that you want to tune. */
@@ -3881,6 +4251,115 @@ export declare interface TunedModelInfo {
     updateTime?: string;
 }
 
+/** Supervised fine-tuning training dataset. */
+export declare interface TuningDataset {
+    /** GCS URI of the file containing training dataset in JSONL format. */
+    gcsUri?: string;
+    /** Inline examples with simple input/output text. */
+    examples?: TuningExample[];
+}
+
+/** The tuning data statistic values for TuningJob. */
+export declare interface TuningDataStats {
+    /** Output only. Statistics for distillation. */
+    distillationDataStats?: DistillationDataStats;
+    /** The SFT Tuning data stats. */
+    supervisedTuningDataStats?: SupervisedTuningDataStats;
+}
+
+export declare interface TuningExample {
+    /** Text model input. */
+    textInput?: string;
+    /** The expected model output. */
+    output?: string;
+}
+
+/** A tuning job. */
+export declare interface TuningJob {
+    /** Output only. Identifier. Resource name of a TuningJob. Format: `projects/{project}/locations/{location}/tuningJobs/{tuning_job}` */
+    name?: string;
+    /** Output only. The detailed state of the job. */
+    state?: JobState;
+    /** Output only. Time when the TuningJob was created. */
+    createTime?: string;
+    /** Output only. Time when the TuningJob for the first time entered the `JOB_STATE_RUNNING` state. */
+    startTime?: string;
+    /** Output only. Time when the TuningJob entered any of the following JobStates: `JOB_STATE_SUCCEEDED`, `JOB_STATE_FAILED`, `JOB_STATE_CANCELLED`, `JOB_STATE_EXPIRED`. */
+    endTime?: string;
+    /** Output only. Time when the TuningJob was most recently updated. */
+    updateTime?: string;
+    /** Output only. Only populated when job's state is `JOB_STATE_FAILED` or `JOB_STATE_CANCELLED`. */
+    error?: GoogleRpcStatus;
+    /** Optional. The description of the TuningJob. */
+    description?: string;
+    /** The base model that is being tuned, e.g., "gemini-1.0-pro-002". . */
+    baseModel?: string;
+    /** Output only. The tuned model resources associated with this TuningJob. */
+    tunedModel?: TunedModel;
+    /** Tuning Spec for Supervised Fine Tuning. */
+    supervisedTuningSpec?: SupervisedTuningSpec;
+    /** Output only. The tuning data statistics associated with this TuningJob. */
+    tuningDataStats?: TuningDataStats;
+    /** Customer-managed encryption key options for a TuningJob. If this is set, then all resources created by the TuningJob will be encrypted with the provided encryption key. */
+    encryptionSpec?: EncryptionSpec;
+    /** Tuning Spec for open sourced and third party Partner models. */
+    partnerModelTuningSpec?: PartnerModelTuningSpec;
+    /** Tuning Spec for Distillation. */
+    distillationSpec?: DistillationSpec;
+    /** Output only. The Experiment associated with this TuningJob. */
+    experiment?: string;
+    /** Optional. The labels with user-defined metadata to organize TuningJob and generated resources such as Model and Endpoint. Label keys and values can be no longer than 64 characters (Unicode codepoints), can only contain lowercase letters, numeric characters, underscores and dashes. International characters are allowed. See https://goo.gl/xmQnxf for more information and examples of labels. */
+    labels?: Record<string, string>;
+    /** Output only. The resource name of the PipelineJob associated with the TuningJob. Format: `projects/{project}/locations/{location}/pipelineJobs/{pipeline_job}`. */
+    pipelineJob?: string;
+    /** Optional. The display name of the TunedModel. The name can be up to 128 characters long and can consist of any UTF-8 characters. */
+    tunedModelDisplayName?: string;
+}
+
+declare class Tunings extends BaseModule {
+    private readonly apiClient;
+    constructor(apiClient: ApiClient);
+    /**
+     * Gets a TuningJob.
+     *
+     * @param name - The resource name of the tuning job.
+     * @return - A TuningJob object.
+     *
+     * @experimental - The SDK's tuning implementation is experimental, and may
+     * change in future versions.
+     */
+    get: (params: types.GetTuningJobParameters) => Promise<types.TuningJob>;
+    /**
+     * Lists tuning jobs.
+     *
+     * @param config - The configuration for the list request.
+     * @return - A list of tuning jobs.
+     *
+     * @experimental - The SDK's tuning implementation is experimental, and may
+     * change in future versions.
+     */
+    list: (params?: types.ListTuningJobsParameters) => Promise<Pager<types.TuningJob>>;
+    /**
+     * Creates a supervised fine-tuning job.
+     *
+     * @param params - The parameters for the tuning job.
+     * @return - A TuningJob operation.
+     *
+     * @experimental - The SDK's tuning implementation is experimental, and may
+     * change in future versions.
+     */
+    tune: (params: types.CreateTuningJobParameters) => Promise<types.TuningJob>;
+    private getInternal;
+    private listInternal;
+    private tuneInternal;
+    private tuneMldevInternal;
+}
+
+export declare interface TuningValidationDataset {
+    /** GCS URI of the file containing validation dataset in JSONL format. */
+    gcsUri?: string;
+}
+
 /** Options about which input is included in the user's turn. */
 export declare enum TurnCoverage {
     TURN_COVERAGE_UNSPECIFIED = "TURN_COVERAGE_UNSPECIFIED",
@@ -3912,11 +4391,11 @@ declare namespace types {
         createModelContent,
         Outcome,
         Language,
-        Type,
         HarmCategory,
         HarmBlockMethod,
         HarmBlockThreshold,
         Mode,
+        Type,
         FinishReason,
         HarmProbability,
         HarmSeverity,
@@ -3924,6 +4403,8 @@ declare namespace types {
         TrafficType,
         Modality,
         MediaResolution,
+        JobState,
+        AdapterSize,
         FeatureSelectionPreference,
         DynamicRetrievalConfigMode,
         FunctionCallingConfigMode,
@@ -3950,10 +4431,8 @@ declare namespace types {
         Part,
         Content,
         HttpOptions,
-        Schema,
         ModelSelectionConfig,
         SafetySetting,
-        FunctionDeclaration,
         GoogleSearch,
         DynamicRetrievalConfig,
         GoogleSearchRetrieval,
@@ -3968,6 +4447,8 @@ declare namespace types {
         VertexRagStore,
         Retrieval,
         ToolCodeExecution,
+        Schema,
+        FunctionDeclaration,
         Tool,
         FunctionCallingConfig,
         ToolConfig,
@@ -4017,6 +4498,11 @@ declare namespace types {
         Endpoint,
         TunedModelInfo,
         Model,
+        UpdateModelConfig,
+        UpdateModelParameters,
+        DeleteModelConfig,
+        DeleteModelParameters,
+        DeleteModelResponse,
         GenerationConfig,
         CountTokensConfig,
         CountTokensParameters,
@@ -4031,6 +4517,34 @@ declare namespace types {
         GeneratedVideo,
         GenerateVideosResponse,
         GenerateVideosOperation,
+        GetTuningJobConfig,
+        GetTuningJobParameters,
+        TunedModel,
+        GoogleRpcStatus,
+        SupervisedHyperParameters,
+        SupervisedTuningSpec,
+        DatasetDistributionDistributionBucket,
+        DatasetDistribution,
+        DatasetStats,
+        DistillationDataStats,
+        SupervisedTuningDatasetDistributionDatasetBucket,
+        SupervisedTuningDatasetDistribution,
+        SupervisedTuningDataStats,
+        TuningDataStats,
+        EncryptionSpec,
+        PartnerModelTuningSpec,
+        DistillationHyperParameters,
+        DistillationSpec,
+        TuningJob,
+        ListTuningJobsConfig,
+        ListTuningJobsParameters,
+        ListTuningJobsResponse,
+        TuningExample,
+        TuningDataset,
+        TuningValidationDataset,
+        CreateTuningJobConfig,
+        CreateTuningJobParameters,
+        Operation,
         CreateCachedContentConfig,
         CreateCachedContentParameters,
         CachedContentUsageMetadata,
@@ -4151,6 +4665,27 @@ export declare interface UpdateCachedContentParameters {
     config?: UpdateCachedContentConfig;
 }
 
+/** Configuration for updating a tuned model. */
+export declare interface UpdateModelConfig {
+    /** Used to override HTTP request options. */
+    httpOptions?: HttpOptions;
+    /** Abort signal which can be used to cancel the request.
+
+     NOTE: AbortSignal is a client-only operation. Using it to cancel an
+     operation will not cancel the request in the service. You will still
+     be charged usage for any applicable operations.
+     */
+    abortSignal?: AbortSignal;
+    displayName?: string;
+    description?: string;
+}
+
+/** Configuration for updating a tuned model. */
+export declare interface UpdateModelParameters {
+    model: string;
+    config?: UpdateModelConfig;
+}
+
 declare interface Uploader {
     /**
      * Uploads a file to the given upload url.