Set SafetyRating.blocked to false if missing in response

Author: dlarocque

packages/vertexai/src/methods/generate-content.ts

@@ -59,6 +59,14 @@ export async function generateContent(
     requestOptions
   );
   const responseJson: GenerateContentResponse = await response.json();
+  responseJson.candidates?.forEach(candidate => {
+    candidate.safetyRatings?.forEach(safetyRating => {
+      // The blocked property is only defined when true
+      if (!safetyRating.blocked) {
+        safetyRating.blocked = false;
+      }
+    })
+  })
   const enhancedResponse = createEnhancedContentResponse(responseJson);
   return {
     response: enhancedResponse

packages/vertexai/src/requests/stream-reader.ts

@@ -77,6 +77,15 @@ async function* generateResponseSequence(
       break;
     }
 
+    value.candidates?.forEach(candidate => {
+      candidate.safetyRatings?.forEach(safetyRating => {
+        // The blocked property is only defined when true
+        if (!safetyRating.blocked) {
+          safetyRating.blocked = false;
+        }
+      })
+    })
+
     const enhancedResponse = createEnhancedContentResponse(value);
     yield enhancedResponse;
   }
@@ -209,6 +218,13 @@ export function aggregateResponses(
             );
           }
         }
+
+        candidate.safetyRatings?.forEach(safetyRating => {
+          // The blocked property will only be defined when true
+          if (!safetyRating.blocked) {
+            safetyRating.blocked = false;
+          }
+        })
       }
     }
   }