@firebase/ai 2.10.0 → 2.11.0-20260408221811

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.10.0";
+var version = "2.11.0-20260408221811";
 
 /**
  * @license
@@ -609,6 +609,11 @@ const SchemaType = {
  * and the {@link https://cloud.google.com/vertex-ai/generative-ai/docs/image/responsible-ai-imagen#safety-filters | Responsible AI and usage guidelines}
  * for more details.
  *
+ * @deprecated All Imagen models are deprecated and will shut down as
+ * early as June 2026. As a replacement, you can
+ * {@link https://firebase.google.com/docs/ai-logic/imagen-models-migration |
+ * migrate your apps to use Gemini Image models (the "Nano Banana" models)}.
+ *
  * @public
  */
 const ImagenSafetyFilterLevel = {
@@ -638,6 +643,11 @@ const ImagenSafetyFilterLevel = {
  * See the <a href="http://firebase.google.com/docs/vertex-ai/generate-images">personGeneration</a>
  * documentation for more details.
  *
+ * @deprecated All Imagen models are deprecated and will shut down as
+ * early as June 2026. As a replacement, you can
+ * {@link https://firebase.google.com/docs/ai-logic/imagen-models-migration |
+ * migrate your apps to use Gemini Image models (the "Nano Banana" models)}.
+ *
  * @public
  */
 const ImagenPersonFilterLevel = {
@@ -671,6 +681,11 @@ const ImagenPersonFilterLevel = {
  * See the {@link http://firebase.google.com/docs/vertex-ai/generate-images | documentation }
  * for more details and examples of the supported aspect ratios.
  *
+ * @deprecated All Imagen models are deprecated and will shut down as
+ * early as June 2026. As a replacement, you can
+ * {@link https://firebase.google.com/docs/ai-logic/imagen-models-migration |
+ * migrate your apps to use Gemini Image models (the "Nano Banana" models)}.
+ *
  * @public
  */
 const ImagenAspectRatio = {
@@ -2271,95 +2286,7 @@ function createPredictRequestBody(prompt, { gcsURI, imageFormat, addWatermark, n
 
 /**
  * @license
- * Copyright 2024 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.
- */
-// https://ai.google.dev/api/rest/v1beta/Content#part
-const VALID_PART_FIELDS = [
-    'text',
-    'inlineData',
-    'functionCall',
-    'functionResponse',
-    'thought',
-    'thoughtSignature'
-];
-const VALID_PARTS_PER_ROLE = {
-    user: ['text', 'inlineData'],
-    function: ['functionResponse'],
-    model: ['text', 'functionCall', 'thought', 'thoughtSignature'],
-    // System instructions shouldn't be in history anyway.
-    system: ['text']
-};
-const VALID_PREVIOUS_CONTENT_ROLES = {
-    user: ['model'],
-    function: ['model'],
-    model: ['user', 'function'],
-    // System instructions shouldn't be in history.
-    system: []
-};
-function validateChatHistory(history) {
-    let prevContent = null;
-    for (const currContent of history) {
-        const { role, parts } = currContent;
-        if (!prevContent && role !== 'user') {
-            throw new AIError(AIErrorCode.INVALID_CONTENT, `First Content should be with role 'user', got ${role}`);
-        }
-        if (!POSSIBLE_ROLES.includes(role)) {
-            throw new AIError(AIErrorCode.INVALID_CONTENT, `Each item should include role field. Got ${role} but valid roles are: ${JSON.stringify(POSSIBLE_ROLES)}`);
-        }
-        if (!Array.isArray(parts)) {
-            throw new AIError(AIErrorCode.INVALID_CONTENT, `Content should have 'parts' property with an array of Parts`);
-        }
-        if (parts.length === 0) {
-            throw new AIError(AIErrorCode.INVALID_CONTENT, `Each Content should have at least one part`);
-        }
-        const countFields = {
-            text: 0,
-            inlineData: 0,
-            functionCall: 0,
-            functionResponse: 0,
-            thought: 0,
-            thoughtSignature: 0,
-            executableCode: 0,
-            codeExecutionResult: 0
-        };
-        for (const part of parts) {
-            for (const key of VALID_PART_FIELDS) {
-                if (key in part) {
-                    countFields[key] += 1;
-                }
-            }
-        }
-        const validParts = VALID_PARTS_PER_ROLE[role];
-        for (const key of VALID_PART_FIELDS) {
-            if (!validParts.includes(key) && countFields[key] > 0) {
-                throw new AIError(AIErrorCode.INVALID_CONTENT, `Content with role '${role}' can't contain '${key}' part`);
-            }
-        }
-        if (prevContent) {
-            const validPreviousContentRoles = VALID_PREVIOUS_CONTENT_ROLES[role];
-            if (!validPreviousContentRoles.includes(prevContent.role)) {
-                throw new AIError(AIErrorCode.INVALID_CONTENT, `Content with role '${role}' can't follow '${prevContent.role}'. Valid previous roles: ${JSON.stringify(VALID_PREVIOUS_CONTENT_ROLES)}`);
-            }
-        }
-        prevContent = currContent;
-    }
-}
-
-/**
- * @license
- * Copyright 2024 Google LLC
+ * Copyright 2026 Google LLC
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -2384,28 +2311,23 @@ const SILENT_ERROR = 'SILENT_ERROR';
  */
 const DEFAULT_MAX_SEQUENTIAL_FUNCTION_CALLS = 10;
 /**
- * ChatSession class that enables sending chat messages and stores
- * history of sent and received messages so far.
+ * Base class for various `ChatSession` classes that enables sending chat
+ * messages and stores history of sent and received messages so far.
  *
  * @public
  */
-class ChatSession {
-    constructor(apiSettings, model, chromeAdapter, params, requestOptions) {
-        this.model = model;
-        this.chromeAdapter = chromeAdapter;
+class ChatSessionBase {
+    constructor(apiSettings, params, requestOptions) {
         this.params = params;
         this.requestOptions = requestOptions;
         this._history = [];
         /**
          * Ensures sequential execution of chat messages to maintain history order.
          * Each call waits for the previous one to settle before proceeding.
+         * @internal
          */
         this._sendPromise = Promise.resolve();
         this._apiSettings = apiSettings;
-        if (params?.history) {
-            validateChatHistory(params.history);
-            this._history = params.history;
-        }
     }
     /**
      * Gets the chat history so far. Blocked prompts are not added to history.
@@ -2416,26 +2338,12 @@ class ChatSession {
         await this._sendPromise;
         return this._history;
     }
-    /**
-     * Format Content into a request for generateContent or
-     * generateContentStream.
-     * @internal
-     */
-    _formatRequest(incomingContent, tempHistory) {
-        return {
-            safetySettings: this.params?.safetySettings,
-            generationConfig: this.params?.generationConfig,
-            tools: this.params?.tools,
-            toolConfig: this.params?.toolConfig,
-            systemInstruction: this.params?.systemInstruction,
-            contents: [...this._history, ...tempHistory, incomingContent]
-        };
-    }
     /**
      * Sends a chat message and receives a non-streaming
      * {@link GenerateContentResult}
+     * @internal
      */
-    async sendMessage(request, singleRequestOptions) {
+    async _sendMessage(request, singleRequestOptions) {
         let finalResult = {};
         await this._sendPromise;
         /**
@@ -2447,7 +2355,7 @@ class ChatSession {
         this._sendPromise = this._sendPromise.then(async () => {
             let functionCalls;
             let functionCallTurnCount = 0;
-            const functionCallMaxTurns = this.requestOptions?.maxSequentalFunctionCalls ??
+            const functionCallMaxTurns = this.requestOptions?.maxSequentialFunctionCalls ??
                 DEFAULT_MAX_SEQUENTIAL_FUNCTION_CALLS;
             // Repeats until model returns a response with no function calls
             // or until `functionCallMaxTurns` is met or exceeded.
@@ -2463,10 +2371,7 @@ class ChatSession {
                 }
                 const formattedRequest = this._formatRequest(formattedContent, tempHistory);
                 tempHistory.push(formattedContent);
-                const result = await generateContent(this._apiSettings, this.model, formattedRequest, this.chromeAdapter, {
-                    ...this.requestOptions,
-                    ...singleRequestOptions
-                });
+                const result = await this._callGenerateContent(formattedRequest, singleRequestOptions);
                 if (result) {
                     finalResult = result;
                     functionCalls = this._getCallableFunctionCalls(result.response);
@@ -2506,8 +2411,9 @@ class ChatSession {
      * Sends a chat message and receives the response as a
      * {@link GenerateContentStreamResult} containing an iterable stream
      * and a response promise.
+     * @internal
      */
-    async sendMessageStream(request, singleRequestOptions) {
+    async _sendMessageStream(request, singleRequestOptions) {
         await this._sendPromise;
         /**
          * Temporarily store multiple turns for cases like automatic function
@@ -2518,7 +2424,7 @@ class ChatSession {
         const callGenerateContentStream = async () => {
             let functionCalls;
             let functionCallTurnCount = 0;
-            const functionCallMaxTurns = this.requestOptions?.maxSequentalFunctionCalls ??
+            const functionCallMaxTurns = this.requestOptions?.maxSequentialFunctionCalls ??
                 DEFAULT_MAX_SEQUENTIAL_FUNCTION_CALLS;
             let result;
             // Repeats until model returns a response with no function calls
@@ -2535,10 +2441,7 @@ class ChatSession {
                 }
                 tempHistory.push(formattedContent);
                 const formattedRequest = this._formatRequest(formattedContent, tempHistory);
-                result = await generateContentStream(this._apiSettings, this.model, formattedRequest, this.chromeAdapter, {
-                    ...this.requestOptions,
-                    ...singleRequestOptions
-                });
+                result = await this._callGenerateContentStream(formattedRequest, singleRequestOptions);
                 functionCalls = this._getCallableFunctionCalls(result.firstValue);
                 if (functionCalls &&
                     result.firstValue &&
@@ -2632,7 +2535,7 @@ class ChatSession {
      * @internal
      */
     async _callFunctionsAsNeeded(functionCalls) {
-        const activeCallList = new Map();
+        const activeCallList = [];
         const promiseList = [];
         const functionDeclarationsTool = this.params?.tools?.find(tool => tool.functionDeclarations);
         if (functionDeclarationsTool &&
@@ -2645,7 +2548,8 @@ class ChatSession {
                         wrappedError.stack = e.stack;
                         throw wrappedError;
                     });
-                    activeCallList.set(functionCall.name, {
+                    activeCallList.push({
+                        name: functionCall.name,
                         id: functionCall.id,
                         results
                     });
@@ -2655,12 +2559,16 @@ class ChatSession {
             // Wait for promises to finish.
             await Promise.all(promiseList);
             const functionResponseParts = [];
-            for (const [name, callData] of activeCallList) {
+            for (const { name, id, results } of activeCallList) {
+                const functionResponse = {
+                    name,
+                    response: await results
+                };
+                if (id) {
+                    functionResponse.id = id;
+                }
                 functionResponseParts.push({
-                    functionResponse: {
-                        name,
-                        response: await callData.results
-                    }
+                    functionResponse
                 });
             }
             return functionResponseParts;
@@ -2671,6 +2579,182 @@ class ChatSession {
     }
 }
 
+/**
+ * @license
+ * Copyright 2024 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.
+ */
+// https://ai.google.dev/api/rest/v1beta/Content#part
+const VALID_PART_FIELDS = [
+    'text',
+    'inlineData',
+    'functionCall',
+    'functionResponse',
+    'thought',
+    'thoughtSignature'
+];
+const VALID_PARTS_PER_ROLE = {
+    user: ['text', 'inlineData'],
+    function: ['functionResponse'],
+    model: ['text', 'functionCall', 'thought', 'thoughtSignature'],
+    // System instructions shouldn't be in history anyway.
+    system: ['text']
+};
+const VALID_PREVIOUS_CONTENT_ROLES = {
+    user: ['model'],
+    function: ['model'],
+    model: ['user', 'function'],
+    // System instructions shouldn't be in history.
+    system: []
+};
+function validateChatHistory(history) {
+    let prevContent = null;
+    for (const currContent of history) {
+        const { role, parts } = currContent;
+        if (!prevContent && role !== 'user') {
+            throw new AIError(AIErrorCode.INVALID_CONTENT, `First Content should be with role 'user', got ${role}`);
+        }
+        if (!POSSIBLE_ROLES.includes(role)) {
+            throw new AIError(AIErrorCode.INVALID_CONTENT, `Each item should include role field. Got ${role} but valid roles are: ${JSON.stringify(POSSIBLE_ROLES)}`);
+        }
+        if (!Array.isArray(parts)) {
+            throw new AIError(AIErrorCode.INVALID_CONTENT, `Content should have 'parts' property with an array of Parts`);
+        }
+        if (parts.length === 0) {
+            throw new AIError(AIErrorCode.INVALID_CONTENT, `Each Content should have at least one part`);
+        }
+        const countFields = {
+            text: 0,
+            inlineData: 0,
+            functionCall: 0,
+            functionResponse: 0,
+            thought: 0,
+            thoughtSignature: 0,
+            executableCode: 0,
+            codeExecutionResult: 0
+        };
+        for (const part of parts) {
+            for (const key of VALID_PART_FIELDS) {
+                if (key in part) {
+                    countFields[key] += 1;
+                }
+            }
+        }
+        const validParts = VALID_PARTS_PER_ROLE[role];
+        for (const key of VALID_PART_FIELDS) {
+            if (!validParts.includes(key) && countFields[key] > 0) {
+                throw new AIError(AIErrorCode.INVALID_CONTENT, `Content with role '${role}' can't contain '${key}' part`);
+            }
+        }
+        if (prevContent) {
+            const validPreviousContentRoles = VALID_PREVIOUS_CONTENT_ROLES[role];
+            if (!validPreviousContentRoles.includes(prevContent.role)) {
+                throw new AIError(AIErrorCode.INVALID_CONTENT, `Content with role '${role}' can't follow '${prevContent.role}'. Valid previous roles: ${JSON.stringify(VALID_PREVIOUS_CONTENT_ROLES)}`);
+            }
+        }
+        prevContent = currContent;
+    }
+}
+
+/**
+ * @license
+ * Copyright 2024 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.
+ */
+/**
+ * ChatSession class that enables sending chat messages and stores
+ * history of sent and received messages so far.
+ *
+ * @public
+ */
+class ChatSession extends ChatSessionBase {
+    constructor(apiSettings, model, chromeAdapter, params, requestOptions) {
+        super(apiSettings, params, requestOptions);
+        this.model = model;
+        this.chromeAdapter = chromeAdapter;
+        this.params = params;
+        this.requestOptions = requestOptions;
+        if (params?.history) {
+            validateChatHistory(params.history);
+            this._history = params.history;
+        }
+    }
+    /**
+     * Format Content into a request for generateContent or
+     * generateContentStream.
+     * @internal
+     */
+    _formatRequest(incomingContent, tempHistory) {
+        return {
+            safetySettings: this.params?.safetySettings,
+            generationConfig: this.params?.generationConfig,
+            tools: this.params?.tools,
+            toolConfig: this.params?.toolConfig,
+            systemInstruction: this.params?.systemInstruction,
+            contents: [...this._history, ...tempHistory, incomingContent]
+        };
+    }
+    /**
+     * Calls default generateContent() (versus a specialized one like
+     * templateGenerateContent).
+     * @internal
+     */
+    _callGenerateContent(formattedRequest, singleRequestOptions) {
+        return generateContent(this._apiSettings, this.model, formattedRequest, this.chromeAdapter, {
+            ...this.requestOptions,
+            ...singleRequestOptions
+        });
+    }
+    /**
+     * Calls default generateContentStream() (versus a specialized one like
+     * templateGenerateContentStream).
+     * @internal
+     */
+    _callGenerateContentStream(formattedRequest, singleRequestOptions) {
+        return generateContentStream(this._apiSettings, this.model, formattedRequest, this.chromeAdapter, {
+            ...this.requestOptions,
+            ...singleRequestOptions
+        });
+    }
+    /**
+     * Sends a chat message and receives a non-streaming
+     * {@link GenerateContentResult}
+     */
+    async sendMessage(request, singleRequestOptions) {
+        return this._sendMessage(request, singleRequestOptions);
+    }
+    /**
+     * Sends a chat message and receives the response as a
+     * {@link GenerateContentStreamResult} containing an iterable stream
+     * and a response promise.
+     */
+    async sendMessageStream(request, singleRequestOptions) {
+        return this._sendMessageStream(request, singleRequestOptions);
+    }
+}
+
 /**
  * @license
  * Copyright 2024 Google LLC
@@ -2838,8 +2922,9 @@ function validateGenerationConfig(generationConfig) {
     }
     if ((generationConfig.responseSchema != null ||
         generationConfig.responseJsonSchema != null) &&
-        generationConfig.responseMimeType) {
-        throw new AIError(AIErrorCode.UNSUPPORTED, `responseMimeType must be set if responseSchema or responseJsonSchema are set.`);
+        generationConfig.responseMimeType !== 'application/json') {
+        throw new AIError(AIErrorCode.UNSUPPORTED, `responseMimeType must be set to "application/json" if` +
+            ` responseSchema or responseJsonSchema are set.`);
     }
 }
 
@@ -3271,6 +3356,11 @@ class LiveGenerativeModel extends AIModel {
  * }
  * ```
  *
+ * @deprecated All Imagen models are deprecated and will shut down as
+ * early as June 2026. As a replacement, you can
+ * {@link https://firebase.google.com/docs/ai-logic/imagen-models-migration |
+ * migrate your apps to use Gemini Image models (the "Nano Banana" models)}.
+ *
  * @public
  */
 class ImagenModel extends AIModel {
@@ -3514,6 +3604,112 @@ class WebSocketHandlerImpl {
     }
 }
 
+/**
+ * @license
+ * Copyright 2026 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.
+ */
+/**
+ * `ChatSession` class for use with server prompt templates that
+ * enables sending chat messages and stores history of sent and
+ * received messages so far.
+ *
+ * @beta
+ */
+class TemplateChatSession extends ChatSessionBase {
+    constructor(apiSettings, params, requestOptions) {
+        super(apiSettings, params, requestOptions);
+        this.params = params;
+        this.requestOptions = requestOptions;
+        if (params.history) {
+            validateChatHistory(params.history);
+            this._history = params.history;
+        }
+    }
+    /**
+     * Format the internal state to the body payload for `templateGenerateContent`.
+     * @internal
+     */
+    _formatRequest(incomingContent, tempHistory) {
+        const request = {
+            history: [...this._history, ...tempHistory, incomingContent]
+        };
+        if (this.params.templateVariables) {
+            request.inputs = this.params.templateVariables;
+        }
+        if (this.params.tools) {
+            request.tools = this.params.tools?.map(tool => {
+                if (tool.functionDeclarations) {
+                    return {
+                        templateFunctions: tool.functionDeclarations.map(declaration => {
+                            if (declaration.parameters) {
+                                const newDeclaration = { ...declaration };
+                                delete newDeclaration.parameters;
+                                newDeclaration.inputSchema = declaration.parameters;
+                                return newDeclaration;
+                            }
+                            return declaration;
+                        })
+                    };
+                }
+                return tool;
+            });
+        }
+        if (this.params.toolConfig) {
+            request.toolConfig = this.params.toolConfig;
+        }
+        return request;
+    }
+    /**
+     * Calls the specific templateGenerateContent() function needed for
+     * this specialized TemplateChatSession.
+     * @internal
+     */
+    _callGenerateContent(formattedRequest, singleRequestOptions) {
+        return templateGenerateContent(this._apiSettings, this.params.templateId, formattedRequest, {
+            ...this.requestOptions,
+            ...singleRequestOptions
+        });
+    }
+    /**
+     * Calls the specific templateGenerateContentStream() function needed for
+     * this specialized TemplateChatSession.
+     * @internal
+     */
+    _callGenerateContentStream(formattedRequest, singleRequestOptions) {
+        return templateGenerateContentStream(this._apiSettings, this.params.templateId, formattedRequest, {
+            ...this.requestOptions,
+            ...singleRequestOptions
+        });
+    }
+    /**
+     * Sends a chat message and receives a non-streaming
+     * {@link GenerateContentResult}
+     */
+    async sendMessage(request, singleRequestOptions) {
+        return this._sendMessage(request, singleRequestOptions);
+    }
+    /**
+     * Sends a chat message and receives the response as a
+     * {@link GenerateContentStreamResult} containing an iterable stream
+     * and a response promise.
+     */
+    async sendMessageStream(request, singleRequestOptions) {
+        return this._sendMessageStream(request, singleRequestOptions);
+    }
+}
+
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -3579,6 +3775,18 @@ class TemplateGenerativeModel {
             ...singleRequestOptions
         });
     }
+    /**
+     * Starts a {@link TemplateChatSession} that will use this template to
+     * respond to messages.
+     *
+     * @param params - Configurations for the chat, including the template
+     * ID and input variables.
+     *
+     * @beta
+     */
+    startChat(params) {
+        return new TemplateChatSession(this._apiSettings, params, this.requestOptions);
+    }
 }
 
 /**
@@ -3602,7 +3810,10 @@ class TemplateGenerativeModel {
  *
  * This class should only be instantiated with {@link getTemplateImagenModel}.
  *
- * @beta
+ * @deprecated All Imagen models are deprecated and will shut down as
+ * early as June 2026. As a replacement, you can
+ * {@link https://firebase.google.com/docs/ai-logic/imagen-models-migration |
+ * migrate your apps to use Gemini Image models (the "Nano Banana" models)}.
  */
 class TemplateImagenModel {
     /**
@@ -3911,6 +4122,11 @@ class AnyOfSchema extends Schema {
  * }
  * ```
  *
+ * @deprecated All Imagen models are deprecated and will shut down as
+ * early as June 2026. As a replacement, you can
+ * {@link https://firebase.google.com/docs/ai-logic/imagen-models-migration |
+ * migrate your apps to use Gemini Image models (the "Nano Banana" models)}.
+ *
  * @public
  */
 class ImagenImageFormat {
@@ -4411,6 +4627,11 @@ function getGenerativeModel(ai, modelParams, requestOptions) {
  *
  * Only Imagen 3 models (named `imagen-3.0-*`) are supported.
  *
+ * @deprecated All Imagen models are deprecated and will shut down as
+ * early as June 2026. As a replacement, you can
+ * {@link https://firebase.google.com/docs/ai-logic/imagen-models-migration |
+ * migrate your apps to use Gemini Image models (the "Nano Banana" models)}.
+ *
  * @param ai - An {@link AI} instance.
  * @param modelParams - Parameters to use when making Imagen requests.
  * @param requestOptions - Additional options to use when making requests.
@@ -4461,10 +4682,13 @@ function getTemplateGenerativeModel(ai, requestOptions) {
  * Returns a {@link TemplateImagenModel} class for executing server-side
  * Imagen templates.
  *
+ * @deprecated All Imagen models are deprecated and will shut down as
+ * early as June 2026. As a replacement, you can
+ * {@link https://firebase.google.com/docs/ai-logic/imagen-models-migration |
+ * migrate your apps to use Gemini Image models (the "Nano Banana" models)}.
+ *
  * @param ai - An {@link AI} instance.
  * @param requestOptions - Additional options to use when making requests.
- *
- * @beta
  */
 function getTemplateImagenModel(ai, requestOptions) {
     return new TemplateImagenModel(ai, requestOptions);
@@ -4493,6 +4717,7 @@ exports.BackendType = BackendType;
 exports.BlockReason = BlockReason;
 exports.BooleanSchema = BooleanSchema;
 exports.ChatSession = ChatSession;
+exports.ChatSessionBase = ChatSessionBase;
 exports.FinishReason = FinishReason;
 exports.FunctionCallingMode = FunctionCallingMode;
 exports.GenerativeModel = GenerativeModel;
@@ -4523,6 +4748,7 @@ exports.ResponseModality = ResponseModality;
 exports.Schema = Schema;
 exports.SchemaType = SchemaType;
 exports.StringSchema = StringSchema;
+exports.TemplateChatSession = TemplateChatSession;
 exports.TemplateGenerativeModel = TemplateGenerativeModel;
 exports.TemplateImagenModel = TemplateImagenModel;
 exports.ThinkingLevel = ThinkingLevel;