Updated prompts, tests and log statements for sanitization APIs

Author: rudrakhsha-crest

model-armor/package.json

@@ -14,7 +14,8 @@
       "test": "c8 mocha -p -j 2 --recursive test/ --timeout=60000"
     },
     "dependencies": {
-      "@google-cloud/modelarmor": "^0.1.0"
+      "@google-cloud/modelarmor": "^0.1.0",
+      "@google-cloud/dlp": "^5.0.0"
     },
     "devDependencies": {
       "c8": "^10.0.0",

model-armor/snippets/sanitizeModelResponse.js

@@ -38,7 +38,7 @@ async function main(projectId, locationId, templateId, modelResponse) {
     };
 
     const [response] = await client.sanitizeModelResponse(request);
-    console.log('Sanitized model response:', response);
+    console.log(JSON.stringify(response, null, 2));
   }
 
   sanitizeModelResponse();

model-armor/snippets/sanitizeModelResponseWithUserPrompt.js

@@ -46,7 +46,7 @@ async function main(
     };
 
     const [response] = await client.sanitizeModelResponse(request);
-    console.log('Sanitized model response with user prompt:', response);
+    console.log(JSON.stringify(response, null, 2));
   }
 
   sanitizeModelResponseWithUserPrompt();

model-armor/snippets/sanitizeUserPrompt.js

@@ -38,7 +38,7 @@ async function main(projectId, locationId, templateId, userPrompt) {
     };
 
     const [response] = await client.sanitizeUserPrompt(request);
-    console.log('Sanitized user prompt:', response);
+    console.log(JSON.stringify(response, null, 2));
   }
 
   sanitizeUserPrompt();

model-armor/snippets/screenPdfFile.js

@@ -15,22 +15,22 @@
 'use strict';
 
 /**
- * Sanitize/Screen PDF content using the Model Armor API.
+ * Sanitize/Screen PDF file content using the Model Armor API.
  *
  * @param {string} projectId - Google Cloud project ID.
  * @param {string} locationId - Google Cloud location.
  * @param {string} templateId - The template ID used for sanitization.
- * @param {string} pdfContentBase64 - Base64-encoded PDF content to sanitize.
+ * @param {string} pdfContentFilename - Path to a PDF file.
  */
-async function main(projectId, locationId, templateId, pdfContentBase64) {
+async function main(projectId, locationId, templateId, pdfContentFilename) {
   // [START modelarmor_screen_pdf_file]
   /**
    * TODO(developer): Uncomment these variables before running the sample.
    */
   // const projectId = process.env.PROJECT_ID || 'your-project-id';
   // const locationId = process.env.LOCATION_ID || 'us-central1';
   // const templateId = process.env.TEMPLATE_ID || 'template-id';
-  // const pdfContentBase64 = process.env.PDF_CONTENT_BASE64 || 'BASE64_ENCODED_PDF_CONTENT';
+  // const pdfContentFilename = 'path/to/file.pdf';
 
   // Imports the Model Armor library
   const modelarmor = require('@google-cloud/modelarmor');
@@ -39,7 +39,11 @@ async function main(projectId, locationId, templateId, pdfContentBase64) {
   const ByteItemType =
     protos.google.cloud.modelarmor.v1.ByteDataItem.ByteItemType;
 
-  // Instantiates a client
+  const fs = require('fs');
+
+  const pdfContent = fs.readFileSync(pdfContentFilename);
+  const pdfContentBase64 = pdfContent.toString('base64');
+
   const client = new ModelArmorClient({
     apiEndpoint: `modelarmor.${locationId}.rep.googleapis.com`,
   });
@@ -55,7 +59,7 @@ async function main(projectId, locationId, templateId, pdfContentBase64) {
   };
 
   const [response] = await client.sanitizeUserPrompt(request);
-  console.log('PDF Sanitization Result:', response);
+  console.log(JSON.stringify(response, null, 2));
   // [END modelarmor_screen_pdf_file]
 }