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

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 = "1.4.1-canary.2d720995d";
+var version = "1.4.1-canary.5200f7bb7";
 
 /**
  * @license
@@ -2311,12 +2311,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;
@@ -2363,6 +2370,9 @@ class Schema {
     static boolean(booleanParams) {
         return new BooleanSchema(booleanParams);
     }
+    static anyOf(anyOfParams) {
+        return new AnyOfSchema(anyOfParams);
+    }
 }
 /**
  * Schema class for "integer" types.
@@ -2490,6 +2500,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
@@ -2667,6 +2705,7 @@ registerAI();
 exports.AIError = AIError;
 exports.AIErrorCode = AIErrorCode;
 exports.AIModel = AIModel;
+exports.AnyOfSchema = AnyOfSchema;
 exports.ArraySchema = ArraySchema;
 exports.Backend = Backend;
 exports.BackendType = BackendType;