@google/genai 0.11.0 → 0.13.0

package/dist/index.js

@@ -491,15 +491,81 @@ function tBytes(apiClient, fromImageBytes) {
     // TODO(b/389133914): Remove dummy bytes converter.
     return fromImageBytes;
 }
+function _isFile(origin) {
+    return (origin !== null &&
+        origin !== undefined &&
+        typeof origin === 'object' &&
+        'name' in origin);
+}
+function isGeneratedVideo(origin) {
+    return (origin !== null &&
+        origin !== undefined &&
+        typeof origin === 'object' &&
+        'video' in origin);
+}
+function isVideo(origin) {
+    return (origin !== null &&
+        origin !== undefined &&
+        typeof origin === 'object' &&
+        'uri' in origin);
+}
 function tFileName(apiClient, fromName) {
-    if (typeof fromName !== 'string') {
-        throw new Error('fromName must be a string');
+    var _a;
+    let name;
+    if (_isFile(fromName)) {
+        name = fromName.name;
+    }
+    if (isVideo(fromName)) {
+        name = fromName.uri;
+        if (name === undefined) {
+            return undefined;
+        }
+    }
+    if (isGeneratedVideo(fromName)) {
+        name = (_a = fromName.video) === null || _a === void 0 ? void 0 : _a.uri;
+        if (name === undefined) {
+            return undefined;
+        }
+    }
+    if (typeof fromName === 'string') {
+        name = fromName;
+    }
+    if (name === undefined) {
+        throw new Error('Could not extract file name from the provided input.');
+    }
+    if (name.startsWith('https://')) {
+        const suffix = name.split('files/')[1];
+        const match = suffix.match(/[a-z0-9]+/);
+        if (match === null) {
+            throw new Error(`Could not extract file name from URI ${name}`);
+        }
+        name = match[0];
+    }
+    else if (name.startsWith('files/')) {
+        name = name.split('files/')[1];
     }
-    // Remove the files/ prefx for MLdev urls to get the actual name of the file.
-    if (fromName.startsWith('files/')) {
-        return fromName.split('files/')[1];
+    return name;
+}
+function tModelsUrl(apiClient, baseModels) {
+    let res;
+    if (apiClient.isVertexAI()) {
+        res = baseModels ? 'publishers/google/models' : 'models';
+    }
+    else {
+        res = baseModels ? 'models' : 'tunedModels';
+    }
+    return res;
+}
+function tExtractModels(apiClient, response) {
+    for (const key of ['models', 'tunedModels', 'publisherModels']) {
+        if (hasField(response, key)) {
+            return response[key];
+        }
     }
-    return fromName;
+    return [];
+}
+function hasField(data, fieldName) {
+    return data !== null && typeof data === 'object' && fieldName in data;
 }
 
 /**
@@ -613,6 +679,12 @@ function toolToMldev$2(apiClient, fromObject) {
     if (fromGoogleSearchRetrieval != null) {
         setValueByPath(toObject, ['googleSearchRetrieval'], googleSearchRetrievalToMldev$2(apiClient, fromGoogleSearchRetrieval));
     }
+    if (getValueByPath(fromObject, ['enterpriseWebSearch']) !== undefined) {
+        throw new Error('enterpriseWebSearch parameter is not supported in Gemini API.');
+    }
+    if (getValueByPath(fromObject, ['googleMaps']) !== undefined) {
+        throw new Error('googleMaps parameter is not supported in Gemini API.');
+    }
     const fromCodeExecution = getValueByPath(fromObject, [
         'codeExecution',
     ]);
@@ -649,6 +721,9 @@ function toolConfigToMldev$1(apiClient, fromObject) {
     if (fromFunctionCallingConfig != null) {
         setValueByPath(toObject, ['functionCallingConfig'], functionCallingConfigToMldev$1(apiClient, fromFunctionCallingConfig));
     }
+    if (getValueByPath(fromObject, ['retrievalConfig']) !== undefined) {
+        throw new Error('retrievalConfig parameter is not supported in Gemini API.');
+    }
     return toObject;
 }
 function createCachedContentConfigToMldev(apiClient, fromObject, parentObject) {
@@ -871,6 +946,58 @@ function googleSearchRetrievalToVertex$2(apiClient, fromObject) {
     }
     return toObject;
 }
+function enterpriseWebSearchToVertex$2() {
+    const toObject = {};
+    return toObject;
+}
+function apiKeyConfigToVertex$2(apiClient, fromObject) {
+    const toObject = {};
+    const fromApiKeyString = getValueByPath(fromObject, ['apiKeyString']);
+    if (fromApiKeyString != null) {
+        setValueByPath(toObject, ['apiKeyString'], fromApiKeyString);
+    }
+    return toObject;
+}
+function authConfigToVertex$2(apiClient, fromObject) {
+    const toObject = {};
+    const fromApiKeyConfig = getValueByPath(fromObject, ['apiKeyConfig']);
+    if (fromApiKeyConfig != null) {
+        setValueByPath(toObject, ['apiKeyConfig'], apiKeyConfigToVertex$2(apiClient, fromApiKeyConfig));
+    }
+    const fromAuthType = getValueByPath(fromObject, ['authType']);
+    if (fromAuthType != null) {
+        setValueByPath(toObject, ['authType'], fromAuthType);
+    }
+    const fromGoogleServiceAccountConfig = getValueByPath(fromObject, [
+        'googleServiceAccountConfig',
+    ]);
+    if (fromGoogleServiceAccountConfig != null) {
+        setValueByPath(toObject, ['googleServiceAccountConfig'], fromGoogleServiceAccountConfig);
+    }
+    const fromHttpBasicAuthConfig = getValueByPath(fromObject, [
+        'httpBasicAuthConfig',
+    ]);
+    if (fromHttpBasicAuthConfig != null) {
+        setValueByPath(toObject, ['httpBasicAuthConfig'], fromHttpBasicAuthConfig);
+    }
+    const fromOauthConfig = getValueByPath(fromObject, ['oauthConfig']);
+    if (fromOauthConfig != null) {
+        setValueByPath(toObject, ['oauthConfig'], fromOauthConfig);
+    }
+    const fromOidcConfig = getValueByPath(fromObject, ['oidcConfig']);
+    if (fromOidcConfig != null) {
+        setValueByPath(toObject, ['oidcConfig'], fromOidcConfig);
+    }
+    return toObject;
+}
+function googleMapsToVertex$2(apiClient, fromObject) {
+    const toObject = {};
+    const fromAuthConfig = getValueByPath(fromObject, ['authConfig']);
+    if (fromAuthConfig != null) {
+        setValueByPath(toObject, ['authConfig'], authConfigToVertex$2(apiClient, fromAuthConfig));
+    }
+    return toObject;
+}
 function toolToVertex$2(apiClient, fromObject) {
     const toObject = {};
     const fromRetrieval = getValueByPath(fromObject, ['retrieval']);
@@ -887,6 +1014,16 @@ function toolToVertex$2(apiClient, fromObject) {
     if (fromGoogleSearchRetrieval != null) {
         setValueByPath(toObject, ['googleSearchRetrieval'], googleSearchRetrievalToVertex$2(apiClient, fromGoogleSearchRetrieval));
     }
+    const fromEnterpriseWebSearch = getValueByPath(fromObject, [
+        'enterpriseWebSearch',
+    ]);
+    if (fromEnterpriseWebSearch != null) {
+        setValueByPath(toObject, ['enterpriseWebSearch'], enterpriseWebSearchToVertex$2());
+    }
+    const fromGoogleMaps = getValueByPath(fromObject, ['googleMaps']);
+    if (fromGoogleMaps != null) {
+        setValueByPath(toObject, ['googleMaps'], googleMapsToVertex$2(apiClient, fromGoogleMaps));
+    }
     const fromCodeExecution = getValueByPath(fromObject, [
         'codeExecution',
     ]);
@@ -915,6 +1052,26 @@ function functionCallingConfigToVertex$1(apiClient, fromObject) {
     }
     return toObject;
 }
+function latLngToVertex$1(apiClient, fromObject) {
+    const toObject = {};
+    const fromLatitude = getValueByPath(fromObject, ['latitude']);
+    if (fromLatitude != null) {
+        setValueByPath(toObject, ['latitude'], fromLatitude);
+    }
+    const fromLongitude = getValueByPath(fromObject, ['longitude']);
+    if (fromLongitude != null) {
+        setValueByPath(toObject, ['longitude'], fromLongitude);
+    }
+    return toObject;
+}
+function retrievalConfigToVertex$1(apiClient, fromObject) {
+    const toObject = {};
+    const fromLatLng = getValueByPath(fromObject, ['latLng']);
+    if (fromLatLng != null) {
+        setValueByPath(toObject, ['latLng'], latLngToVertex$1(apiClient, fromLatLng));
+    }
+    return toObject;
+}
 function toolConfigToVertex$1(apiClient, fromObject) {
     const toObject = {};
     const fromFunctionCallingConfig = getValueByPath(fromObject, [
@@ -923,6 +1080,12 @@ function toolConfigToVertex$1(apiClient, fromObject) {
     if (fromFunctionCallingConfig != null) {
         setValueByPath(toObject, ['functionCallingConfig'], functionCallingConfigToVertex$1(apiClient, fromFunctionCallingConfig));
     }
+    const fromRetrievalConfig = getValueByPath(fromObject, [
+        'retrievalConfig',
+    ]);
+    if (fromRetrievalConfig != null) {
+        setValueByPath(toObject, ['retrievalConfig'], retrievalConfigToVertex$1(apiClient, fromRetrievalConfig));
+    }
     return toObject;
 }
 function createCachedContentConfigToVertex(apiClient, fromObject, parentObject) {
@@ -1402,6 +1565,17 @@ exports.Mode = void 0;
     Mode["MODE_UNSPECIFIED"] = "MODE_UNSPECIFIED";
     Mode["MODE_DYNAMIC"] = "MODE_DYNAMIC";
 })(exports.Mode || (exports.Mode = {}));
+/** Type of auth scheme. */
+exports.AuthType = void 0;
+(function (AuthType) {
+    AuthType["AUTH_TYPE_UNSPECIFIED"] = "AUTH_TYPE_UNSPECIFIED";
+    AuthType["NO_AUTH"] = "NO_AUTH";
+    AuthType["API_KEY_AUTH"] = "API_KEY_AUTH";
+    AuthType["HTTP_BASIC_AUTH"] = "HTTP_BASIC_AUTH";
+    AuthType["GOOGLE_SERVICE_ACCOUNT_AUTH"] = "GOOGLE_SERVICE_ACCOUNT_AUTH";
+    AuthType["OAUTH"] = "OAUTH";
+    AuthType["OIDC_AUTH"] = "OIDC_AUTH";
+})(exports.AuthType || (exports.AuthType = {}));
 /** Optional. The type of the data. */
 exports.Type = void 0;
 (function (Type) {
@@ -2009,6 +2183,8 @@ class EmbedContentResponse {
 /** The output images response. */
 class GenerateImagesResponse {
 }
+class ListModelsResponse {
+}
 class DeleteModelResponse {
 }
 /** Response for counting tokens. */
@@ -2056,6 +2232,70 @@ class DeleteFileResponse {
 /** Represents a single response in a replay. */
 class ReplayResponse {
 }
+/** Response message for API call. */
+class LiveServerMessage {
+    /**
+     * 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() {
+        var _a, _b, _c;
+        let text = '';
+        let anyTextPartFound = false;
+        const nonTextParts = [];
+        for (const part of (_c = (_b = (_a = this.serverContent) === null || _a === void 0 ? void 0 : _a.modelTurn) === null || _b === void 0 ? void 0 : _b.parts) !== null && _c !== void 0 ? _c : []) {
+            for (const [fieldName, fieldValue] of Object.entries(part)) {
+                if (fieldName !== 'text' &&
+                    fieldName !== 'thought' &&
+                    fieldValue !== null) {
+                    nonTextParts.push(fieldName);
+                }
+            }
+            if (typeof part.text === 'string') {
+                if (typeof part.thought === 'boolean' && part.thought) {
+                    continue;
+                }
+                anyTextPartFound = true;
+                text += part.text;
+            }
+        }
+        if (nonTextParts.length > 0) {
+            console.warn(`there are non-text parts ${nonTextParts} in the response, returning concatenation of all text parts. Please refer to the non text parts for a full response from model.`);
+        }
+        // part.text === '' is different from part.text is null
+        return anyTextPartFound ? text : 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() {
+        var _a, _b, _c;
+        let data = '';
+        const nonDataParts = [];
+        for (const part of (_c = (_b = (_a = this.serverContent) === null || _a === void 0 ? void 0 : _a.modelTurn) === null || _b === void 0 ? void 0 : _b.parts) !== null && _c !== void 0 ? _c : []) {
+            for (const [fieldName, fieldValue] of Object.entries(part)) {
+                if (fieldName !== 'inlineData' && fieldValue !== null) {
+                    nonDataParts.push(fieldName);
+                }
+            }
+            if (part.inlineData && typeof part.inlineData.data === 'string') {
+                data += atob(part.inlineData.data);
+            }
+        }
+        if (nonDataParts.length > 0) {
+            console.warn(`there are non-data parts ${nonDataParts} in the response, returning concatenation of all data parts. Please refer to the non data parts for a full response from model.`);
+        }
+        return data.length > 0 ? btoa(data) : undefined;
+    }
+}
 /** Client generated response to a `ToolCall` received from the server.
 
   Individual `FunctionResponse` objects are matched to the respective
@@ -2817,7 +3057,7 @@ const CONTENT_TYPE_HEADER = 'Content-Type';
 const SERVER_TIMEOUT_HEADER = 'X-Server-Timeout';
 const USER_AGENT_HEADER = 'User-Agent';
 const GOOGLE_API_CLIENT_HEADER = 'x-goog-api-client';
-const SDK_VERSION = '0.11.0'; // x-release-please-version
+const SDK_VERSION = '0.13.0'; // x-release-please-version
 const LIBRARY_LABEL = `google-genai-sdk/${SDK_VERSION}`;
 const VERTEX_AI_API_DEFAULT_VERSION = 'v1beta1';
 const GOOGLE_AI_API_DEFAULT_VERSION = 'v1beta';
@@ -2864,18 +3104,11 @@ class ApiClient {
         if (this.clientOptions.vertexai) {
             initHttpOptions.apiVersion =
                 (_a = this.clientOptions.apiVersion) !== null && _a !== void 0 ? _a : VERTEX_AI_API_DEFAULT_VERSION;
-            // Assume that proj/api key validation occurs before they are passed in.
-            if (this.getProject() || this.getLocation()) {
-                initHttpOptions.baseUrl = `https://${this.clientOptions.location}-aiplatform.googleapis.com/`;
-                this.clientOptions.apiKey = undefined; // unset API key.
-            }
-            else {
-                initHttpOptions.baseUrl = `https://aiplatform.googleapis.com/`;
-                this.clientOptions.project = undefined; // unset project.
-                this.clientOptions.location = undefined; // unset location.
-            }
+            initHttpOptions.baseUrl = this.baseUrlFromProjectLocation();
+            this.normalizeAuthParameters();
         }
         else {
+            // Gemini API
             initHttpOptions.apiVersion =
                 (_b = this.clientOptions.apiVersion) !== null && _b !== void 0 ? _b : GOOGLE_AI_API_DEFAULT_VERSION;
             initHttpOptions.baseUrl = `https://generativelanguage.googleapis.com/`;
@@ -2886,6 +3119,39 @@ class ApiClient {
             this.clientOptions.httpOptions = this.patchHttpOptions(initHttpOptions, opts.httpOptions);
         }
     }
+    /**
+     * 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
+     */
+    baseUrlFromProjectLocation() {
+        if (this.clientOptions.project &&
+            this.clientOptions.location &&
+            this.clientOptions.location !== 'global') {
+            // Regional endpoint
+            return `https://${this.clientOptions.location}-aiplatform.googleapis.com/`;
+        }
+        // Global endpoint (covers 'global' location and API key usage)
+        return `https://aiplatform.googleapis.com/`;
+    }
+    /**
+     * 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
+     */
+    normalizeAuthParameters() {
+        if (this.clientOptions.project && this.clientOptions.location) {
+            // Using project/location for auth, clear potential API key
+            this.clientOptions.apiKey = undefined;
+            return;
+        }
+        // Using API key for auth (or no auth provided yet), clear project/location
+        this.clientOptions.project = undefined;
+        this.clientOptions.location = undefined;
+    }
     isVertexAI() {
         var _a;
         return (_a = this.clientOptions.vertexai) !== null && _a !== void 0 ? _a : false;
@@ -3223,6 +3489,16 @@ class ApiClient {
         const uploadUrl = await this.fetchUploadUrl(fileToUpload, config);
         return uploader.upload(file, uploadUrl, this);
     }
+    /**
+     * Downloads a file asynchronously to the specified path.
+     *
+     * @params params - The parameters for the download request, see {@link
+     * DownloadFileParameters}
+     */
+    async downloadFile(params) {
+        const downloader = this.clientOptions.downloader;
+        await downloader.download(params, this);
+    }
     async fetchUploadUrl(file, config) {
         var _a;
         let httpOptions = {};
@@ -3309,7 +3585,22 @@ function crossError() {
 `);
 }
 
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class CrossDownloader {
+    async download(_params, _apiClient) {
+        throw crossError();
+    }
+}
+
 const MAX_CHUNK_SIZE = 1024 * 1024 * 8; // bytes
+const MAX_RETRY_COUNT = 3;
+const INITIAL_RETRY_DELAY_MS = 1000;
+const DELAY_MULTIPLIER = 2;
+const X_GOOG_UPLOAD_STATUS_HEADER_FIELD = 'x-goog-upload-status';
 class CrossUploader {
     async upload(file, uploadUrl, apiClient) {
         if (typeof file === 'string') {
@@ -3329,7 +3620,7 @@ class CrossUploader {
     }
 }
 async function uploadBlob(file, uploadUrl, apiClient) {
-    var _a, _b;
+    var _a, _b, _c;
     let fileSize = 0;
     let offset = 0;
     let response = new HttpResponse(new Response());
@@ -3341,24 +3632,34 @@ async function uploadBlob(file, uploadUrl, apiClient) {
         if (offset + chunkSize >= fileSize) {
             uploadCommand += ', finalize';
         }
-        response = await apiClient.request({
-            path: '',
-            body: chunk,
-            httpMethod: 'POST',
-            httpOptions: {
-                apiVersion: '',
-                baseUrl: uploadUrl,
-                headers: {
-                    'X-Goog-Upload-Command': uploadCommand,
-                    'X-Goog-Upload-Offset': String(offset),
-                    'Content-Length': String(chunkSize),
+        let retryCount = 0;
+        let currentDelayMs = INITIAL_RETRY_DELAY_MS;
+        while (retryCount < MAX_RETRY_COUNT) {
+            response = await apiClient.request({
+                path: '',
+                body: chunk,
+                httpMethod: 'POST',
+                httpOptions: {
+                    apiVersion: '',
+                    baseUrl: uploadUrl,
+                    headers: {
+                        'X-Goog-Upload-Command': uploadCommand,
+                        'X-Goog-Upload-Offset': String(offset),
+                        'Content-Length': String(chunkSize),
+                    },
                 },
-            },
-        });
+            });
+            if ((_a = response === null || response === void 0 ? void 0 : response.headers) === null || _a === void 0 ? void 0 : _a[X_GOOG_UPLOAD_STATUS_HEADER_FIELD]) {
+                break;
+            }
+            retryCount++;
+            await sleep(currentDelayMs);
+            currentDelayMs = currentDelayMs * DELAY_MULTIPLIER;
+        }
         offset += chunkSize;
         // The `x-goog-upload-status` header field can be `active`, `final` and
         //`cancelled` in resposne.
-        if (((_a = response === null || response === void 0 ? void 0 : response.headers) === null || _a === void 0 ? void 0 : _a['x-goog-upload-status']) !== 'active') {
+        if (((_b = response === null || response === void 0 ? void 0 : response.headers) === null || _b === void 0 ? void 0 : _b[X_GOOG_UPLOAD_STATUS_HEADER_FIELD]) !== 'active') {
             break;
         }
         // TODO(b/401391430) Investigate why the upload status is not finalized
@@ -3368,7 +3669,7 @@ async function uploadBlob(file, uploadUrl, apiClient) {
         }
     }
     const responseJson = (await (response === null || response === void 0 ? void 0 : response.json()));
-    if (((_b = response === null || response === void 0 ? void 0 : response.headers) === null || _b === void 0 ? void 0 : _b['x-goog-upload-status']) !== 'final') {
+    if (((_c = response === null || response === void 0 ? void 0 : response.headers) === null || _c === void 0 ? void 0 : _c[X_GOOG_UPLOAD_STATUS_HEADER_FIELD]) !== 'final') {
         throw new Error('Failed to upload file: Upload status is not finalized.');
     }
     return responseJson['file'];
@@ -3377,6 +3678,9 @@ async function getBlobStat(file) {
     const fileStat = { size: file.size, type: file.type };
     return fileStat;
 }
+function sleep(ms) {
+    return new Promise((resolvePromise) => setTimeout(resolvePromise, ms));
+}
 
 /**
  * @license
@@ -3725,6 +4029,25 @@ class Files extends BaseModule {
             return 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'});
+     * ```
+     */
+    async download(params) {
+        await this.apiClient.downloadFile(params);
+    }
     async listInternal(params) {
         var _a, _b;
         let response;
@@ -4083,6 +4406,58 @@ function googleSearchRetrievalToVertex$1(apiClient, fromObject) {
     }
     return toObject;
 }
+function enterpriseWebSearchToVertex$1() {
+    const toObject = {};
+    return toObject;
+}
+function apiKeyConfigToVertex$1(apiClient, fromObject) {
+    const toObject = {};
+    const fromApiKeyString = getValueByPath(fromObject, ['apiKeyString']);
+    if (fromApiKeyString != null) {
+        setValueByPath(toObject, ['apiKeyString'], fromApiKeyString);
+    }
+    return toObject;
+}
+function authConfigToVertex$1(apiClient, fromObject) {
+    const toObject = {};
+    const fromApiKeyConfig = getValueByPath(fromObject, ['apiKeyConfig']);
+    if (fromApiKeyConfig != null) {
+        setValueByPath(toObject, ['apiKeyConfig'], apiKeyConfigToVertex$1(apiClient, fromApiKeyConfig));
+    }
+    const fromAuthType = getValueByPath(fromObject, ['authType']);
+    if (fromAuthType != null) {
+        setValueByPath(toObject, ['authType'], fromAuthType);
+    }
+    const fromGoogleServiceAccountConfig = getValueByPath(fromObject, [
+        'googleServiceAccountConfig',
+    ]);
+    if (fromGoogleServiceAccountConfig != null) {
+        setValueByPath(toObject, ['googleServiceAccountConfig'], fromGoogleServiceAccountConfig);
+    }
+    const fromHttpBasicAuthConfig = getValueByPath(fromObject, [
+        'httpBasicAuthConfig',
+    ]);
+    if (fromHttpBasicAuthConfig != null) {
+        setValueByPath(toObject, ['httpBasicAuthConfig'], fromHttpBasicAuthConfig);
+    }
+    const fromOauthConfig = getValueByPath(fromObject, ['oauthConfig']);
+    if (fromOauthConfig != null) {
+        setValueByPath(toObject, ['oauthConfig'], fromOauthConfig);
+    }
+    const fromOidcConfig = getValueByPath(fromObject, ['oidcConfig']);
+    if (fromOidcConfig != null) {
+        setValueByPath(toObject, ['oidcConfig'], fromOidcConfig);
+    }
+    return toObject;
+}
+function googleMapsToVertex$1(apiClient, fromObject) {
+    const toObject = {};
+    const fromAuthConfig = getValueByPath(fromObject, ['authConfig']);
+    if (fromAuthConfig != null) {
+        setValueByPath(toObject, ['authConfig'], authConfigToVertex$1(apiClient, fromAuthConfig));
+    }
+    return toObject;
+}
 function toolToMldev$1(apiClient, fromObject) {
     const toObject = {};
     if (getValueByPath(fromObject, ['retrieval']) !== undefined) {
@@ -4098,6 +4473,12 @@ function toolToMldev$1(apiClient, fromObject) {
     if (fromGoogleSearchRetrieval != null) {
         setValueByPath(toObject, ['googleSearchRetrieval'], googleSearchRetrievalToMldev$1(apiClient, fromGoogleSearchRetrieval));
     }
+    if (getValueByPath(fromObject, ['enterpriseWebSearch']) !== undefined) {
+        throw new Error('enterpriseWebSearch parameter is not supported in Gemini API.');
+    }
+    if (getValueByPath(fromObject, ['googleMaps']) !== undefined) {
+        throw new Error('googleMaps parameter is not supported in Gemini API.');
+    }
     const fromCodeExecution = getValueByPath(fromObject, [
         'codeExecution',
     ]);
@@ -4128,6 +4509,16 @@ function toolToVertex$1(apiClient, fromObject) {
     if (fromGoogleSearchRetrieval != null) {
         setValueByPath(toObject, ['googleSearchRetrieval'], googleSearchRetrievalToVertex$1(apiClient, fromGoogleSearchRetrieval));
     }
+    const fromEnterpriseWebSearch = getValueByPath(fromObject, [
+        'enterpriseWebSearch',
+    ]);
+    if (fromEnterpriseWebSearch != null) {
+        setValueByPath(toObject, ['enterpriseWebSearch'], enterpriseWebSearchToVertex$1());
+    }
+    const fromGoogleMaps = getValueByPath(fromObject, ['googleMaps']);
+    if (fromGoogleMaps != null) {
+        setValueByPath(toObject, ['googleMaps'], googleMapsToVertex$1(apiClient, fromGoogleMaps));
+    }
     const fromCodeExecution = getValueByPath(fromObject, [
         'codeExecution',
     ]);
@@ -4393,8 +4784,11 @@ function liveConnectConfigToMldev(apiClient, fromObject, parentObject) {
     if (parentObject !== undefined && fromSessionResumption != null) {
         setValueByPath(parentObject, ['setup', 'sessionResumption'], sessionResumptionConfigToMldev(apiClient, fromSessionResumption));
     }
-    if (getValueByPath(fromObject, ['inputAudioTranscription']) !== undefined) {
-        throw new Error('inputAudioTranscription parameter is not supported in Gemini API.');
+    const fromInputAudioTranscription = getValueByPath(fromObject, [
+        'inputAudioTranscription',
+    ]);
+    if (parentObject !== undefined && fromInputAudioTranscription != null) {
+        setValueByPath(parentObject, ['setup', 'inputAudioTranscription'], audioTranscriptionConfigToMldev());
     }
     const fromOutputAudioTranscription = getValueByPath(fromObject, [
         'outputAudioTranscription',
@@ -4791,6 +5185,12 @@ function liveServerContentFromMldev(apiClient, fromObject) {
     if (fromInterrupted != null) {
         setValueByPath(toObject, ['interrupted'], fromInterrupted);
     }
+    const fromGroundingMetadata = getValueByPath(fromObject, [
+        'groundingMetadata',
+    ]);
+    if (fromGroundingMetadata != null) {
+        setValueByPath(toObject, ['groundingMetadata'], fromGroundingMetadata);
+    }
     const fromGenerationComplete = getValueByPath(fromObject, [
         'generationComplete',
     ]);
@@ -4825,6 +5225,12 @@ function liveServerContentFromVertex(apiClient, fromObject) {
     if (fromInterrupted != null) {
         setValueByPath(toObject, ['interrupted'], fromInterrupted);
     }
+    const fromGroundingMetadata = getValueByPath(fromObject, [
+        'groundingMetadata',
+    ]);
+    if (fromGroundingMetadata != null) {
+        setValueByPath(toObject, ['groundingMetadata'], fromGroundingMetadata);
+    }
     const fromGenerationComplete = getValueByPath(fromObject, [
         'generationComplete',
     ]);
@@ -5388,6 +5794,12 @@ function toolToMldev(apiClient, fromObject) {
     if (fromGoogleSearchRetrieval != null) {
         setValueByPath(toObject, ['googleSearchRetrieval'], googleSearchRetrievalToMldev(apiClient, fromGoogleSearchRetrieval));
     }
+    if (getValueByPath(fromObject, ['enterpriseWebSearch']) !== undefined) {
+        throw new Error('enterpriseWebSearch parameter is not supported in Gemini API.');
+    }
+    if (getValueByPath(fromObject, ['googleMaps']) !== undefined) {
+        throw new Error('googleMaps parameter is not supported in Gemini API.');
+    }
     const fromCodeExecution = getValueByPath(fromObject, [
         'codeExecution',
     ]);
@@ -5424,6 +5836,9 @@ function toolConfigToMldev(apiClient, fromObject) {
     if (fromFunctionCallingConfig != null) {
         setValueByPath(toObject, ['functionCallingConfig'], functionCallingConfigToMldev(apiClient, fromFunctionCallingConfig));
     }
+    if (getValueByPath(fromObject, ['retrievalConfig']) !== undefined) {
+        throw new Error('retrievalConfig parameter is not supported in Gemini API.');
+    }
     return toObject;
 }
 function prebuiltVoiceConfigToMldev(apiClient, fromObject) {
@@ -5785,6 +6200,34 @@ function getModelParametersToMldev(apiClient, fromObject) {
     }
     return toObject;
 }
+function listModelsConfigToMldev(apiClient, fromObject, parentObject) {
+    const toObject = {};
+    const fromPageSize = getValueByPath(fromObject, ['pageSize']);
+    if (parentObject !== undefined && fromPageSize != null) {
+        setValueByPath(parentObject, ['_query', 'pageSize'], fromPageSize);
+    }
+    const fromPageToken = getValueByPath(fromObject, ['pageToken']);
+    if (parentObject !== undefined && fromPageToken != null) {
+        setValueByPath(parentObject, ['_query', 'pageToken'], fromPageToken);
+    }
+    const fromFilter = getValueByPath(fromObject, ['filter']);
+    if (parentObject !== undefined && fromFilter != null) {
+        setValueByPath(parentObject, ['_query', 'filter'], fromFilter);
+    }
+    const fromQueryBase = getValueByPath(fromObject, ['queryBase']);
+    if (parentObject !== undefined && fromQueryBase != null) {
+        setValueByPath(parentObject, ['_url', 'models_url'], tModelsUrl(apiClient, fromQueryBase));
+    }
+    return toObject;
+}
+function listModelsParametersToMldev(apiClient, fromObject) {
+    const toObject = {};
+    const fromConfig = getValueByPath(fromObject, ['config']);
+    if (fromConfig != null) {
+        setValueByPath(toObject, ['config'], listModelsConfigToMldev(apiClient, fromConfig, toObject));
+    }
+    return toObject;
+}
 function updateModelConfigToMldev(apiClient, fromObject, parentObject) {
     const toObject = {};
     const fromDisplayName = getValueByPath(fromObject, ['displayName']);
@@ -6061,6 +6504,58 @@ function googleSearchRetrievalToVertex(apiClient, fromObject) {
     }
     return toObject;
 }
+function enterpriseWebSearchToVertex() {
+    const toObject = {};
+    return toObject;
+}
+function apiKeyConfigToVertex(apiClient, fromObject) {
+    const toObject = {};
+    const fromApiKeyString = getValueByPath(fromObject, ['apiKeyString']);
+    if (fromApiKeyString != null) {
+        setValueByPath(toObject, ['apiKeyString'], fromApiKeyString);
+    }
+    return toObject;
+}
+function authConfigToVertex(apiClient, fromObject) {
+    const toObject = {};
+    const fromApiKeyConfig = getValueByPath(fromObject, ['apiKeyConfig']);
+    if (fromApiKeyConfig != null) {
+        setValueByPath(toObject, ['apiKeyConfig'], apiKeyConfigToVertex(apiClient, fromApiKeyConfig));
+    }
+    const fromAuthType = getValueByPath(fromObject, ['authType']);
+    if (fromAuthType != null) {
+        setValueByPath(toObject, ['authType'], fromAuthType);
+    }
+    const fromGoogleServiceAccountConfig = getValueByPath(fromObject, [
+        'googleServiceAccountConfig',
+    ]);
+    if (fromGoogleServiceAccountConfig != null) {
+        setValueByPath(toObject, ['googleServiceAccountConfig'], fromGoogleServiceAccountConfig);
+    }
+    const fromHttpBasicAuthConfig = getValueByPath(fromObject, [
+        'httpBasicAuthConfig',
+    ]);
+    if (fromHttpBasicAuthConfig != null) {
+        setValueByPath(toObject, ['httpBasicAuthConfig'], fromHttpBasicAuthConfig);
+    }
+    const fromOauthConfig = getValueByPath(fromObject, ['oauthConfig']);
+    if (fromOauthConfig != null) {
+        setValueByPath(toObject, ['oauthConfig'], fromOauthConfig);
+    }
+    const fromOidcConfig = getValueByPath(fromObject, ['oidcConfig']);
+    if (fromOidcConfig != null) {
+        setValueByPath(toObject, ['oidcConfig'], fromOidcConfig);
+    }
+    return toObject;
+}
+function googleMapsToVertex(apiClient, fromObject) {
+    const toObject = {};
+    const fromAuthConfig = getValueByPath(fromObject, ['authConfig']);
+    if (fromAuthConfig != null) {
+        setValueByPath(toObject, ['authConfig'], authConfigToVertex(apiClient, fromAuthConfig));
+    }
+    return toObject;
+}
 function toolToVertex(apiClient, fromObject) {
     const toObject = {};
     const fromRetrieval = getValueByPath(fromObject, ['retrieval']);
@@ -6077,6 +6572,16 @@ function toolToVertex(apiClient, fromObject) {
     if (fromGoogleSearchRetrieval != null) {
         setValueByPath(toObject, ['googleSearchRetrieval'], googleSearchRetrievalToVertex(apiClient, fromGoogleSearchRetrieval));
     }
+    const fromEnterpriseWebSearch = getValueByPath(fromObject, [
+        'enterpriseWebSearch',
+    ]);
+    if (fromEnterpriseWebSearch != null) {
+        setValueByPath(toObject, ['enterpriseWebSearch'], enterpriseWebSearchToVertex());
+    }
+    const fromGoogleMaps = getValueByPath(fromObject, ['googleMaps']);
+    if (fromGoogleMaps != null) {
+        setValueByPath(toObject, ['googleMaps'], googleMapsToVertex(apiClient, fromGoogleMaps));
+    }
     const fromCodeExecution = getValueByPath(fromObject, [
         'codeExecution',
     ]);
@@ -6105,6 +6610,26 @@ function functionCallingConfigToVertex(apiClient, fromObject) {
     }
     return toObject;
 }
+function latLngToVertex(apiClient, fromObject) {
+    const toObject = {};
+    const fromLatitude = getValueByPath(fromObject, ['latitude']);
+    if (fromLatitude != null) {
+        setValueByPath(toObject, ['latitude'], fromLatitude);
+    }
+    const fromLongitude = getValueByPath(fromObject, ['longitude']);
+    if (fromLongitude != null) {
+        setValueByPath(toObject, ['longitude'], fromLongitude);
+    }
+    return toObject;
+}
+function retrievalConfigToVertex(apiClient, fromObject) {
+    const toObject = {};
+    const fromLatLng = getValueByPath(fromObject, ['latLng']);
+    if (fromLatLng != null) {
+        setValueByPath(toObject, ['latLng'], latLngToVertex(apiClient, fromLatLng));
+    }
+    return toObject;
+}
 function toolConfigToVertex(apiClient, fromObject) {
     const toObject = {};
     const fromFunctionCallingConfig = getValueByPath(fromObject, [
@@ -6113,6 +6638,12 @@ function toolConfigToVertex(apiClient, fromObject) {
     if (fromFunctionCallingConfig != null) {
         setValueByPath(toObject, ['functionCallingConfig'], functionCallingConfigToVertex(apiClient, fromFunctionCallingConfig));
     }
+    const fromRetrievalConfig = getValueByPath(fromObject, [
+        'retrievalConfig',
+    ]);
+    if (fromRetrievalConfig != null) {
+        setValueByPath(toObject, ['retrievalConfig'], retrievalConfigToVertex(apiClient, fromRetrievalConfig));
+    }
     return toObject;
 }
 function prebuiltVoiceConfigToVertex(apiClient, fromObject) {
@@ -6491,6 +7022,34 @@ function getModelParametersToVertex(apiClient, fromObject) {
     }
     return toObject;
 }
+function listModelsConfigToVertex(apiClient, fromObject, parentObject) {
+    const toObject = {};
+    const fromPageSize = getValueByPath(fromObject, ['pageSize']);
+    if (parentObject !== undefined && fromPageSize != null) {
+        setValueByPath(parentObject, ['_query', 'pageSize'], fromPageSize);
+    }
+    const fromPageToken = getValueByPath(fromObject, ['pageToken']);
+    if (parentObject !== undefined && fromPageToken != null) {
+        setValueByPath(parentObject, ['_query', 'pageToken'], fromPageToken);
+    }
+    const fromFilter = getValueByPath(fromObject, ['filter']);
+    if (parentObject !== undefined && fromFilter != null) {
+        setValueByPath(parentObject, ['_query', 'filter'], fromFilter);
+    }
+    const fromQueryBase = getValueByPath(fromObject, ['queryBase']);
+    if (parentObject !== undefined && fromQueryBase != null) {
+        setValueByPath(parentObject, ['_url', 'models_url'], tModelsUrl(apiClient, fromQueryBase));
+    }
+    return toObject;
+}
+function listModelsParametersToVertex(apiClient, fromObject) {
+    const toObject = {};
+    const fromConfig = getValueByPath(fromObject, ['config']);
+    if (fromConfig != null) {
+        setValueByPath(toObject, ['config'], listModelsConfigToVertex(apiClient, fromConfig, toObject));
+    }
+    return toObject;
+}
 function updateModelConfigToVertex(apiClient, fromObject, parentObject) {
     const toObject = {};
     const fromDisplayName = getValueByPath(fromObject, ['displayName']);
@@ -7001,6 +7560,26 @@ function modelFromMldev(apiClient, fromObject) {
     }
     return toObject;
 }
+function listModelsResponseFromMldev(apiClient, fromObject) {
+    const toObject = {};
+    const fromNextPageToken = getValueByPath(fromObject, [
+        'nextPageToken',
+    ]);
+    if (fromNextPageToken != null) {
+        setValueByPath(toObject, ['nextPageToken'], fromNextPageToken);
+    }
+    const fromModels = getValueByPath(fromObject, ['_self']);
+    if (fromModels != null) {
+        let transformedList = tExtractModels(apiClient, fromModels);
+        if (Array.isArray(transformedList)) {
+            transformedList = transformedList.map((item) => {
+                return modelFromMldev(apiClient, item);
+            });
+        }
+        setValueByPath(toObject, ['models'], transformedList);
+    }
+    return toObject;
+}
 function deleteModelResponseFromMldev() {
     const toObject = {};
     return toObject;
@@ -7473,6 +8052,26 @@ function modelFromVertex(apiClient, fromObject) {
     }
     return toObject;
 }
+function listModelsResponseFromVertex(apiClient, fromObject) {
+    const toObject = {};
+    const fromNextPageToken = getValueByPath(fromObject, [
+        'nextPageToken',
+    ]);
+    if (fromNextPageToken != null) {
+        setValueByPath(toObject, ['nextPageToken'], fromNextPageToken);
+    }
+    const fromModels = getValueByPath(fromObject, ['_self']);
+    if (fromModels != null) {
+        let transformedList = tExtractModels(apiClient, fromModels);
+        if (Array.isArray(transformedList)) {
+            transformedList = transformedList.map((item) => {
+                return modelFromVertex(apiClient, item);
+            });
+        }
+        setValueByPath(toObject, ['models'], transformedList);
+    }
+    return toObject;
+}
 function deleteModelResponseFromVertex() {
     const toObject = {};
     return toObject;
@@ -7590,7 +8189,7 @@ const FUNCTION_RESPONSE_REQUIRES_ID = 'FunctionResponse request must have an `id
  * @param event The MessageEvent from the WebSocket.
  */
 async function handleWebSocketMessage(apiClient, onmessage, event) {
-    let serverMessage;
+    const serverMessage = new LiveServerMessage();
     let data;
     if (event.data instanceof Blob) {
         data = JSON.parse(await event.data.text());
@@ -7599,10 +8198,12 @@ async function handleWebSocketMessage(apiClient, onmessage, event) {
         data = JSON.parse(event.data);
     }
     if (apiClient.isVertexAI()) {
-        serverMessage = liveServerMessageFromVertex(apiClient, data);
+        const resp = liveServerMessageFromVertex(apiClient, data);
+        Object.assign(serverMessage, resp);
     }
     else {
-        serverMessage = liveServerMessageFromMldev(apiClient, data);
+        const resp = liveServerMessageFromMldev(apiClient, data);
+        Object.assign(serverMessage, resp);
     }
     onmessage(serverMessage);
 }
@@ -8109,6 +8710,27 @@ class Models extends BaseModule {
                 return response;
             });
         };
+        this.list = async (params) => {
+            var _a;
+            const defaultConfig = {
+                queryBase: true,
+            };
+            const actualConfig = Object.assign(Object.assign({}, defaultConfig), params === null || params === void 0 ? void 0 : params.config);
+            const actualParams = {
+                config: actualConfig,
+            };
+            if (this.apiClient.isVertexAI()) {
+                if (!actualParams.config.queryBase) {
+                    if ((_a = actualParams.config) === null || _a === void 0 ? void 0 : _a.filter) {
+                        throw new Error('Filtering tuned models list for Vertex AI is not currently supported');
+                    }
+                    else {
+                        actualParams.config.filter = 'labels.tune-type:*';
+                    }
+                }
+            }
+            return new Pager(exports.PagedItem.PAGED_ITEM_MODELS, (x) => this.listInternal(x), await this.listInternal(actualParams), actualParams);
+        };
     }
     async generateContentInternal(params) {
         var _a, _b, _c, _d;
@@ -8472,6 +9094,64 @@ class Models extends BaseModule {
             });
         }
     }
+    async listInternal(params) {
+        var _a, _b, _c, _d;
+        let response;
+        let path = '';
+        let queryParams = {};
+        if (this.apiClient.isVertexAI()) {
+            const body = listModelsParametersToVertex(this.apiClient, params);
+            path = formatMap('{models_url}', body['_url']);
+            queryParams = body['_query'];
+            delete body['config'];
+            delete body['_url'];
+            delete body['_query'];
+            response = this.apiClient
+                .request({
+                path: path,
+                queryParams: queryParams,
+                body: JSON.stringify(body),
+                httpMethod: 'GET',
+                httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
+            })
+                .then((httpResponse) => {
+                return httpResponse.json();
+            });
+            return response.then((apiResponse) => {
+                const resp = listModelsResponseFromVertex(this.apiClient, apiResponse);
+                const typedResp = new ListModelsResponse();
+                Object.assign(typedResp, resp);
+                return typedResp;
+            });
+        }
+        else {
+            const body = listModelsParametersToMldev(this.apiClient, params);
+            path = formatMap('{models_url}', body['_url']);
+            queryParams = body['_query'];
+            delete body['config'];
+            delete body['_url'];
+            delete body['_query'];
+            response = this.apiClient
+                .request({
+                path: path,
+                queryParams: queryParams,
+                body: JSON.stringify(body),
+                httpMethod: 'GET',
+                httpOptions: (_c = params.config) === null || _c === void 0 ? void 0 : _c.httpOptions,
+                abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
+            })
+                .then((httpResponse) => {
+                return httpResponse.json();
+            });
+            return response.then((apiResponse) => {
+                const resp = listModelsResponseFromMldev(this.apiClient, apiResponse);
+                const typedResp = new ListModelsResponse();
+                Object.assign(typedResp, resp);
+                return typedResp;
+            });
+        }
+    }
     /**
      * Updates a tuned model by its name.
      *
@@ -9962,6 +10642,7 @@ class GoogleGenAI {
             httpOptions: options.httpOptions,
             userAgentExtra: LANGUAGE_LABEL_PREFIX + 'cross',
             uploader: new CrossUploader(),
+            downloader: new CrossDownloader(),
         });
         this.models = new Models(this.apiClient);
         this.live = new Live(this.apiClient, auth, new CrossWebSocketFactory());
@@ -9994,10 +10675,12 @@ exports.GoogleGenAI = GoogleGenAI;
 exports.HttpResponse = HttpResponse;
 exports.ListCachedContentsResponse = ListCachedContentsResponse;
 exports.ListFilesResponse = ListFilesResponse;
+exports.ListModelsResponse = ListModelsResponse;
 exports.ListTuningJobsResponse = ListTuningJobsResponse;
 exports.Live = Live;
 exports.LiveClientToolResponse = LiveClientToolResponse;
 exports.LiveSendToolResponseParameters = LiveSendToolResponseParameters;
+exports.LiveServerMessage = LiveServerMessage;
 exports.Models = Models;
 exports.Operations = Operations;
 exports.Pager = Pager;