@google/genai 0.9.0 → 0.10.0

package/dist/web/index.mjs

@@ -1,3 +1,63 @@
+/**
+ * @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
@@ -173,6 +233,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');
@@ -296,38 +386,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') {
@@ -1534,6 +1597,7 @@ var FinishReason;
     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";
@@ -1916,6 +1980,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.
      *
@@ -2171,7 +2271,7 @@ class Caches extends BaseModule {
      * ```
      */
     async create(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -2189,6 +2289,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();
@@ -2211,7 +2312,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();
@@ -2234,7 +2336,7 @@ class Caches extends BaseModule {
      * ```
      */
     async get(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -2252,6 +2354,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();
@@ -2274,7 +2377,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();
@@ -2297,7 +2401,7 @@ class Caches extends BaseModule {
      * ```
      */
     async delete(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -2315,6 +2419,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();
@@ -2339,7 +2444,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();
@@ -2367,7 +2473,7 @@ class Caches extends BaseModule {
      * ```
      */
     async update(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -2385,6 +2491,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();
@@ -2407,7 +2514,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();
@@ -2419,7 +2527,7 @@ class Caches extends BaseModule {
         }
     }
     async listInternal(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -2437,6 +2545,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();
@@ -2461,7 +2570,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();
@@ -3170,7 +3280,7 @@ class Files extends BaseModule {
         });
     }
     async listInternal(params) {
-        var _a;
+        var _a, _b;
         let response;
         let path = '';
         let queryParams = {};
@@ -3191,6 +3301,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();
@@ -3204,7 +3315,7 @@ class Files extends BaseModule {
         }
     }
     async createInternal(params) {
-        var _a;
+        var _a, _b;
         let response;
         let path = '';
         let queryParams = {};
@@ -3225,6 +3336,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();
@@ -3253,7 +3365,7 @@ class Files extends BaseModule {
      * ```
      */
     async get(params) {
-        var _a;
+        var _a, _b;
         let response;
         let path = '';
         let queryParams = {};
@@ -3274,6 +3386,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();
@@ -3298,7 +3411,7 @@ class Files extends BaseModule {
      * ```
      */
     async delete(params) {
-        var _a;
+        var _a, _b;
         let response;
         let path = '';
         let queryParams = {};
@@ -3319,6 +3432,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();
@@ -4129,6 +4243,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;
@@ -7476,21 +7675,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) {
@@ -7598,10 +7782,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));
     }
     /**
@@ -7822,7 +8013,7 @@ class Models extends BaseModule {
         };
     }
     async generateContentInternal(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -7840,6 +8031,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();
@@ -7864,7 +8056,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();
@@ -7878,7 +8071,7 @@ class Models extends BaseModule {
         }
     }
     async generateContentStreamInternal(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -7896,6 +8089,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* () {
@@ -7934,7 +8128,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* () {
@@ -7983,7 +8178,7 @@ class Models extends BaseModule {
      * ```
      */
     async embedContent(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -8001,6 +8196,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();
@@ -8025,7 +8221,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();
@@ -8058,7 +8255,7 @@ class Models extends BaseModule {
      * ```
      */
     async generateImagesInternal(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -8076,6 +8273,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();
@@ -8100,7 +8298,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();
@@ -8122,7 +8321,7 @@ class Models extends BaseModule {
      * ```
      */
     async get(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -8140,6 +8339,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();
@@ -8162,7 +8362,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();
@@ -8190,7 +8391,7 @@ class Models extends BaseModule {
      * ```
      */
     async countTokens(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -8208,6 +8409,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();
@@ -8232,7 +8434,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();
@@ -8264,7 +8467,7 @@ class Models extends BaseModule {
      * ```
      */
     async computeTokens(params) {
-        var _a;
+        var _a, _b;
         let response;
         let path = '';
         let queryParams = {};
@@ -8282,6 +8485,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();
@@ -8321,7 +8525,7 @@ class Models extends BaseModule {
      * ```
      */
     async generateVideos(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -8339,6 +8543,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();
@@ -8361,7 +8566,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();
@@ -8628,7 +8834,7 @@ class Operations extends BaseModule {
         }
     }
     async getVideosOperationInternal(params) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         let response;
         let path = '';
         let queryParams = {};
@@ -8646,6 +8852,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();
@@ -8668,7 +8875,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();
@@ -8680,7 +8888,7 @@ class Operations extends BaseModule {
         }
     }
     async fetchPredictVideosOperationInternal(params) {
-        var _a;
+        var _a, _b;
         let response;
         let path = '';
         let queryParams = {};
@@ -8698,6 +8906,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();
@@ -8722,7 +8931,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';
@@ -8851,7 +9060,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) {
@@ -8915,7 +9124,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) {
@@ -8949,14 +9158,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);
@@ -9011,6 +9227,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) {
@@ -9149,7 +9389,7 @@ async function throwErrorIfNotOK(response) {
         else {
             errorBody = {
                 error: {
-                    message: 'exception parsing response',
+                    message: await response.text(),
                     code: response.status,
                     status: response.statusText,
                 },
@@ -9349,6 +9589,17 @@ class GoogleGenAI {
         }
         this.vertexai = (_a = options.vertexai) !== null && _a !== void 0 ? _a : false;
         this.apiKey = options.apiKey;
+        const baseUrl = getBaseUrl(options, 
+        /*vertexBaseUrlFromEnv*/ undefined, 
+        /*geminiBaseUrlFromEnv*/ undefined);
+        if (baseUrl) {
+            if (options.httpOptions) {
+                options.httpOptions.baseUrl = baseUrl;
+            }
+            else {
+                options.httpOptions = { baseUrl: baseUrl };
+            }
+        }
         this.apiVersion = options.apiVersion;
         const auth = new WebAuth(this.apiKey);
         this.apiClient = new ApiClient({
@@ -9369,5 +9620,5 @@ class GoogleGenAI {
     }
 }
 
-export { ActivityHandling, BlockedReason, Caches, Chat, Chats, ComputeTokensResponse, ControlReferenceType, CountTokensResponse, CreateFileResponse, DeleteCachedContentResponse, DeleteFileResponse, DynamicRetrievalConfigMode, EmbedContentResponse, EndSensitivity, FeatureSelectionPreference, FileSource, FileState, Files, FinishReason, FunctionCallingConfigMode, FunctionResponse, GenerateContentResponse, GenerateContentResponsePromptFeedback, GenerateContentResponseUsageMetadata, GenerateImagesResponse, GenerateVideosResponse, GoogleGenAI, HarmBlockMethod, HarmBlockThreshold, HarmCategory, HarmProbability, HarmSeverity, HttpResponse, ImagePromptLanguage, Language, ListCachedContentsResponse, ListFilesResponse, Live, LiveClientToolResponse, LiveSendToolResponseParameters, MaskReferenceMode, MediaModality, MediaResolution, Modality, Mode, Models, Operations, Outcome, PagedItem, Pager, PersonGeneration, ReplayResponse, SafetyFilterLevel, Session, StartSensitivity, SubjectReferenceType, TrafficType, TurnCoverage, Type, createModelContent, createPartFromBase64, createPartFromCodeExecutionResult, createPartFromExecutableCode, createPartFromFunctionCall, createPartFromFunctionResponse, createPartFromText, createPartFromUri, createUserContent };
+export { ActivityHandling, BlockedReason, Caches, Chat, Chats, ComputeTokensResponse, ControlReferenceType, CountTokensResponse, CreateFileResponse, DeleteCachedContentResponse, DeleteFileResponse, DynamicRetrievalConfigMode, EmbedContentResponse, EndSensitivity, FeatureSelectionPreference, FileSource, FileState, Files, FinishReason, FunctionCallingConfigMode, FunctionResponse, GenerateContentResponse, GenerateContentResponsePromptFeedback, GenerateContentResponseUsageMetadata, GenerateImagesResponse, GenerateVideosResponse, GoogleGenAI, HarmBlockMethod, HarmBlockThreshold, HarmCategory, HarmProbability, HarmSeverity, HttpResponse, ImagePromptLanguage, Language, ListCachedContentsResponse, ListFilesResponse, Live, LiveClientToolResponse, LiveSendToolResponseParameters, MaskReferenceMode, MediaModality, MediaResolution, Modality, Mode, Models, Operations, Outcome, PagedItem, Pager, PersonGeneration, ReplayResponse, SafetyFilterLevel, Session, StartSensitivity, SubjectReferenceType, TrafficType, TurnCoverage, Type, createModelContent, createPartFromBase64, createPartFromCodeExecutionResult, createPartFromExecutableCode, createPartFromFunctionCall, createPartFromFunctionResponse, createPartFromText, createPartFromUri, createUserContent, setDefaultBaseUrls };
 //# sourceMappingURL=index.mjs.map