@google/genai 0.12.0 → 0.14.0

package/dist/node/node.d.ts

@@ -41,6 +41,21 @@ export declare enum AdapterSize {
 declare class ApiClient {
     readonly clientOptions: ApiClientInitOptions;
     constructor(opts: ApiClientInitOptions);
+    /**
+     * Determines the base URL for Vertex AI based on project and location.
+     * Uses the global endpoint if location is 'global' or if project/location
+     * are not specified (implying API key usage).
+     * @private
+     */
+    private baseUrlFromProjectLocation;
+    /**
+     * Normalizes authentication parameters for Vertex AI.
+     * If project and location are provided, API key is cleared.
+     * If project and location are not provided (implying API key usage),
+     * project and location are cleared.
+     * @private
+     */
+    private normalizeAuthParameters;
     isVertexAI(): boolean;
     getProject(): string | undefined;
     getLocation(): string | undefined;
@@ -77,6 +92,13 @@ declare class ApiClient {
      * @throws An error if the `mimeType` is not provided and can not be inferred,
      */
     uploadFile(file: string | Blob, config?: UploadFileConfig): Promise<File_2>;
+    /**
+     * Downloads a file asynchronously to the specified path.
+     *
+     * @params params - The parameters for the download request, see {@link
+     * DownloadFileParameters}
+     */
+    downloadFile(params: DownloadFileParameters): Promise<void>;
     private fetchUploadUrl;
 }
 
@@ -95,6 +117,12 @@ declare interface ApiClientInitOptions {
      * creating a client, will be set through the Node_client or Web_client.
      */
     uploader: Uploader;
+    /**
+     * Optional. The downloader to use for downloading files. This field is
+     * required for creating a client, will be set through the Node_client or
+     * Web_client.
+     */
+    downloader: Downloader;
     /**
      * Optional. The Google Cloud project ID for Vertex AI users.
      * It is not the numeric project name.
@@ -132,6 +160,12 @@ declare interface ApiClientInitOptions {
     userAgentExtra?: string;
 }
 
+/** Config for authentication with API key. */
+export declare interface ApiKeyConfig {
+    /** The API key to be used in the request directly. */
+    apiKeyString?: string;
+}
+
 /** The audio transcription configuration in Setup. */
 export declare interface AudioTranscriptionConfig {
 }
@@ -153,6 +187,61 @@ declare interface Auth {
     addAuthHeaders(headers: Headers): Promise<void>;
 }
 
+/** Auth configuration to run the extension. */
+export declare interface AuthConfig {
+    /** Config for API key auth. */
+    apiKeyConfig?: ApiKeyConfig;
+    /** Type of auth scheme. */
+    authType?: AuthType;
+    /** Config for Google Service Account auth. */
+    googleServiceAccountConfig?: AuthConfigGoogleServiceAccountConfig;
+    /** Config for HTTP Basic auth. */
+    httpBasicAuthConfig?: AuthConfigHttpBasicAuthConfig;
+    /** Config for user oauth. */
+    oauthConfig?: AuthConfigOauthConfig;
+    /** Config for user OIDC auth. */
+    oidcConfig?: AuthConfigOidcConfig;
+}
+
+/** Config for Google Service Account Authentication. */
+export declare interface AuthConfigGoogleServiceAccountConfig {
+    /** Optional. The service account that the extension execution service runs as. - If the service account is specified, the `iam.serviceAccounts.getAccessToken` permission should be granted to Vertex AI Extension Service Agent (https://cloud.google.com/vertex-ai/docs/general/access-control#service-agents) on the specified service account. - If not specified, the Vertex AI Extension Service Agent will be used to execute the Extension. */
+    serviceAccount?: string;
+}
+
+/** Config for HTTP Basic Authentication. */
+export declare interface AuthConfigHttpBasicAuthConfig {
+    /** Required. The name of the SecretManager secret version resource storing the base64 encoded credentials. Format: `projects/{project}/secrets/{secrete}/versions/{version}` - If specified, the `secretmanager.versions.access` permission should be granted to Vertex AI Extension Service Agent (https://cloud.google.com/vertex-ai/docs/general/access-control#service-agents) on the specified resource. */
+    credentialSecret?: string;
+}
+
+/** Config for user oauth. */
+export declare interface AuthConfigOauthConfig {
+    /** Access token for extension endpoint. Only used to propagate token from [[ExecuteExtensionRequest.runtime_auth_config]] at request time. */
+    accessToken?: string;
+    /** The service account used to generate access tokens for executing the Extension. - If the service account is specified, the `iam.serviceAccounts.getAccessToken` permission should be granted to Vertex AI Extension Service Agent (https://cloud.google.com/vertex-ai/docs/general/access-control#service-agents) on the provided service account. */
+    serviceAccount?: string;
+}
+
+/** Config for user OIDC auth. */
+export declare interface AuthConfigOidcConfig {
+    /** OpenID Connect formatted ID token for extension endpoint. Only used to propagate token from [[ExecuteExtensionRequest.runtime_auth_config]] at request time. */
+    idToken?: string;
+    /** The service account used to generate an OpenID Connect (OIDC)-compatible JWT token signed by the Google OIDC Provider (accounts.google.com) for extension endpoint (https://cloud.google.com/iam/docs/create-short-lived-credentials-direct#sa-credentials-oidc). - The audience for the token will be set to the URL in the server url defined in the OpenApi spec. - If the service account is provided, the service account should grant `iam.serviceAccounts.getOpenIdToken` permission to Vertex AI Extension Service Agent (https://cloud.google.com/vertex-ai/docs/general/access-control#service-agents). */
+    serviceAccount?: string;
+}
+
+/** Type of auth scheme. */
+export declare enum AuthType {
+    AUTH_TYPE_UNSPECIFIED = "AUTH_TYPE_UNSPECIFIED",
+    NO_AUTH = "NO_AUTH",
+    API_KEY_AUTH = "API_KEY_AUTH",
+    HTTP_BASIC_AUTH = "HTTP_BASIC_AUTH",
+    GOOGLE_SERVICE_ACCOUNT_AUTH = "GOOGLE_SERVICE_ACCOUNT_AUTH",
+    OAUTH = "OAUTH",
+    OIDC_AUTH = "OIDC_AUTH"
+}
+
 /** Configures automatic detection of activity. */
 export declare interface AutomaticActivityDetection {
     /** If enabled, detected voice and text input count as activity. If disabled, the client must send activity signals. */
@@ -185,6 +274,8 @@ export declare interface BaseUrlParameters {
 
 /** Content blob. */
 declare interface Blob_2 {
+    /** Optional. Display name of the blob. Used to provide a label or filename to distinguish blobs. This field is not currently used in the Gemini GenerateContent calls. */
+    displayName?: string;
     /** Required. Raw bytes. */
     data?: string;
     /** Required. The IANA standard MIME type of the source data. */
@@ -471,6 +562,19 @@ export declare class Chats {
     create(params: types.CreateChatParameters): Chat;
 }
 
+/** Describes the machine learning model version checkpoint. */
+export declare interface Checkpoint {
+    /** The ID of the checkpoint.
+     */
+    checkpointId?: string;
+    /** The epoch of the checkpoint.
+     */
+    epoch?: string;
+    /** The step of the checkpoint.
+     */
+    step?: string;
+}
+
 /** Source attributions for content. */
 export declare interface Citation {
     /** Output only. End index into the content. */
@@ -600,7 +704,7 @@ export declare interface ControlReferenceConfig {
  A control image is an image that represents a sketch image of areas for the
  model to fill in based on the prompt.
  */
-export declare interface ControlReferenceImage {
+export declare class ControlReferenceImage {
     /** The reference image for the editing operation. */
     referenceImage?: Image_2;
     /** The id of the reference image. */
@@ -609,6 +713,8 @@ export declare interface ControlReferenceImage {
     referenceType?: string;
     /** Configuration for the control reference image. */
     config?: ControlReferenceConfig;
+    /** Internal method to convert to ReferenceImageAPIInternal. */
+    toReferenceImageAPI(): any;
 }
 
 /** Enum representing the control type of a control reference image. */
@@ -823,6 +929,8 @@ export declare interface CreateTuningJobConfig {
     epochCount?: number;
     /** Multiplier for adjusting the default learning rate. */
     learningRateMultiplier?: number;
+    /** If set to true, disable intermediate checkpoints for SFT and only the last checkpoint will be exported. Otherwise, enable intermediate checkpoints for SFT. */
+    exportLastCheckpointOnly?: boolean;
     /** 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. */
@@ -1005,6 +1113,19 @@ export declare interface DistillationSpec {
     validationDatasetUri?: string;
 }
 
+export declare type DownloadableFileUnion = string | File_2 | GeneratedVideo | Video;
+
+declare interface Downloader {
+    /**
+     * Downloads a file to the given location.
+     *
+     * @param params The parameters for downloading the file.
+     * @param apiClient The ApiClient to use for uploading.
+     * @return A Promises that resolves when the download is complete.
+     */
+    download(params: DownloadFileParameters, apiClient: ApiClient): Promise<void>;
+}
+
 /** Used to override the default configuration. */
 export declare interface DownloadFileConfig {
     /** Used to override HTTP request options. */
@@ -1018,6 +1139,16 @@ export declare interface DownloadFileConfig {
     abortSignal?: AbortSignal;
 }
 
+/** Parameters used to download a file. */
+export declare interface DownloadFileParameters {
+    /** The file to download. It can be a file name, a file object or a generated video. */
+    file: DownloadableFileUnion;
+    /** Location where the file should be downloaded to. */
+    downloadPath: string;
+    /** Configuration to for the download operation. */
+    config?: DownloadFileConfig;
+}
+
 /** Describes the options to customize dynamic retrieval. */
 export declare interface DynamicRetrievalConfig {
     /** The mode of the predictor to be used in dynamic retrieval. */
@@ -1032,6 +1163,100 @@ export declare enum DynamicRetrievalConfigMode {
     MODE_DYNAMIC = "MODE_DYNAMIC"
 }
 
+/** Configuration for editing an image. */
+export declare interface EditImageConfig {
+    /** 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 URI used to store the generated images.
+     */
+    outputGcsUri?: string;
+    /** Description of what to discourage in the generated images.
+     */
+    negativePrompt?: string;
+    /** Number of images to generate.
+     */
+    numberOfImages?: number;
+    /** Aspect ratio of the generated images.
+     */
+    aspectRatio?: string;
+    /** Controls how much the model adheres to the text prompt. Large
+     values increase output and prompt alignment, but may compromise image
+     quality.
+     */
+    guidanceScale?: number;
+    /** Random seed for image generation. This is not available when
+     ``add_watermark`` is set to true.
+     */
+    seed?: number;
+    /** Filter level for safety filtering.
+     */
+    safetyFilterLevel?: SafetyFilterLevel;
+    /** Allows generation of people by the model.
+     */
+    personGeneration?: PersonGeneration;
+    /** Whether to report the safety scores of each generated image and
+     the positive prompt in the response.
+     */
+    includeSafetyAttributes?: boolean;
+    /** Whether to include the Responsible AI filter reason if the image
+     is filtered out of the response.
+     */
+    includeRaiReason?: boolean;
+    /** Language of the text in the prompt.
+     */
+    language?: ImagePromptLanguage;
+    /** MIME type of the generated image.
+     */
+    outputMimeType?: string;
+    /** Compression quality of the generated image (for ``image/jpeg``
+     only).
+     */
+    outputCompressionQuality?: number;
+    /** Describes the editing mode for the request. */
+    editMode?: EditMode;
+    /** The number of sampling steps. A higher value has better image
+     quality, while a lower value has better latency. */
+    baseSteps?: number;
+}
+
+/** Parameters for the request to edit an image. */
+export declare interface EditImageParameters {
+    /** The model to use. */
+    model: string;
+    /** A text description of the edit to apply to the image. */
+    prompt: string;
+    /** The reference images for Imagen 3 editing. */
+    referenceImages: ReferenceImage[];
+    /** Configuration for editing. */
+    config?: EditImageConfig;
+}
+
+/** Response for the request to edit an image. */
+export declare class EditImageResponse {
+    /** Generated images. */
+    generatedImages?: GeneratedImage[];
+}
+
+/** Enum representing the Imagen 3 Edit mode. */
+export declare enum EditMode {
+    EDIT_MODE_DEFAULT = "EDIT_MODE_DEFAULT",
+    EDIT_MODE_INPAINT_REMOVAL = "EDIT_MODE_INPAINT_REMOVAL",
+    EDIT_MODE_INPAINT_INSERTION = "EDIT_MODE_INPAINT_INSERTION",
+    EDIT_MODE_OUTPAINT = "EDIT_MODE_OUTPAINT",
+    EDIT_MODE_CONTROLLED_EDITING = "EDIT_MODE_CONTROLLED_EDITING",
+    EDIT_MODE_STYLE = "EDIT_MODE_STYLE",
+    EDIT_MODE_BGSWAP = "EDIT_MODE_BGSWAP",
+    EDIT_MODE_PRODUCT_IMAGE = "EDIT_MODE_PRODUCT_IMAGE"
+}
+
+/** Optional parameters for the embed_content method. */
 export declare interface EmbedContentConfig {
     /** Used to override HTTP request options. */
     httpOptions?: HttpOptions;
@@ -1118,6 +1343,10 @@ export declare enum EndSensitivity {
     END_SENSITIVITY_LOW = "END_SENSITIVITY_LOW"
 }
 
+/** Tool to search public web data, powered by Vertex AI Search and Sec4 compliance. */
+export declare interface EnterpriseWebSearch {
+}
+
 /** Code generated by the model that is meant to be executed, and the result returned to the model. Generated when using the [FunctionDeclaration] tool and [FunctionCallingConfig] mode is set to [Mode.CODE]. */
 export declare interface ExecutableCode {
     /** Required. The code to be executed. */
@@ -1261,6 +1490,23 @@ export declare class Files extends BaseModule {
      * ```
      */
     upload(params: types.UploadFileParameters): Promise<types.File>;
+    /**
+     * Downloads a remotely stored file asynchronously to a location specified in
+     * the `params` object. This method only works on Node environment, to
+     * download files in the browser, use a browser compliant method like an <a>
+     * tag.
+     *
+     * @param params - The parameters for the download request.
+     *
+     * @example
+     * The following code downloads an example file named "files/mehozpxf877d" as
+     * "file.txt".
+     *
+     * ```ts
+     * await ai.files.download({file: file.name, downloadPath: 'file.txt'});
+     * ```
+     */
+    download(params: types.DownloadFileParameters): Promise<void>;
     private listInternal;
     private createInternal;
     /**
@@ -1477,10 +1723,21 @@ export declare interface GenerateContentConfig {
      random number is used.
      */
     seed?: number;
-    /** Output response media type of the generated candidate text.
+    /** Output response mimetype of the generated candidate text.
+     Supported mimetype:
+     - `text/plain`: (default) Text output.
+     - `application/json`: JSON response in the candidates.
+     The model needs to be prompted to output the appropriate response type,
+     otherwise the behavior is undefined.
+     This is a preview feature.
      */
     responseMimeType?: string;
-    /** Schema that the generated candidate text must adhere to.
+    /** The `Schema` object allows the definition of input and output data types.
+     These types can be objects, but also primitives and arrays.
+     Represents a select subset of an [OpenAPI 3.0 schema
+     object](https://spec.openapis.org/oas/v3.0.3#schema).
+     If set, a compatible response_mime_type must also be set.
+     Compatible mimetypes: `application/json`: Schema for JSON response.
      */
     responseSchema?: SchemaUnion;
     /** Configuration for model router requests.
@@ -2186,6 +2443,12 @@ export declare interface GoogleGenAIOptions {
     httpOptions?: HttpOptions;
 }
 
+/** Tool to support Google Maps in Model. */
+export declare interface GoogleMaps {
+    /** Optional. Auth config for the Google Maps tool. */
+    authConfig?: AuthConfig;
+}
+
 /** 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. */
@@ -2434,6 +2697,20 @@ export declare enum Language {
     PYTHON = "PYTHON"
 }
 
+/** An object that represents a latitude/longitude pair.
+
+ This is expressed as a pair of doubles to represent degrees latitude and
+ degrees longitude. Unless specified otherwise, this object must conform to the
+ <a href="https://en.wikipedia.org/wiki/World_Geodetic_System#1984_version">
+ WGS84 standard</a>. Values must be within normalized ranges.
+ */
+export declare interface LatLng {
+    /** The latitude in degrees. It must be in the range [-90.0, +90.0]. */
+    latitude?: number;
+    /** The longitude in degrees. It must be in the range [-180.0, +180.0] */
+    longitude?: number;
+}
+
 /** Config for caches.list method. */
 export declare interface ListCachedContentsConfig {
     /** Used to override HTTP request options. */
@@ -2492,6 +2769,32 @@ export declare class ListFilesResponse {
     files?: File_2[];
 }
 
+export declare interface ListModelsConfig {
+    /** 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;
+    /** Set true to list base models, false to list tuned models. */
+    queryBase?: boolean;
+}
+
+export declare interface ListModelsParameters {
+    config?: ListModelsConfig;
+}
+
+export declare class ListModelsResponse {
+    nextPageToken?: string;
+    models?: Model[];
+}
+
 /** Configuration for the list tuning jobs method. */
 export declare interface ListTuningJobsConfig {
     /** Used to override HTTP request options. */
@@ -2888,7 +3191,7 @@ export declare interface LiveServerGoAway {
 }
 
 /** Response message for API call. */
-export declare interface LiveServerMessage {
+export declare class LiveServerMessage {
     /** Sent in response to a `LiveClientSetup` message from the client. */
     setupComplete?: LiveServerSetupComplete;
     /** Content generated by the model in response to client messages. */
@@ -2903,6 +3206,23 @@ export declare interface LiveServerMessage {
     goAway?: LiveServerGoAway;
     /** Update of the session resumption state. */
     sessionResumptionUpdate?: LiveServerSessionResumptionUpdate;
+    /**
+     * Returns the concatenation of all text parts from the server content if present.
+     *
+     * @remarks
+     * If there are non-text parts in the response, the concatenation of all text
+     * parts will be returned, and a warning will be logged.
+     */
+    get text(): string | undefined;
+    /**
+     * Returns the concatenation of all inline data parts from the server content if present.
+     *
+     * @remarks
+     * If there are non-inline data parts in the
+     * response, the concatenation of all inline data parts will be returned, and
+     * a warning will be logged.
+     */
+    get data(): string | undefined;
 }
 
 /** Update of the session resumption state.
@@ -2922,7 +3242,6 @@ export declare interface LiveServerSessionResumptionUpdate {
     lastConsumedClientMessageIndex?: string;
 }
 
-/** Sent in response to a `LiveGenerateContentSetup` message from the client. */
 export declare interface LiveServerSetupComplete {
 }
 
@@ -2990,7 +3309,7 @@ export declare interface MaskReferenceConfig {
  image. If the user provides a mask image, the mask must be in the same
  dimensions as the raw image.
  */
-export declare interface MaskReferenceImage {
+export declare class MaskReferenceImage {
     /** The reference image for the editing operation. */
     referenceImage?: Image_2;
     /** The id of the reference image. */
@@ -2999,6 +3318,8 @@ export declare interface MaskReferenceImage {
     referenceType?: string;
     /** Configuration for the mask reference image. */
     config?: MaskReferenceConfig;
+    /** Internal method to convert to ReferenceImageAPIInternal. */
+    toReferenceImageAPI(): any;
 }
 
 /** Enum representing the mask mode of a mask reference image. */
@@ -3076,6 +3397,11 @@ export declare interface Model {
     outputTokenLimit?: number;
     /** List of actions that are supported by the model. */
     supportedActions?: string[];
+    /** The default checkpoint id of a model version.
+     */
+    defaultCheckpointId?: string;
+    /** The checkpoints of the model. */
+    checkpoints?: Checkpoint[];
 }
 
 export declare class Models extends BaseModule {
@@ -3165,9 +3491,7 @@ export declare class Models extends BaseModule {
     /**
      * Generates an image based on a text description and configuration.
      *
-     * @param model - The model to use.
-     * @param prompt - A text description of the image to generate.
-     * @param [config] - The config for image generation.
+     * @param params - The parameters for generating images.
      * @return The response from the API.
      *
      * @example
@@ -3184,6 +3508,49 @@ export declare class Models extends BaseModule {
      * ```
      */
     generateImages: (params: types.GenerateImagesParameters) => Promise<types.GenerateImagesResponse>;
+    list: (params?: types.ListModelsParameters) => Promise<Pager<types.Model>>;
+    /**
+     * Edits an image based on a prompt, list of reference images, and configuration.
+     *
+     * @param params - The parameters for editing an image.
+     * @return The response from the API.
+     *
+     * @example
+     * ```ts
+     * const response = await client.models.editImage({
+     *  model: 'imagen-3.0-capability-001',
+     *  prompt: 'Generate an image containing a mug with the product logo [1] visible on the side of the mug.',
+     *  referenceImages: [subjectReferenceImage]
+     *  config: {
+     *    numberOfImages: 1,
+     *    includeRaiReason: true,
+     *  },
+     * });
+     * console.log(response?.generatedImages?.[0]?.image?.imageBytes);
+     * ```
+     */
+    editImage: (params: types.EditImageParameters) => Promise<types.EditImageResponse>;
+    /**
+     * Upscales an image based on an image, upscale factor, and configuration.
+     * Only supported in Vertex AI currently.
+     *
+     * @param params - The parameters for upscaling an image.
+     * @return The response from the API.
+     *
+     * @example
+     * ```ts
+     * const response = await client.models.upscaleImage({
+     *  model: 'imagen-3.0-generate-002',
+     *  image: image,
+     *  upscaleFactor: 'x2',
+     *  config: {
+     *    includeRaiReason: true,
+     *  },
+     * });
+     * console.log(response?.generatedImages?.[0]?.image?.imageBytes);
+     * ```
+     */
+    upscaleImage: (params: types.UpscaleImageParameters) => Promise<types.UpscaleImageResponse>;
     private generateContentInternal;
     private generateContentStreamInternal;
     /**
@@ -3228,6 +3595,8 @@ export declare class Models extends BaseModule {
      * ```
      */
     private generateImagesInternal;
+    private editImageInternal;
+    private upscaleImageInternal;
     /**
      * Fetches information about a model by name.
      *
@@ -3237,6 +3606,7 @@ export declare class Models extends BaseModule {
      * ```
      */
     get(params: types.GetModelParameters): Promise<types.Model>;
+    private listInternal;
     /**
      * Updates a tuned model by its name.
      *
@@ -3520,6 +3890,8 @@ export declare interface Part {
     videoMetadata?: VideoMetadata;
     /** Indicates if the part is thought from the model. */
     thought?: boolean;
+    /** Optional. Inlined bytes data. */
+    inlineData?: Blob_2;
     /** Optional. Result of executing the [ExecutableCode]. */
     codeExecutionResult?: CodeExecutionResult;
     /** Optional. Code generated by the model that is meant to be executed. */
@@ -3530,8 +3902,6 @@ export declare interface Part {
     functionCall?: FunctionCall;
     /** Optional. The result output of a [FunctionCall] that contains a string representing the [FunctionDeclaration.name] and a structured JSON object containing any output from the function call. It is used as context to the model. */
     functionResponse?: FunctionResponse;
-    /** Optional. Inlined bytes data. */
-    inlineData?: Blob_2;
     /** Optional. Text part (can be code). */
     text?: string;
 }
@@ -3618,13 +3988,15 @@ export declare interface RagRetrievalConfigRankingRankService {
  It can optionally be provided in addition to a mask reference image or
  a style reference image.
  */
-export declare interface RawReferenceImage {
+export declare class RawReferenceImage {
     /** The reference image for the editing operation. */
     referenceImage?: Image_2;
     /** The id of the reference image. */
     referenceId?: number;
     /** The type of the reference image. Only set by the SDK. */
     referenceType?: string;
+    /** Internal method to convert to ReferenceImageAPIInternal. */
+    toReferenceImageAPI(): any;
 }
 
 /** Marks the end of user activity.
@@ -3641,6 +4013,8 @@ export declare interface RealtimeInputConfig {
     turnCoverage?: TurnCoverage;
 }
 
+export declare type ReferenceImage = RawReferenceImage | MaskReferenceImage | ControlReferenceImage | StyleReferenceImage | SubjectReferenceImage;
+
 /** Represents a recorded session. */
 export declare interface ReplayFile {
     replayId?: string;
@@ -3679,6 +4053,13 @@ export declare interface Retrieval {
     vertexRagStore?: VertexRagStore;
 }
 
+/** Retrieval config.
+ */
+export declare interface RetrievalConfig {
+    /** Optional. The location of the user. */
+    latLng?: LatLng;
+}
+
 /** Metadata related to retrieval in the grounding flow. */
 export declare interface RetrievalMetadata {
     /** Optional. Score indicating how likely information from Google Search could help answer the prompt. The score is in the range `[0, 1]`, where 0 is the least likely and 1 is the most likely. This score is only populated when Google Search grounding and dynamic retrieval is enabled. It will be compared to the threshold to determine whether to trigger Google Search. */
@@ -4031,7 +4412,7 @@ export declare interface StyleReferenceConfig {
  A raw reference image can also be provided as a destination for the style to
  be applied to.
  */
-export declare interface StyleReferenceImage {
+export declare class StyleReferenceImage {
     /** The reference image for the editing operation. */
     referenceImage?: Image_2;
     /** The id of the reference image. */
@@ -4040,6 +4421,8 @@ export declare interface StyleReferenceImage {
     referenceType?: string;
     /** Configuration for the style reference image. */
     config?: StyleReferenceConfig;
+    /** Internal method to convert to ReferenceImageAPIInternal. */
+    toReferenceImageAPI(): any;
 }
 
 /** Configuration for a Subject reference image. */
@@ -4058,7 +4441,7 @@ export declare interface SubjectReferenceConfig {
  A raw reference image can also be provided as a destination for the subject to
  be applied to.
  */
-export declare interface SubjectReferenceImage {
+export declare class SubjectReferenceImage {
     /** The reference image for the editing operation. */
     referenceImage?: Image_2;
     /** The id of the reference image. */
@@ -4067,6 +4450,7 @@ export declare interface SubjectReferenceImage {
     referenceType?: string;
     /** Configuration for the subject reference image. */
     config?: SubjectReferenceConfig;
+    toReferenceImageAPI(): any;
 }
 
 /** Enum representing the subject type of a subject reference image. */
@@ -4153,6 +4537,8 @@ export declare interface SupervisedTuningSpec {
     trainingDatasetUri?: string;
     /** Optional. Cloud Storage path to file containing validation dataset for tuning. The dataset must be formatted as a JSONL file. */
     validationDatasetUri?: string;
+    /** Optional. If set to true, disable intermediate checkpoints for SFT and only the last checkpoint will be exported. */
+    exportLastCheckpointOnly?: boolean;
 }
 
 export declare interface TestTableFile {
@@ -4210,6 +4596,12 @@ export declare interface Tool {
     googleSearch?: GoogleSearch;
     /** Optional. GoogleSearchRetrieval tool type. Specialized retrieval tool that is powered by Google search. */
     googleSearchRetrieval?: GoogleSearchRetrieval;
+    /** Optional. Enterprise web search tool type. Specialized retrieval
+     tool that is powered by Vertex AI Search and Sec4 compliance. */
+    enterpriseWebSearch?: EnterpriseWebSearch;
+    /** Optional. Google Maps tool type. Specialized retrieval tool
+     that is powered by Google Maps. */
+    googleMaps?: GoogleMaps;
     /** 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. */
@@ -4227,6 +4619,8 @@ export declare interface ToolCodeExecution {
 export declare interface ToolConfig {
     /** Optional. Function calling config. */
     functionCallingConfig?: FunctionCallingConfig;
+    /** Optional. Retrieval config. */
+    retrievalConfig?: RetrievalConfig;
 }
 
 export declare type ToolListUnion = Tool[];
@@ -4253,6 +4647,27 @@ export declare interface TunedModel {
     model?: string;
     /** Output only. A resource name of an Endpoint. Format: `projects/{project}/locations/{location}/endpoints/{endpoint}`. */
     endpoint?: string;
+    /** The checkpoints associated with this TunedModel.
+     This field is only populated for tuning jobs that enable intermediate
+     checkpoints. */
+    checkpoints?: TunedModelCheckpoint[];
+}
+
+/** TunedModelCheckpoint for the Tuned Model of a Tuning Job. */
+export declare interface TunedModelCheckpoint {
+    /** The ID of the checkpoint.
+     */
+    checkpointId?: string;
+    /** The epoch of the checkpoint.
+     */
+    epoch?: string;
+    /** The step of the checkpoint.
+     */
+    step?: string;
+    /** The Endpoint resource name that the checkpoint is deployed to.
+     Format: `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+     */
+    endpoint?: string;
 }
 
 /** A tuned machine learning model. */
@@ -4409,6 +4824,7 @@ declare namespace types {
         HarmBlockMethod,
         HarmBlockThreshold,
         Mode,
+        AuthType,
         Type,
         FinishReason,
         HarmProbability,
@@ -4425,23 +4841,24 @@ declare namespace types {
         SafetyFilterLevel,
         PersonGeneration,
         ImagePromptLanguage,
-        FileState,
-        FileSource,
         MaskReferenceMode,
         ControlReferenceType,
         SubjectReferenceType,
+        EditMode,
+        FileState,
+        FileSource,
         MediaModality,
         StartSensitivity,
         EndSensitivity,
         ActivityHandling,
         TurnCoverage,
+        Blob_2 as Blob,
         VideoMetadata,
         CodeExecutionResult,
         ExecutableCode,
         FileData,
         FunctionCall,
         FunctionResponse,
-        Blob_2 as Blob,
         Part,
         Content,
         HttpOptions,
@@ -4450,6 +4867,14 @@ declare namespace types {
         GoogleSearch,
         DynamicRetrievalConfig,
         GoogleSearchRetrieval,
+        EnterpriseWebSearch,
+        ApiKeyConfig,
+        AuthConfigGoogleServiceAccountConfig,
+        AuthConfigHttpBasicAuthConfig,
+        AuthConfigOauthConfig,
+        AuthConfigOidcConfig,
+        AuthConfig,
+        GoogleMaps,
         VertexAISearch,
         VertexRagStoreRagResource,
         RagRetrievalConfigFilter,
@@ -4465,6 +4890,8 @@ declare namespace types {
         FunctionDeclaration,
         Tool,
         FunctionCallingConfig,
+        LatLng,
+        RetrievalConfig,
         ToolConfig,
         PrebuiltVoiceConfig,
         VoiceConfig,
@@ -4495,6 +4922,8 @@ declare namespace types {
         ModalityTokenCount,
         GenerateContentResponseUsageMetadata,
         GenerateContentResponse,
+        ReferenceImage,
+        EditImageParameters,
         EmbedContentConfig,
         EmbedContentParameters,
         ContentEmbeddingStatistics,
@@ -4507,11 +4936,22 @@ declare namespace types {
         SafetyAttributes,
         GeneratedImage,
         GenerateImagesResponse,
+        MaskReferenceConfig,
+        ControlReferenceConfig,
+        StyleReferenceConfig,
+        SubjectReferenceConfig,
+        EditImageConfig,
+        EditImageResponse,
+        UpscaleImageResponse,
         GetModelConfig,
         GetModelParameters,
         Endpoint,
         TunedModelInfo,
+        Checkpoint,
         Model,
+        ListModelsConfig,
+        ListModelsParameters,
+        ListModelsResponse,
         UpdateModelConfig,
         UpdateModelParameters,
         DeleteModelConfig,
@@ -4533,6 +4973,7 @@ declare namespace types {
         GenerateVideosOperation,
         GetTuningJobConfig,
         GetTuningJobParameters,
+        TunedModelCheckpoint,
         TunedModel,
         GoogleRpcStatus,
         SupervisedHyperParameters,
@@ -4601,16 +5042,13 @@ declare namespace types {
         ReplayFile,
         UploadFileConfig,
         DownloadFileConfig,
+        DownloadFileParameters,
         UpscaleImageConfig,
         UpscaleImageParameters,
         RawReferenceImage,
-        MaskReferenceConfig,
         MaskReferenceImage,
-        ControlReferenceConfig,
         ControlReferenceImage,
-        StyleReferenceConfig,
         StyleReferenceImage,
-        SubjectReferenceConfig,
         SubjectReferenceImage,
         LiveServerSetupComplete,
         Transcription,
@@ -4632,6 +5070,7 @@ declare namespace types {
         ActivityStart,
         ActivityEnd,
         LiveClientRealtimeInput,
+        LiveSendRealtimeInputParameters,
         LiveClientToolResponse,
         LiveClientMessage,
         LiveConnectConfig,
@@ -4639,7 +5078,6 @@ declare namespace types {
         CreateChatParameters,
         SendMessageParameters,
         LiveSendClientContentParameters,
-        LiveSendRealtimeInputParameters,
         LiveSendToolResponseParameters,
         OperationGetParameters,
         BlobImageUnion,
@@ -4649,7 +5087,8 @@ declare namespace types {
         ContentListUnion,
         SchemaUnion,
         SpeechConfigUnion,
-        ToolListUnion
+        ToolListUnion,
+        DownloadableFileUnion
     }
 }
 
@@ -4692,6 +5131,7 @@ export declare interface UpdateModelConfig {
     abortSignal?: AbortSignal;
     displayName?: string;
     description?: string;
+    defaultCheckpointId?: string;
 }
 
 /** Configuration for updating a tuned model. */
@@ -4788,6 +5228,11 @@ export declare interface UpscaleImageParameters {
     config?: UpscaleImageConfig;
 }
 
+export declare class UpscaleImageResponse {
+    /** Generated images. */
+    generatedImages?: GeneratedImage[];
+}
+
 /** Usage metadata about response(s). */
 export declare interface UsageMetadata {
     /** Number of tokens in the prompt. When `cached_content` is set, this is still the total effective prompt size meaning this includes the number of tokens in the cached content. */