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

package/dist/index.node.mjs

@@ -4,7 +4,7 @@ import { FirebaseError, getModularInstance } from '@firebase/util';
 import { Logger } from '@firebase/logger';
 
 var name = "@firebase/ai";
-var version = "1.4.1-canary.2d720995d";
+var version = "1.4.1-canary.5200f7bb7";
 
 /**
  * @license
@@ -2307,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;
@@ -2359,6 +2366,9 @@ class Schema {
     static boolean(booleanParams) {
         return new BooleanSchema(booleanParams);
     }
+    static anyOf(anyOfParams) {
+        return new AnyOfSchema(anyOfParams);
+    }
 }
 /**
  * Schema class for "integer" types.
@@ -2486,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
@@ -2660,5 +2698,5 @@ function registerAI() {
 }
 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