@google/genai 0.2.0 → 0.4.0

package/dist/index.js

@@ -2,7 +2,7 @@
 
 /**
  * @license
- * Copyright 2024 Google LLC
+ * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 class BaseModule {
@@ -129,7 +129,7 @@ function getValueByPath(data, keys) {
 
 /**
  * @license
- * Copyright 2024 Google LLC
+ * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 function tModel(apiClient, model) {
@@ -354,9 +354,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);
@@ -497,14 +494,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";
@@ -519,6 +514,7 @@ var PagedItem;
 class Pager {
     constructor(name, request, response, params) {
         this.pageInternal = [];
+        this.paramsInternal = {};
         this.requestInternal = request;
         this.init(name, response, params);
     }
@@ -675,7 +671,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.
@@ -911,17 +907,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.
  */
@@ -1264,7 +1249,7 @@ class SessionSendToolResponseParameters {
 
 /**
  * @license
- * Copyright 2024 Google LLC
+ * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 class Caches extends BaseModule {
@@ -1279,7 +1264,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);
          * }
@@ -1298,7 +1283,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,
@@ -1369,7 +1354,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) {
@@ -1432,7 +1417,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) {
@@ -1459,7 +1444,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;
@@ -1484,7 +1469,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;
@@ -1499,7 +1484,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'}
      * });
@@ -1757,18 +1742,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);
@@ -1787,16 +1764,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);
@@ -1817,10 +1794,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);
@@ -1884,11 +1869,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;
 }
@@ -2387,11 +2372,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;
 }
@@ -2503,7 +2488,7 @@ typeof SuppressedError === "function" ? SuppressedError : function (error, suppr
 
 /**
  * @license
- * Copyright 2024 Google LLC
+ * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 /**
@@ -2608,16 +2593,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 = []) {
@@ -2641,6 +2645,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;
@@ -2672,6 +2685,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;
@@ -2763,13 +2787,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';
@@ -2981,7 +3005,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) => {
@@ -2996,7 +3020,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) => {
@@ -3141,7 +3165,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');
@@ -3263,7 +3287,7 @@ class CrossWebSocketFactory {
 
 /**
  * @license
- * Copyright 2024 Google LLC
+ * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 class Files extends BaseModule {
@@ -3281,7 +3305,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);
          * }
@@ -3314,63 +3338,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);
@@ -3385,7 +3389,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();
@@ -3399,34 +3403,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);
@@ -3441,7 +3423,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();
@@ -3465,37 +3447,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);
@@ -3510,7 +3472,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();
@@ -3542,13 +3504,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']);
@@ -3641,16 +3596,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']);
@@ -3663,16 +3608,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']);
@@ -3753,10 +3688,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, [
@@ -3778,10 +3709,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']);
@@ -3790,18 +3717,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 {
@@ -3831,14 +3750,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: {
@@ -3875,14 +3792,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: {
@@ -4041,14 +3956,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?',
@@ -4120,14 +4033,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: {
@@ -4198,14 +4109,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.'
      * });
@@ -4274,14 +4183,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?'
      * });
@@ -4289,7 +4196,7 @@ class Models extends BaseModule {
      * ```
      */
     async computeTokens(params) {
-        var _a, _b;
+        var _a;
         let response;
         let path = '';
         let queryParams = {};
@@ -4319,29 +4226,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.');
         }
     }
 }
@@ -4486,15 +4371,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.');
     }
@@ -4507,12 +4386,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);
@@ -4533,10 +4413,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);
@@ -4571,18 +4459,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);
@@ -4601,16 +4481,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);
@@ -4631,10 +4511,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);
@@ -4729,11 +4617,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;
 }
@@ -5682,21 +5570,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']);
@@ -6066,7 +5939,7 @@ function contentEmbeddingFromVertex(apiClient, fromObject) {
     }
     return toObject;
 }
-function embedContentMetadataFromMldev(apiClient, fromObject) {
+function embedContentMetadataFromMldev() {
     const toObject = {};
     return toObject;
 }
@@ -6153,6 +6026,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']);
@@ -6165,6 +6074,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) {
@@ -6179,6 +6092,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);
@@ -6241,14 +6158,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']);
@@ -6260,7 +6169,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.';
@@ -6521,6 +6430,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.
@@ -6558,20 +6473,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],
@@ -6595,9 +6504,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;
@@ -6673,8 +6579,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));
                 }
@@ -6714,8 +6619,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)) {
@@ -6723,7 +6630,7 @@ class Session {
             }
         }
         const clientMessage = {
-            toolResponse: { functionResponses: functionResponses }
+            toolResponse: { functionResponses: functionResponses },
         };
         return clientMessage;
     }
@@ -6841,7 +6748,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],
@@ -6992,7 +6899,6 @@ exports.createPartFromFunctionCall = createPartFromFunctionCall;
 exports.createPartFromFunctionResponse = createPartFromFunctionResponse;
 exports.createPartFromText = createPartFromText;
 exports.createPartFromUri = createPartFromUri;
-exports.createPartFromVideoMetadata = createPartFromVideoMetadata;
 exports.createUserContent = createUserContent;
 exports.toolToMldev = toolToMldev$1;
 exports.toolToVertex = toolToVertex$1;