@firebase/ai 2.1.0-canary.84b8bed35 → 2.1.0-canary.c5f08a9bc

package/dist/esm/src/types/imagen/internal.d.ts

@@ -59,6 +59,14 @@ export interface ImagenResponseInternal {
          * The reason why the image was filtered.
          */
         raiFilteredReason?: string;
+        /**
+         * The safety attributes.
+         *
+         * This type is currently unused in the SDK. It is sent back because our requests set
+         * `includeSafetyAttributes`. This property is currently only used to avoid throwing an error
+         * when encountering this unsupported prediction type.
+         */
+        safetyAttributes?: unknown;
     }>;
 }
 /**
@@ -81,6 +89,7 @@ export interface ImagenResponseInternal {
  *     "personGeneration": "allow_all",
  *     "sampleCount": 2,
  *     "includeRaiReason": true,
+ *     "includeSafetyAttributes": true,
  *     "aspectRatio": "9:16"
  *   }
  * }
@@ -108,6 +117,7 @@ export interface PredictRequestBody {
         safetyFilterLevel?: string;
         personGeneration?: string;
         includeRaiReason: boolean;
+        includeSafetyAttributes: boolean;
     };
 }
 /**

package/dist/index.cjs.js

@@ -8,7 +8,7 @@ var util = require('@firebase/util');
 var logger$1 = require('@firebase/logger');
 
 var name = "@firebase/ai";
-var version = "2.1.0-canary.84b8bed35";
+var version = "2.1.0-canary.c5f08a9bc";
 
 /**
  * @license
@@ -1328,6 +1328,7 @@ async function handlePredictResponse(response) {
                 gcsURI: prediction.gcsUri
             });
         }
+        else if (prediction.safetyAttributes) ;
         else {
             throw new AIError(AIErrorCode.RESPONSE_ERROR, `Unexpected element in 'predictions' array in response: '${JSON.stringify(prediction)}'`);
         }
@@ -1870,7 +1871,8 @@ function createPredictRequestBody(prompt, { gcsURI, imageFormat, addWatermark, n
             addWatermark,
             safetyFilterLevel,
             personGeneration: personFilterLevel,
-            includeRaiReason: true
+            includeRaiReason: true,
+            includeSafetyAttributes: true
         }
     };
     return body;