chore(modelarmor): refactored floor settings snippets

mihirvala08 · mihirvala08 · commit 664fd3e8972a · 2025-06-13T17:17:45.000+05:30
diff --git a/model-armor/snippets/getFolderFloorSettings.js b/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 getFolderFloorSettings(folderId) {
+async function main(folderId) {
   // [START modelarmor_get_folder_floor_settings]
   /**
    * TODO(developer): Uncomment these variables before running the sample.
@@ -29,22 +29,27 @@ async function getFolderFloorSettings(folderId) {
   const name = `folders/${folderId}/locations/global/floorSetting`;
 
   // Imports the Modelarmor library
-  const {ModelArmorClient} = require('@google-cloud/modelarmor').v1;
+  const { ModelArmorClient } = require('@google-cloud/modelarmor').v1;
 
   // Instantiates a client
   const modelarmorClient = new ModelArmorClient();
 
-  // Construct request
-  const request = {
-    name,
-  };
+  async function getFolderFloorSettings() {
+    // Construct request
+    const request = {
+      name,
+    };
 
-  const [response] = await modelarmorClient.getFloorSetting(request);
-  return response;
+    const [response] = await modelarmorClient.getFloorSetting(request);
+    return response;
+
+  }
+
+  return await getFolderFloorSettings();
   // [END modelarmor_get_folder_floor_settings]
 }
 
-module.exports = getFolderFloorSettings;
+module.exports.main = main;
 
 /* c8 ignore next 10 */
 if (require.main === module) {
diff --git a/model-armor/snippets/getOrganizationFloorSettings.js b/model-armor/snippets/getOrganizationFloorSettings.js
@@ -20,33 +20,37 @@
  * @param {string} organizationId - The ID of the Google Cloud organization for which to retrieve
  *                                 floor settings.
  */
-async function getOrganizationFloorSettings(organizationId) {
+async function main(organizationId) {
   // [START modelarmor_get_organization_floor_settings]
   /**
    * TODO(developer): Uncomment these variables before running the sample.
    */
   // const organizationId = 'your-organization-id';
-
   const name = `organizations/${organizationId}/locations/global/floorSetting`;
 
   // Imports the Modelarmor library
-  const {ModelArmorClient} = require('@google-cloud/modelarmor').v1;
+  const { ModelArmorClient } = require('@google-cloud/modelarmor').v1;
 
   // Instantiates a client
   const modelarmorClient = new ModelArmorClient();
 
-  // Construct request
-  const request = {
-    name,
-  };
+  async function getOrganizationFloorSettings() {
+    // Construct request
+    const request = {
+      name,
+    };
+
+    // Run request
+    const [response] = await modelarmorClient.getFloorSetting(request);
+    return response;
+
+  }
 
-  // Run request
-  const [response] = await modelarmorClient.getFloorSetting(request);
-  return response;
+  return await getOrganizationFloorSettings();
   // [END modelarmor_get_organization_floor_settings]
 }
 
-module.exports = getOrganizationFloorSettings;
+module.exports.main = main;
 
 /* c8 ignore next 10 */
 if (require.main === module) {
diff --git a/model-armor/snippets/getProjectFloorSettings.js b/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 getProjectFloorSettings(projectId) {
+async function main(projectId) {
   // [START modelarmor_get_project_floor_settings]
   /**
    * TODO(developer): Uncomment these variables before running the sample.
@@ -30,23 +30,27 @@ async function getProjectFloorSettings(projectId) {
   const name = `projects/${projectId}/locations/global/floorSetting`;
 
   // Imports the Modelarmor library
-  const {ModelArmorClient} = require('@google-cloud/modelarmor').v1;
+  const { ModelArmorClient } = require('@google-cloud/modelarmor').v1;
 
   // Instantiates a client
   const modelarmorClient = new ModelArmorClient();
 
-  // Construct request
-  const request = {
-    name,
-  };
+  async function getProjectFloorSettings() {
+    // Construct request
+    const request = {
+      name,
+    };
 
-  // Run request
-  const [response] = await modelarmorClient.getFloorSetting(request);
-  return response;
+    // Run request
+    const [response] = await modelarmorClient.getFloorSetting(request);
+    return response;
+  }
+
+  return await getProjectFloorSettings();
   // [END modelarmor_get_project_floor_settings]
 }
 
-module.exports = getProjectFloorSettings;
+module.exports.main = main;
 
 /* c8 ignore next 10 */
 if (require.main === module) {
diff --git a/model-armor/snippets/updateFolderFloorSettings.js b/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 updateFolderFloorSettings(folderId) {
+async function main(folderId) {
   // [START modelarmor_update_folder_floor_settings]
   /**
    * TODO(developer): Uncomment these variables before running the sample.
@@ -28,51 +28,57 @@ async function updateFolderFloorSettings(folderId) {
 
   // Imports the Model Armor library
   const modelarmor = require('@google-cloud/modelarmor');
-  const {ModelArmorClient} = modelarmor.v1;
-  const {protos} = modelarmor;
+  const { ModelArmorClient } = modelarmor.v1;
+  const { protos } = modelarmor;
 
   // Instantiates a client
   const client = new ModelArmorClient();
 
-  const floorSettingsName = `folders/${folderId}/locations/global/floorSetting`;
+  async function updateFolderFloorSettings() {
 
-  // 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.HARASSMENT,
-            confidenceLevel:
-              protos.google.cloud.modelarmor.v1.DetectionConfidenceLevel
-                .LOW_AND_ABOVE,
-          },
-          {
-            filterType:
-              protos.google.cloud.modelarmor.v1.RaiFilterType.SEXUALLY_EXPLICIT,
-            confidenceLevel:
-              protos.google.cloud.modelarmor.v1.DetectionConfidenceLevel.HIGH,
-          },
-        ],
+    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.HARASSMENT,
+              confidenceLevel:
+                protos.google.cloud.modelarmor.v1.DetectionConfidenceLevel
+                  .LOW_AND_ABOVE,
+            },
+            {
+              filterType:
+                protos.google.cloud.modelarmor.v1.RaiFilterType.SEXUALLY_EXPLICIT,
+              confidenceLevel:
+                protos.google.cloud.modelarmor.v1.DetectionConfidenceLevel.HIGH,
+            },
+          ],
+        },
       },
-    },
-    enableFloorSettingEnforcement: true,
-  };
+      enableFloorSettingEnforcement: true,
+    };
+
+    const request = {
+      floorSetting: floorSetting,
+    };
+
+    const [response] = await client.updateFloorSetting(request);
+    return response;
 
-  const request = {
-    floorSetting: floorSetting,
-  };
+  }
 
-  const [response] = await client.updateFloorSetting(request);
-  return response;
+  return await updateFolderFloorSettings();
   // [END modelarmor_update_folder_floor_settings]
 }
 
-module.exports = updateFolderFloorSettings;
+module.exports.main = main;
 
 /* c8 ignore next 10 */
 if (require.main === module) {
diff --git a/model-armor/snippets/updateOrganizationFloorSettings.js b/model-armor/snippets/updateOrganizationFloorSettings.js
@@ -19,58 +19,64 @@
  *
  * @param {string} organizationId - Google Cloud organization ID for which floor settings need to be updated.
  */
-async function updateOrganizationFloorSettings(organizationId) {
+async function main(organizationId) {
+
   // [START modelarmor_update_organization_floor_settings]
   /**
    * TODO(developer): Uncomment these variables before running the sample.
    */
   // const organizationId = 'your-organization-id';
 
   const modelarmor = require('@google-cloud/modelarmor');
-  const {ModelArmorClient} = modelarmor.v1;
-  const {protos} = modelarmor;
+  const { ModelArmorClient } = modelarmor.v1;
+  const { protos } = modelarmor;
 
   const client = new ModelArmorClient();
 
-  const floorSettingsName = `organizations/${organizationId}/locations/global/floorSetting`;
+  async function updateOrganizationFloorSettings() {
+
+    const floorSettingsName = `organizations/${organizationId}/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.HARASSMENT,
-            confidenceLevel:
-              protos.google.cloud.modelarmor.v1.DetectionConfidenceLevel
-                .LOW_AND_ABOVE,
-          },
-          {
-            filterType:
-              protos.google.cloud.modelarmor.v1.RaiFilterType.SEXUALLY_EXPLICIT,
-            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.HARASSMENT,
+              confidenceLevel:
+                protos.google.cloud.modelarmor.v1.DetectionConfidenceLevel
+                  .LOW_AND_ABOVE,
+            },
+            {
+              filterType:
+                protos.google.cloud.modelarmor.v1.RaiFilterType.SEXUALLY_EXPLICIT,
+              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);
+    return response;
+  }
 
-  const [response] = await client.updateFloorSetting(request);
-  return response;
+  return await updateOrganizationFloorSettings();
   // [END modelarmor_update_organization_floor_settings]
 }
 
-module.exports = updateOrganizationFloorSettings;
+module.exports.main = main;
 
 /* c8 ignore next 10 */
 if (require.main === module) {
diff --git a/model-armor/snippets/updateProjectFloorSettings.js b/model-armor/snippets/updateProjectFloorSettings.js
