feat: [google-cloud-modelarmor] MultiLanguage Detection in Model Armor Floor Setting

packages/google-cloud-modelarmor/google/cloud/modelarmor/__init__.py

@@ -23,6 +23,7 @@
 )
 from google.cloud.modelarmor_v1.services.model_armor.client import ModelArmorClient
 from google.cloud.modelarmor_v1.types.service import (
+    AiPlatformFloorSetting,
     ByteDataItem,
     CreateTemplateRequest,
     CsamFilterResult,
@@ -72,6 +73,7 @@
 __all__ = (
     "ModelArmorClient",
     "ModelArmorAsyncClient",
+    "AiPlatformFloorSetting",
     "ByteDataItem",
     "CreateTemplateRequest",
     "CsamFilterResult",

packages/google-cloud-modelarmor/google/cloud/modelarmor/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "0.2.6"  # {x-release-please-version}
+__version__ = "0.0.0"  # {x-release-please-version}

packages/google-cloud-modelarmor/google/cloud/modelarmor_v1/__init__.py

@@ -20,6 +20,7 @@
 
 from .services.model_armor import ModelArmorAsyncClient, ModelArmorClient
 from .types.service import (
+    AiPlatformFloorSetting,
     ByteDataItem,
     CreateTemplateRequest,
     CsamFilterResult,
@@ -68,6 +69,7 @@
 
 __all__ = (
     "ModelArmorAsyncClient",
+    "AiPlatformFloorSetting",
     "ByteDataItem",
     "CreateTemplateRequest",
     "CsamFilterResult",

packages/google-cloud-modelarmor/google/cloud/modelarmor_v1/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "0.2.6"  # {x-release-please-version}
+__version__ = "0.0.0"  # {x-release-please-version}

packages/google-cloud-modelarmor/google/cloud/modelarmor_v1/types/__init__.py

@@ -14,6 +14,7 @@
 # limitations under the License.
 #
 from .service import (
+    AiPlatformFloorSetting,
     ByteDataItem,
     CreateTemplateRequest,
     CsamFilterResult,
@@ -61,6 +62,7 @@
 )
 
 __all__ = (
+    "AiPlatformFloorSetting",
     "ByteDataItem",
     "CreateTemplateRequest",
     "CsamFilterResult",

packages/google-cloud-modelarmor/google/cloud/modelarmor_v1/types/service.py

@@ -32,6 +32,7 @@
         "InvocationResult",
         "Template",
         "FloorSetting",
+        "AiPlatformFloorSetting",
         "ListTemplatesRequest",
         "ListTemplatesResponse",
         "GetTemplateRequest",
@@ -246,11 +247,32 @@ class TemplateMetadata(proto.Message):
                 operations.
             log_sanitize_operations (bool):
                 Optional. If true, log sanitize operations.
+            enforcement_type (google.cloud.modelarmor_v1.types.Template.TemplateMetadata.EnforcementType):
+                Optional. Enforcement type for Model Armor
+                filters.
             multi_language_detection (google.cloud.modelarmor_v1.types.Template.TemplateMetadata.MultiLanguageDetection):
                 Optional. Metadata for multi language
                 detection.
         """
 
+        class EnforcementType(proto.Enum):
+            r"""Enforcement type for Model Armor filters.
+
+            Values:
+                ENFORCEMENT_TYPE_UNSPECIFIED (0):
+                    Default value. Same as INSPECT_AND_BLOCK.
+                INSPECT_ONLY (1):
+                    Model Armor filters will run in inspect only
+                    mode. No action will be taken on the request.
+                INSPECT_AND_BLOCK (2):
+                    Model Armor filters will run in inspect and
+                    block mode. Requests that trip Model Armor
+                    filters will be blocked.
+            """
+            ENFORCEMENT_TYPE_UNSPECIFIED = 0
+            INSPECT_ONLY = 1
+            INSPECT_AND_BLOCK = 2
+
         class MultiLanguageDetection(proto.Message):
             r"""Metadata to enable multi language detection via template.
 
@@ -293,6 +315,11 @@ class MultiLanguageDetection(proto.Message):
             proto.BOOL,
             number=7,
         )
+        enforcement_type: "Template.TemplateMetadata.EnforcementType" = proto.Field(
+            proto.ENUM,
+            number=8,
+            enum="Template.TemplateMetadata.EnforcementType",
+        )
         multi_language_detection: "Template.TemplateMetadata.MultiLanguageDetection" = (
             proto.Field(
                 proto.MESSAGE,
@@ -350,8 +377,32 @@ class FloorSetting(proto.Message):
             Optional. Floor Settings enforcement status.
 
             This field is a member of `oneof`_ ``_enable_floor_setting_enforcement``.
+        integrated_services (MutableSequence[google.cloud.modelarmor_v1.types.FloorSetting.IntegratedService]):
+            Optional. List of integrated services for
+            which the floor setting is applicable.
+        ai_platform_floor_setting (google.cloud.modelarmor_v1.types.AiPlatformFloorSetting):
+            Optional. AI Platform floor setting.
+
+            This field is a member of `oneof`_ ``_ai_platform_floor_setting``.
+        floor_setting_metadata (google.cloud.modelarmor_v1.types.FloorSetting.FloorSettingMetadata):
+            Optional. Metadata for FloorSetting
     """
 
+    class IntegratedService(proto.Enum):
+        r"""Integrated service for which the floor setting is applicable.
+
+        Values:
+            INTEGRATED_SERVICE_UNSPECIFIED (0):
+                Unspecified integrated service.
+            AI_PLATFORM (1):
+                AI Platform.
+        """
+        INTEGRATED_SERVICE_UNSPECIFIED = 0
+        AI_PLATFORM = 1
+
+    class FloorSettingMetadata(proto.Message):
+        r"""message describing FloorSetting Metadata"""
+
     name: str = proto.Field(
         proto.STRING,
         number=1,
@@ -376,6 +427,66 @@ class FloorSetting(proto.Message):
         number=5,
         optional=True,
     )
+    integrated_services: MutableSequence[IntegratedService] = proto.RepeatedField(
+        proto.ENUM,
+        number=6,
+        enum=IntegratedService,
+    )
+    ai_platform_floor_setting: "AiPlatformFloorSetting" = proto.Field(
+        proto.MESSAGE,
+        number=7,
+        optional=True,
+        message="AiPlatformFloorSetting",
+    )
+    floor_setting_metadata: FloorSettingMetadata = proto.Field(
+        proto.MESSAGE,
+        number=8,
+        message=FloorSettingMetadata,
+    )
+
+
+class AiPlatformFloorSetting(proto.Message):
+    r"""message describing AiPlatformFloorSetting
+
+    This message has `oneof`_ fields (mutually exclusive fields).
+    For each oneof, at most one member field can be set at the same time.
+    Setting any member of the oneof automatically clears all other
+    members.
+
+    .. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
+
+    Attributes:
+        inspect_only (bool):
+            Optional. If true, Model Armor filters will
+            be run in inspect only mode. No action will be
+            taken on the request.
+
+            This field is a member of `oneof`_ ``enforcement_type``.
+        inspect_and_block (bool):
+            Optional. If true, Model Armor filters will
+            be run in inspect and block mode. Requests that
+            trip Model Armor filters will be blocked.
+
+            This field is a member of `oneof`_ ``enforcement_type``.
+        enable_cloud_logging (bool):
+            Optional. If true, log Model Armor filter
+            results to Cloud Logging.
+    """
+
+    inspect_only: bool = proto.Field(
+        proto.BOOL,
+        number=1,
+        oneof="enforcement_type",
+    )
+    inspect_and_block: bool = proto.Field(
+        proto.BOOL,
+        number=2,
+        oneof="enforcement_type",
+    )
+    enable_cloud_logging: bool = proto.Field(
+        proto.BOOL,
+        number=3,
+    )
 
 
 class ListTemplatesRequest(proto.Message):

packages/google-cloud-modelarmor/google/cloud/modelarmor_v1beta/__init__.py

@@ -20,6 +20,7 @@
 
 from .services.model_armor import ModelArmorAsyncClient, ModelArmorClient
 from .types.service import (
+    AiPlatformFloorSetting,
     ByteDataItem,
     CreateTemplateRequest,
     CsamFilterResult,
@@ -68,6 +69,7 @@
 
 __all__ = (
     "ModelArmorAsyncClient",
+    "AiPlatformFloorSetting",
     "ByteDataItem",
     "CreateTemplateRequest",
     "CsamFilterResult",

packages/google-cloud-modelarmor/google/cloud/modelarmor_v1beta/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "0.2.6"  # {x-release-please-version}
+__version__ = "0.0.0"  # {x-release-please-version}

packages/google-cloud-modelarmor/google/cloud/modelarmor_v1beta/types/__init__.py

@@ -14,6 +14,7 @@
 # limitations under the License.
 #
 from .service import (
+    AiPlatformFloorSetting,
     ByteDataItem,
     CreateTemplateRequest,
     CsamFilterResult,
@@ -61,6 +62,7 @@
 )
 
 __all__ = (
+    "AiPlatformFloorSetting",
     "ByteDataItem",
     "CreateTemplateRequest",
     "CsamFilterResult",