fix(modelarmor): Updated floor setting snippets/tests for intermittent behavior of floorsettings API

Author: mihirvala08

model_armor/snippets/update_folder_floor_settings.py

@@ -15,6 +15,9 @@
 Sample code for updating the model armor folder settings of a folder.
 """
 
+# TODO: Update to change rai filter configs using floor setting
+# once floor settings API works correctly.
+
 from google.cloud import modelarmor_v1
 
 
@@ -50,12 +53,7 @@ def update_folder_floor_settings(folder_id: str) -> modelarmor_v1.FloorSetting:
                 name=floor_settings_name,
                 filter_config=modelarmor_v1.FilterConfig(
                     rai_settings=modelarmor_v1.RaiFilterSettings(
-                        rai_filters=[
-                            modelarmor_v1.RaiFilterSettings.RaiFilter(
-                                filter_type=modelarmor_v1.RaiFilterType.HATE_SPEECH,
-                                confidence_level=modelarmor_v1.DetectionConfidenceLevel.HIGH,
-                            )
-                        ]
+                        rai_filters=[]
                     ),
                 ),
                 enable_floor_setting_enforcement=True,

model_armor/snippets/update_organizations_floor_settings.py

@@ -15,6 +15,9 @@
 Sample code for updating the model armor floor settings of an organization.
 """
 
+# TODO: Update to change rai filter configs using floor setting
+# once floor settings API works correctly.
+
 from google.cloud import modelarmor_v1
 
 
@@ -54,12 +57,7 @@ def update_organization_floor_settings(
                 name=floor_settings_name,
                 filter_config=modelarmor_v1.FilterConfig(
                     rai_settings=modelarmor_v1.RaiFilterSettings(
-                        rai_filters=[
-                            modelarmor_v1.RaiFilterSettings.RaiFilter(
-                                filter_type=modelarmor_v1.RaiFilterType.HATE_SPEECH,
-                                confidence_level=modelarmor_v1.DetectionConfidenceLevel.HIGH,
-                            )
-                        ]
+                        rai_filters=[]
                     ),
                 ),
                 enable_floor_setting_enforcement=True,

model_armor/snippets/update_project_floor_settings.py

@@ -15,10 +15,15 @@
 Sample code for updating the model armor project floor settings.
 """
 
+# TODO: Update to change rai filter configs using floor setting
+# once floor settings API works correctly.
+
 from google.cloud import modelarmor_v1
 
 
-def update_project_floor_settings(project_id: str) -> modelarmor_v1.FloorSetting:
+def update_project_floor_settings(
+    project_id: str,
+) -> modelarmor_v1.FloorSetting:
     """Update the floor settings of a project.
 
     Args:
@@ -50,12 +55,7 @@ def update_project_floor_settings(project_id: str) -> modelarmor_v1.FloorSetting
                 name=floor_settings_name,
                 filter_config=modelarmor_v1.FilterConfig(
                     rai_settings=modelarmor_v1.RaiFilterSettings(
-                        rai_filters=[
-                            modelarmor_v1.RaiFilterSettings.RaiFilter(
-                                filter_type=modelarmor_v1.RaiFilterType.HATE_SPEECH,
-                                confidence_level=modelarmor_v1.DetectionConfidenceLevel.HIGH,
-                            )
-                        ]
+                        rai_filters=[]
                     ),
                 ),
                 enable_floor_setting_enforcement=True,