Updated model armor code snippets to use module exports

Author: rudrakhsha-crest

model-armor/snippets/getFolderFloorSettings.js

@@ -19,7 +19,7 @@
  *
  * @param {string} folderId - The ID of the Google Cloud folder for which to retrieve floor settings.
  */
-async function main(folderId) {
+async function getFolderFloorSettings(folderId) {
   // [START modelarmor_get_folder_floor_settings]
   /**
    * TODO(developer): Uncomment these variables before running the sample.
@@ -34,19 +34,14 @@ async function main(folderId) {
   // Instantiates a client
   const modelarmorClient = new ModelArmorClient();
 
-  async function getFolderFloorSettings() {
-    // Construct request
-    const request = {
-      name,
-    };
+  // Construct request
+  const request = {
+    name,
+  };
 
-    const response = await modelarmorClient.getFloorSetting(request);
-    console.log(response);
-  }
-
-  getFolderFloorSettings();
+  const [response] = await modelarmorClient.getFloorSetting(request);
+  return response;
   // [END modelarmor_get_folder_floor_settings]
 }
 
-const args = process.argv.slice(2);
-main(...args).catch(console.error);
+module.exports = getFolderFloorSettings;

model-armor/snippets/getOrganizationFloorSettings.js

@@ -20,7 +20,7 @@
  * @param {string} organizationId - The ID of the Google Cloud organization for which to retrieve
  *                                 floor settings.
  */
-async function main(organizationId) {
+async function getOrganizationFloorSettings(organizationId) {
   // [START modelarmor_get_organization_floor_settings]
   /**
    * TODO(developer): Uncomment these variables before running the sample.
@@ -35,20 +35,15 @@ async function main(organizationId) {
   // Instantiates a client
   const modelarmorClient = new ModelArmorClient();
 
-  async function getOrganizationFloorSettings() {
-    // Construct request
-    const request = {
-      name,
-    };
+  // Construct request
+  const request = {
+    name,
+  };
 
-    // Run request
-    const response = await modelarmorClient.getFloorSetting(request);
-    console.log(response);
-  }
-
-  getOrganizationFloorSettings();
+  // Run request
+  const [response] = await modelarmorClient.getFloorSetting(request);
+  return response;
   // [END modelarmor_get_organization_floor_settings]
 }
 
-const args = process.argv.slice(2);
-main(...args).catch(console.error);
+module.exports = getOrganizationFloorSettings;

model-armor/snippets/getProjectFloorSettings.js

@@ -20,7 +20,7 @@
  * @param {string} projectId - The ID of the Google Cloud project for which to retrieve
  *                            floor settings.
  */
-async function main(projectId) {
+async function getProjectFloorSettings(projectId) {
   // [START modelarmor_get_project_floor_settings]
   /**
    * TODO(developer): Uncomment these variables before running the sample.
@@ -35,20 +35,15 @@ async function main(projectId) {
   // Instantiates a client
   const modelarmorClient = new ModelArmorClient();
 
-  async function getProjectFloorSettings() {
-    // Construct request
-    const request = {
-      name,
-    };
+  // Construct request
+  const request = {
+    name,
+  };
 
-    // Run request
-    const response = await modelarmorClient.getFloorSetting(request);
-    console.log(response);
-  }
-
-  getProjectFloorSettings();
+  // Run request
+  const [response] = await modelarmorClient.getFloorSetting(request);
+  return response;
   // [END modelarmor_get_project_floor_settings]
 }
 
-const args = process.argv.slice(2);
-main(...args).catch(console.error);
+module.exports = getProjectFloorSettings;

model-armor/snippets/updateFolderFloorSettings.js

@@ -19,7 +19,7 @@
  *
  * @param {string} folderId - Google Cloud folder ID for which floor settings need to be updated.
  */
-async function main(folderId) {
+async function updateFolderFloorSettings(folderId) {
   // [START modelarmor_update_folder_floor_settings]
   /**
    * TODO(developer): Uncomment these variables before running the sample.
@@ -34,40 +34,35 @@ async function main(folderId) {
   // Instantiates a client
   const client = new ModelArmorClient();
 
-  async function updateFolderFloorSettings() {
-    const floorSettingsName = `folders/${folderId}/locations/global/floorSetting`;
+  const floorSettingsName = `folders/${folderId}/locations/global/floorSetting`;
 
-    // Build the floor settings with your preferred filters
-    // For more details on filters, please refer to the following doc:
-    // https://cloud.google.com/security-command-center/docs/key-concepts-model-armor#ma-filters
-    const floorSetting = {
-      name: floorSettingsName,
-      filterConfig: {
-        raiSettings: {
-          raiFilters: [
-            {
-              filterType:
-                protos.google.cloud.modelarmor.v1.RaiFilterType.HATE_SPEECH,
-              confidenceLevel:
-                protos.google.cloud.modelarmor.v1.DetectionConfidenceLevel.HIGH,
-            },
-          ],
-        },
+  // Build the floor settings with your preferred filters
+  // For more details on filters, please refer to the following doc:
+  // https://cloud.google.com/security-command-center/docs/key-concepts-model-armor#ma-filters
+  const floorSetting = {
+    name: floorSettingsName,
+    filterConfig: {
+      raiSettings: {
+        raiFilters: [
+          {
+            filterType:
+              protos.google.cloud.modelarmor.v1.RaiFilterType.HATE_SPEECH,
+            confidenceLevel:
+              protos.google.cloud.modelarmor.v1.DetectionConfidenceLevel.HIGH,
+          },
+        ],
       },
-      enableFloorSettingEnforcement: true,
-    };
+    },
+    enableFloorSettingEnforcement: true,
+  };
 
-    const request = {
-      floorSetting: floorSetting,
-    };
+  const request = {
+    floorSetting: floorSetting,
+  };
 
-    const [response] = await client.updateFloorSetting(request);
-    console.log('Updated folder floor settings', response);
-  }
-
-  updateFolderFloorSettings();
+  const [response] = await client.updateFloorSetting(request);
+  return response;
   // [END modelarmor_update_folder_floor_settings]
 }
 
-const args = process.argv.slice(2);
-main(...args).catch(console.error);
+module.exports = updateFolderFloorSettings;

model-armor/snippets/updateOrganizationFloorSettings.js

@@ -19,7 +19,7 @@
  *
  * @param {string} organizationId - Google Cloud organization ID for which floor settings need to be updated.
  */
-async function main(organizationId) {
+async function updateOrganizationFloorSettings(organizationId) {
   // [START modelarmor_update_organization_floor_settings]
   /**
    * TODO(developer): Uncomment these variables before running the sample.
@@ -59,10 +59,8 @@ async function main(organizationId) {
   };
 
   const [response] = await client.updateFloorSetting(request);
-  console.log('Updated organization floor settings:', response);
-
+  return response;
   // [END modelarmor_update_organization_floor_settings]
 }
 
-const args = process.argv.slice(2);
-main(...args).catch(console.error);
+module.exports = updateOrganizationFloorSettings;

model-armor/snippets/updateProjectFloorSettings.js

@@ -19,7 +19,7 @@
  *
  * @param {string} projectId - Google Cloud project ID for which floor settings need to be updated.
  */
-async function main(projectId) {
+async function updateProjectFloorSettings(projectId) {
   // [START modelarmor_update_project_floor_settings]
   /**
    * TODO(developer): Uncomment these variables before running the sample.
@@ -59,9 +59,8 @@ async function main(projectId) {
   };
 
   const [response] = await client.updateFloorSetting(request);
-  console.log('Updated project floor settings:', response);
+  return response;
   // [END modelarmor_update_project_floor_settings]
 }
 
-const args = process.argv.slice(2);
-main(...args).catch(console.error);
+module.exports = updateProjectFloorSettings;

model-armor/test/modelarmor.test.js

@@ -92,7 +92,10 @@ async function disableFloorSettings() {
           paths: ['enable_floor_setting_enforcement'],
         },
       });
-      console.log('Disabled project floor settings:', updatedProjectSettings.name);
+      console.log(
+        'Disabled project floor settings:',
+        updatedProjectSettings.name
+      );
     }
 
     // Disable folder floor settings if folderId is available
@@ -111,7 +114,10 @@ async function disableFloorSettings() {
             paths: ['enable_floor_setting_enforcement'],
           },
         });
-        console.log('Disabled folder floor settings:', updatedFolderSettings.name);
+        console.log(
+          'Disabled folder floor settings:',
+          updatedFolderSettings.name
+        );
       }
     }
 
@@ -131,7 +137,10 @@ async function disableFloorSettings() {
             paths: ['enable_floor_setting_enforcement'],
           },
         });
-        console.log('Disabled organization floor settings:', updatedOrgSettings.name);
+        console.log(
+          'Disabled organization floor settings:',
+          updatedOrgSettings.name
+        );
       }
     }
   } catch (error) {
@@ -230,7 +239,7 @@ describe('Model Armor tests', () => {
   after(async () => {
     // Disable floor settings to restore original state
     await disableFloorSettings();
-    
+
     // Clean up all templates
     const directTemplates = [emptyTemplateId, basicTemplateId];
     for (const templateId of directTemplates) {
@@ -316,4 +325,4 @@ describe('Model Armor tests', () => {
     // Check that the response contains enableFloorSettingEnforcement=true
     assert.match(output, /enableFloorSettingEnforcement:\s*true/);
   });
-});
+});