@firebase/ai 2.11.1 → 2.12.0-20260505164105

package/dist/index.node.cjs.js

@@ -8,7 +8,7 @@ var util = require('@firebase/util');
 var logger$1 = require('@firebase/logger');
 
 var name = "@firebase/ai";
-var version = "2.11.1";
+var version = "2.12.0-20260505164105";
 
 /**
  * @license
@@ -281,7 +281,78 @@ const FinishReason = {
     /**
      * The function call generated by the model was invalid.
      */
-    MALFORMED_FUNCTION_CALL: 'MALFORMED_FUNCTION_CALL'
+    MALFORMED_FUNCTION_CALL: 'MALFORMED_FUNCTION_CALL',
+    /**
+     * Token generation stopped because generated images contain safety violations.
+     */
+    IMAGE_SAFETY: 'IMAGE_SAFETY',
+    /**
+     * Image generation stopped because generated images have other prohibited content.
+     */
+    IMAGE_PROHIBITED_CONTENT: 'IMAGE_PROHIBITED_CONTENT',
+    /**
+     * Image generation stopped because of other miscellaneous issue.
+     */
+    IMAGE_OTHER: 'IMAGE_OTHER',
+    /**
+     * The model was expected to generate an image, but none was generated.
+     */
+    NO_IMAGE: 'NO_IMAGE',
+    /**
+     * Image generation stopped due to recitation.
+     */
+    IMAGE_RECITATION: 'IMAGE_RECITATION',
+    /**
+     * The response candidate content was flagged for using an unsupported language.
+     */
+    LANGUAGE: 'LANGUAGE',
+    /**
+     * Model generated a tool call but no tools were enabled in the request.
+     */
+    UNEXPECTED_TOOL_CALL: 'UNEXPECTED_TOOL_CALL',
+    /**
+     * Model called too many tools consecutively, thus the system exited execution.
+     */
+    TOO_MANY_TOOL_CALLS: 'TOO_MANY_TOOL_CALLS',
+    /**
+     * Request has at least one thought signature missing.
+     */
+    MISSING_THOUGHT_SIGNATURE: 'MISSING_THOUGHT_SIGNATURE',
+    /**
+     * Finished due to malformed response.
+     */
+    MALFORMED_RESPONSE: 'MALFORMED_RESPONSE'
+};
+/**
+ * Aspect ratios for generated images.
+ * @public
+ */
+/* eslint-disable camelcase */
+const ImageConfigAspectRatio = {
+    SQUARE_1x1: '1:1',
+    PORTRAIT_9x16: '9:16',
+    LANDSCAPE_16x9: '16:9',
+    PORTRAIT_3x4: '3:4',
+    LANDSCAPE_4x3: '4:3',
+    PORTRAIT_2x3: '2:3',
+    LANDSCAPE_3x2: '3:2',
+    PORTRAIT_4x5: '4:5',
+    LANDSCAPE_5x4: '5:4',
+    PORTRAIT_1x4: '1:4',
+    LANDSCAPE_4x1: '4:1',
+    PORTRAIT_1x8: '1:8',
+    LANDSCAPE_8x1: '8:1',
+    ULTRAWIDE_21x9: '21:9'
+};
+/**
+ * Sizes for generated images. For example, '1K' is 1024px, '2K' is 2048px, and '4K' is 4096px.
+ * @public
+ */
+const ImageConfigImageSize = {
+    SIZE_512: '512',
+    SIZE_1K: '1K',
+    SIZE_2K: '2K',
+    SIZE_4K: '4K'
 };
 /**
  * @public
@@ -491,7 +562,8 @@ const LiveResponseType = {
     SERVER_CONTENT: 'serverContent',
     TOOL_CALL: 'toolCall',
     TOOL_CALL_CANCELLATION: 'toolCallCancellation',
-    GOING_AWAY_NOTICE: 'goingAwayNotice'
+    GOING_AWAY_NOTICE: 'goingAwayNotice',
+    SESSION_RESUMPTION_UPDATE: 'sessionResumptionUpdate'
 };
 
 /**
@@ -1526,7 +1598,24 @@ function getInlineDataParts(response) {
         return undefined;
     }
 }
-const badFinishReasons = [FinishReason.RECITATION, FinishReason.SAFETY];
+const badFinishReasons = [
+    FinishReason.RECITATION,
+    FinishReason.SAFETY,
+    FinishReason.BLOCKLIST,
+    FinishReason.PROHIBITED_CONTENT,
+    FinishReason.SPII,
+    FinishReason.MALFORMED_FUNCTION_CALL,
+    FinishReason.IMAGE_SAFETY,
+    FinishReason.IMAGE_PROHIBITED_CONTENT,
+    FinishReason.IMAGE_OTHER,
+    FinishReason.NO_IMAGE,
+    FinishReason.IMAGE_RECITATION,
+    FinishReason.LANGUAGE,
+    FinishReason.UNEXPECTED_TOOL_CALL,
+    FinishReason.TOO_MANY_TOOL_CALLS,
+    FinishReason.MISSING_THOUGHT_SIGNATURE,
+    FinishReason.MALFORMED_RESPONSE
+];
 function hadBadFinishReason(candidate) {
     return (!!candidate.finishReason &&
         badFinishReasons.some(reason => reason === candidate.finishReason));
@@ -2928,6 +3017,153 @@ function validateGenerationConfig(generationConfig) {
     }
 }
 
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+/**
+ * A wrapper for the native `WebSocket` available in both Browsers and Node >= 22.
+ *
+ * @internal
+ */
+class WebSocketHandlerImpl {
+    constructor() {
+        if (typeof WebSocket === 'undefined') {
+            throw new AIError(AIErrorCode.UNSUPPORTED, 'The WebSocket API is not available in this environment. ' +
+                'The "Live" feature is not supported here. It is supported in ' +
+                'modern browser windows, Web Workers with WebSocket support, and Node >= 22.');
+        }
+    }
+    connect(url) {
+        return new Promise((resolve, reject) => {
+            this.ws = new WebSocket(url);
+            this.ws.binaryType = 'blob'; // Only important to set in Node
+            this.ws.addEventListener('open', () => resolve(), { once: true });
+            this.ws.addEventListener('error', () => reject(new AIError(AIErrorCode.FETCH_ERROR, `Error event raised on WebSocket`)), { once: true });
+            this.ws.addEventListener('close', (closeEvent) => {
+                if (closeEvent.reason) {
+                    logger.warn(`WebSocket connection closed by server. Reason: '${closeEvent.reason}'`);
+                }
+            });
+        });
+    }
+    send(data) {
+        if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
+            throw new AIError(AIErrorCode.REQUEST_ERROR, 'WebSocket is not open.');
+        }
+        this.ws.send(data);
+    }
+    async *listen() {
+        if (!this.ws) {
+            throw new AIError(AIErrorCode.REQUEST_ERROR, 'WebSocket is not connected.');
+        }
+        const messageQueue = [];
+        const errorQueue = [];
+        let resolvePromise = null;
+        let isClosed = false;
+        const messageListener = async (event) => {
+            let data;
+            if (event.data instanceof Blob) {
+                data = await event.data.text();
+            }
+            else if (typeof event.data === 'string') {
+                data = event.data;
+            }
+            else {
+                errorQueue.push(new AIError(AIErrorCode.PARSE_FAILED, `Failed to parse WebSocket response. Expected data to be a Blob or string, but was ${typeof event.data}.`));
+                if (resolvePromise) {
+                    resolvePromise();
+                    resolvePromise = null;
+                }
+                return;
+            }
+            try {
+                const obj = JSON.parse(data);
+                messageQueue.push(obj);
+            }
+            catch (e) {
+                const err = e;
+                errorQueue.push(new AIError(AIErrorCode.PARSE_FAILED, `Error parsing WebSocket message to JSON: ${err.message}`));
+            }
+            if (resolvePromise) {
+                resolvePromise();
+                resolvePromise = null;
+            }
+        };
+        const errorListener = () => {
+            errorQueue.push(new AIError(AIErrorCode.FETCH_ERROR, 'WebSocket connection error.'));
+            if (resolvePromise) {
+                resolvePromise();
+                resolvePromise = null;
+            }
+        };
+        const closeListener = (event) => {
+            if (event.reason) {
+                logger.warn(`WebSocket connection closed by the server with reason: ${event.reason}`);
+            }
+            isClosed = true;
+            if (resolvePromise) {
+                resolvePromise();
+                resolvePromise = null;
+            }
+            // Clean up listeners to prevent memory leaks
+            this.ws?.removeEventListener('message', messageListener);
+            this.ws?.removeEventListener('close', closeListener);
+            this.ws?.removeEventListener('error', errorListener);
+        };
+        this.ws.addEventListener('message', messageListener);
+        this.ws.addEventListener('close', closeListener);
+        this.ws.addEventListener('error', errorListener);
+        while (!isClosed) {
+            if (errorQueue.length > 0) {
+                const error = errorQueue.shift();
+                throw error;
+            }
+            if (messageQueue.length > 0) {
+                yield messageQueue.shift();
+            }
+            else {
+                await new Promise(resolve => {
+                    resolvePromise = resolve;
+                });
+            }
+        }
+        // If the loop terminated because isClosed is true, check for any final errors
+        if (errorQueue.length > 0) {
+            const error = errorQueue.shift();
+            throw error;
+        }
+    }
+    close(code, reason) {
+        return new Promise(resolve => {
+            if (!this.ws) {
+                return resolve();
+            }
+            this.ws.addEventListener('close', () => resolve(), { once: true });
+            // Calling 'close' during these states results in an error.
+            if (this.ws.readyState === WebSocket.CLOSED ||
+                this.ws.readyState === WebSocket.CONNECTING) {
+                return resolve();
+            }
+            if (this.ws.readyState !== WebSocket.CLOSING) {
+                this.ws.close(code, reason);
+            }
+        });
+    }
+}
+
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -2955,9 +3191,10 @@ class LiveSession {
     /**
      * @internal
      */
-    constructor(webSocketHandler, serverMessages) {
-        this.webSocketHandler = webSocketHandler;
-        this.serverMessages = serverMessages;
+    constructor(_setupMessage, _apiSettings, _sessionResumption, webSocketHandler) {
+        this._setupMessage = _setupMessage;
+        this._apiSettings = _apiSettings;
+        this._sessionResumption = _sessionResumption;
         /**
          * Indicates whether this Live session is closed.
          *
@@ -2970,6 +3207,64 @@ class LiveSession {
          * @beta
          */
         this.inConversation = false;
+        /**
+         * Generator yielding WebSocket messages from the server.
+         */
+        this._serverMessages = null;
+        this._webSocketHandler = webSocketHandler || new WebSocketHandlerImpl();
+        this.connectionPromise = this._connectSession(this._sessionResumption);
+    }
+    /**
+     * Initializes connection to the WebSocket. Should be called immediately
+     * after instantiation.
+     *
+     * @internal
+     */
+    async _connectSession(sessionResumption) {
+        const url = new WebSocketUrl(this._apiSettings);
+        await this._webSocketHandler.connect(url.toString());
+        try {
+            // Begin listening for server messages, and begin the handshake by sending the 'setupMessage'
+            this._serverMessages = this._webSocketHandler.listen();
+            const setupMessage = { ...this._setupMessage };
+            if (sessionResumption) {
+                setupMessage.setup.sessionResumption = sessionResumption;
+            }
+            this._webSocketHandler.send(JSON.stringify(setupMessage));
+            // Verify we received the handshake response 'setupComplete'
+            const firstMessage = (await this._serverMessages.next()).value;
+            if (!firstMessage ||
+                !(typeof firstMessage === 'object') ||
+                !('setupComplete' in firstMessage)) {
+                await this._webSocketHandler.close(1011, 'Handshake failure');
+                throw new AIError(AIErrorCode.RESPONSE_ERROR, 'Server connection handshake failed. The server did not respond with a setupComplete message.');
+            }
+            this.isClosed = false;
+        }
+        catch (e) {
+            // Ensure connection is closed on any setup error
+            await this._webSocketHandler.close();
+            throw e;
+        }
+    }
+    /**
+     * Resumes an existing live session with the server.
+     *
+     * This closes the current WebSocket connection and establishes a new one using
+     * the same configuration (URI, headers, model, system instruction, tools, etc.)
+     * as the original session.
+     *
+     * @param sessionResumption - The configuration for session resumption, such as the handle to the previous session state to restore.
+     * @throws If the session resumption configuration is unsupported.
+     *
+     * @beta
+     */
+    async resumeSession(sessionResumption) {
+        if (!this._sessionResumption) {
+            throw new AIError(AIErrorCode.UNSUPPORTED, 'Cannot resume session: no sessionResumption config provided');
+        }
+        await this.close();
+        await this._connectSession(sessionResumption);
     }
     /**
      * Sends content to the server.
@@ -2991,7 +3286,7 @@ class LiveSession {
                 turnComplete
             }
         };
-        this.webSocketHandler.send(JSON.stringify(message));
+        this._webSocketHandler.send(JSON.stringify(message));
     }
     /**
      * Sends text to the server in realtime.
@@ -3015,7 +3310,7 @@ class LiveSession {
                 text
             }
         };
-        this.webSocketHandler.send(JSON.stringify(message));
+        this._webSocketHandler.send(JSON.stringify(message));
     }
     /**
      * Sends audio data to the server in realtime.
@@ -3044,7 +3339,7 @@ class LiveSession {
                 audio: blob
             }
         };
-        this.webSocketHandler.send(JSON.stringify(message));
+        this._webSocketHandler.send(JSON.stringify(message));
     }
     /**
      * Sends video data to the server in realtime.
@@ -3072,7 +3367,7 @@ class LiveSession {
                 video: blob
             }
         };
-        this.webSocketHandler.send(JSON.stringify(message));
+        this._webSocketHandler.send(JSON.stringify(message));
     }
     /**
      * Sends function responses to the server.
@@ -3091,7 +3386,7 @@ class LiveSession {
                 functionResponses
             }
         };
-        this.webSocketHandler.send(JSON.stringify(message));
+        this._webSocketHandler.send(JSON.stringify(message));
     }
     /**
      * Yields messages received from the server.
@@ -3106,42 +3401,50 @@ class LiveSession {
         if (this.isClosed) {
             throw new AIError(AIErrorCode.SESSION_CLOSED, 'Cannot read from a Live session that is closed. Try starting a new Live session.');
         }
-        for await (const message of this.serverMessages) {
-            if (message && typeof message === 'object') {
-                if (LiveResponseType.SERVER_CONTENT in message) {
-                    yield {
-                        type: 'serverContent',
-                        ...message
-                            .serverContent
-                    };
-                }
-                else if (LiveResponseType.TOOL_CALL in message) {
-                    yield {
-                        type: 'toolCall',
-                        ...message
-                            .toolCall
-                    };
-                }
-                else if (LiveResponseType.TOOL_CALL_CANCELLATION in message) {
-                    yield {
-                        type: 'toolCallCancellation',
-                        ...message.toolCallCancellation
-                    };
-                }
-                else if ('goAway' in message) {
-                    const notice = message.goAway;
-                    yield {
-                        type: LiveResponseType.GOING_AWAY_NOTICE,
-                        timeLeft: parseDuration(notice.timeLeft)
-                    };
+        if (this._serverMessages) {
+            for await (const message of this._serverMessages) {
+                if (message && typeof message === 'object') {
+                    if (LiveResponseType.SERVER_CONTENT in message) {
+                        yield {
+                            type: 'serverContent',
+                            ...message
+                                .serverContent
+                        };
+                    }
+                    else if (LiveResponseType.TOOL_CALL in message) {
+                        yield {
+                            type: 'toolCall',
+                            ...message
+                                .toolCall
+                        };
+                    }
+                    else if (LiveResponseType.TOOL_CALL_CANCELLATION in message) {
+                        yield {
+                            type: 'toolCallCancellation',
+                            ...message.toolCallCancellation
+                        };
+                    }
+                    else if ('goAway' in message) {
+                        const notice = message.goAway;
+                        yield {
+                            type: LiveResponseType.GOING_AWAY_NOTICE,
+                            timeLeft: parseDuration(notice.timeLeft)
+                        };
+                    }
+                    else if (LiveResponseType.SESSION_RESUMPTION_UPDATE in message) {
+                        yield {
+                            type: LiveResponseType.SESSION_RESUMPTION_UPDATE,
+                            ...message.sessionResumptionUpdate
+                        };
+                    }
+                    else {
+                        logger.warn(`Received an unknown message type from the server: ${JSON.stringify(message)}`);
+                    }
                 }
                 else {
-                    logger.warn(`Received an unknown message type from the server: ${JSON.stringify(message)}`);
+                    logger.warn(`Received an invalid message from the server: ${JSON.stringify(message)}`);
                 }
             }
-            else {
-                logger.warn(`Received an invalid message from the server: ${JSON.stringify(message)}`);
-            }
         }
     }
     /**
@@ -3153,7 +3456,7 @@ class LiveSession {
     async close() {
         if (!this.isClosed) {
             this.isClosed = true;
-            await this.webSocketHandler.close(1000, 'Client closed session.');
+            await this._webSocketHandler.close(1000, 'Client closed session.');
         }
     }
     /**
@@ -3176,7 +3479,7 @@ class LiveSession {
             const message = {
                 realtimeInput: { mediaChunks: [mediaChunk] }
             };
-            this.webSocketHandler.send(JSON.stringify(message));
+            this._webSocketHandler.send(JSON.stringify(message));
         });
     }
     /**
@@ -3256,6 +3559,7 @@ class LiveGenerativeModel extends AIModel {
      */
     constructor(ai, modelParams, 
     /**
+     * For testing injection
      * @internal
      */
     _webSocketHandler) {
@@ -3274,9 +3578,7 @@ class LiveGenerativeModel extends AIModel {
      *
      * @beta
      */
-    async connect() {
-        const url = new WebSocketUrl(this._apiSettings);
-        await this._webSocketHandler.connect(url.toString());
+    async connect(sessionResumption) {
         let fullModelPath;
         if (this._apiSettings.backend.backendType === BackendType.GOOGLE_AI) {
             fullModelPath = `projects/${this._apiSettings.project}/${this.model}`;
@@ -3287,36 +3589,24 @@ class LiveGenerativeModel extends AIModel {
         // inputAudioTranscription and outputAudioTranscription are on the generation config in the public API,
         // but the backend expects them to be in the `setup` message.
         const { inputAudioTranscription, outputAudioTranscription, ...generationConfig } = this.generationConfig;
+        const contextWindowCompression = generationConfig.contextWindowCompression;
+        delete generationConfig.contextWindowCompression;
         const setupMessage = {
             setup: {
                 model: fullModelPath,
                 generationConfig,
+                contextWindowCompression,
                 tools: this.tools,
                 toolConfig: this.toolConfig,
                 systemInstruction: this.systemInstruction,
                 inputAudioTranscription,
-                outputAudioTranscription
+                outputAudioTranscription,
+                sessionResumption
             }
         };
-        try {
-            // Begin listening for server messages, and begin the handshake by sending the 'setupMessage'
-            const serverMessages = this._webSocketHandler.listen();
-            this._webSocketHandler.send(JSON.stringify(setupMessage));
-            // Verify we received the handshake response 'setupComplete'
-            const firstMessage = (await serverMessages.next()).value;
-            if (!firstMessage ||
-                !(typeof firstMessage === 'object') ||
-                !('setupComplete' in firstMessage)) {
-                await this._webSocketHandler.close(1011, 'Handshake failure');
-                throw new AIError(AIErrorCode.RESPONSE_ERROR, 'Server connection handshake failed. The server did not respond with a setupComplete message.');
-            }
-            return new LiveSession(this._webSocketHandler, serverMessages);
-        }
-        catch (e) {
-            // Ensure connection is closed on any setup error
-            await this._webSocketHandler.close();
-            throw e;
-        }
+        const session = new LiveSession(setupMessage, this._apiSettings, sessionResumption, this._webSocketHandler);
+        await session.connectionPromise;
+        return session;
     }
 }
 
@@ -3457,153 +3747,6 @@ class ImagenModel extends AIModel {
     }
 }
 
-/**
- * @license
- * Copyright 2025 Google LLC
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *   http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-/**
- * A wrapper for the native `WebSocket` available in both Browsers and Node >= 22.
- *
- * @internal
- */
-class WebSocketHandlerImpl {
-    constructor() {
-        if (typeof WebSocket === 'undefined') {
-            throw new AIError(AIErrorCode.UNSUPPORTED, 'The WebSocket API is not available in this environment. ' +
-                'The "Live" feature is not supported here. It is supported in ' +
-                'modern browser windows, Web Workers with WebSocket support, and Node >= 22.');
-        }
-    }
-    connect(url) {
-        return new Promise((resolve, reject) => {
-            this.ws = new WebSocket(url);
-            this.ws.binaryType = 'blob'; // Only important to set in Node
-            this.ws.addEventListener('open', () => resolve(), { once: true });
-            this.ws.addEventListener('error', () => reject(new AIError(AIErrorCode.FETCH_ERROR, `Error event raised on WebSocket`)), { once: true });
-            this.ws.addEventListener('close', (closeEvent) => {
-                if (closeEvent.reason) {
-                    logger.warn(`WebSocket connection closed by server. Reason: '${closeEvent.reason}'`);
-                }
-            });
-        });
-    }
-    send(data) {
-        if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
-            throw new AIError(AIErrorCode.REQUEST_ERROR, 'WebSocket is not open.');
-        }
-        this.ws.send(data);
-    }
-    async *listen() {
-        if (!this.ws) {
-            throw new AIError(AIErrorCode.REQUEST_ERROR, 'WebSocket is not connected.');
-        }
-        const messageQueue = [];
-        const errorQueue = [];
-        let resolvePromise = null;
-        let isClosed = false;
-        const messageListener = async (event) => {
-            let data;
-            if (event.data instanceof Blob) {
-                data = await event.data.text();
-            }
-            else if (typeof event.data === 'string') {
-                data = event.data;
-            }
-            else {
-                errorQueue.push(new AIError(AIErrorCode.PARSE_FAILED, `Failed to parse WebSocket response. Expected data to be a Blob or string, but was ${typeof event.data}.`));
-                if (resolvePromise) {
-                    resolvePromise();
-                    resolvePromise = null;
-                }
-                return;
-            }
-            try {
-                const obj = JSON.parse(data);
-                messageQueue.push(obj);
-            }
-            catch (e) {
-                const err = e;
-                errorQueue.push(new AIError(AIErrorCode.PARSE_FAILED, `Error parsing WebSocket message to JSON: ${err.message}`));
-            }
-            if (resolvePromise) {
-                resolvePromise();
-                resolvePromise = null;
-            }
-        };
-        const errorListener = () => {
-            errorQueue.push(new AIError(AIErrorCode.FETCH_ERROR, 'WebSocket connection error.'));
-            if (resolvePromise) {
-                resolvePromise();
-                resolvePromise = null;
-            }
-        };
-        const closeListener = (event) => {
-            if (event.reason) {
-                logger.warn(`WebSocket connection closed by the server with reason: ${event.reason}`);
-            }
-            isClosed = true;
-            if (resolvePromise) {
-                resolvePromise();
-                resolvePromise = null;
-            }
-            // Clean up listeners to prevent memory leaks
-            this.ws?.removeEventListener('message', messageListener);
-            this.ws?.removeEventListener('close', closeListener);
-            this.ws?.removeEventListener('error', errorListener);
-        };
-        this.ws.addEventListener('message', messageListener);
-        this.ws.addEventListener('close', closeListener);
-        this.ws.addEventListener('error', errorListener);
-        while (!isClosed) {
-            if (errorQueue.length > 0) {
-                const error = errorQueue.shift();
-                throw error;
-            }
-            if (messageQueue.length > 0) {
-                yield messageQueue.shift();
-            }
-            else {
-                await new Promise(resolve => {
-                    resolvePromise = resolve;
-                });
-            }
-        }
-        // If the loop terminated because isClosed is true, check for any final errors
-        if (errorQueue.length > 0) {
-            const error = errorQueue.shift();
-            throw error;
-        }
-    }
-    close(code, reason) {
-        return new Promise(resolve => {
-            if (!this.ws) {
-                return resolve();
-            }
-            this.ws.addEventListener('close', () => resolve(), { once: true });
-            // Calling 'close' during these states results in an error.
-            if (this.ws.readyState === WebSocket.CLOSED ||
-                this.ws.readyState === WebSocket.CONNECTING) {
-                return resolve();
-            }
-            if (this.ws.readyState !== WebSocket.CLOSING) {
-                this.ws.close(code, reason);
-            }
-        });
-    }
-}
-
 /**
  * @license
  * Copyright 2026 Google LLC
@@ -3748,11 +3891,16 @@ class TemplateGenerativeModel {
      * @param templateId - The ID of the server-side template to execute.
      * @param templateVariables - A key-value map of variables to populate the
      * template with.
+     * @param singleRequestOptions - Optional. Options to use for this request.
+     * @param templateToolConfig - Optional. Configuration for tools to use with this request.
      *
      * @beta
      */
-    async generateContent(templateId, templateVariables, singleRequestOptions) {
-        return templateGenerateContent(this._apiSettings, templateId, { inputs: templateVariables }, {
+    async generateContent(templateId, templateVariables, singleRequestOptions, templateToolConfig) {
+        return templateGenerateContent(this._apiSettings, templateId, {
+            inputs: templateVariables,
+            ...(templateToolConfig && { toolConfig: templateToolConfig })
+        }, {
             ...this.requestOptions,
             ...singleRequestOptions
         });
@@ -3766,11 +3914,16 @@ class TemplateGenerativeModel {
      * @param templateId - The ID of the server-side template to execute.
      * @param templateVariables - A key-value map of variables to populate the
      * template with.
+     * @param singleRequestOptions - Optional.Options to use for this request.
+     * @param templateToolConfig - Optional. Configuration for tools to use with this request.
      *
      * @beta
      */
-    async generateContentStream(templateId, templateVariables, singleRequestOptions) {
-        return templateGenerateContentStream(this._apiSettings, templateId, { inputs: templateVariables }, {
+    async generateContentStream(templateId, templateVariables, singleRequestOptions, templateToolConfig) {
+        return templateGenerateContentStream(this._apiSettings, templateId, {
+            inputs: templateVariables,
+            ...(templateToolConfig && { toolConfig: templateToolConfig })
+        }, {
             ...this.requestOptions,
             ...singleRequestOptions
         });
@@ -4663,8 +4816,7 @@ function getLiveGenerativeModel(ai, modelParams) {
     if (!modelParams.model) {
         throw new AIError(AIErrorCode.NO_MODEL, `Must provide a model name for getLiveGenerativeModel. Example: getLiveGenerativeModel(ai, { model: 'my-model-name' })`);
     }
-    const webSocketHandler = new WebSocketHandlerImpl();
-    return new LiveGenerativeModel(ai, modelParams, webSocketHandler);
+    return new LiveGenerativeModel(ai, modelParams);
 }
 /**
  * Returns a {@link TemplateGenerativeModel} class for executing server-side
@@ -4727,6 +4879,8 @@ exports.HarmBlockThreshold = HarmBlockThreshold;
 exports.HarmCategory = HarmCategory;
 exports.HarmProbability = HarmProbability;
 exports.HarmSeverity = HarmSeverity;
+exports.ImageConfigAspectRatio = ImageConfigAspectRatio;
+exports.ImageConfigImageSize = ImageConfigImageSize;
 exports.ImagenAspectRatio = ImagenAspectRatio;
 exports.ImagenImageFormat = ImagenImageFormat;
 exports.ImagenModel = ImagenModel;