feat(modelarmor): Added snippets for floor settings

model-armor/package.json

@@ -1,27 +1,26 @@
 {
-    "name": "nodejs-model-armor-samples",
-    "private": true,
-    "license": "Apache-2.0",
-    "files": [
-      "*.js"
-    ],
-    "author": "Google LLC",
-    "repository": "googleapis/nodejs-model-armor",
-    "engines": {
-      "node": ">=16.0.0"
-    },
-    "scripts": {
-      "test": "c8 mocha -p -j 2 --recursive test/ --timeout=60000"
-    },
-    "dependencies": {
-      "@google-cloud/modelarmor": "^0.1.0",
-      "@google-cloud/dlp": "^5.0.0"
-    },
-    "devDependencies": {
-      "c8": "^10.0.0",
-      "chai": "^4.5.0",
-      "mocha": "^10.0.0",
-      "uuid": "^10.0.0"
-    }
+  "name": "nodejs-model-armor-samples",
+  "private": true,
+  "license": "Apache-2.0",
+  "files": [
+    "*.js"
+  ],
+  "author": "Google LLC",
+  "repository": "googleapis/nodejs-model-armor",
+  "engines": {
+    "node": ">=16.0.0"
+  },
+  "scripts": {
+    "test": "c8 mocha -p -j 2 --recursive test/ --timeout=60000"
+  },
+  "dependencies": {
+    "@google-cloud/modelarmor": "^0.1.0",
+    "@google-cloud/dlp": "^5.0.0"
+  },
+  "devDependencies": {
+    "c8": "^10.0.0",
+    "chai": "^4.5.0",
+    "mocha": "^10.0.0",
+    "uuid": "^10.0.0"
+  }
 }
-

model-armor/snippets/ci-setup.json

@@ -0,0 +1,6 @@
+{
+    "env": {
+        "MA_FOLDER_ID": 695279264361,
+        "MA_ORG_ID": 951890214235
+    }
+}

model-armor/snippets/createTemplateWithMetadata.js

@@ -66,7 +66,7 @@ async function createTemplateWithMetadata(projectId, locationId, templateId) {
       },
     },
     templateMetadata: {
-      ignorePartialInvocationFailures: true,
+      logTemplateOperations: true,
       logSanitizeOperations: true,
     },
   };

model-armor/snippets/getFolderFloorSettings.js

@@ -0,0 +1,47 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+'use strict';
+
+/**
+ * Retrieves the floor settings for a Google Cloud folder.
+ *
+ * @param {string} folderId - The ID of the Google Cloud folder for which to retrieve floor settings.
+ */
+async function getFolderFloorSettings(folderId) {
+  // [START modelarmor_get_folder_floor_settings]
+  /**
+   * TODO(developer): Uncomment these variables before running the sample.
+   */
+  // const folderId = 'your-folder-id';
+
+  const name = `folders/${folderId}/locations/global/floorSetting`;
+
+  // Imports the Modelarmor library
+  const {ModelArmorClient} = require('@google-cloud/modelarmor').v1;
+
+  // Instantiates a client
+  const modelarmorClient = new ModelArmorClient();
+
+  // Construct request
+  const request = {
+    name,
+  };
+
+  const [response] = await modelarmorClient.getFloorSetting(request);
+  return response;
+  // [END modelarmor_get_folder_floor_settings]
+}
+
+module.exports = getFolderFloorSettings;

model-armor/snippets/getOrganizationFloorSettings.js

@@ -0,0 +1,49 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+'use strict';
+
+/**
+ * Retrieves the floor settings for a Google Cloud organization.
+ *
+ * @param {string} organizationId - The ID of the Google Cloud organization for which to retrieve
+ *                                 floor settings.
+ */
+async function getOrganizationFloorSettings(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;
+
+  // Instantiates a client
+  const modelarmorClient = new ModelArmorClient();
+
+  // Construct request
+  const request = {
+    name,
+  };
+
+  // Run request
+  const [response] = await modelarmorClient.getFloorSetting(request);
+  return response;
+  // [END modelarmor_get_organization_floor_settings]
+}
+
+module.exports = getOrganizationFloorSettings;

model-armor/snippets/getProjectFloorSettings.js

@@ -0,0 +1,49 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+'use strict';
+
+/**
+ * Retrieves the floor settings for a Google Cloud project.
+ *
+ * @param {string} projectId - The ID of the Google Cloud project for which to retrieve
+ *                            floor settings.
+ */
+async function getProjectFloorSettings(projectId) {
+  // [START modelarmor_get_project_floor_settings]
+  /**
+   * TODO(developer): Uncomment these variables before running the sample.
+   */
+  // const projectId = 'your-project-id';
+
+  const name = `projects/${projectId}/locations/global/floorSetting`;
+
+  // Imports the Modelarmor library
+  const {ModelArmorClient} = require('@google-cloud/modelarmor').v1;
+
+  // Instantiates a client
+  const modelarmorClient = new ModelArmorClient();
+
+  // Construct request
+  const request = {
+    name,
+  };
+
+  // Run request
+  const [response] = await modelarmorClient.getFloorSetting(request);
+  return response;
+  // [END modelarmor_get_project_floor_settings]
+}
+
+module.exports = getProjectFloorSettings;

model-armor/snippets/updateFolderFloorSettings.js

@@ -0,0 +1,68 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+'use strict';
+
+/**
+ * Updates the floor settings of a folder in Model Armor.
+ *
+ * @param {string} folderId - Google Cloud folder ID for which floor settings need to be updated.
+ */
+async function updateFolderFloorSettings(folderId) {
+  // [START modelarmor_update_folder_floor_settings]
+  /**
+   * TODO(developer): Uncomment these variables before running the sample.
+   */
+  // const folderId = 'your-folder-id';
+
+  // Imports the Model Armor library
+  const modelarmor = require('@google-cloud/modelarmor');
+  const {ModelArmorClient} = modelarmor.v1;
+  const {protos} = modelarmor;
+
+  // Instantiates a client
+  const client = new ModelArmorClient();
+
+  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,
+          },
+        ],
+      },
+    },
+    enableFloorSettingEnforcement: true,
+  };
+
+  const request = {
+    floorSetting: floorSetting,
+  };
+
+  const [response] = await client.updateFloorSetting(request);
+  return response;
+  // [END modelarmor_update_folder_floor_settings]
+}
+
+module.exports = updateFolderFloorSettings;

model-armor/snippets/updateOrganizationFloorSettings.js

@@ -0,0 +1,66 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+'use strict';
+
+/**
+ * Updates the floor settings of an organization in Model Armor.
+ *
+ * @param {string} organizationId - Google Cloud organization ID for which floor settings need to be updated.
+ */
+async function updateOrganizationFloorSettings(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 client = new ModelArmorClient();
+
+  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.HATE_SPEECH,
+            confidenceLevel:
+              protos.google.cloud.modelarmor.v1.DetectionConfidenceLevel.HIGH,
+          },
+        ],
+      },
+    },
+    enableFloorSettingEnforcement: true,
+  };
+
+  const request = {
+    floorSetting: floorSetting,
+  };
+
+  const [response] = await client.updateFloorSetting(request);
+  return response;
+  // [END modelarmor_update_organization_floor_settings]
+}
+
+module.exports = updateOrganizationFloorSettings;