fix(ai): Add 'includeSafetyAttributes' to Predict request payloads

Author: dlarocque

.changeset/good-cooks-deliver.md

@@ -0,0 +1,5 @@
+---
+'@firebase/ai': patch
+---
+
+Add 'includeSafetyAttributes' field to Predict request payloads.

packages/ai/src/requests/request-helpers.test.ts

@@ -214,6 +214,7 @@ describe('request formatting methods', () => {
       expect(body.instances[0].prompt).to.equal(prompt);
       expect(body.parameters.sampleCount).to.equal(1);
       expect(body.parameters.includeRaiReason).to.be.true;
+      expect(body.parameters.includeSafetyAttributes).to.be.true;
 
       // Parameters without default values should be undefined
       expect(body.parameters.storageUri).to.be.undefined;
@@ -258,6 +259,7 @@ describe('request formatting methods', () => {
       personGeneration: safetySettings.personFilterLevel,
       aspectRatio,
       includeRaiReason: true,
+      includeSafetyAttributes: true,
       storageUri: undefined
     });
   });

packages/ai/src/requests/request-helpers.ts

@@ -156,7 +156,8 @@ export function createPredictRequestBody(
       addWatermark,
       safetyFilterLevel,
       personGeneration: personFilterLevel,
-      includeRaiReason: true
+      includeRaiReason: true,
+      includeSafetyAttributes: true
     }
   };
   return body;

packages/ai/src/types/imagen/internal.ts

@@ -84,6 +84,7 @@ export interface ImagenResponseInternal {
  *     "personGeneration": "allow_all",
  *     "sampleCount": 2,
  *     "includeRaiReason": true,
+ *     "includeSafetyAttributes": true,
  *     "aspectRatio": "9:16"
  *   }
  * }
@@ -111,6 +112,7 @@ export interface PredictRequestBody {
     safetyFilterLevel?: string;
     personGeneration?: string; // Maps to personFilterLevel
     includeRaiReason: boolean;
+    includeSafetyAttributes: boolean;
   };
 }