chore(modelarmor): reverted update floor setting snippets changes

Author: mihirvala08

model_armor/snippets/update_folder_floor_settings.py

@@ -15,9 +15,6 @@
 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
 
 
@@ -53,7 +50,12 @@ 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=[]
+                        rai_filters=[
+                            modelarmor_v1.RaiFilterSettings.RaiFilter(
+                                filter_type=modelarmor_v1.RaiFilterType.HATE_SPEECH,
+                                confidence_level=modelarmor_v1.DetectionConfidenceLevel.HIGH,
+                            )
+                        ]
                     ),
                 ),
                 enable_floor_setting_enforcement=True,

model_armor/snippets/update_organizations_floor_settings.py

@@ -15,9 +15,6 @@
 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
 
 
@@ -57,7 +54,12 @@ def update_organization_floor_settings(
                 name=floor_settings_name,
                 filter_config=modelarmor_v1.FilterConfig(
                     rai_settings=modelarmor_v1.RaiFilterSettings(
-                        rai_filters=[]
+                        rai_filters=[
+                            modelarmor_v1.RaiFilterSettings.RaiFilter(
+                                filter_type=modelarmor_v1.RaiFilterType.HATE_SPEECH,
+                                confidence_level=modelarmor_v1.DetectionConfidenceLevel.HIGH,
+                            )
+                        ]
                     ),
                 ),
                 enable_floor_setting_enforcement=True,

model_armor/snippets/update_project_floor_settings.py

@@ -15,15 +15,10 @@
 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:
@@ -55,7 +50,12 @@ def update_project_floor_settings(
                 name=floor_settings_name,
                 filter_config=modelarmor_v1.FilterConfig(
                     rai_settings=modelarmor_v1.RaiFilterSettings(
-                        rai_filters=[]
+                        rai_filters=[
+                            modelarmor_v1.RaiFilterSettings.RaiFilter(
+                                filter_type=modelarmor_v1.RaiFilterType.HATE_SPEECH,
+                                confidence_level=modelarmor_v1.DetectionConfidenceLevel.HIGH,
+                            )
+                        ]
                     ),
                 ),
                 enable_floor_setting_enforcement=True,