Ignore safetyAttributes in response parsing

Author: dlarocque

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

@@ -296,6 +296,8 @@ export async function handlePredictResponse<
         mimeType: prediction.mimeType,
         gcsURI: prediction.gcsUri
       } as T);
+    } else if (prediction.safetyAttributes) {
+      // Ignore safetyAttributes "prediction" to avoid throwing an error below.
     } else {
       throw new AIError(
         AIErrorCode.RESPONSE_ERROR,

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

@@ -61,6 +61,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 field is currently only used to avoid logging unsupported
+     * prediction types.
+     */
+    safetyAttributes?: unknown;
   }>;
 }