google-cloud-modelarmor 0.5.0__tar.gz → 0.6.0__tar.gz

{google_cloud_modelarmor-0.5.0/google_cloud_modelarmor.egg-info → google_cloud_modelarmor-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-modelarmor
-Version: 0.5.0
+Version: 0.6.0
 Summary: Google Cloud Modelarmor API client library
 Home-page: https://github.com/googleapis/google-cloud-python/tree/main/packages/google-cloud-modelarmor
 Author: Google LLC
@@ -41,12 +41,14 @@ Dynamic: requires-dist
 Dynamic: requires-python
 Dynamic: summary
 
-Python Client for Model Armor API
-=================================
+Python Client for Model Armor
+=============================
 
 |preview| |pypi| |versions|
 
-`Model Armor API`_: Model Armor helps you protect against risks like prompt injection, harmful content, and data leakage in generative AI applications by letting you define policies that filter user prompts and model responses.
+`Model Armor`_: Model Armor helps you protect against risks like prompt injection, harmful
+content, and data leakage in generative AI applications by letting you
+define policies that filter user prompts and model responses.
 
 - `Client Library Documentation`_
 - `Product Documentation`_
@@ -57,9 +59,9 @@ Python Client for Model Armor API
    :target: https://pypi.org/project/google-cloud-modelarmor/
 .. |versions| image:: https://img.shields.io/pypi/pyversions/google-cloud-modelarmor.svg
    :target: https://pypi.org/project/google-cloud-modelarmor/
-.. _Model Armor API: https://cloud.google.com/security-command-center/docs/model-armor-overview
+.. _Model Armor: https://cloud.google.com/security-command-center/
 .. _Client Library Documentation: https://cloud.google.com/python/docs/reference/google-cloud-modelarmor/latest/summary_overview
-.. _Product Documentation:  https://cloud.google.com/security-command-center/docs/model-armor-overview
+.. _Product Documentation:  https://cloud.google.com/security-command-center/
 
 Quick Start
 -----------
@@ -68,12 +70,12 @@ In order to use this library, you first need to go through the following steps:
 
 1. `Select or create a Cloud Platform project.`_
 2. `Enable billing for your project.`_
-3. `Enable the Model Armor API.`_
+3. `Enable the Model Armor.`_
 4. `Set up Authentication.`_
 
 .. _Select or create a Cloud Platform project.: https://console.cloud.google.com/project
 .. _Enable billing for your project.: https://cloud.google.com/billing/docs/how-to/modify-project#enable_billing_for_a_project
-.. _Enable the Model Armor API.:  https://cloud.google.com/security-command-center/docs/model-armor-overview
+.. _Enable the Model Armor.:  https://cloud.google.com/security-command-center/
 .. _Set up Authentication.: https://googleapis.dev/python/google-api-core/latest/auth.html
 
 Installation
@@ -140,14 +142,14 @@ Windows
 Next Steps
 ~~~~~~~~~~
 
--  Read the `Client Library Documentation`_ for Model Armor API
+-  Read the `Client Library Documentation`_ for Model Armor
    to see other available methods on the client.
--  Read the `Model Armor API Product documentation`_ to learn
+-  Read the `Model Armor Product documentation`_ to learn
    more about the product and see How-to Guides.
 -  View this `README`_ to see the full list of Cloud
    APIs that we cover.
 
-.. _Model Armor API Product documentation:  https://cloud.google.com/security-command-center/docs/model-armor-overview
+.. _Model Armor Product documentation:  https://cloud.google.com/security-command-center/
 .. _README: https://github.com/googleapis/google-cloud-python/blob/main/README.rst
 
 Logging

{google_cloud_modelarmor-0.5.0 → google_cloud_modelarmor-0.6.0}/README.rst

@@ -1,9 +1,11 @@
-Python Client for Model Armor API
-=================================
+Python Client for Model Armor
+=============================
 
 |preview| |pypi| |versions|
 
-`Model Armor API`_: Model Armor helps you protect against risks like prompt injection, harmful content, and data leakage in generative AI applications by letting you define policies that filter user prompts and model responses.
+`Model Armor`_: Model Armor helps you protect against risks like prompt injection, harmful
+content, and data leakage in generative AI applications by letting you
+define policies that filter user prompts and model responses.
 
 - `Client Library Documentation`_
 - `Product Documentation`_
@@ -14,9 +16,9 @@ Python Client for Model Armor API
    :target: https://pypi.org/project/google-cloud-modelarmor/
 .. |versions| image:: https://img.shields.io/pypi/pyversions/google-cloud-modelarmor.svg
    :target: https://pypi.org/project/google-cloud-modelarmor/
-.. _Model Armor API: https://cloud.google.com/security-command-center/docs/model-armor-overview
+.. _Model Armor: https://cloud.google.com/security-command-center/
 .. _Client Library Documentation: https://cloud.google.com/python/docs/reference/google-cloud-modelarmor/latest/summary_overview
-.. _Product Documentation:  https://cloud.google.com/security-command-center/docs/model-armor-overview
+.. _Product Documentation:  https://cloud.google.com/security-command-center/
 
 Quick Start
 -----------
@@ -25,12 +27,12 @@ In order to use this library, you first need to go through the following steps:
 
 1. `Select or create a Cloud Platform project.`_
 2. `Enable billing for your project.`_
-3. `Enable the Model Armor API.`_
+3. `Enable the Model Armor.`_
 4. `Set up Authentication.`_
 
 .. _Select or create a Cloud Platform project.: https://console.cloud.google.com/project
 .. _Enable billing for your project.: https://cloud.google.com/billing/docs/how-to/modify-project#enable_billing_for_a_project
-.. _Enable the Model Armor API.:  https://cloud.google.com/security-command-center/docs/model-armor-overview
+.. _Enable the Model Armor.:  https://cloud.google.com/security-command-center/
 .. _Set up Authentication.: https://googleapis.dev/python/google-api-core/latest/auth.html
 
 Installation
@@ -97,14 +99,14 @@ Windows
 Next Steps
 ~~~~~~~~~~
 
--  Read the `Client Library Documentation`_ for Model Armor API
+-  Read the `Client Library Documentation`_ for Model Armor
    to see other available methods on the client.
--  Read the `Model Armor API Product documentation`_ to learn
+-  Read the `Model Armor Product documentation`_ to learn
    more about the product and see How-to Guides.
 -  View this `README`_ to see the full list of Cloud
    APIs that we cover.
 
-.. _Model Armor API Product documentation:  https://cloud.google.com/security-command-center/docs/model-armor-overview
+.. _Model Armor Product documentation:  https://cloud.google.com/security-command-center/
 .. _README: https://github.com/googleapis/google-cloud-python/blob/main/README.rst
 
 Logging

{google_cloud_modelarmor-0.5.0 → google_cloud_modelarmor-0.6.0}/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.5.0"  # {x-release-please-version}
+__version__ = "0.6.0"  # {x-release-please-version}

{google_cloud_modelarmor-0.5.0/google/cloud/modelarmor_v1beta → google_cloud_modelarmor-0.6.0/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.5.0"  # {x-release-please-version}
+__version__ = "0.6.0"  # {x-release-please-version}

{google_cloud_modelarmor-0.5.0 → google_cloud_modelarmor-0.6.0}/google/cloud/modelarmor_v1beta/__init__.py

@@ -50,6 +50,7 @@ from .types.service import (
     ListTemplatesResponse,
     MaliciousUriFilterResult,
     MaliciousUriFilterSettings,
+    McpServerFloorSetting,
     MessageItem,
     MultiLanguageDetectionMetadata,
     PiAndJailbreakFilterResult,
@@ -71,6 +72,7 @@ from .types.service import (
     SdpFinding,
     SdpFindingLikelihood,
     SdpInspectResult,
+    StreamingMode,
     Template,
     UpdateFloorSettingRequest,
     UpdateTemplateRequest,
@@ -193,6 +195,7 @@ __all__ = (
     "ListTemplatesResponse",
     "MaliciousUriFilterResult",
     "MaliciousUriFilterSettings",
+    "McpServerFloorSetting",
     "MessageItem",
     "ModelArmorClient",
     "MultiLanguageDetectionMetadata",
@@ -215,6 +218,7 @@ __all__ = (
     "SdpFinding",
     "SdpFindingLikelihood",
     "SdpInspectResult",
+    "StreamingMode",
     "Template",
     "UpdateFloorSettingRequest",
     "UpdateTemplateRequest",

{google_cloud_modelarmor-0.5.0 → google_cloud_modelarmor-0.6.0}/google/cloud/modelarmor_v1beta/gapic_metadata.json

@@ -45,6 +45,16 @@
                 "sanitize_user_prompt"
               ]
             },
+            "StreamSanitizeModelResponse": {
+              "methods": [
+                "stream_sanitize_model_response"
+              ]
+            },
+            "StreamSanitizeUserPrompt": {
+              "methods": [
+                "stream_sanitize_user_prompt"
+              ]
+            },
             "UpdateFloorSetting": {
               "methods": [
                 "update_floor_setting"
@@ -95,6 +105,16 @@
                 "sanitize_user_prompt"
               ]
             },
+            "StreamSanitizeModelResponse": {
+              "methods": [
+                "stream_sanitize_model_response"
+              ]
+            },
+            "StreamSanitizeUserPrompt": {
+              "methods": [
+                "stream_sanitize_user_prompt"
+              ]
+            },
             "UpdateFloorSetting": {
               "methods": [
                 "update_floor_setting"
@@ -145,6 +165,16 @@
                 "sanitize_user_prompt"
               ]
             },
+            "StreamSanitizeModelResponse": {
+              "methods": [
+                "stream_sanitize_model_response"
+              ]
+            },
+            "StreamSanitizeUserPrompt": {
+              "methods": [
+                "stream_sanitize_user_prompt"
+              ]
+            },
             "UpdateFloorSetting": {
               "methods": [
                 "update_floor_setting"

{google_cloud_modelarmor-0.5.0/google/cloud/modelarmor_v1 → google_cloud_modelarmor-0.6.0/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.5.0"  # {x-release-please-version}
+__version__ = "0.6.0"  # {x-release-please-version}

{google_cloud_modelarmor-0.5.0 → google_cloud_modelarmor-0.6.0}/google/cloud/modelarmor_v1beta/services/model_armor/async_client.py

@@ -17,6 +17,9 @@ import logging as std_logging
 import re
 from collections import OrderedDict
 from typing import (
+    AsyncIterable,
+    AsyncIterator,
+    Awaitable,
     Callable,
     Dict,
     Mapping,
@@ -1280,6 +1283,180 @@ class ModelArmorAsyncClient:
         # Done; return the response.
         return response
 
+    def stream_sanitize_user_prompt(
+        self,
+        requests: Optional[AsyncIterator[service.SanitizeUserPromptRequest]] = None,
+        *,
+        retry: OptionalRetry = gapic_v1.method.DEFAULT,
+        timeout: Union[float, object] = gapic_v1.method.DEFAULT,
+        metadata: Sequence[Tuple[str, Union[str, bytes]]] = (),
+    ) -> Awaitable[AsyncIterable[service.SanitizeUserPromptResponse]]:
+        r"""Streaming version of Sanitize User Prompt.
+
+        .. code-block:: python
+
+            # This snippet has been automatically generated and should be regarded as a
+            # code template only.
+            # It will require modifications to work:
+            # - It may require correct/in-range values for request initialization.
+            # - It may require specifying regional endpoints when creating the service
+            #   client as shown in:
+            #   https://googleapis.dev/python/google-api-core/latest/client_options.html
+            from google.cloud import modelarmor_v1beta
+
+            async def sample_stream_sanitize_user_prompt():
+                # Create a client
+                client = modelarmor_v1beta.ModelArmorAsyncClient()
+
+                # Initialize request argument(s)
+                user_prompt_data = modelarmor_v1beta.DataItem()
+                user_prompt_data.text = "text_value"
+
+                request = modelarmor_v1beta.SanitizeUserPromptRequest(
+                    name="name_value",
+                    user_prompt_data=user_prompt_data,
+                )
+
+                # This method expects an iterator which contains
+                # 'modelarmor_v1beta.SanitizeUserPromptRequest' objects
+                # Here we create a generator that yields a single `request` for
+                # demonstrative purposes.
+                requests = [request]
+
+                def request_generator():
+                    for request in requests:
+                        yield request
+
+                # Make the request
+                stream = await client.stream_sanitize_user_prompt(requests=request_generator())
+
+                # Handle the response
+                async for response in stream:
+                    print(response)
+
+        Args:
+            requests (AsyncIterator[`google.cloud.modelarmor_v1beta.types.SanitizeUserPromptRequest`]):
+                The request object AsyncIterator. Sanitize User Prompt request.
+            retry (google.api_core.retry_async.AsyncRetry): Designation of what errors, if any,
+                should be retried.
+            timeout (float): The timeout for this request.
+            metadata (Sequence[Tuple[str, Union[str, bytes]]]): Key/value pairs which should be
+                sent along with the request as metadata. Normally, each value must be of type `str`,
+                but for metadata keys ending with the suffix `-bin`, the corresponding values must
+                be of type `bytes`.
+
+        Returns:
+            AsyncIterable[google.cloud.modelarmor_v1beta.types.SanitizeUserPromptResponse]:
+                Sanitized User Prompt Response.
+        """
+
+        # Wrap the RPC method; this adds retry and timeout information,
+        # and friendly error handling.
+        rpc = self._client._transport._wrapped_methods[
+            self._client._transport.stream_sanitize_user_prompt
+        ]
+
+        # Validate the universe domain.
+        self._client._validate_universe_domain()
+
+        # Send the request.
+        response = rpc(
+            requests,
+            retry=retry,
+            timeout=timeout,
+            metadata=metadata,
+        )
+
+        # Done; return the response.
+        return response
+
+    def stream_sanitize_model_response(
+        self,
+        requests: Optional[AsyncIterator[service.SanitizeModelResponseRequest]] = None,
+        *,
+        retry: OptionalRetry = gapic_v1.method.DEFAULT,
+        timeout: Union[float, object] = gapic_v1.method.DEFAULT,
+        metadata: Sequence[Tuple[str, Union[str, bytes]]] = (),
+    ) -> Awaitable[AsyncIterable[service.SanitizeModelResponseResponse]]:
+        r"""Streaming version of Sanitizes Model Response.
+
+        .. code-block:: python
+
+            # This snippet has been automatically generated and should be regarded as a
+            # code template only.
+            # It will require modifications to work:
+            # - It may require correct/in-range values for request initialization.
+            # - It may require specifying regional endpoints when creating the service
+            #   client as shown in:
+            #   https://googleapis.dev/python/google-api-core/latest/client_options.html
+            from google.cloud import modelarmor_v1beta
+
+            async def sample_stream_sanitize_model_response():
+                # Create a client
+                client = modelarmor_v1beta.ModelArmorAsyncClient()
+
+                # Initialize request argument(s)
+                model_response_data = modelarmor_v1beta.DataItem()
+                model_response_data.text = "text_value"
+
+                request = modelarmor_v1beta.SanitizeModelResponseRequest(
+                    name="name_value",
+                    model_response_data=model_response_data,
+                )
+
+                # This method expects an iterator which contains
+                # 'modelarmor_v1beta.SanitizeModelResponseRequest' objects
+                # Here we create a generator that yields a single `request` for
+                # demonstrative purposes.
+                requests = [request]
+
+                def request_generator():
+                    for request in requests:
+                        yield request
+
+                # Make the request
+                stream = await client.stream_sanitize_model_response(requests=request_generator())
+
+                # Handle the response
+                async for response in stream:
+                    print(response)
+
+        Args:
+            requests (AsyncIterator[`google.cloud.modelarmor_v1beta.types.SanitizeModelResponseRequest`]):
+                The request object AsyncIterator. Sanitize Model Response request.
+            retry (google.api_core.retry_async.AsyncRetry): Designation of what errors, if any,
+                should be retried.
+            timeout (float): The timeout for this request.
+            metadata (Sequence[Tuple[str, Union[str, bytes]]]): Key/value pairs which should be
+                sent along with the request as metadata. Normally, each value must be of type `str`,
+                but for metadata keys ending with the suffix `-bin`, the corresponding values must
+                be of type `bytes`.
+
+        Returns:
+            AsyncIterable[google.cloud.modelarmor_v1beta.types.SanitizeModelResponseResponse]:
+                Sanitized Model Response Response.
+        """
+
+        # Wrap the RPC method; this adds retry and timeout information,
+        # and friendly error handling.
+        rpc = self._client._transport._wrapped_methods[
+            self._client._transport.stream_sanitize_model_response
+        ]
+
+        # Validate the universe domain.
+        self._client._validate_universe_domain()
+
+        # Send the request.
+        response = rpc(
+            requests,
+            retry=retry,
+            timeout=timeout,
+            metadata=metadata,
+        )
+
+        # Done; return the response.
+        return response
+
     async def get_location(
         self,
         request: Optional[Union[locations_pb2.GetLocationRequest, dict]] = None,

{google_cloud_modelarmor-0.5.0 → google_cloud_modelarmor-0.6.0}/google/cloud/modelarmor_v1beta/services/model_armor/client.py

@@ -23,6 +23,8 @@ from http import HTTPStatus
 from typing import (
     Callable,
     Dict,
+    Iterable,
+    Iterator,
     Mapping,
     MutableMapping,
     MutableSequence,
@@ -1710,6 +1712,180 @@ class ModelArmorClient(metaclass=ModelArmorClientMeta):
         # Done; return the response.
         return response
 
+    def stream_sanitize_user_prompt(
+        self,
+        requests: Optional[Iterator[service.SanitizeUserPromptRequest]] = None,
+        *,
+        retry: OptionalRetry = gapic_v1.method.DEFAULT,
+        timeout: Union[float, object] = gapic_v1.method.DEFAULT,
+        metadata: Sequence[Tuple[str, Union[str, bytes]]] = (),
+    ) -> Iterable[service.SanitizeUserPromptResponse]:
+        r"""Streaming version of Sanitize User Prompt.
+
+        .. code-block:: python
+
+            # This snippet has been automatically generated and should be regarded as a
+            # code template only.
+            # It will require modifications to work:
+            # - It may require correct/in-range values for request initialization.
+            # - It may require specifying regional endpoints when creating the service
+            #   client as shown in:
+            #   https://googleapis.dev/python/google-api-core/latest/client_options.html
+            from google.cloud import modelarmor_v1beta
+
+            def sample_stream_sanitize_user_prompt():
+                # Create a client
+                client = modelarmor_v1beta.ModelArmorClient()
+
+                # Initialize request argument(s)
+                user_prompt_data = modelarmor_v1beta.DataItem()
+                user_prompt_data.text = "text_value"
+
+                request = modelarmor_v1beta.SanitizeUserPromptRequest(
+                    name="name_value",
+                    user_prompt_data=user_prompt_data,
+                )
+
+                # This method expects an iterator which contains
+                # 'modelarmor_v1beta.SanitizeUserPromptRequest' objects
+                # Here we create a generator that yields a single `request` for
+                # demonstrative purposes.
+                requests = [request]
+
+                def request_generator():
+                    for request in requests:
+                        yield request
+
+                # Make the request
+                stream = client.stream_sanitize_user_prompt(requests=request_generator())
+
+                # Handle the response
+                for response in stream:
+                    print(response)
+
+        Args:
+            requests (Iterator[google.cloud.modelarmor_v1beta.types.SanitizeUserPromptRequest]):
+                The request object iterator. Sanitize User Prompt request.
+            retry (google.api_core.retry.Retry): Designation of what errors, if any,
+                should be retried.
+            timeout (float): The timeout for this request.
+            metadata (Sequence[Tuple[str, Union[str, bytes]]]): Key/value pairs which should be
+                sent along with the request as metadata. Normally, each value must be of type `str`,
+                but for metadata keys ending with the suffix `-bin`, the corresponding values must
+                be of type `bytes`.
+
+        Returns:
+            Iterable[google.cloud.modelarmor_v1beta.types.SanitizeUserPromptResponse]:
+                Sanitized User Prompt Response.
+        """
+
+        # Wrap the RPC method; this adds retry and timeout information,
+        # and friendly error handling.
+        rpc = self._transport._wrapped_methods[
+            self._transport.stream_sanitize_user_prompt
+        ]
+
+        # Validate the universe domain.
+        self._validate_universe_domain()
+
+        # Send the request.
+        response = rpc(
+            requests,
+            retry=retry,
+            timeout=timeout,
+            metadata=metadata,
+        )
+
+        # Done; return the response.
+        return response
+
+    def stream_sanitize_model_response(
+        self,
+        requests: Optional[Iterator[service.SanitizeModelResponseRequest]] = None,
+        *,
+        retry: OptionalRetry = gapic_v1.method.DEFAULT,
+        timeout: Union[float, object] = gapic_v1.method.DEFAULT,
+        metadata: Sequence[Tuple[str, Union[str, bytes]]] = (),
+    ) -> Iterable[service.SanitizeModelResponseResponse]:
+        r"""Streaming version of Sanitizes Model Response.
+
+        .. code-block:: python
+
+            # This snippet has been automatically generated and should be regarded as a
+            # code template only.
+            # It will require modifications to work:
+            # - It may require correct/in-range values for request initialization.
+            # - It may require specifying regional endpoints when creating the service
+            #   client as shown in:
+            #   https://googleapis.dev/python/google-api-core/latest/client_options.html
+            from google.cloud import modelarmor_v1beta
+
+            def sample_stream_sanitize_model_response():
+                # Create a client
+                client = modelarmor_v1beta.ModelArmorClient()
+
+                # Initialize request argument(s)
+                model_response_data = modelarmor_v1beta.DataItem()
+                model_response_data.text = "text_value"
+
+                request = modelarmor_v1beta.SanitizeModelResponseRequest(
+                    name="name_value",
+                    model_response_data=model_response_data,
+                )
+
+                # This method expects an iterator which contains
+                # 'modelarmor_v1beta.SanitizeModelResponseRequest' objects
+                # Here we create a generator that yields a single `request` for
+                # demonstrative purposes.
+                requests = [request]
+
+                def request_generator():
+                    for request in requests:
+                        yield request
+
+                # Make the request
+                stream = client.stream_sanitize_model_response(requests=request_generator())
+
+                # Handle the response
+                for response in stream:
+                    print(response)
+
+        Args:
+            requests (Iterator[google.cloud.modelarmor_v1beta.types.SanitizeModelResponseRequest]):
+                The request object iterator. Sanitize Model Response request.
+            retry (google.api_core.retry.Retry): Designation of what errors, if any,
+                should be retried.
+            timeout (float): The timeout for this request.
+            metadata (Sequence[Tuple[str, Union[str, bytes]]]): Key/value pairs which should be
+                sent along with the request as metadata. Normally, each value must be of type `str`,
+                but for metadata keys ending with the suffix `-bin`, the corresponding values must
+                be of type `bytes`.
+
+        Returns:
+            Iterable[google.cloud.modelarmor_v1beta.types.SanitizeModelResponseResponse]:
+                Sanitized Model Response Response.
+        """
+
+        # Wrap the RPC method; this adds retry and timeout information,
+        # and friendly error handling.
+        rpc = self._transport._wrapped_methods[
+            self._transport.stream_sanitize_model_response
+        ]
+
+        # Validate the universe domain.
+        self._validate_universe_domain()
+
+        # Send the request.
+        response = rpc(
+            requests,
+            retry=retry,
+            timeout=timeout,
+            metadata=metadata,
+        )
+
+        # Done; return the response.
+        return response
+
     def __enter__(self) -> "ModelArmorClient":
         return self
 

{google_cloud_modelarmor-0.5.0 → google_cloud_modelarmor-0.6.0}/google/cloud/modelarmor_v1beta/services/model_armor/transports/base.py

@@ -234,6 +234,16 @@ class ModelArmorTransport(abc.ABC):
                 default_timeout=60.0,
                 client_info=client_info,
             ),
+            self.stream_sanitize_user_prompt: gapic_v1.method.wrap_method(
+                self.stream_sanitize_user_prompt,
+                default_timeout=None,
+                client_info=client_info,
+            ),
+            self.stream_sanitize_model_response: gapic_v1.method.wrap_method(
+                self.stream_sanitize_model_response,
+                default_timeout=None,
+                client_info=client_info,
+            ),
             self.get_location: gapic_v1.method.wrap_method(
                 self.get_location,
                 default_timeout=None,
@@ -342,6 +352,30 @@ class ModelArmorTransport(abc.ABC):
     ]:
         raise NotImplementedError()
 
+    @property
+    def stream_sanitize_user_prompt(
+        self,
+    ) -> Callable[
+        [service.SanitizeUserPromptRequest],
+        Union[
+            service.SanitizeUserPromptResponse,
+            Awaitable[service.SanitizeUserPromptResponse],
+        ],
+    ]:
+        raise NotImplementedError()
+
+    @property
+    def stream_sanitize_model_response(
+        self,
+    ) -> Callable[
+        [service.SanitizeModelResponseRequest],
+        Union[
+            service.SanitizeModelResponseResponse,
+            Awaitable[service.SanitizeModelResponseResponse],
+        ],
+    ]:
+        raise NotImplementedError()
+
     @property
     def get_location(
         self,