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]