@google/genai 0.9.0 → 0.10.0

package/dist/node/index.js

@@ -24,6 +24,66 @@ function _interopNamespaceDefault(e) {
 var NodeWs__namespace = /*#__PURE__*/_interopNamespaceDefault(NodeWs);
 var fs__namespace = /*#__PURE__*/_interopNamespaceDefault(fs);
 
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+let _defaultBaseGeminiUrl = undefined;
+let _defaultBaseVertexUrl = undefined;
+/**
+ * Overrides the base URLs for the Gemini API and Vertex AI API.
+ *
+ * @remarks This function should be called before initializing the SDK. If the
+ * base URLs are set after initializing the SDK, the base URLs will not be
+ * updated. Base URLs provided in the HttpOptions will also take precedence over
+ * URLs set here.
+ *
+ * @example
+ * ```ts
+ * import {GoogleGenAI, setDefaultBaseUrls} from '@google/genai';
+ * // Override the base URL for the Gemini API.
+ * setDefaultBaseUrls({geminiUrl:'https://gemini.google.com'});
+ *
+ * // Override the base URL for the Vertex AI API.
+ * setDefaultBaseUrls({vertexUrl: 'https://vertexai.googleapis.com'});
+ *
+ * const ai = new GoogleGenAI({apiKey: 'GEMINI_API_KEY'});
+ * ```
+ */
+function setDefaultBaseUrls(baseUrlParams) {
+    _defaultBaseGeminiUrl = baseUrlParams.geminiUrl;
+    _defaultBaseVertexUrl = baseUrlParams.vertexUrl;
+}
+/**
+ * Returns the default base URLs for the Gemini API and Vertex AI API.
+ */
+function getDefaultBaseUrls() {
+    return {
+        geminiUrl: _defaultBaseGeminiUrl,
+        vertexUrl: _defaultBaseVertexUrl,
+    };
+}
+/**
+ * Returns the default base URL based on the following priority:
+ *   1. Base URLs set via HttpOptions.
+ *   2. Base URLs set via the latest call to setDefaultBaseUrls.
+ *   3. Base URLs set via environment variables.
+ */
+function getBaseUrl(options, vertexBaseUrlFromEnv, geminiBaseUrlFromEnv) {
+    var _a, _b, _c;
+    if (!((_a = options.httpOptions) === null || _a === void 0 ? void 0 : _a.baseUrl)) {
+        const defaultBaseUrls = getDefaultBaseUrls();
+        if (options.vertexai) {
+            return (_b = defaultBaseUrls.vertexUrl) !== null && _b !== void 0 ? _b : vertexBaseUrlFromEnv;
+        }
+        else {
+            return (_c = defaultBaseUrls.geminiUrl) !== null && _c !== void 0 ? _c : geminiBaseUrlFromEnv;
+        }
+    }
+    return options.httpOptions.baseUrl;
+}
+
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -199,6 +259,36 @@ function tCachesModel(apiClient, model) {
         return transformedModel;
     }
 }
+function tBlobs(apiClient, blobs) {
+    if (Array.isArray(blobs)) {
+        return blobs.map((blob) => tBlob(apiClient, blob));
+    }
+    else {
+        return [tBlob(apiClient, blobs)];
+    }
+}
+function tBlob(apiClient, blob) {
+    if (typeof blob === 'object' && blob !== null) {
+        return blob;
+    }
+    throw new Error(`Could not parse input as Blob. Unsupported blob type: ${typeof blob}`);
+}
+function tImageBlob(apiClient, blob) {
+    const transformedBlob = tBlob(apiClient, blob);
+    if (transformedBlob.mimeType &&
+        transformedBlob.mimeType.startsWith('image/')) {
+        return transformedBlob;
+    }
+    throw new Error(`Unsupported mime type: ${transformedBlob.mimeType}`);
+}
+function tAudioBlob(apiClient, blob) {
+    const transformedBlob = tBlob(apiClient, blob);
+    if (transformedBlob.mimeType &&
+        transformedBlob.mimeType.startsWith('audio/')) {
+        return transformedBlob;
+    }
+    throw new Error(`Unsupported mime type: ${transformedBlob.mimeType}`);
+}
 function tPart(apiClient, origin) {
     if (origin === null || origin === undefined) {
         throw new Error('PartUnion is required');
@@ -322,38 +412,11 @@ function tContents(apiClient, origin) {
     }
     return result;
 }
-function processSchema(apiClient, schema) {
-    if (!apiClient.isVertexAI()) {
-        if ('default' in schema) {
-            throw new Error('Default value is not supported in the response schema for the Gemini API.');
-        }
-    }
-    if ('anyOf' in schema) {
-        if (schema['anyOf'] !== undefined) {
-            for (const subSchema of schema['anyOf']) {
-                processSchema(apiClient, subSchema);
-            }
-        }
-    }
-    if ('items' in schema) {
-        if (schema['items'] !== undefined) {
-            processSchema(apiClient, schema['items']);
-        }
-    }
-    if ('properties' in schema) {
-        if (schema['properties'] !== undefined) {
-            for (const subSchema of Object.values(schema['properties'])) {
-                processSchema(apiClient, subSchema);
-            }
-        }
-    }
-}
 function tSchema(apiClient, schema) {
-    processSchema(apiClient, schema);
     return schema;
 }
 function tSpeechConfig(apiClient, speechConfig) {
-    if (typeof speechConfig === 'object' && 'voiceConfig' in speechConfig) {
+    if (typeof speechConfig === 'object') {
         return speechConfig;
     }
     else if (typeof speechConfig === 'string') {
@@ -1560,6 +1623,7 @@ exports.FinishReason = void 0;
     FinishReason["MAX_TOKENS"] = "MAX_TOKENS";
     FinishReason["SAFETY"] = "SAFETY";
     FinishReason["RECITATION"] = "RECITATION";
+    FinishReason["LANGUAGE"] = "LANGUAGE";
     FinishReason["OTHER"] = "OTHER";
     FinishReason["BLOCKLIST"] = "BLOCKLIST";
     FinishReason["PROHIBITED_CONTENT"] = "PROHIBITED_CONTENT";
@@ -1942,6 +2006,42 @@ class GenerateContentResponse {
         // part.text === '' is different from part.text is null
         return anyTextPartText ? text : undefined;
     }
+    /**
+     * Returns the concatenation of all inline data parts from the first candidate
+     * in the response.
+     *
+     * @remarks
+     * If there are multiple candidates in the response, the inline data from the
+     * first one will be returned. If there are non-inline data parts in the
+     * response, the concatenation of all inline data parts will be returned, and
+     * a warning will be logged.
+     */
+    get data() {
+        var _a, _b, _c, _d, _e, _f, _g, _h;
+        if (((_d = (_c = (_b = (_a = this.candidates) === null || _a === void 0 ? void 0 : _a[0]) === null || _b === void 0 ? void 0 : _b.content) === null || _c === void 0 ? void 0 : _c.parts) === null || _d === void 0 ? void 0 : _d.length) === 0) {
+            return undefined;
+        }
+        if (this.candidates && this.candidates.length > 1) {
+            console.warn('there are multiple candidates in the response, returning data from the first one.');
+        }
+        let data = '';
+        const nonDataParts = [];
+        for (const part of (_h = (_g = (_f = (_e = this.candidates) === null || _e === void 0 ? void 0 : _e[0]) === null || _f === void 0 ? void 0 : _f.content) === null || _g === void 0 ? void 0 : _g.parts) !== null && _h !== void 0 ? _h : []) {
+            for (const [fieldName, fieldValue] of Object.entries(part)) {
+                if (fieldName !== 'inlineData' &&
+                    (fieldValue !== null || fieldValue !== undefined)) {
+                    nonDataParts.push(fieldName);
+                }
+            }
+            if (part.inlineData && typeof part.inlineData.data === 'string') {
+                data += atob(part.inlineData.data);
+            }
+        }
+        if (nonDataParts.length > 0) {
+            console.warn(`there are non-data parts ${nonDataParts} in the response, returning concatenation of all data parts. Please refer to the non data parts for a full response from model.`);
+        }
+        return data.length > 0 ? btoa(data) : undefined;
+    }
     /**
      * Returns the function calls from the first candidate in the response.
      *
@@ -2197,7 +2297,7 @@ class Caches extends BaseModule {
      * ```
      */
     async create(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -2215,6 +2315,7 @@ class Caches extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -2237,7 +2338,8 @@ class Caches extends BaseModule {
                 queryParams: queryParams,
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
+                httpOptions: (_c = params.config) === null || _c === void 0 ? void 0 : _c.httpOptions,
+                abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -2260,7 +2362,7 @@ class Caches extends BaseModule {
      * ```
      */
     async get(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -2278,6 +2380,7 @@ class Caches extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'GET',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -2300,7 +2403,8 @@ class Caches extends BaseModule {
                 queryParams: queryParams,
                 body: JSON.stringify(body),
                 httpMethod: 'GET',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
+                httpOptions: (_c = params.config) === null || _c === void 0 ? void 0 : _c.httpOptions,
+                abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -2323,7 +2427,7 @@ class Caches extends BaseModule {
      * ```
      */
     async delete(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -2341,6 +2445,7 @@ class Caches extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'DELETE',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -2365,7 +2470,8 @@ class Caches extends BaseModule {
                 queryParams: queryParams,
                 body: JSON.stringify(body),
                 httpMethod: 'DELETE',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
+                httpOptions: (_c = params.config) === null || _c === void 0 ? void 0 : _c.httpOptions,
+                abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -2393,7 +2499,7 @@ class Caches extends BaseModule {
      * ```
      */
     async update(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -2411,6 +2517,7 @@ class Caches extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'PATCH',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -2433,7 +2540,8 @@ class Caches extends BaseModule {
                 queryParams: queryParams,
                 body: JSON.stringify(body),
                 httpMethod: 'PATCH',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
+                httpOptions: (_c = params.config) === null || _c === void 0 ? void 0 : _c.httpOptions,
+                abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -2445,7 +2553,7 @@ class Caches extends BaseModule {
         }
     }
     async listInternal(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -2463,6 +2571,7 @@ class Caches extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'GET',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -2487,7 +2596,8 @@ class Caches extends BaseModule {
                 queryParams: queryParams,
                 body: JSON.stringify(body),
                 httpMethod: 'GET',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
+                httpOptions: (_c = params.config) === null || _c === void 0 ? void 0 : _c.httpOptions,
+                abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -3196,7 +3306,7 @@ class Files extends BaseModule {
         });
     }
     async listInternal(params) {
-        var _a;
+        var _a, _b;
         let response;
         let path = '';
         let queryParams = {};
@@ -3217,6 +3327,7 @@ class Files extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'GET',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -3230,7 +3341,7 @@ class Files extends BaseModule {
         }
     }
     async createInternal(params) {
-        var _a;
+        var _a, _b;
         let response;
         let path = '';
         let queryParams = {};
@@ -3251,6 +3362,7 @@ class Files extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -3279,7 +3391,7 @@ class Files extends BaseModule {
      * ```
      */
     async get(params) {
-        var _a;
+        var _a, _b;
         let response;
         let path = '';
         let queryParams = {};
@@ -3300,6 +3412,7 @@ class Files extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'GET',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -3324,7 +3437,7 @@ class Files extends BaseModule {
      * ```
      */
     async delete(params) {
-        var _a;
+        var _a, _b;
         let response;
         let path = '';
         let queryParams = {};
@@ -3345,6 +3458,7 @@ class Files extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'DELETE',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -4155,6 +4269,91 @@ function liveConnectParametersToVertex(apiClient, fromObject) {
     }
     return toObject;
 }
+function activityStartToMldev() {
+    const toObject = {};
+    return toObject;
+}
+function activityStartToVertex() {
+    const toObject = {};
+    return toObject;
+}
+function activityEndToMldev() {
+    const toObject = {};
+    return toObject;
+}
+function activityEndToVertex() {
+    const toObject = {};
+    return toObject;
+}
+function liveSendRealtimeInputParametersToMldev(apiClient, fromObject) {
+    const toObject = {};
+    const fromMedia = getValueByPath(fromObject, ['media']);
+    if (fromMedia != null) {
+        setValueByPath(toObject, ['mediaChunks'], tBlobs(apiClient, fromMedia));
+    }
+    const fromAudio = getValueByPath(fromObject, ['audio']);
+    if (fromAudio != null) {
+        setValueByPath(toObject, ['audio'], tAudioBlob(apiClient, fromAudio));
+    }
+    const fromAudioStreamEnd = getValueByPath(fromObject, [
+        'audioStreamEnd',
+    ]);
+    if (fromAudioStreamEnd != null) {
+        setValueByPath(toObject, ['audioStreamEnd'], fromAudioStreamEnd);
+    }
+    const fromVideo = getValueByPath(fromObject, ['video']);
+    if (fromVideo != null) {
+        setValueByPath(toObject, ['video'], tImageBlob(apiClient, fromVideo));
+    }
+    const fromText = getValueByPath(fromObject, ['text']);
+    if (fromText != null) {
+        setValueByPath(toObject, ['text'], fromText);
+    }
+    const fromActivityStart = getValueByPath(fromObject, [
+        'activityStart',
+    ]);
+    if (fromActivityStart != null) {
+        setValueByPath(toObject, ['activityStart'], activityStartToMldev());
+    }
+    const fromActivityEnd = getValueByPath(fromObject, ['activityEnd']);
+    if (fromActivityEnd != null) {
+        setValueByPath(toObject, ['activityEnd'], activityEndToMldev());
+    }
+    return toObject;
+}
+function liveSendRealtimeInputParametersToVertex(apiClient, fromObject) {
+    const toObject = {};
+    const fromMedia = getValueByPath(fromObject, ['media']);
+    if (fromMedia != null) {
+        setValueByPath(toObject, ['mediaChunks'], tBlobs(apiClient, fromMedia));
+    }
+    if (getValueByPath(fromObject, ['audio']) !== undefined) {
+        throw new Error('audio parameter is not supported in Vertex AI.');
+    }
+    const fromAudioStreamEnd = getValueByPath(fromObject, [
+        'audioStreamEnd',
+    ]);
+    if (fromAudioStreamEnd != null) {
+        setValueByPath(toObject, ['audioStreamEnd'], fromAudioStreamEnd);
+    }
+    if (getValueByPath(fromObject, ['video']) !== undefined) {
+        throw new Error('video parameter is not supported in Vertex AI.');
+    }
+    if (getValueByPath(fromObject, ['text']) !== undefined) {
+        throw new Error('text parameter is not supported in Vertex AI.');
+    }
+    const fromActivityStart = getValueByPath(fromObject, [
+        'activityStart',
+    ]);
+    if (fromActivityStart != null) {
+        setValueByPath(toObject, ['activityStart'], activityStartToVertex());
+    }
+    const fromActivityEnd = getValueByPath(fromObject, ['activityEnd']);
+    if (fromActivityEnd != null) {
+        setValueByPath(toObject, ['activityEnd'], activityEndToVertex());
+    }
+    return toObject;
+}
 function liveServerSetupCompleteFromMldev() {
     const toObject = {};
     return toObject;
@@ -7502,21 +7701,6 @@ class Session {
             clientContent: { turnComplete: params.turnComplete },
         };
     }
-    tLiveClientRealtimeInput(apiClient, params) {
-        let clientMessage = {};
-        if (!('media' in params) || !params.media) {
-            throw new Error(`Failed to convert realtime input "media", type: '${typeof params.media}'`);
-        }
-        // LiveClientRealtimeInput
-        clientMessage = {
-            realtimeInput: {
-                mediaChunks: [params.media],
-                activityStart: params.activityStart,
-                activityEnd: params.activityEnd,
-            },
-        };
-        return clientMessage;
-    }
     tLiveClienttToolResponse(apiClient, params) {
         let functionResponses = [];
         if (params.functionResponses == null) {
@@ -7624,10 +7808,17 @@ class Session {
       of audio and image mimetypes are allowed.
      */
     sendRealtimeInput(params) {
-        if (params.media == null) {
-            throw new Error('Media is required.');
+        let clientMessage = {};
+        if (this.apiClient.isVertexAI()) {
+            clientMessage = {
+                'realtimeInput': liveSendRealtimeInputParametersToVertex(this.apiClient, params),
+            };
+        }
+        else {
+            clientMessage = {
+                'realtimeInput': liveSendRealtimeInputParametersToMldev(this.apiClient, params),
+            };
         }
-        const clientMessage = this.tLiveClientRealtimeInput(this.apiClient, params);
         this.conn.send(JSON.stringify(clientMessage));
     }
     /**
@@ -7848,7 +8039,7 @@ class Models extends BaseModule {
         };
     }
     async generateContentInternal(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -7866,6 +8057,7 @@ class Models extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -7890,7 +8082,8 @@ class Models extends BaseModule {
                 queryParams: queryParams,
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
+                httpOptions: (_c = params.config) === null || _c === void 0 ? void 0 : _c.httpOptions,
+                abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -7904,7 +8097,7 @@ class Models extends BaseModule {
         }
     }
     async generateContentStreamInternal(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -7922,6 +8115,7 @@ class Models extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             });
             return response.then(function (apiResponse) {
                 return __asyncGenerator(this, arguments, function* () {
@@ -7960,7 +8154,8 @@ class Models extends BaseModule {
                 queryParams: queryParams,
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
+                httpOptions: (_c = params.config) === null || _c === void 0 ? void 0 : _c.httpOptions,
+                abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
             });
             return response.then(function (apiResponse) {
                 return __asyncGenerator(this, arguments, function* () {
@@ -8009,7 +8204,7 @@ class Models extends BaseModule {
      * ```
      */
     async embedContent(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -8027,6 +8222,7 @@ class Models extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -8051,7 +8247,8 @@ class Models extends BaseModule {
                 queryParams: queryParams,
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
+                httpOptions: (_c = params.config) === null || _c === void 0 ? void 0 : _c.httpOptions,
+                abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -8084,7 +8281,7 @@ class Models extends BaseModule {
      * ```
      */
     async generateImagesInternal(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -8102,6 +8299,7 @@ class Models extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -8126,7 +8324,8 @@ class Models extends BaseModule {
                 queryParams: queryParams,
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
+                httpOptions: (_c = params.config) === null || _c === void 0 ? void 0 : _c.httpOptions,
+                abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -8148,7 +8347,7 @@ class Models extends BaseModule {
      * ```
      */
     async get(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -8166,6 +8365,7 @@ class Models extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'GET',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -8188,7 +8388,8 @@ class Models extends BaseModule {
                 queryParams: queryParams,
                 body: JSON.stringify(body),
                 httpMethod: 'GET',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
+                httpOptions: (_c = params.config) === null || _c === void 0 ? void 0 : _c.httpOptions,
+                abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -8216,7 +8417,7 @@ class Models extends BaseModule {
      * ```
      */
     async countTokens(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -8234,6 +8435,7 @@ class Models extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -8258,7 +8460,8 @@ class Models extends BaseModule {
                 queryParams: queryParams,
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
+                httpOptions: (_c = params.config) === null || _c === void 0 ? void 0 : _c.httpOptions,
+                abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -8290,7 +8493,7 @@ class Models extends BaseModule {
      * ```
      */
     async computeTokens(params) {
-        var _a;
+        var _a, _b;
         let response;
         let path = '';
         let queryParams = {};
@@ -8308,6 +8511,7 @@ class Models extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -8347,7 +8551,7 @@ class Models extends BaseModule {
      * ```
      */
     async generateVideos(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -8365,6 +8569,7 @@ class Models extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -8387,7 +8592,8 @@ class Models extends BaseModule {
                 queryParams: queryParams,
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
+                httpOptions: (_c = params.config) === null || _c === void 0 ? void 0 : _c.httpOptions,
+                abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -8654,7 +8860,7 @@ class Operations extends BaseModule {
         }
     }
     async getVideosOperationInternal(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -8672,6 +8878,7 @@ class Operations extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'GET',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -8694,7 +8901,8 @@ class Operations extends BaseModule {
                 queryParams: queryParams,
                 body: JSON.stringify(body),
                 httpMethod: 'GET',
-                httpOptions: (_b = params.config) === null || _b === void 0 ? void 0 : _b.httpOptions,
+                httpOptions: (_c = params.config) === null || _c === void 0 ? void 0 : _c.httpOptions,
+                abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -8706,7 +8914,7 @@ class Operations extends BaseModule {
         }
     }
     async fetchPredictVideosOperationInternal(params) {
-        var _a;
+        var _a, _b;
         let response;
         let path = '';
         let queryParams = {};
@@ -8724,6 +8932,7 @@ class Operations extends BaseModule {
                 body: JSON.stringify(body),
                 httpMethod: 'POST',
                 httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
                 return httpResponse.json();
@@ -8748,7 +8957,7 @@ const CONTENT_TYPE_HEADER = 'Content-Type';
 const SERVER_TIMEOUT_HEADER = 'X-Server-Timeout';
 const USER_AGENT_HEADER = 'User-Agent';
 const GOOGLE_API_CLIENT_HEADER = 'x-goog-api-client';
-const SDK_VERSION = '0.9.0'; // x-release-please-version
+const SDK_VERSION = '0.10.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';
@@ -8877,7 +9086,7 @@ class ApiClient {
     getWebsocketBaseUrl() {
         const baseUrl = this.getBaseUrl();
         const urlParts = new URL(baseUrl);
-        urlParts.protocol = 'wss';
+        urlParts.protocol = urlParts.protocol == 'http:' ? 'ws' : 'wss';
         return urlParts.toString();
     }
     setBaseUrl(url) {
@@ -8941,7 +9150,7 @@ class ApiClient {
         else {
             requestInit.body = request.body;
         }
-        requestInit = await this.includeExtraHttpOptionsToRequestInit(requestInit, patchedHttpOptions);
+        requestInit = await this.includeExtraHttpOptionsToRequestInit(requestInit, patchedHttpOptions, request.abortSignal);
         return this.unaryApiCall(url, requestInit, request.httpMethod);
     }
     patchHttpOptions(baseHttpOptions, requestHttpOptions) {
@@ -8975,14 +9184,21 @@ class ApiClient {
         }
         let requestInit = {};
         requestInit.body = request.body;
-        requestInit = await this.includeExtraHttpOptionsToRequestInit(requestInit, patchedHttpOptions);
+        requestInit = await this.includeExtraHttpOptionsToRequestInit(requestInit, patchedHttpOptions, request.abortSignal);
         return this.streamApiCall(url, requestInit, request.httpMethod);
     }
-    async includeExtraHttpOptionsToRequestInit(requestInit, httpOptions) {
-        if (httpOptions && httpOptions.timeout && httpOptions.timeout > 0) {
+    async includeExtraHttpOptionsToRequestInit(requestInit, httpOptions, abortSignal) {
+        if ((httpOptions && httpOptions.timeout) || abortSignal) {
             const abortController = new AbortController();
             const signal = abortController.signal;
-            setTimeout(() => abortController.abort(), httpOptions.timeout);
+            if (httpOptions.timeout && (httpOptions === null || httpOptions === void 0 ? void 0 : httpOptions.timeout) > 0) {
+                setTimeout(() => abortController.abort(), httpOptions.timeout);
+            }
+            if (abortSignal) {
+                abortSignal.addEventListener('abort', () => {
+                    abortController.abort();
+                });
+            }
             requestInit.signal = signal;
         }
         requestInit.headers = await this.getHeadersInternal(httpOptions);
@@ -9037,6 +9253,30 @@ class ApiClient {
                         break;
                     }
                     const chunkString = decoder.decode(value);
+                    // Parse and throw an error if the chunk contains an error.
+                    try {
+                        const chunkJson = JSON.parse(chunkString);
+                        if ('error' in chunkJson) {
+                            const errorJson = JSON.parse(JSON.stringify(chunkJson['error']));
+                            const status = errorJson['status'];
+                            const code = errorJson['code'];
+                            const errorMessage = `got status: ${status}. ${JSON.stringify(chunkJson)}`;
+                            if (code >= 400 && code < 500) {
+                                const clientError = new ClientError(errorMessage);
+                                throw clientError;
+                            }
+                            else if (code >= 500 && code < 600) {
+                                const serverError = new ServerError(errorMessage);
+                                throw serverError;
+                            }
+                        }
+                    }
+                    catch (e) {
+                        const error = e;
+                        if (error.name === 'ClientError' || error.name === 'ServerError') {
+                            throw e;
+                        }
+                    }
                     buffer += chunkString;
                     let match = buffer.match(responseLineRE);
                     while (match) {
@@ -9175,7 +9415,7 @@ async function throwErrorIfNotOK(response) {
         else {
             errorBody = {
                 error: {
-                    message: 'exception parsing response',
+                    message: await response.text(),
                     code: response.status,
                     status: response.statusText,
                 },
@@ -9619,6 +9859,15 @@ class GoogleGenAI {
                 this.apiKey = undefined;
             }
         }
+        const baseUrl = getBaseUrl(options, getEnv('GOOGLE_VERTEX_BASE_URL'), getEnv('GOOGLE_GEMINI_BASE_URL'));
+        if (baseUrl) {
+            if (options.httpOptions) {
+                options.httpOptions.baseUrl = baseUrl;
+            }
+            else {
+                options.httpOptions = { baseUrl: baseUrl };
+            }
+        }
         this.apiVersion = options.apiVersion;
         const auth = new NodeAuth({
             apiKey: this.apiKey,
@@ -9694,4 +9943,5 @@ exports.createPartFromFunctionResponse = createPartFromFunctionResponse;
 exports.createPartFromText = createPartFromText;
 exports.createPartFromUri = createPartFromUri;
 exports.createUserContent = createUserContent;
+exports.setDefaultBaseUrls = setDefaultBaseUrls;
 //# sourceMappingURL=index.js.map