@google/genai 1.51.0 → 2.0.0

package/dist/vertex_internal/index.d.ts

@@ -1319,6 +1319,10 @@ declare interface CreateFileSearchStoreConfig {
     /** The human-readable display name for the file search store.
      */
     displayName?: string;
+    /** The embedding model to use for the FileSearchStore.
+     Format: `models/{model}`. If not specified, the default embedding model will be used.
+     */
+    embeddingModel?: string;
 }
 
 /** Config for file_search_stores.create parameters. */
@@ -1842,6 +1846,19 @@ declare interface DownloadFileParameters {
     config?: DownloadFileConfig;
 }
 
+/** Used to override the default configuration. */
+declare interface DownloadMediaConfig {
+    /** Used to override HTTP request options. */
+    httpOptions?: HttpOptions;
+    /** Abort signal which can be used to cancel the request.
+
+     NOTE: AbortSignal is a client-only operation. Using it to cancel an
+     operation will not cancel the request in the service. You will still
+     be charged usage for any applicable operations.
+     */
+    abortSignal?: AbortSignal;
+}
+
 /** Describes the options to customize dynamic retrieval. */
 declare interface DynamicRetrievalConfig {
     /** Optional. The threshold to be used in dynamic retrieval. If not set, a system default value is used. */
@@ -2314,6 +2331,8 @@ declare interface FileSearchStore {
     /** The size of raw bytes ingested into the FileSearchStore. This is the
      total size of all the documents in the FileSearchStore. */
     sizeBytes?: string;
+    /** The embedding model used by the FileSearchStore. */
+    embeddingModel?: string;
 }
 
 /** Source of the File. */
@@ -3660,6 +3679,8 @@ declare interface GroundingChunkRetrievedContext {
     fileSearchStore?: string;
     /** Optional. Page number of the retrieved context. This field is not supported in Vertex AI. */
     pageNumber?: number;
+    /** Optional. Media ID. This field is not supported in Vertex AI. */
+    mediaId?: string;
 }
 
 /** A list of string values. This data type is not supported in Vertex AI. */
@@ -7804,6 +7825,7 @@ declare namespace types {
         EmbedContentParameters,
         UploadToFileSearchStoreResponse,
         UploadToFileSearchStoreOperation,
+        DownloadMediaConfig,
         BlobImageUnion,
         PartUnion,
         PartListUnion,

package/dist/vertex_internal/index.js

@@ -274,60 +274,6 @@ function _moveValueRecursive(data, sourceKeys, destKeys, keyIdx, excludeKeys) {
     }
 }
 
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-// Code generated by the Google Gen AI SDK generator DO NOT EDIT.
-function uploadToFileSearchStoreConfigToMldev(fromObject, parentObject) {
-    const toObject = {};
-    const fromMimeType = getValueByPath(fromObject, ['mimeType']);
-    if (parentObject !== undefined && fromMimeType != null) {
-        setValueByPath(parentObject, ['mimeType'], fromMimeType);
-    }
-    const fromDisplayName = getValueByPath(fromObject, ['displayName']);
-    if (parentObject !== undefined && fromDisplayName != null) {
-        setValueByPath(parentObject, ['displayName'], fromDisplayName);
-    }
-    const fromCustomMetadata = getValueByPath(fromObject, [
-        'customMetadata',
-    ]);
-    if (parentObject !== undefined && fromCustomMetadata != null) {
-        let transformedList = fromCustomMetadata;
-        if (Array.isArray(transformedList)) {
-            transformedList = transformedList.map((item) => {
-                return item;
-            });
-        }
-        setValueByPath(parentObject, ['customMetadata'], transformedList);
-    }
-    const fromChunkingConfig = getValueByPath(fromObject, [
-        'chunkingConfig',
-    ]);
-    if (parentObject !== undefined && fromChunkingConfig != null) {
-        setValueByPath(parentObject, ['chunkingConfig'], fromChunkingConfig);
-    }
-    return toObject;
-}
-
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-/**
- * API errors raised by the GenAI API.
- */
-class ApiError extends Error {
-    constructor(options) {
-        super(options.message);
-        this.name = 'ApiError';
-        this.status = options.status;
-        Object.setPrototypeOf(this, ApiError.prototype);
-    }
-}
-
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -1981,6 +1927,120 @@ class UploadToFileSearchStoreOperation {
     }
 }
 
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+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(fromName) {
+    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];
+    }
+    return name;
+}
+
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+function uploadToFileSearchStoreConfigToMldev(fromObject, parentObject) {
+    const toObject = {};
+    const fromMimeType = getValueByPath(fromObject, ['mimeType']);
+    if (parentObject !== undefined && fromMimeType != null) {
+        setValueByPath(parentObject, ['mimeType'], fromMimeType);
+    }
+    const fromDisplayName = getValueByPath(fromObject, ['displayName']);
+    if (parentObject !== undefined && fromDisplayName != null) {
+        setValueByPath(parentObject, ['displayName'], fromDisplayName);
+    }
+    const fromCustomMetadata = getValueByPath(fromObject, [
+        'customMetadata',
+    ]);
+    if (parentObject !== undefined && fromCustomMetadata != null) {
+        let transformedList = fromCustomMetadata;
+        if (Array.isArray(transformedList)) {
+            transformedList = transformedList.map((item) => {
+                return item;
+            });
+        }
+        setValueByPath(parentObject, ['customMetadata'], transformedList);
+    }
+    const fromChunkingConfig = getValueByPath(fromObject, [
+        'chunkingConfig',
+    ]);
+    if (parentObject !== undefined && fromChunkingConfig != null) {
+        setValueByPath(parentObject, ['chunkingConfig'], fromChunkingConfig);
+    }
+    return toObject;
+}
+
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * API errors raised by the GenAI API.
+ */
+class ApiError extends Error {
+    constructor(options) {
+        super(options.message);
+        this.name = 'ApiError';
+        this.status = options.status;
+        Object.setPrototypeOf(this, ApiError.prototype);
+    }
+}
+
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -1990,7 +2050,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 = '1.51.0'; // x-release-please-version
+const SDK_VERSION = '2.0.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';
@@ -2255,6 +2315,24 @@ class ApiClient {
             const abortController = new AbortController();
             const signal = abortController.signal;
             if (httpOptions.timeout && (httpOptions === null || httpOptions === void 0 ? void 0 : httpOptions.timeout) > 0) {
+                // In Node > 18, the built-in fetch is backed by Undici. Undici sets a global
+                // dispatcher on the global scope which tracks its internal headersTimeout and
+                // bodyTimeout using Symbol properties.
+                const dispatcherSymbol = Symbol.for('undici.globalDispatcher.1');
+                const globalDispatcher = globalThis[dispatcherSymbol];
+                if (globalDispatcher) {
+                    const symbols = Object.getOwnPropertySymbols(globalDispatcher);
+                    for (const sym of symbols) {
+                        const desc = sym.description;
+                        if ((desc === null || desc === void 0 ? void 0 : desc.includes('headers timeout')) ||
+                            (desc === null || desc === void 0 ? void 0 : desc.includes('body timeout'))) {
+                            const currentTimeout = globalDispatcher[sym];
+                            if (typeof currentTimeout === 'number') {
+                                globalDispatcher[sym] = Math.max(currentTimeout, httpOptions.timeout);
+                            }
+                        }
+                    }
+                }
                 const timeoutHandle = setTimeout(() => abortController.abort(), httpOptions.timeout);
                 if (timeoutHandle &&
                     typeof timeoutHandle.unref ===
@@ -2288,7 +2366,7 @@ class ApiClient {
                 throw e;
             }
             else {
-                throw new Error(JSON.stringify(e));
+                throw new Error(`exception ${e} sending request`, { cause: e });
             }
         });
     }
@@ -2303,7 +2381,7 @@ class ApiClient {
                 throw e;
             }
             else {
-                throw new Error(JSON.stringify(e));
+                throw new Error(`exception ${e} sending request`, { cause: e });
             }
         });
     }
@@ -2431,11 +2509,14 @@ class ApiClient {
             for (const [key, value] of Object.entries(httpOptions.headers)) {
                 headers.append(key, value);
             }
-            // Append a timeout header if it is set, note that the timeout option is
-            // in milliseconds but the header is in seconds.
-            if (httpOptions.timeout && httpOptions.timeout > 0) {
-                headers.append(SERVER_TIMEOUT_HEADER, String(Math.ceil(httpOptions.timeout / 1000)));
-            }
+        }
+        // Append a timeout header if it is set, note that the timeout option is
+        // in milliseconds but the header is in seconds.
+        // NOTE: This is intentionally outside the httpOptions.headers guard above
+        // so that the X-Server-Timeout header is always sent whenever a timeout
+        // is configured, even if the caller did not supply any custom headers.
+        if ((httpOptions === null || httpOptions === void 0 ? void 0 : httpOptions.timeout) && httpOptions.timeout > 0) {
+            headers.append(SERVER_TIMEOUT_HEADER, String(Math.ceil(httpOptions.timeout / 1000)));
         }
         await this.clientOptions.auth.addAuthHeaders(headers, url);
         return headers;
@@ -2739,67 +2820,6 @@ function buildGoogleAuthOptions(googleAuthOptions) {
     }
 }
 
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-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(fromName) {
-    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];
-    }
-    return name;
-}
-
 /**
  * @license
  * Copyright 2025 Google LLC