@google/genai 0.2.0 → 0.4.0

package/dist/index.mjs

@@ -1,6 +1,6 @@
 /**
  * @license
- * Copyright 2024 Google LLC
+ * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 class BaseModule {
@@ -127,7 +127,7 @@ function getValueByPath(data, keys) {
 
 /**
  * @license
- * Copyright 2024 Google LLC
+ * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 function tModel(apiClient, model) {
@@ -352,9 +352,6 @@ function processSchema(apiClient, schema) {
         }
     }
     if ('anyOf' in schema) {
-        if (!apiClient.isVertexAI()) {
-            throw new Error('AnyOf is not supported in the response schema for the Gemini API.');
-        }
         if (schema['anyOf'] !== undefined) {
             for (const subSchema of schema['anyOf']) {
                 processSchema(apiClient, subSchema);
@@ -495,14 +492,12 @@ function tFileName(apiClient, fromName) {
 
 /**
  * @license
- * Copyright 2024 Google LLC
+ * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 /**
  * @fileoverview Pagers for the GenAI List APIs.
  */
-// TODO(b/401555348): Re-enable linting after enforcing strong types.
-/*  eslint-disable @typescript-eslint/no-explicit-any */
 var PagedItem;
 (function (PagedItem) {
     PagedItem["PAGED_ITEM_BATCH_JOBS"] = "batchJobs";
@@ -517,6 +512,7 @@ var PagedItem;
 class Pager {
     constructor(name, request, response, params) {
         this.pageInternal = [];
+        this.paramsInternal = {};
         this.requestInternal = request;
         this.init(name, response, params);
     }
@@ -673,7 +669,7 @@ class Pager {
 
 /**
  * @license
- * Copyright 2024 Google LLC
+ * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 // Code generated by the Google Gen AI SDK generator DO NOT EDIT.
@@ -909,17 +905,6 @@ function createPartFromBase64(data, mimeType) {
         },
     };
 }
-/**
- * Creates a `Part` object from the `startOffset` and `endOffset` of a `VideoMetadata` object.
- */
-function createPartFromVideoMetadata(startOffset, endOffset) {
-    return {
-        videoMetadata: {
-            startOffset: startOffset,
-            endOffset: endOffset,
-        },
-    };
-}
 /**
  * Creates a `Part` object from the `outcome` and `output` of a `CodeExecutionResult` object.
  */
@@ -1262,7 +1247,7 @@ class SessionSendToolResponseParameters {
 
 /**
  * @license
- * Copyright 2024 Google LLC
+ * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 class Caches extends BaseModule {
@@ -1277,7 +1262,7 @@ class Caches extends BaseModule {
          *
          * @example
          * ```ts
-         * const cachedContents = await client.caches.list({config: {'pageSize': 2}});
+         * const cachedContents = await ai.caches.list({config: {'pageSize': 2}});
          * for (const cachedContent of cachedContents) {
          *   console.log(cachedContent);
          * }
@@ -1296,7 +1281,7 @@ class Caches extends BaseModule {
      * @example
      * ```ts
      * const contents = ...; // Initialize the content to cache.
-     * const response = await client.caches.create({
+     * const response = await ai.caches.create({
      *   model: 'gemini-2.0-flash',
      *   config: {
      *    'contents': contents,
@@ -1367,7 +1352,7 @@ class Caches extends BaseModule {
      *
      * @example
      * ```ts
-     * await client.caches.get({name: 'gemini-1.5-flash'});
+     * await ai.caches.get({name: 'gemini-1.5-flash'});
      * ```
      */
     async get(params) {
@@ -1430,7 +1415,7 @@ class Caches extends BaseModule {
      *
      * @example
      * ```ts
-     * await client.caches.delete({name: 'gemini-1.5-flash'});
+     * await ai.caches.delete({name: 'gemini-1.5-flash'});
      * ```
      */
     async delete(params) {
@@ -1457,7 +1442,7 @@ class Caches extends BaseModule {
                 return httpResponse.json();
             });
             return response.then((apiResponse) => {
-                const resp = deleteCachedContentResponseFromVertex(this.apiClient);
+                const resp = deleteCachedContentResponseFromVertex();
                 const typedResp = new DeleteCachedContentResponse();
                 Object.assign(typedResp, resp);
                 return typedResp;
@@ -1482,7 +1467,7 @@ class Caches extends BaseModule {
                 return httpResponse.json();
             });
             return response.then((apiResponse) => {
-                const resp = deleteCachedContentResponseFromMldev(this.apiClient);
+                const resp = deleteCachedContentResponseFromMldev();
                 const typedResp = new DeleteCachedContentResponse();
                 Object.assign(typedResp, resp);
                 return typedResp;
@@ -1497,7 +1482,7 @@ class Caches extends BaseModule {
      *
      * @example
      * ```ts
-     * const response = await client.caches.update({
+     * const response = await ai.caches.update({
      *   name: 'gemini-1.5-flash',
      *   config: {'ttl': '7600s'}
      * });
@@ -1755,18 +1740,10 @@ function schemaToVertex$1(apiClient, fromObject) {
     if (fromPattern != null) {
         setValueByPath(toObject, ['pattern'], fromPattern);
     }
-    const fromMinimum = getValueByPath(fromObject, ['minimum']);
-    if (fromMinimum != null) {
-        setValueByPath(toObject, ['minimum'], fromMinimum);
-    }
     const fromDefault = getValueByPath(fromObject, ['default']);
     if (fromDefault != null) {
         setValueByPath(toObject, ['default'], fromDefault);
     }
-    const fromAnyOf = getValueByPath(fromObject, ['anyOf']);
-    if (fromAnyOf != null) {
-        setValueByPath(toObject, ['anyOf'], fromAnyOf);
-    }
     const fromMaxLength = getValueByPath(fromObject, ['maxLength']);
     if (fromMaxLength != null) {
         setValueByPath(toObject, ['maxLength'], fromMaxLength);
@@ -1785,16 +1762,16 @@ function schemaToVertex$1(apiClient, fromObject) {
     if (fromMinProperties != null) {
         setValueByPath(toObject, ['minProperties'], fromMinProperties);
     }
-    const fromMaximum = getValueByPath(fromObject, ['maximum']);
-    if (fromMaximum != null) {
-        setValueByPath(toObject, ['maximum'], fromMaximum);
-    }
     const fromMaxProperties = getValueByPath(fromObject, [
         'maxProperties',
     ]);
     if (fromMaxProperties != null) {
         setValueByPath(toObject, ['maxProperties'], fromMaxProperties);
     }
+    const fromAnyOf = getValueByPath(fromObject, ['anyOf']);
+    if (fromAnyOf != null) {
+        setValueByPath(toObject, ['anyOf'], fromAnyOf);
+    }
     const fromDescription = getValueByPath(fromObject, ['description']);
     if (fromDescription != null) {
         setValueByPath(toObject, ['description'], fromDescription);
@@ -1815,10 +1792,18 @@ function schemaToVertex$1(apiClient, fromObject) {
     if (fromMaxItems != null) {
         setValueByPath(toObject, ['maxItems'], fromMaxItems);
     }
+    const fromMaximum = getValueByPath(fromObject, ['maximum']);
+    if (fromMaximum != null) {
+        setValueByPath(toObject, ['maximum'], fromMaximum);
+    }
     const fromMinItems = getValueByPath(fromObject, ['minItems']);
     if (fromMinItems != null) {
         setValueByPath(toObject, ['minItems'], fromMinItems);
     }
+    const fromMinimum = getValueByPath(fromObject, ['minimum']);
+    if (fromMinimum != null) {
+        setValueByPath(toObject, ['minimum'], fromMinimum);
+    }
     const fromNullable = getValueByPath(fromObject, ['nullable']);
     if (fromNullable != null) {
         setValueByPath(toObject, ['nullable'], fromNullable);
@@ -1882,11 +1867,11 @@ function functionDeclarationToVertex$1(apiClient, fromObject) {
     }
     return toObject;
 }
-function googleSearchToMldev$1(apiClient, fromObject) {
+function googleSearchToMldev$1() {
     const toObject = {};
     return toObject;
 }
-function googleSearchToVertex$1(apiClient, fromObject) {
+function googleSearchToVertex$1() {
     const toObject = {};
     return toObject;
 }
@@ -2385,11 +2370,11 @@ function cachedContentFromVertex(apiClient, fromObject) {
     }
     return toObject;
 }
-function deleteCachedContentResponseFromMldev(apiClient, fromObject) {
+function deleteCachedContentResponseFromMldev() {
     const toObject = {};
     return toObject;
 }
-function deleteCachedContentResponseFromVertex(apiClient, fromObject) {
+function deleteCachedContentResponseFromVertex() {
     const toObject = {};
     return toObject;
 }
@@ -2501,7 +2486,7 @@ typeof SuppressedError === "function" ? SuppressedError : function (error, suppr
 
 /**
  * @license
- * Copyright 2024 Google LLC
+ * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 /**
@@ -2606,16 +2591,35 @@ class Chats {
     /**
      * Creates a new chat session.
      *
+     * @remarks
+     * The config in the params will be used for all requests within the chat
+     * session unless overridden by a per-request `config` in
+     * {@link ./types.SendMessageParameters}.
+     *
      * @param params - Parameters for creating a chat session.
      * @returns A new chat session.
+     *
+     * @example
+     * ```ts
+     * const chat = ai.chats.create({
+     *   model: 'gemini-2.0-flash'
+     *   config: {
+     *     temperature: 0.5,
+     *     maxOutputTokens: 1024,
+     *   }
+     * });
+     * ```
      */
     create(params) {
         return new Chat(this.apiClient, this.modelsModule, params.model, params.config, params.history);
     }
 }
 /**
- * Chat session that enables sending messages and stores the chat history so
- * far.
+ * Chat session that enables sending messages to the model with previous
+ * conversation context.
+ *
+ * @remarks
+ * The session maintains all the turns between user and model.
  */
 class Chat {
     constructor(apiClient, modelsModule, model, config = {}, history = []) {
@@ -2639,6 +2643,15 @@ class Chat {
      * @see {@link Chat#sendMessageStream} for streaming method.
      * @param params - parameters for sending messages within a chat session.
      * @returns The model's response.
+     *
+     * @example
+     * ```ts
+     * const chat = ai.chats.create({model: 'gemini-2.0-flash'});
+     * const response = await chat.sendMessage({
+     *   message: 'Why is the sky blue?'
+     * });
+     * console.log(response.text);
+     * ```
      */
     async sendMessage(params) {
         var _a;
@@ -2670,6 +2683,17 @@ class Chat {
      * @see {@link Chat#sendMessage} for non-streaming method.
      * @param params - parameters for sending the message.
      * @return The model's response.
+     *
+     * @example
+     * ```ts
+     * const chat = ai.chats.create({model: 'gemini-2.0-flash'});
+     * const response = await chat.sendMessageStream({
+     *   message: 'Why is the sky blue?'
+     * });
+     * for await (const chunk of response) {
+     *   console.log(chunk.text);
+     * }
+     * ```
      */
     async sendMessageStream(params) {
         var _a;
@@ -2761,13 +2785,13 @@ class Chat {
 
 /**
  * @license
- * Copyright 2024 Google LLC
+ * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 const CONTENT_TYPE_HEADER = 'Content-Type';
 const USER_AGENT_HEADER = 'User-Agent';
 const GOOGLE_API_CLIENT_HEADER = 'x-goog-api-client';
-const SDK_VERSION = '0.2.0'; // x-release-please-version
+const SDK_VERSION = '0.4.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';
@@ -2979,7 +3003,7 @@ class ApiClient {
     async unaryApiCall(url, requestInit, httpMethod) {
         return this.apiCall(url.toString(), Object.assign(Object.assign({}, requestInit), { method: httpMethod }))
             .then(async (response) => {
-            await throwErrorIfNotOK(response, url.toString());
+            await throwErrorIfNotOK(response);
             return new HttpResponse(response);
         })
             .catch((e) => {
@@ -2994,7 +3018,7 @@ class ApiClient {
     async streamApiCall(url, requestInit, httpMethod) {
         return this.apiCall(url.toString(), Object.assign(Object.assign({}, requestInit), { method: httpMethod }))
             .then(async (response) => {
-            await throwErrorIfNotOK(response, url.toString());
+            await throwErrorIfNotOK(response);
             return this.processStreamResponse(response);
         })
             .catch((e) => {
@@ -3139,7 +3163,7 @@ class ApiClient {
         return uploadUrl;
     }
 }
-async function throwErrorIfNotOK(response, url, requestInit) {
+async function throwErrorIfNotOK(response) {
     var _a;
     if (response === undefined) {
         throw new ServerError('response is undefined');
@@ -3261,7 +3285,7 @@ class CrossWebSocketFactory {
 
 /**
  * @license
- * Copyright 2024 Google LLC
+ * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 class Files extends BaseModule {
@@ -3279,7 +3303,7 @@ class Files extends BaseModule {
          * size of each page is 10.
          *
          * ```ts
-         * const listResponse = await client.files.list({config: {'pageSize': 10}});
+         * const listResponse = await ai.files.list({config: {'pageSize': 10}});
          * for await (const file of listResponse) {
          *   console.log(file.name);
          * }
@@ -3312,63 +3336,43 @@ class Files extends BaseModule {
      *
      * This section can contain multiple paragraphs and code examples.
      *
-     * @param file The string path to the file to be uploaded or a Blob object.
-     * @param config Optional parameters specified in the `types.UploadFileConfig`
-     *     interface. Optional @see {@link types.UploadFileConfig}
+     * @param params - Optional parameters specified in the
+     *        `common.UploadFileParameters` interface.
+     *        Optional @see {@link common.UploadFileParameters}
      * @return A promise that resolves to a `types.File` object.
      * @throws An error if called on a Vertex AI client.
      * @throws An error if the `mimeType` is not provided and can not be inferred,
-     * the `mimeType` can be provided in the `config` parameter.
+     * the `mimeType` can be provided in the `params.config` parameter.
      * @throws An error occurs if a suitable upload location cannot be established.
      *
      * @example
      * The following code uploads a file to Gemini API.
      *
      * ```ts
-     * const file = await client.files.upload('file.txt', {
+     * const file = await ai.files.upload({file: 'file.txt', config: {
      *   mimeType: 'text/plain',
-     * });
+     * }});
      * console.log(file.name);
      * ```
      */
-    async upload(file, config) {
+    async upload(params) {
         if (this.apiClient.isVertexAI()) {
             throw new Error('Vertex AI does not support uploading files. You can share files through a GCS bucket.');
         }
-        return this.apiClient.uploadFile(file, config).then((response) => {
+        return this.apiClient
+            .uploadFile(params.file, params.config)
+            .then((response) => {
             const file = fileFromMldev(this.apiClient, response);
             return file;
         });
     }
     async listInternal(params) {
-        var _a, _b;
+        var _a;
         let response;
         let path = '';
         let queryParams = {};
         if (this.apiClient.isVertexAI()) {
-            const body = listFilesParametersToVertex(this.apiClient, params);
-            path = formatMap('None', 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,
-            })
-                .then((httpResponse) => {
-                return httpResponse.json();
-            });
-            return response.then((apiResponse) => {
-                const resp = listFilesResponseFromVertex(this.apiClient);
-                const typedResp = new ListFilesResponse();
-                Object.assign(typedResp, resp);
-                return typedResp;
-            });
+            throw new Error('This method is only supported by the Gemini Developer API.');
         }
         else {
             const body = listFilesParametersToMldev(this.apiClient, params);
@@ -3383,7 +3387,7 @@ class Files extends BaseModule {
                 queryParams: queryParams,
                 body: JSON.stringify(body),
                 httpMethod: 'GET',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
+                httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -3397,34 +3401,12 @@ class Files extends BaseModule {
         }
     }
     async createInternal(params) {
-        var _a, _b;
+        var _a;
         let response;
         let path = '';
         let queryParams = {};
         if (this.apiClient.isVertexAI()) {
-            const body = createFileParametersToVertex(this.apiClient, params);
-            path = formatMap('', 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: 'POST',
-                httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
-            })
-                .then((httpResponse) => {
-                return httpResponse.json();
-            });
-            return response.then((apiResponse) => {
-                const resp = createFileResponseFromVertex(this.apiClient, apiResponse);
-                const typedResp = new CreateFileResponse();
-                Object.assign(typedResp, resp);
-                return typedResp;
-            });
+            throw new Error('This method is only supported by the Gemini Developer API.');
         }
         else {
             const body = createFileParametersToMldev(this.apiClient, params);
@@ -3439,7 +3421,7 @@ class Files extends BaseModule {
                 queryParams: queryParams,
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
+                httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -3463,37 +3445,17 @@ class Files extends BaseModule {
      * const config: GetFileParameters = {
      *   name: fileName,
      * };
-     * file = await client.files.get(config);
+     * file = await ai.files.get(config);
      * console.log(file.name);
      * ```
      */
     async get(params) {
-        var _a, _b;
+        var _a;
         let response;
         let path = '';
         let queryParams = {};
         if (this.apiClient.isVertexAI()) {
-            const body = getFileParametersToVertex(this.apiClient, params);
-            path = formatMap('None', 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,
-            })
-                .then((httpResponse) => {
-                return httpResponse.json();
-            });
-            return response.then((apiResponse) => {
-                const resp = fileFromVertex(this.apiClient);
-                return resp;
-            });
+            throw new Error('This method is only supported by the Gemini Developer API.');
         }
         else {
             const body = getFileParametersToMldev(this.apiClient, params);
@@ -3508,7 +3470,7 @@ class Files extends BaseModule {
                 queryParams: queryParams,
                 body: JSON.stringify(body),
                 httpMethod: 'GET',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
+                httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -3540,13 +3502,6 @@ function listFilesParametersToMldev(apiClient, fromObject) {
     }
     return toObject;
 }
-function listFilesParametersToVertex(apiClient, fromObject) {
-    const toObject = {};
-    if (getValueByPath(fromObject, ['config']) !== undefined) {
-        throw new Error('config parameter is not supported in Vertex AI.');
-    }
-    return toObject;
-}
 function fileStatusToMldev(apiClient, fromObject) {
     const toObject = {};
     const fromDetails = getValueByPath(fromObject, ['details']);
@@ -3639,16 +3594,6 @@ function createFileParametersToMldev(apiClient, fromObject) {
     }
     return toObject;
 }
-function createFileParametersToVertex(apiClient, fromObject) {
-    const toObject = {};
-    if (getValueByPath(fromObject, ['file']) !== undefined) {
-        throw new Error('file parameter is not supported in Vertex AI.');
-    }
-    if (getValueByPath(fromObject, ['config']) !== undefined) {
-        throw new Error('config parameter is not supported in Vertex AI.');
-    }
-    return toObject;
-}
 function getFileParametersToMldev(apiClient, fromObject) {
     const toObject = {};
     const fromName = getValueByPath(fromObject, ['name']);
@@ -3661,16 +3606,6 @@ function getFileParametersToMldev(apiClient, fromObject) {
     }
     return toObject;
 }
-function getFileParametersToVertex(apiClient, fromObject) {
-    const toObject = {};
-    if (getValueByPath(fromObject, ['name']) !== undefined) {
-        throw new Error('name parameter is not supported in Vertex AI.');
-    }
-    if (getValueByPath(fromObject, ['config']) !== undefined) {
-        throw new Error('config parameter is not supported in Vertex AI.');
-    }
-    return toObject;
-}
 function fileStatusFromMldev(apiClient, fromObject) {
     const toObject = {};
     const fromDetails = getValueByPath(fromObject, ['details']);
@@ -3751,10 +3686,6 @@ function fileFromMldev(apiClient, fromObject) {
     }
     return toObject;
 }
-function fileFromVertex(apiClient, fromObject) {
-    const toObject = {};
-    return toObject;
-}
 function listFilesResponseFromMldev(apiClient, fromObject) {
     const toObject = {};
     const fromNextPageToken = getValueByPath(fromObject, [
@@ -3776,10 +3707,6 @@ function listFilesResponseFromMldev(apiClient, fromObject) {
     }
     return toObject;
 }
-function listFilesResponseFromVertex(apiClient, fromObject) {
-    const toObject = {};
-    return toObject;
-}
 function createFileResponseFromMldev(apiClient, fromObject) {
     const toObject = {};
     const fromHttpHeaders = getValueByPath(fromObject, ['httpHeaders']);
@@ -3788,18 +3715,10 @@ function createFileResponseFromMldev(apiClient, fromObject) {
     }
     return toObject;
 }
-function createFileResponseFromVertex(apiClient, fromObject) {
-    const toObject = {};
-    const fromHttpHeaders = getValueByPath(fromObject, ['httpHeaders']);
-    if (fromHttpHeaders != null) {
-        setValueByPath(toObject, ['httpHeaders'], fromHttpHeaders);
-    }
-    return toObject;
-}
 
 /**
  * @license
- * Copyright 2024 Google LLC
+ * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 class Models extends BaseModule {
@@ -3829,14 +3748,12 @@ class Models extends BaseModule {
          *
          * Some models support multimodal input and output.
          *
-         * @param model - The model to use for generating content.
-         * @param contents - The input contents to use for generating content.
-         * @param [config] - The configuration for generating content.
+         * @param params - The parameters for generating content.
          * @return The response from generating content.
          *
          * @example
          * ```ts
-         * const response = await client.models.generateContent({
+         * const response = await ai.models.generateContent({
          *   model: 'gemini-2.0-flash',
          *   contents: 'why is the sky blue?',
          *   config: {
@@ -3873,14 +3790,12 @@ class Models extends BaseModule {
          *
          * Some models support multimodal input and output.
          *
-         * @param model - The model to use for generating content.
-         * @param contents - The input contents to use for generating content.
-         * @param [config] - The configuration for generating content.
+         * @param params - The parameters for generating content with streaming response.
          * @return The response from generating content.
          *
          * @example
          * ```ts
-         * const response = await client.models.generateContentStream({
+         * const response = await ai.models.generateContentStream({
          *   model: 'gemini-2.0-flash',
          *   contents: 'why is the sky blue?',
          *   config: {
@@ -4039,14 +3954,12 @@ class Models extends BaseModule {
     /**
      * Calculates embeddings for the given contents. Only text is supported.
      *
-     * @param model - The model to use.
-     * @param contents - The contents to embed.
-     * @param [config] - The config for embedding contents.
+     * @param params - The parameters for embedding contents.
      * @return The response from the API.
      *
      * @example
      * ```ts
-     * const response = await client.models.embedContent({
+     * const response = await ai.models.embedContent({
      *  model: 'text-embedding-004',
      *  contents: [
      *    'What is your name?',
@@ -4118,14 +4031,12 @@ 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
      * ```ts
-     * const response = await client.models.generateImages({
+     * const response = await ai.models.generateImages({
      *  model: 'imagen-3.0-generate-002',
      *  prompt: 'Robot holding a red skateboard',
      *  config: {
@@ -4196,14 +4107,12 @@ class Models extends BaseModule {
      * Counts the number of tokens in the given contents. Multimodal input is
      * supported for Gemini models.
      *
-     * @param model - The model to use for counting tokens.
-     * @param contents - The contents to count tokens for.
-     * @param [config] - The config for counting tokens.
+     * @param params - The parameters for counting tokens.
      * @return The response from the API.
      *
      * @example
      * ```ts
-     * const response = await client.models.countTokens({
+     * const response = await ai.models.countTokens({
      *  model: 'gemini-2.0-flash',
      *  contents: 'The quick brown fox jumps over the lazy dog.'
      * });
@@ -4272,14 +4181,12 @@ class Models extends BaseModule {
      *
      * This method is not supported by the Gemini Developer API.
      *
-     * @param model - The model to use.
-     * @param contents - The content to compute tokens for.
-     * @param [config] - The config for computing tokens.
+     * @param params - The parameters for computing tokens.
      * @return The response from the API.
      *
      * @example
      * ```ts
-     * const response = await client.models.computeTokens({
+     * const response = await ai.models.computeTokens({
      *  model: 'gemini-2.0-flash',
      *  contents: 'What is your name?'
      * });
@@ -4287,7 +4194,7 @@ class Models extends BaseModule {
      * ```
      */
     async computeTokens(params) {
-        var _a, _b;
+        var _a;
         let response;
         let path = '';
         let queryParams = {};
@@ -4317,29 +4224,7 @@ class Models extends BaseModule {
             });
         }
         else {
-            const body = computeTokensParametersToMldev(this.apiClient, params);
-            path = formatMap('None', 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: 'POST',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
-            })
-                .then((httpResponse) => {
-                return httpResponse.json();
-            });
-            return response.then((apiResponse) => {
-                const resp = computeTokensResponseFromMldev(this.apiClient, apiResponse);
-                const typedResp = new ComputeTokensResponse();
-                Object.assign(typedResp, resp);
-                return typedResp;
-            });
+            throw new Error('This method is only supported by the Vertex AI.');
         }
     }
 }
@@ -4484,15 +4369,9 @@ function schemaToMldev(apiClient, fromObject) {
     if (getValueByPath(fromObject, ['pattern']) !== undefined) {
         throw new Error('pattern parameter is not supported in Gemini API.');
     }
-    if (getValueByPath(fromObject, ['minimum']) !== undefined) {
-        throw new Error('minimum parameter is not supported in Gemini API.');
-    }
     if (getValueByPath(fromObject, ['default']) !== undefined) {
         throw new Error('default parameter is not supported in Gemini API.');
     }
-    if (getValueByPath(fromObject, ['anyOf']) !== undefined) {
-        throw new Error('anyOf parameter is not supported in Gemini API.');
-    }
     if (getValueByPath(fromObject, ['maxLength']) !== undefined) {
         throw new Error('maxLength parameter is not supported in Gemini API.');
     }
@@ -4505,12 +4384,13 @@ function schemaToMldev(apiClient, fromObject) {
     if (getValueByPath(fromObject, ['minProperties']) !== undefined) {
         throw new Error('minProperties parameter is not supported in Gemini API.');
     }
-    if (getValueByPath(fromObject, ['maximum']) !== undefined) {
-        throw new Error('maximum parameter is not supported in Gemini API.');
-    }
     if (getValueByPath(fromObject, ['maxProperties']) !== undefined) {
         throw new Error('maxProperties parameter is not supported in Gemini API.');
     }
+    const fromAnyOf = getValueByPath(fromObject, ['anyOf']);
+    if (fromAnyOf != null) {
+        setValueByPath(toObject, ['anyOf'], fromAnyOf);
+    }
     const fromDescription = getValueByPath(fromObject, ['description']);
     if (fromDescription != null) {
         setValueByPath(toObject, ['description'], fromDescription);
@@ -4531,10 +4411,18 @@ function schemaToMldev(apiClient, fromObject) {
     if (fromMaxItems != null) {
         setValueByPath(toObject, ['maxItems'], fromMaxItems);
     }
+    const fromMaximum = getValueByPath(fromObject, ['maximum']);
+    if (fromMaximum != null) {
+        setValueByPath(toObject, ['maximum'], fromMaximum);
+    }
     const fromMinItems = getValueByPath(fromObject, ['minItems']);
     if (fromMinItems != null) {
         setValueByPath(toObject, ['minItems'], fromMinItems);
     }
+    const fromMinimum = getValueByPath(fromObject, ['minimum']);
+    if (fromMinimum != null) {
+        setValueByPath(toObject, ['minimum'], fromMinimum);
+    }
     const fromNullable = getValueByPath(fromObject, ['nullable']);
     if (fromNullable != null) {
         setValueByPath(toObject, ['nullable'], fromNullable);
@@ -4569,18 +4457,10 @@ function schemaToVertex(apiClient, fromObject) {
     if (fromPattern != null) {
         setValueByPath(toObject, ['pattern'], fromPattern);
     }
-    const fromMinimum = getValueByPath(fromObject, ['minimum']);
-    if (fromMinimum != null) {
-        setValueByPath(toObject, ['minimum'], fromMinimum);
-    }
     const fromDefault = getValueByPath(fromObject, ['default']);
     if (fromDefault != null) {
         setValueByPath(toObject, ['default'], fromDefault);
     }
-    const fromAnyOf = getValueByPath(fromObject, ['anyOf']);
-    if (fromAnyOf != null) {
-        setValueByPath(toObject, ['anyOf'], fromAnyOf);
-    }
     const fromMaxLength = getValueByPath(fromObject, ['maxLength']);
     if (fromMaxLength != null) {
         setValueByPath(toObject, ['maxLength'], fromMaxLength);
@@ -4599,16 +4479,16 @@ function schemaToVertex(apiClient, fromObject) {
     if (fromMinProperties != null) {
         setValueByPath(toObject, ['minProperties'], fromMinProperties);
     }
-    const fromMaximum = getValueByPath(fromObject, ['maximum']);
-    if (fromMaximum != null) {
-        setValueByPath(toObject, ['maximum'], fromMaximum);
-    }
     const fromMaxProperties = getValueByPath(fromObject, [
         'maxProperties',
     ]);
     if (fromMaxProperties != null) {
         setValueByPath(toObject, ['maxProperties'], fromMaxProperties);
     }
+    const fromAnyOf = getValueByPath(fromObject, ['anyOf']);
+    if (fromAnyOf != null) {
+        setValueByPath(toObject, ['anyOf'], fromAnyOf);
+    }
     const fromDescription = getValueByPath(fromObject, ['description']);
     if (fromDescription != null) {
         setValueByPath(toObject, ['description'], fromDescription);
@@ -4629,10 +4509,18 @@ function schemaToVertex(apiClient, fromObject) {
     if (fromMaxItems != null) {
         setValueByPath(toObject, ['maxItems'], fromMaxItems);
     }
+    const fromMaximum = getValueByPath(fromObject, ['maximum']);
+    if (fromMaximum != null) {
+        setValueByPath(toObject, ['maximum'], fromMaximum);
+    }
     const fromMinItems = getValueByPath(fromObject, ['minItems']);
     if (fromMinItems != null) {
         setValueByPath(toObject, ['minItems'], fromMinItems);
     }
+    const fromMinimum = getValueByPath(fromObject, ['minimum']);
+    if (fromMinimum != null) {
+        setValueByPath(toObject, ['minimum'], fromMinimum);
+    }
     const fromNullable = getValueByPath(fromObject, ['nullable']);
     if (fromNullable != null) {
         setValueByPath(toObject, ['nullable'], fromNullable);
@@ -4727,11 +4615,11 @@ function functionDeclarationToVertex(apiClient, fromObject) {
     }
     return toObject;
 }
-function googleSearchToMldev(apiClient, fromObject) {
+function googleSearchToMldev() {
     const toObject = {};
     return toObject;
 }
-function googleSearchToVertex(apiClient, fromObject) {
+function googleSearchToVertex() {
     const toObject = {};
     return toObject;
 }
@@ -5680,21 +5568,6 @@ function countTokensParametersToVertex(apiClient, fromObject) {
     }
     return toObject;
 }
-function computeTokensParametersToMldev(apiClient, fromObject) {
-    const toObject = {};
-    const fromModel = getValueByPath(fromObject, ['model']);
-    if (fromModel != null) {
-        setValueByPath(toObject, ['_url', 'model'], tModel(apiClient, fromModel));
-    }
-    if (getValueByPath(fromObject, ['contents']) !== undefined) {
-        throw new Error('contents parameter is not supported in Gemini API.');
-    }
-    const fromConfig = getValueByPath(fromObject, ['config']);
-    if (fromConfig != null) {
-        setValueByPath(toObject, ['config'], fromConfig);
-    }
-    return toObject;
-}
 function computeTokensParametersToVertex(apiClient, fromObject) {
     const toObject = {};
     const fromModel = getValueByPath(fromObject, ['model']);
@@ -6064,7 +5937,7 @@ function contentEmbeddingFromVertex(apiClient, fromObject) {
     }
     return toObject;
 }
-function embedContentMetadataFromMldev(apiClient, fromObject) {
+function embedContentMetadataFromMldev() {
     const toObject = {};
     return toObject;
 }
@@ -6151,6 +6024,42 @@ function imageFromVertex(apiClient, fromObject) {
     }
     return toObject;
 }
+function safetyAttributesFromMldev(apiClient, fromObject) {
+    const toObject = {};
+    const fromCategories = getValueByPath(fromObject, [
+        'safetyAttributes',
+        'categories',
+    ]);
+    if (fromCategories != null) {
+        setValueByPath(toObject, ['categories'], fromCategories);
+    }
+    const fromScores = getValueByPath(fromObject, [
+        'safetyAttributes',
+        'scores',
+    ]);
+    if (fromScores != null) {
+        setValueByPath(toObject, ['scores'], fromScores);
+    }
+    return toObject;
+}
+function safetyAttributesFromVertex(apiClient, fromObject) {
+    const toObject = {};
+    const fromCategories = getValueByPath(fromObject, [
+        'safetyAttributes',
+        'categories',
+    ]);
+    if (fromCategories != null) {
+        setValueByPath(toObject, ['categories'], fromCategories);
+    }
+    const fromScores = getValueByPath(fromObject, [
+        'safetyAttributes',
+        'scores',
+    ]);
+    if (fromScores != null) {
+        setValueByPath(toObject, ['scores'], fromScores);
+    }
+    return toObject;
+}
 function generatedImageFromMldev(apiClient, fromObject) {
     const toObject = {};
     const fromImage = getValueByPath(fromObject, ['_self']);
@@ -6163,6 +6072,10 @@ function generatedImageFromMldev(apiClient, fromObject) {
     if (fromRaiFilteredReason != null) {
         setValueByPath(toObject, ['raiFilteredReason'], fromRaiFilteredReason);
     }
+    const fromSafetyAttributes = getValueByPath(fromObject, ['_self']);
+    if (fromSafetyAttributes != null) {
+        setValueByPath(toObject, ['safetyAttributes'], safetyAttributesFromMldev(apiClient, fromSafetyAttributes));
+    }
     return toObject;
 }
 function generatedImageFromVertex(apiClient, fromObject) {
@@ -6177,6 +6090,10 @@ function generatedImageFromVertex(apiClient, fromObject) {
     if (fromRaiFilteredReason != null) {
         setValueByPath(toObject, ['raiFilteredReason'], fromRaiFilteredReason);
     }
+    const fromSafetyAttributes = getValueByPath(fromObject, ['_self']);
+    if (fromSafetyAttributes != null) {
+        setValueByPath(toObject, ['safetyAttributes'], safetyAttributesFromVertex(apiClient, fromSafetyAttributes));
+    }
     const fromEnhancedPrompt = getValueByPath(fromObject, ['prompt']);
     if (fromEnhancedPrompt != null) {
         setValueByPath(toObject, ['enhancedPrompt'], fromEnhancedPrompt);
@@ -6239,14 +6156,6 @@ function countTokensResponseFromVertex(apiClient, fromObject) {
     }
     return toObject;
 }
-function computeTokensResponseFromMldev(apiClient, fromObject) {
-    const toObject = {};
-    const fromTokensInfo = getValueByPath(fromObject, ['tokensInfo']);
-    if (fromTokensInfo != null) {
-        setValueByPath(toObject, ['tokensInfo'], fromTokensInfo);
-    }
-    return toObject;
-}
 function computeTokensResponseFromVertex(apiClient, fromObject) {
     const toObject = {};
     const fromTokensInfo = getValueByPath(fromObject, ['tokensInfo']);
@@ -6258,7 +6167,7 @@ function computeTokensResponseFromVertex(apiClient, fromObject) {
 
 /**
  * @license
- * Copyright 2024 Google LLC
+ * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 const FUNCTION_RESPONSE_REQUIRES_ID = 'FunctionResponse request must have an `id` field from the response of a ToolCall.FunctionalCalls in Google AI.';
@@ -6519,6 +6428,12 @@ function liveServerMessageFromVertex(apiClient, fromObject) {
 /**
  * Handles incoming messages from the WebSocket.
  *
+ * @remarks
+ * This function is responsible for parsing incoming messages, transforming them
+ * into LiveServerMessages, and then calling the onmessage callback. Note that
+ * the first message which is received from the server is a setupComplete
+ * message.
+ *
  * @param apiClient The ApiClient instance.
  * @param onmessage The user-provided onmessage callback (if any).
  * @param event The MessageEvent from the WebSocket.
@@ -6556,20 +6471,14 @@ class Live {
        Establishes a connection to the specified model with the given
        configuration and returns a Session object representing that connection.
   
-      > [!CAUTION] This SDK does not yet support the live API for **Google Vertex AI**.
-  
        @experimental
   
-       @param model - Model to use for the Live session.
-       @param config - Configuration parameters for the Live session.
-       @param callbacks - Optional callbacks for websocket events. If not
-           provided, default no-op callbacks will be used. Generally, prefer to
-           provide explicit callbacks to allow for proper handling of websocket
-           events (e.g. connection errors).
+       @param params - The parameters for establishing a connection to the model.
+       @return A live session.
   
        @example
        ```ts
-       const session = await client.live.connect({
+       const session = await ai.live.connect({
          model: 'gemini-2.0-flash-exp',
          config: {
            responseModalities: [Modality.AUDIO],
@@ -6593,9 +6502,6 @@ class Live {
       */
     async connect(params) {
         var _a, _b;
-        if (this.apiClient.isVertexAI()) {
-            throw new Error('This SDK does not yet support the live API for **Google Vertex AI**.');
-        }
         const websocketBaseUrl = this.apiClient.getWebsocketBaseUrl();
         const apiVersion = this.apiClient.getApiVersion();
         let url;
@@ -6671,8 +6577,7 @@ class Session {
         if (params.turns !== null && params.turns !== undefined) {
             let contents = [];
             try {
-                contents =
-                    tContents(apiClient, params.turns);
+                contents = tContents(apiClient, params.turns);
                 if (apiClient.isVertexAI()) {
                     contents = contents.map((item) => contentToVertex(apiClient, item));
                 }
@@ -6712,8 +6617,10 @@ class Session {
             throw new Error('functionResponses is required.');
         }
         for (const functionResponse of functionResponses) {
-            if (typeof functionResponse !== 'object' || functionResponse === null ||
-                !('name' in functionResponse) || !('response' in functionResponse)) {
+            if (typeof functionResponse !== 'object' ||
+                functionResponse === null ||
+                !('name' in functionResponse) ||
+                !('response' in functionResponse)) {
                 throw new Error(`Could not parse function response, type '${typeof functionResponse}'.`);
             }
             if (!apiClient.isVertexAI() && !('id' in functionResponse)) {
@@ -6721,7 +6628,7 @@ class Session {
             }
         }
         const clientMessage = {
-            toolResponse: { functionResponses: functionResponses }
+            toolResponse: { functionResponses: functionResponses },
         };
         return clientMessage;
     }
@@ -6839,7 +6746,7 @@ class Session {
   
        @example
        ```ts
-       const session = await client.live.connect({
+       const session = await ai.live.connect({
          model: 'gemini-2.0-flash-exp',
          config: {
            responseModalities: [Modality.AUDIO],
@@ -6957,5 +6864,5 @@ class GoogleGenAI {
     }
 }
 
-export { BlockedReason, Caches, Chat, Chats, ComputeTokensResponse, ControlReferenceType, CountTokensResponse, CreateFileResponse, DeleteCachedContentResponse, DynamicRetrievalConfigMode, EmbedContentResponse, FileSource, FileState, FinishReason, FunctionCallingConfigMode, FunctionResponse, GenerateContentResponse, GenerateContentResponsePromptFeedback, GenerateContentResponseUsageMetadata, GenerateImagesResponse, GoogleGenAI, HarmBlockMethod, HarmBlockThreshold, HarmCategory, HarmProbability, HarmSeverity, HttpResponse, ImagePromptLanguage, Language, ListCachedContentsResponse, ListFilesResponse, Live, LiveClientToolResponse, MaskReferenceMode, MediaResolution, Modality, Mode, Models, Outcome, PersonGeneration, ReplayResponse, SafetyFilterLevel, Session, SessionSendToolResponseParameters, State, SubjectReferenceType, Type, contentToMldev$1 as contentToMldev, contentToVertex$1 as contentToVertex, createModelContent, createPartFromBase64, createPartFromCodeExecutionResult, createPartFromExecutableCode, createPartFromFunctionCall, createPartFromFunctionResponse, createPartFromText, createPartFromUri, createPartFromVideoMetadata, createUserContent, toolToMldev$1 as toolToMldev, toolToVertex$1 as toolToVertex };
+export { BlockedReason, Caches, Chat, Chats, ComputeTokensResponse, ControlReferenceType, CountTokensResponse, CreateFileResponse, DeleteCachedContentResponse, DynamicRetrievalConfigMode, EmbedContentResponse, FileSource, FileState, FinishReason, FunctionCallingConfigMode, FunctionResponse, GenerateContentResponse, GenerateContentResponsePromptFeedback, GenerateContentResponseUsageMetadata, GenerateImagesResponse, GoogleGenAI, HarmBlockMethod, HarmBlockThreshold, HarmCategory, HarmProbability, HarmSeverity, HttpResponse, ImagePromptLanguage, Language, ListCachedContentsResponse, ListFilesResponse, Live, LiveClientToolResponse, MaskReferenceMode, MediaResolution, Modality, Mode, Models, Outcome, PersonGeneration, ReplayResponse, SafetyFilterLevel, Session, SessionSendToolResponseParameters, State, SubjectReferenceType, Type, contentToMldev$1 as contentToMldev, contentToVertex$1 as contentToVertex, createModelContent, createPartFromBase64, createPartFromCodeExecutionResult, createPartFromExecutableCode, createPartFromFunctionCall, createPartFromFunctionResponse, createPartFromText, createPartFromUri, createUserContent, toolToMldev$1 as toolToMldev, toolToVertex$1 as toolToVertex };
 //# sourceMappingURL=index.mjs.map