From 420b5a774852501f04c716f74b6c58f466bb71df Mon Sep 17 00:00:00 2001 From: Google Team Member Date: Sat, 13 Jun 2026 02:11:32 -0700 Subject: [PATCH] feat: expose Computer Use API fields Removing visibility restrictions because we already handle non-eap users in the code. PiperOrigin-RevId: 931571802 --- google/genai/_interactions/types/tool.py | 5 ++++- google/genai/_interactions/types/tool_param.py | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/google/genai/_interactions/types/tool.py b/google/genai/_interactions/types/tool.py index 9e29c6fde..11ff6948d 100644 --- a/google/genai/_interactions/types/tool.py +++ b/google/genai/_interactions/types/tool.py @@ -64,7 +64,10 @@ class ComputerUse(BaseModel): type: Literal["computer_use"] - environment: Optional[Literal["browser"]] = None + enable_prompt_injection_detection: Optional[bool] = None + """Whether enable the prompt injection detection check on computer-use request.""" + + environment: Optional[Literal["browser", "mobile", "desktop"]] = None """The environment being operated.""" excluded_predefined_functions: Optional[List[str]] = None diff --git a/google/genai/_interactions/types/tool_param.py b/google/genai/_interactions/types/tool_param.py index 9042a481a..bc50c4d24 100644 --- a/google/genai/_interactions/types/tool_param.py +++ b/google/genai/_interactions/types/tool_param.py @@ -63,7 +63,10 @@ class ComputerUse(TypedDict, total=False): type: Required[Literal["computer_use"]] - environment: Literal["browser"] + enable_prompt_injection_detection: bool + """Whether enable the prompt injection detection check on computer-use request.""" + + environment: Literal["browser", "mobile", "desktop"] """The environment being operated.""" excluded_predefined_functions: SequenceNotStr[str]