@firebase/ai 1.4.1-canary.2b5731292 → 1.4.1-canary.5200f7bb7

package/dist/index.node.mjs

@@ -2,10 +2,9 @@ import { _isFirebaseServerApp, _getProvider, getApp, _registerComponent, registe
 import { Component } from '@firebase/component';
 import { FirebaseError, getModularInstance } from '@firebase/util';
 import { Logger } from '@firebase/logger';
-import { __asyncGenerator, __await } from 'tslib';
 
 var name = "@firebase/ai";
-var version = "1.4.1-canary.2b5731292";
+var version = "1.4.1-canary.5200f7bb7";
 
 /**
  * @license
@@ -626,8 +625,8 @@ class AIService {
     constructor(app, backend, authProvider, appCheckProvider) {
         this.app = app;
         this.backend = backend;
-        const appCheck = appCheckProvider === null || appCheckProvider === void 0 ? void 0 : appCheckProvider.getImmediate({ optional: true });
-        const auth = authProvider === null || authProvider === void 0 ? void 0 : authProvider.getImmediate({ optional: true });
+        const appCheck = appCheckProvider?.getImmediate({ optional: true });
+        const auth = authProvider?.getImmediate({ optional: true });
         this.auth = auth || null;
         this.appCheck = appCheck || null;
         if (backend instanceof VertexAIBackend) {
@@ -799,14 +798,13 @@ class AIModel {
      * @internal
      */
     constructor(ai, modelName) {
-        var _a, _b, _c, _d, _e, _f;
-        if (!((_b = (_a = ai.app) === null || _a === void 0 ? void 0 : _a.options) === null || _b === void 0 ? void 0 : _b.apiKey)) {
+        if (!ai.app?.options?.apiKey) {
             throw new AIError(AIErrorCode.NO_API_KEY, `The "apiKey" field is empty in the local Firebase config. Firebase AI requires this field to contain a valid API key.`);
         }
-        else if (!((_d = (_c = ai.app) === null || _c === void 0 ? void 0 : _c.options) === null || _d === void 0 ? void 0 : _d.projectId)) {
+        else if (!ai.app?.options?.projectId) {
             throw new AIError(AIErrorCode.NO_PROJECT_ID, `The "projectId" field is empty in the local Firebase config. Firebase AI requires this field to contain a valid project ID.`);
         }
-        else if (!((_f = (_e = ai.app) === null || _e === void 0 ? void 0 : _e.options) === null || _f === void 0 ? void 0 : _f.appId)) {
+        else if (!ai.app?.options?.appId) {
             throw new AIError(AIErrorCode.NO_APP_ID, `The "appId" field is empty in the local Firebase config. Firebase AI requires this field to contain a valid app ID.`);
         }
         else {
@@ -934,8 +932,7 @@ class RequestUrl {
         return url.toString();
     }
     get baseUrl() {
-        var _a;
-        return ((_a = this.requestOptions) === null || _a === void 0 ? void 0 : _a.baseUrl) || DEFAULT_BASE_URL;
+        return this.requestOptions?.baseUrl || DEFAULT_BASE_URL;
     }
     get apiVersion() {
         return DEFAULT_API_VERSION; // TODO: allow user-set options if that feature becomes available
@@ -1011,7 +1008,7 @@ async function makeRequest(model, task, apiSettings, stream, body, requestOption
     try {
         const request = await constructRequest(model, task, apiSettings, stream, body, requestOptions);
         // Timeout is 180s by default
-        const timeoutMillis = (requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.timeout) != null && requestOptions.timeout >= 0
+        const timeoutMillis = requestOptions?.timeout != null && requestOptions.timeout >= 0
             ? requestOptions.timeout
             : DEFAULT_FETCH_TIMEOUT_MS;
         const abortController = new AbortController();
@@ -1035,10 +1032,7 @@ async function makeRequest(model, task, apiSettings, stream, body, requestOption
             if (response.status === 403 &&
                 errorDetails &&
                 errorDetails.some((detail) => detail.reason === 'SERVICE_DISABLED') &&
-                errorDetails.some((detail) => {
-                    var _a, _b;
-                    return (_b = (_a = detail.links) === null || _a === void 0 ? void 0 : _a[0]) === null || _b === void 0 ? void 0 : _b.description.includes('Google developers console API activation');
-                })) {
+                errorDetails.some((detail) => detail.links?.[0]?.description.includes('Google developers console API activation'))) {
                 throw new AIError(AIErrorCode.API_NOT_ENABLED, `The Firebase AI SDK requires the Firebase AI ` +
                     `API ('firebasevertexai.googleapis.com') to be enabled in your ` +
                     `Firebase project. Enable this API by visiting the Firebase Console ` +
@@ -1183,10 +1177,9 @@ function addHelpers(response) {
  * Returns all text found in all parts of first candidate.
  */
 function getText(response) {
-    var _a, _b, _c, _d;
     const textStrings = [];
-    if ((_b = (_a = response.candidates) === null || _a === void 0 ? void 0 : _a[0].content) === null || _b === void 0 ? void 0 : _b.parts) {
-        for (const part of (_d = (_c = response.candidates) === null || _c === void 0 ? void 0 : _c[0].content) === null || _d === void 0 ? void 0 : _d.parts) {
+    if (response.candidates?.[0].content?.parts) {
+        for (const part of response.candidates?.[0].content?.parts) {
             if (part.text) {
                 textStrings.push(part.text);
             }
@@ -1203,10 +1196,9 @@ function getText(response) {
  * Returns {@link FunctionCall}s associated with first candidate.
  */
 function getFunctionCalls(response) {
-    var _a, _b, _c, _d;
     const functionCalls = [];
-    if ((_b = (_a = response.candidates) === null || _a === void 0 ? void 0 : _a[0].content) === null || _b === void 0 ? void 0 : _b.parts) {
-        for (const part of (_d = (_c = response.candidates) === null || _c === void 0 ? void 0 : _c[0].content) === null || _d === void 0 ? void 0 : _d.parts) {
+    if (response.candidates?.[0].content?.parts) {
+        for (const part of response.candidates?.[0].content?.parts) {
             if (part.functionCall) {
                 functionCalls.push(part.functionCall);
             }
@@ -1225,10 +1217,9 @@ function getFunctionCalls(response) {
  * @internal
  */
 function getInlineDataParts(response) {
-    var _a, _b, _c, _d;
     const data = [];
-    if ((_b = (_a = response.candidates) === null || _a === void 0 ? void 0 : _a[0].content) === null || _b === void 0 ? void 0 : _b.parts) {
-        for (const part of (_d = (_c = response.candidates) === null || _c === void 0 ? void 0 : _c[0].content) === null || _d === void 0 ? void 0 : _d.parts) {
+    if (response.candidates?.[0].content?.parts) {
+        for (const part of response.candidates?.[0].content?.parts) {
             if (part.inlineData) {
                 data.push(part);
             }
@@ -1247,19 +1238,18 @@ function hadBadFinishReason(candidate) {
         badFinishReasons.some(reason => reason === candidate.finishReason));
 }
 function formatBlockErrorMessage(response) {
-    var _a, _b, _c;
     let message = '';
     if ((!response.candidates || response.candidates.length === 0) &&
         response.promptFeedback) {
         message += 'Response was blocked';
-        if ((_a = response.promptFeedback) === null || _a === void 0 ? void 0 : _a.blockReason) {
+        if (response.promptFeedback?.blockReason) {
             message += ` due to ${response.promptFeedback.blockReason}`;
         }
-        if ((_b = response.promptFeedback) === null || _b === void 0 ? void 0 : _b.blockReasonMessage) {
+        if (response.promptFeedback?.blockReasonMessage) {
             message += `: ${response.promptFeedback.blockReasonMessage}`;
         }
     }
-    else if ((_c = response.candidates) === null || _c === void 0 ? void 0 : _c[0]) {
+    else if (response.candidates?.[0]) {
         const firstCandidate = response.candidates[0];
         if (hadBadFinishReason(firstCandidate)) {
             message += `Candidate was blocked due to ${firstCandidate.finishReason}`;
@@ -1278,12 +1268,11 @@ function formatBlockErrorMessage(response) {
  * @internal
  */
 async function handlePredictResponse(response) {
-    var _a;
     const responseJson = await response.json();
     const images = [];
     let filteredReason = undefined;
     // The backend should always send a non-empty array of predictions if the response was successful.
-    if (!responseJson.predictions || ((_a = responseJson.predictions) === null || _a === void 0 ? void 0 : _a.length) === 0) {
+    if (!responseJson.predictions || responseJson.predictions?.length === 0) {
         throw new AIError(AIErrorCode.RESPONSE_ERROR, 'No predictions or filtered reason received from Vertex AI. Please report this issue with the full error details at https://github.com/firebase/firebase-js-sdk/issues.');
     }
     for (const prediction of responseJson.predictions) {
@@ -1347,13 +1336,12 @@ async function handlePredictResponse(response) {
  * @internal
  */
 function mapGenerateContentRequest(generateContentRequest) {
-    var _a, _b;
-    (_a = generateContentRequest.safetySettings) === null || _a === void 0 ? void 0 : _a.forEach(safetySetting => {
+    generateContentRequest.safetySettings?.forEach(safetySetting => {
         if (safetySetting.method) {
             throw new AIError(AIErrorCode.UNSUPPORTED, 'SafetySetting.method is not supported in the the Gemini Developer API. Please remove this property.');
         }
     });
-    if ((_b = generateContentRequest.generationConfig) === null || _b === void 0 ? void 0 : _b.topK) {
+    if (generateContentRequest.generationConfig?.topK) {
         const roundedTopK = Math.round(generateContentRequest.generationConfig.topK);
         if (roundedTopK !== generateContentRequest.generationConfig.topK) {
             logger.warn('topK in GenerationConfig has been rounded to the nearest integer to match the format for requests to the Gemini Developer API.');
@@ -1394,7 +1382,10 @@ function mapGenerateContentResponse(googleAIResponse) {
  */
 function mapCountTokensRequest(countTokensRequest, model) {
     const mappedCountTokensRequest = {
-        generateContentRequest: Object.assign({ model }, countTokensRequest)
+        generateContentRequest: {
+            model,
+            ...countTokensRequest
+        }
     };
     return mappedCountTokensRequest;
 }
@@ -1414,7 +1405,6 @@ function mapGenerateContentCandidates(candidates) {
     let mappedSafetyRatings;
     if (mappedCandidates) {
         candidates.forEach(candidate => {
-            var _a;
             // Map citationSources to citations.
             let citationMetadata;
             if (candidate.citationMetadata) {
@@ -1425,14 +1415,18 @@ function mapGenerateContentCandidates(candidates) {
             // Assign missing candidate SafetyRatings properties to their defaults if undefined.
             if (candidate.safetyRatings) {
                 mappedSafetyRatings = candidate.safetyRatings.map(safetyRating => {
-                    var _a, _b, _c;
-                    return Object.assign(Object.assign({}, safetyRating), { severity: (_a = safetyRating.severity) !== null && _a !== void 0 ? _a : HarmSeverity.HARM_SEVERITY_UNSUPPORTED, probabilityScore: (_b = safetyRating.probabilityScore) !== null && _b !== void 0 ? _b : 0, severityScore: (_c = safetyRating.severityScore) !== null && _c !== void 0 ? _c : 0 });
+                    return {
+                        ...safetyRating,
+                        severity: safetyRating.severity ?? HarmSeverity.HARM_SEVERITY_UNSUPPORTED,
+                        probabilityScore: safetyRating.probabilityScore ?? 0,
+                        severityScore: safetyRating.severityScore ?? 0
+                    };
                 });
             }
             // videoMetadata is not supported.
             // Throw early since developers may send a long video as input and only expect to pay
             // for inference on a small portion of the video.
-            if ((_a = candidate.content) === null || _a === void 0 ? void 0 : _a.parts.some(part => part === null || part === void 0 ? void 0 : part.videoMetadata)) {
+            if (candidate.content?.parts.some(part => part?.videoMetadata)) {
                 throw new AIError(AIErrorCode.UNSUPPORTED, 'Part.videoMetadata is not supported in the Gemini Developer API. Please remove this property.');
             }
             const mappedCandidate = {
@@ -1453,13 +1447,12 @@ function mapPromptFeedback(promptFeedback) {
     // Assign missing SafetyRating properties to their defaults if undefined.
     const mappedSafetyRatings = [];
     promptFeedback.safetyRatings.forEach(safetyRating => {
-        var _a, _b, _c;
         mappedSafetyRatings.push({
             category: safetyRating.category,
             probability: safetyRating.probability,
-            severity: (_a = safetyRating.severity) !== null && _a !== void 0 ? _a : HarmSeverity.HARM_SEVERITY_UNSUPPORTED,
-            probabilityScore: (_b = safetyRating.probabilityScore) !== null && _b !== void 0 ? _b : 0,
-            severityScore: (_c = safetyRating.severityScore) !== null && _c !== void 0 ? _c : 0,
+            severity: safetyRating.severity ?? HarmSeverity.HARM_SEVERITY_UNSUPPORTED,
+            probabilityScore: safetyRating.probabilityScore ?? 0,
+            severityScore: safetyRating.severityScore ?? 0,
             blocked: safetyRating.blocked
         });
     });
@@ -1520,24 +1513,22 @@ async function getResponsePromise(stream, apiSettings) {
         allResponses.push(value);
     }
 }
-function generateResponseSequence(stream, apiSettings) {
-    return __asyncGenerator(this, arguments, function* generateResponseSequence_1() {
-        const reader = stream.getReader();
-        while (true) {
-            const { value, done } = yield __await(reader.read());
-            if (done) {
-                break;
-            }
-            let enhancedResponse;
-            if (apiSettings.backend.backendType === BackendType.GOOGLE_AI) {
-                enhancedResponse = createEnhancedContentResponse(mapGenerateContentResponse(value));
-            }
-            else {
-                enhancedResponse = createEnhancedContentResponse(value);
-            }
-            yield yield __await(enhancedResponse);
+async function* generateResponseSequence(stream, apiSettings) {
+    const reader = stream.getReader();
+    while (true) {
+        const { value, done } = await reader.read();
+        if (done) {
+            break;
         }
-    });
+        let enhancedResponse;
+        if (apiSettings.backend.backendType === BackendType.GOOGLE_AI) {
+            enhancedResponse = createEnhancedContentResponse(mapGenerateContentResponse(value));
+        }
+        else {
+            enhancedResponse = createEnhancedContentResponse(value);
+        }
+        yield enhancedResponse;
+    }
 }
 /**
  * Reads a raw stream from the fetch response and join incomplete
@@ -1589,7 +1580,7 @@ function getResponseStream(inputStream) {
 function aggregateResponses(responses) {
     const lastResponse = responses[responses.length - 1];
     const aggregatedResponse = {
-        promptFeedback: lastResponse === null || lastResponse === void 0 ? void 0 : lastResponse.promptFeedback
+        promptFeedback: lastResponse?.promptFeedback
     };
     for (const response of responses) {
         if (response.candidates) {
@@ -1946,7 +1937,7 @@ class ChatSession {
         this._history = [];
         this._sendPromise = Promise.resolve();
         this._apiSettings = apiSettings;
-        if (params === null || params === void 0 ? void 0 : params.history) {
+        if (params?.history) {
             validateChatHistory(params.history);
             this._history = params.history;
         }
@@ -1965,15 +1956,14 @@ class ChatSession {
      * {@link GenerateContentResult}
      */
     async sendMessage(request) {
-        var _a, _b, _c, _d, _e;
         await this._sendPromise;
         const newContent = formatNewContent(request);
         const generateContentRequest = {
-            safetySettings: (_a = this.params) === null || _a === void 0 ? void 0 : _a.safetySettings,
-            generationConfig: (_b = this.params) === null || _b === void 0 ? void 0 : _b.generationConfig,
-            tools: (_c = this.params) === null || _c === void 0 ? void 0 : _c.tools,
-            toolConfig: (_d = this.params) === null || _d === void 0 ? void 0 : _d.toolConfig,
-            systemInstruction: (_e = this.params) === null || _e === void 0 ? void 0 : _e.systemInstruction,
+            safetySettings: this.params?.safetySettings,
+            generationConfig: this.params?.generationConfig,
+            tools: this.params?.tools,
+            toolConfig: this.params?.toolConfig,
+            systemInstruction: this.params?.systemInstruction,
             contents: [...this._history, newContent]
         };
         let finalResult = {};
@@ -1981,14 +1971,13 @@ class ChatSession {
         this._sendPromise = this._sendPromise
             .then(() => generateContent(this._apiSettings, this.model, generateContentRequest, this.requestOptions))
             .then(result => {
-            var _a, _b;
             if (result.response.candidates &&
                 result.response.candidates.length > 0) {
                 this._history.push(newContent);
                 const responseContent = {
-                    parts: ((_a = result.response.candidates) === null || _a === void 0 ? void 0 : _a[0].content.parts) || [],
+                    parts: result.response.candidates?.[0].content.parts || [],
                     // Response seems to come back without a role set.
-                    role: ((_b = result.response.candidates) === null || _b === void 0 ? void 0 : _b[0].content.role) || 'model'
+                    role: result.response.candidates?.[0].content.role || 'model'
                 };
                 this._history.push(responseContent);
             }
@@ -2009,15 +1998,14 @@ class ChatSession {
      * and a response promise.
      */
     async sendMessageStream(request) {
-        var _a, _b, _c, _d, _e;
         await this._sendPromise;
         const newContent = formatNewContent(request);
         const generateContentRequest = {
-            safetySettings: (_a = this.params) === null || _a === void 0 ? void 0 : _a.safetySettings,
-            generationConfig: (_b = this.params) === null || _b === void 0 ? void 0 : _b.generationConfig,
-            tools: (_c = this.params) === null || _c === void 0 ? void 0 : _c.tools,
-            toolConfig: (_d = this.params) === null || _d === void 0 ? void 0 : _d.toolConfig,
-            systemInstruction: (_e = this.params) === null || _e === void 0 ? void 0 : _e.systemInstruction,
+            safetySettings: this.params?.safetySettings,
+            generationConfig: this.params?.generationConfig,
+            tools: this.params?.tools,
+            toolConfig: this.params?.toolConfig,
+            systemInstruction: this.params?.systemInstruction,
             contents: [...this._history, newContent]
         };
         const streamPromise = generateContentStream(this._apiSettings, this.model, generateContentRequest, this.requestOptions);
@@ -2033,7 +2021,7 @@ class ChatSession {
             .then(response => {
             if (response.candidates && response.candidates.length > 0) {
                 this._history.push(newContent);
-                const responseContent = Object.assign({}, response.candidates[0].content);
+                const responseContent = { ...response.candidates[0].content };
                 // Response seems to come back without a role set.
                 if (!responseContent.role) {
                     responseContent.role = 'model';
@@ -2126,7 +2114,14 @@ class GenerativeModel extends AIModel {
      */
     async generateContent(request) {
         const formattedParams = formatGenerateContentInput(request);
-        return generateContent(this._apiSettings, this.model, Object.assign({ generationConfig: this.generationConfig, safetySettings: this.safetySettings, tools: this.tools, toolConfig: this.toolConfig, systemInstruction: this.systemInstruction }, formattedParams), this.requestOptions);
+        return generateContent(this._apiSettings, this.model, {
+            generationConfig: this.generationConfig,
+            safetySettings: this.safetySettings,
+            tools: this.tools,
+            toolConfig: this.toolConfig,
+            systemInstruction: this.systemInstruction,
+            ...formattedParams
+        }, this.requestOptions);
     }
     /**
      * Makes a single streaming call to the model
@@ -2136,14 +2131,33 @@ class GenerativeModel extends AIModel {
      */
     async generateContentStream(request) {
         const formattedParams = formatGenerateContentInput(request);
-        return generateContentStream(this._apiSettings, this.model, Object.assign({ generationConfig: this.generationConfig, safetySettings: this.safetySettings, tools: this.tools, toolConfig: this.toolConfig, systemInstruction: this.systemInstruction }, formattedParams), this.requestOptions);
+        return generateContentStream(this._apiSettings, this.model, {
+            generationConfig: this.generationConfig,
+            safetySettings: this.safetySettings,
+            tools: this.tools,
+            toolConfig: this.toolConfig,
+            systemInstruction: this.systemInstruction,
+            ...formattedParams
+        }, this.requestOptions);
     }
     /**
      * Gets a new {@link ChatSession} instance which can be used for
      * multi-turn chats.
      */
     startChat(startChatParams) {
-        return new ChatSession(this._apiSettings, this.model, Object.assign({ tools: this.tools, toolConfig: this.toolConfig, systemInstruction: this.systemInstruction, generationConfig: this.generationConfig, safetySettings: this.safetySettings }, startChatParams), this.requestOptions);
+        return new ChatSession(this._apiSettings, this.model, {
+            tools: this.tools,
+            toolConfig: this.toolConfig,
+            systemInstruction: this.systemInstruction,
+            generationConfig: this.generationConfig,
+            safetySettings: this.safetySettings,
+            /**
+             * Overrides params inherited from GenerativeModel with those explicitly set in the
+             * StartChatParams. For example, if startChatParams.generationConfig is set, it'll override
+             * this.generationConfig.
+             */
+            ...startChatParams
+        }, this.requestOptions);
     }
     /**
      * Counts the tokens in the provided request.
@@ -2229,7 +2243,10 @@ class ImagenModel extends AIModel {
      * @beta
      */
     async generateImages(prompt) {
-        const body = createPredictRequestBody(prompt, Object.assign(Object.assign({}, this.generationConfig), this.safetySettings));
+        const body = createPredictRequestBody(prompt, {
+            ...this.generationConfig,
+            ...this.safetySettings
+        });
         const response = await makeRequest(this.model, Task.PREDICT, this._apiSettings, 
         /* stream */ false, JSON.stringify(body), this.requestOptions);
         return handlePredictResponse(response);
@@ -2254,7 +2271,11 @@ class ImagenModel extends AIModel {
      * If all images are filtered, the `images` array will be empty.
      */
     async generateImagesGCS(prompt, gcsURI) {
-        const body = createPredictRequestBody(prompt, Object.assign(Object.assign({ gcsURI }, this.generationConfig), this.safetySettings));
+        const body = createPredictRequestBody(prompt, {
+            gcsURI,
+            ...this.generationConfig,
+            ...this.safetySettings
+        });
         const response = await makeRequest(this.model, Task.PREDICT, this._apiSettings, 
         /* stream */ false, JSON.stringify(body), this.requestOptions);
         return handlePredictResponse(response);
@@ -2286,12 +2307,19 @@ class ImagenModel extends AIModel {
  */
 class Schema {
     constructor(schemaParams) {
+        // TODO(dlarocque): Enforce this with union types
+        if (!schemaParams.type && !schemaParams.anyOf) {
+            throw new AIError(AIErrorCode.INVALID_SCHEMA, "A schema must have either a 'type' or an 'anyOf' array of sub-schemas.");
+        }
         // eslint-disable-next-line guard-for-in
         for (const paramKey in schemaParams) {
             this[paramKey] = schemaParams[paramKey];
         }
         // Ensure these are explicitly set to avoid TS errors.
         this.type = schemaParams.type;
+        this.format = schemaParams.hasOwnProperty('format')
+            ? schemaParams.format
+            : undefined;
         this.nullable = schemaParams.hasOwnProperty('nullable')
             ? !!schemaParams.nullable
             : false;
@@ -2338,6 +2366,9 @@ class Schema {
     static boolean(booleanParams) {
         return new BooleanSchema(booleanParams);
     }
+    static anyOf(anyOfParams) {
+        return new AnyOfSchema(anyOfParams);
+    }
 }
 /**
  * Schema class for "integer" types.
@@ -2345,7 +2376,10 @@ class Schema {
  */
 class IntegerSchema extends Schema {
     constructor(schemaParams) {
-        super(Object.assign({ type: SchemaType.INTEGER }, schemaParams));
+        super({
+            type: SchemaType.INTEGER,
+            ...schemaParams
+        });
     }
 }
 /**
@@ -2354,7 +2388,10 @@ class IntegerSchema extends Schema {
  */
 class NumberSchema extends Schema {
     constructor(schemaParams) {
-        super(Object.assign({ type: SchemaType.NUMBER }, schemaParams));
+        super({
+            type: SchemaType.NUMBER,
+            ...schemaParams
+        });
     }
 }
 /**
@@ -2363,7 +2400,10 @@ class NumberSchema extends Schema {
  */
 class BooleanSchema extends Schema {
     constructor(schemaParams) {
-        super(Object.assign({ type: SchemaType.BOOLEAN }, schemaParams));
+        super({
+            type: SchemaType.BOOLEAN,
+            ...schemaParams
+        });
     }
 }
 /**
@@ -2373,7 +2413,10 @@ class BooleanSchema extends Schema {
  */
 class StringSchema extends Schema {
     constructor(schemaParams, enumValues) {
-        super(Object.assign({ type: SchemaType.STRING }, schemaParams));
+        super({
+            type: SchemaType.STRING,
+            ...schemaParams
+        });
         this.enum = enumValues;
     }
     /**
@@ -2395,7 +2438,10 @@ class StringSchema extends Schema {
  */
 class ArraySchema extends Schema {
     constructor(schemaParams, items) {
-        super(Object.assign({ type: SchemaType.ARRAY }, schemaParams));
+        super({
+            type: SchemaType.ARRAY,
+            ...schemaParams
+        });
         this.items = items;
     }
     /**
@@ -2414,7 +2460,10 @@ class ArraySchema extends Schema {
  */
 class ObjectSchema extends Schema {
     constructor(schemaParams, properties, optionalProperties = []) {
-        super(Object.assign({ type: SchemaType.OBJECT }, schemaParams));
+        super({
+            type: SchemaType.OBJECT,
+            ...schemaParams
+        });
         this.properties = properties;
         this.optionalProperties = optionalProperties;
     }
@@ -2423,7 +2472,7 @@ class ObjectSchema extends Schema {
      */
     toJSON() {
         const obj = super.toJSON();
-        obj.properties = Object.assign({}, this.properties);
+        obj.properties = { ...this.properties };
         const required = [];
         if (this.optionalProperties) {
             for (const propertyKey of this.optionalProperties) {
@@ -2447,6 +2496,34 @@ class ObjectSchema extends Schema {
         return obj;
     }
 }
+/**
+ * Schema class representing a value that can conform to any of the provided sub-schemas. This is
+ * useful when a field can accept multiple distinct types or structures.
+ * @public
+ */
+class AnyOfSchema extends Schema {
+    constructor(schemaParams) {
+        if (schemaParams.anyOf.length === 0) {
+            throw new AIError(AIErrorCode.INVALID_SCHEMA, "The 'anyOf' array must not be empty.");
+        }
+        super({
+            ...schemaParams,
+            type: undefined // anyOf schemas do not have an explicit type
+        });
+        this.anyOf = schemaParams.anyOf;
+    }
+    /**
+     * @internal
+     */
+    toJSON() {
+        const obj = super.toJSON();
+        // Ensure the 'anyOf' property contains serialized SchemaRequest objects.
+        if (this.anyOf && Array.isArray(this.anyOf)) {
+            obj.anyOf = this.anyOf.map(s => s.toJSON());
+        }
+        return obj;
+    }
+}
 
 /**
  * @license
@@ -2616,10 +2693,10 @@ function registerAI() {
         return new AIService(app, backend, auth, appCheckProvider);
     }, "PUBLIC" /* ComponentType.PUBLIC */).setMultipleInstances(true));
     registerVersion(name, version, 'node');
-    // BUILD_TARGET will be replaced by values like esm2017, cjs2017, etc during the compilation
-    registerVersion(name, version, 'esm2017');
+    // BUILD_TARGET will be replaced by values like esm, cjs, etc during the compilation
+    registerVersion(name, version, 'esm2020');
 }
 registerAI();
 
-export { AIError, AIErrorCode, AIModel, ArraySchema, Backend, BackendType, BlockReason, BooleanSchema, ChatSession, FinishReason, FunctionCallingMode, GenerativeModel, GoogleAIBackend, HarmBlockMethod, HarmBlockThreshold, HarmCategory, HarmProbability, HarmSeverity, ImagenAspectRatio, ImagenImageFormat, ImagenModel, ImagenPersonFilterLevel, ImagenSafetyFilterLevel, IntegerSchema, Modality, NumberSchema, ObjectSchema, POSSIBLE_ROLES, ResponseModality, Schema, SchemaType, StringSchema, VertexAIBackend, getAI, getGenerativeModel, getImagenModel };
+export { AIError, AIErrorCode, AIModel, AnyOfSchema, ArraySchema, Backend, BackendType, BlockReason, BooleanSchema, ChatSession, FinishReason, FunctionCallingMode, GenerativeModel, GoogleAIBackend, HarmBlockMethod, HarmBlockThreshold, HarmCategory, HarmProbability, HarmSeverity, ImagenAspectRatio, ImagenImageFormat, ImagenModel, ImagenPersonFilterLevel, ImagenSafetyFilterLevel, IntegerSchema, Modality, NumberSchema, ObjectSchema, POSSIBLE_ROLES, ResponseModality, Schema, SchemaType, StringSchema, VertexAIBackend, getAI, getGenerativeModel, getImagenModel };
 //# sourceMappingURL=index.node.mjs.map