feat: add enable_prompt_injection_detection for Computer Use feature for the Gemini API.

PiperOrigin-RevId: 919308344

Author: yyyu-google

google/genai/_live_converters.py

@@ -23,6 +23,14 @@
 from ._common import set_value_by_path as setv
 
 
+def _Environment_to_vertex_enum_validate(enum_value: Any) -> None:
+  if enum_value in set(['ENVIRONMENT_MOBILE', 'ENVIRONMENT_DESKTOP']):
+    raise ValueError(
+        f'{enum_value} enum value is only supported in Gemini Developer API'
+        ' mode, not in Gemini Enterprise Agent Platform mode.'
+    )
+
+
 def _AudioTranscriptionConfig_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -124,6 +132,32 @@ def _CodeExecutionResult_to_vertex(
   return to_object
 
 
+def _ComputerUse_to_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+  to_object: dict[str, Any] = {}
+  if getv(from_object, ['environment']) is not None:
+    _Environment_to_vertex_enum_validate(getv(from_object, ['environment']))
+    setv(to_object, ['environment'], getv(from_object, ['environment']))
+
+  if getv(from_object, ['excluded_predefined_functions']) is not None:
+    setv(
+        to_object,
+        ['excludedPredefinedFunctions'],
+        getv(from_object, ['excluded_predefined_functions']),
+    )
+
+  if getv(from_object, ['enable_prompt_injection_detection']) is not None:
+    raise ValueError(
+        'enable_prompt_injection_detection parameter is only supported in'
+        ' Gemini Developer API mode, not in Gemini Enterprise Agent Platform'
+        ' mode.'
+    )
+
+  return to_object
+
+
 def _Content_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -1892,7 +1926,11 @@ def _Tool_to_vertex(
     setv(to_object, ['retrieval'], getv(from_object, ['retrieval']))
 
   if getv(from_object, ['computer_use']) is not None:
-    setv(to_object, ['computerUse'], getv(from_object, ['computer_use']))
+    setv(
+        to_object,
+        ['computerUse'],
+        _ComputerUse_to_vertex(getv(from_object, ['computer_use']), to_object),
+    )
 
   if getv(from_object, ['file_search']) is not None:
     raise ValueError(

google/genai/caches.py

@@ -32,6 +32,14 @@
 logger = logging.getLogger('google_genai.caches')
 
 
+def _Environment_to_vertex_enum_validate(enum_value: Any) -> None:
+  if enum_value in set(['ENVIRONMENT_MOBILE', 'ENVIRONMENT_DESKTOP']):
+    raise ValueError(
+        f'{enum_value} enum value is only supported in Gemini Developer API'
+        ' mode, not in Gemini Enterprise Agent Platform mode.'
+    )
+
+
 def _AuthConfig_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -119,6 +127,32 @@ def _CodeExecutionResult_to_vertex(
   return to_object
 
 
+def _ComputerUse_to_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+  to_object: dict[str, Any] = {}
+  if getv(from_object, ['environment']) is not None:
+    _Environment_to_vertex_enum_validate(getv(from_object, ['environment']))
+    setv(to_object, ['environment'], getv(from_object, ['environment']))
+
+  if getv(from_object, ['excluded_predefined_functions']) is not None:
+    setv(
+        to_object,
+        ['excludedPredefinedFunctions'],
+        getv(from_object, ['excluded_predefined_functions']),
+    )
+
+  if getv(from_object, ['enable_prompt_injection_detection']) is not None:
+    raise ValueError(
+        'enable_prompt_injection_detection parameter is only supported in'
+        ' Gemini Developer API mode, not in Gemini Enterprise Agent Platform'
+        ' mode.'
+    )
+
+  return to_object
+
+
 def _Content_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -960,7 +994,11 @@ def _Tool_to_vertex(
     setv(to_object, ['retrieval'], getv(from_object, ['retrieval']))
 
   if getv(from_object, ['computer_use']) is not None:
-    setv(to_object, ['computerUse'], getv(from_object, ['computer_use']))
+    setv(
+        to_object,
+        ['computerUse'],
+        _ComputerUse_to_vertex(getv(from_object, ['computer_use']), to_object),
+    )
 
   if getv(from_object, ['file_search']) is not None:
     raise ValueError(

google/genai/models.py

@@ -36,6 +36,14 @@
 logger = logging.getLogger('google_genai.models')
 
 
+def _Environment_to_vertex_enum_validate(enum_value: Any) -> None:
+  if enum_value in set(['ENVIRONMENT_MOBILE', 'ENVIRONMENT_DESKTOP']):
+    raise ValueError(
+        f'{enum_value} enum value is only supported in Gemini Developer API'
+        ' mode, not in Gemini Enterprise Agent Platform mode.'
+    )
+
+
 def _PersonGeneration_to_mldev_enum_validate(enum_value: Any) -> None:
   if enum_value in set(['ALLOW_ALL']):
     raise ValueError(
@@ -273,6 +281,33 @@ def _ComputeTokensResponse_from_vertex(
   return to_object
 
 
+def _ComputerUse_to_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+    root_object: Optional[Union[dict[str, Any], object]] = None,
+) -> dict[str, Any]:
+  to_object: dict[str, Any] = {}
+  if getv(from_object, ['environment']) is not None:
+    _Environment_to_vertex_enum_validate(getv(from_object, ['environment']))
+    setv(to_object, ['environment'], getv(from_object, ['environment']))
+
+  if getv(from_object, ['excluded_predefined_functions']) is not None:
+    setv(
+        to_object,
+        ['excludedPredefinedFunctions'],
+        getv(from_object, ['excluded_predefined_functions']),
+    )
+
+  if getv(from_object, ['enable_prompt_injection_detection']) is not None:
+    raise ValueError(
+        'enable_prompt_injection_detection parameter is only supported in'
+        ' Gemini Developer API mode, not in Gemini Enterprise Agent Platform'
+        ' mode.'
+    )
+
+  return to_object
+
+
 def _ContentEmbeddingStatistics_from_vertex(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -4350,7 +4385,13 @@ def _Tool_to_vertex(
     setv(to_object, ['retrieval'], getv(from_object, ['retrieval']))
 
   if getv(from_object, ['computer_use']) is not None:
-    setv(to_object, ['computerUse'], getv(from_object, ['computer_use']))
+    setv(
+        to_object,
+        ['computerUse'],
+        _ComputerUse_to_vertex(
+            getv(from_object, ['computer_use']), to_object, root_object
+        ),
+    )
 
   if getv(from_object, ['file_search']) is not None:
     raise ValueError(

google/genai/types.py

@@ -202,6 +202,10 @@ class Environment(_common.CaseInSensitiveEnum):
   """Defaults to browser."""
   ENVIRONMENT_BROWSER = 'ENVIRONMENT_BROWSER'
   """Operates in a web browser."""
+  ENVIRONMENT_MOBILE = 'ENVIRONMENT_MOBILE'
+  """Operates in a mobile environment."""
+  ENVIRONMENT_DESKTOP = 'ENVIRONMENT_DESKTOP'
+  """Operates in a desktop environment."""
 
 
 class AuthType(_common.CaseInSensitiveEnum):
@@ -3295,6 +3299,11 @@ class ComputerUse(_common.BaseModel):
       1. Using a more restricted / different action space.
       2. Improving the definitions / instructions of predefined functions.""",
   )
+  enable_prompt_injection_detection: Optional[bool] = Field(
+      default=None,
+      description="""Optional. Whether enable the prompt injection detection check on computer-use request.
+      """,
+  )
 
 
 class ComputerUseDict(TypedDict, total=False):
@@ -3310,6 +3319,10 @@ class ComputerUseDict(TypedDict, total=False):
       1. Using a more restricted / different action space.
       2. Improving the definitions / instructions of predefined functions."""
 
+  enable_prompt_injection_detection: Optional[bool]
+  """Optional. Whether enable the prompt injection detection check on computer-use request.
+      """
+
 
 ComputerUseOrDict = Union[ComputerUse, ComputerUseDict]