inspect-ai 0.3.71__py3-none-any.whl → 0.3.72__py3-none-any.whl

inspect_ai/tool/_tools/_computer/_common.py

@@ -11,19 +11,6 @@ from inspect_ai.tool import ToolError, ToolResult
 from inspect_ai.util._sandbox.context import sandbox_with
 from inspect_ai.util._sandbox.environment import SandboxEnvironment
 
-Action = Literal[
-    "key",
-    "type",
-    "mouse_move",
-    "left_click",
-    "left_click_drag",
-    "right_click",
-    "middle_click",
-    "double_click",
-    "screenshot",
-    "cursor_position",
-]
-
 
 class ToolExecResult(BaseModel):
     output: str | None = Field(default=None)
@@ -31,6 +18,122 @@ class ToolExecResult(BaseModel):
     base64_image: str | None = Field(default=None)
 
 
+async def cursor_position(timeout: int | None = None) -> ToolResult:
+    return await _send_cmd(["cursor_position"], timeout=timeout)
+
+
+async def screenshot(timeout: int | None = None) -> ToolResult:
+    return await _send_cmd(["screenshot"], timeout=timeout)
+
+
+async def wait(duration: int, timeout: int | None = None) -> ToolResult:
+    return await _send_cmd(["wait", "--duration", f"{duration}"], timeout=timeout)
+
+
+async def mouse_move(coordinate: list[int], timeout: int | None = None) -> ToolResult:
+    return await _send_cmd(
+        ["mouse_move", "--coordinate", f"{coordinate[0]}", f"{coordinate[1]}"],
+        timeout=timeout,
+    )
+
+
+async def left_mouse_down(timeout: int | None = None) -> ToolResult:
+    return await _send_cmd(["left_mouse_down"], timeout=timeout)
+
+
+async def left_mouse_up(timeout: int | None = None) -> ToolResult:
+    return await _send_cmd(["left_mouse_up"], timeout=timeout)
+
+
+async def left_click(coordinate: list[int], timeout: int | None = None) -> ToolResult:
+    return await _send_cmd(
+        ["left_click", "--coordinate", f"{coordinate[0]}", f"{coordinate[1]}"],
+        timeout=timeout,
+    )
+
+
+async def left_click_drag(
+    start_coordinate: list[int], coordinate: list[int], timeout: int | None = None
+) -> ToolResult:
+    return await _send_cmd(
+        [
+            "left_click_drag",
+            "--start_coordinate",
+            f"{start_coordinate[0]}",
+            f"{start_coordinate[1]}",
+            "--coordinate",
+            f"{coordinate[0]}",
+            f"{coordinate[1]}",
+        ],
+        timeout=timeout,
+    )
+
+
+async def right_click(coordinate: list[int], timeout: int | None = None) -> ToolResult:
+    return await _send_cmd(
+        ["right_click", "--coordinate", f"{coordinate[0]}", f"{coordinate[1]}"],
+        timeout=timeout,
+    )
+
+
+async def middle_click(coordinate: list[int], timeout: int | None = None) -> ToolResult:
+    return await _send_cmd(
+        ["middle_click", "--coordinate", f"{coordinate[0]}", f"{coordinate[1]}"],
+        timeout=timeout,
+    )
+
+
+async def double_click(coordinate: list[int], timeout: int | None = None) -> ToolResult:
+    return await _send_cmd(
+        ["double_click", "--coordinate", f"{coordinate[0]}", f"{coordinate[1]}"],
+        timeout=timeout,
+    )
+
+
+async def triple_click(coordinate: list[int], timeout: int | None = None) -> ToolResult:
+    return await _send_cmd(
+        ["triple_click", "--coordinate", f"{coordinate[0]}", f"{coordinate[1]}"],
+        timeout=timeout,
+    )
+
+
+async def scroll(
+    scroll_amount: int,
+    scroll_direction: Literal["up", "down", "left", "right"],
+    coordinate: list[int] | None,
+    timeout: int | None = None,
+) -> ToolResult:
+    return await _send_cmd(
+        [
+            "scroll",
+            "--scroll_amount",
+            f"{scroll_amount}",
+            "--scroll_direction",
+            f"{scroll_direction}",
+        ]
+        + (
+            ["--coordinate", f"{coordinate[0]}", f"{coordinate[1]}"]
+            if coordinate
+            else []
+        ),
+        timeout=timeout,
+    )
+
+
+async def press_key(key: str, timeout: int | None = None) -> ToolResult:
+    return await _send_cmd(["key", "--text", key], timeout=timeout)
+
+
+async def hold_key(key: str, duration: int, timeout: int | None = None) -> ToolResult:
+    return await _send_cmd(
+        ["hold_key", "--text", key, "--duration", f"{duration}"], timeout=timeout
+    )
+
+
+async def type(text: str, timeout: int | None = None) -> ToolResult:
+    return await _send_cmd(["type", "--text", text], timeout=timeout)
+
+
 async def _send_cmd(cmdTail: list[str], timeout: int | None = None) -> ToolResult:
     from inspect_ai.log._samples import sample_active
 
@@ -39,7 +142,7 @@ async def _send_cmd(cmdTail: list[str], timeout: int | None = None) -> ToolResul
     sample_id = sample.sample.id
     assert sample_id
 
-    cmd = ["python3", "/opt/inspect/tool/computer_tool.py", "--action"] + cmdTail
+    cmd = ["python3", "/opt/inspect/tool/computer_tool.py"] + cmdTail
 
     raw_exec_result = await (await computer_sandbox()).exec(cmd, timeout=timeout)
 
@@ -72,50 +175,6 @@ async def _send_cmd(cmdTail: list[str], timeout: int | None = None) -> ToolResul
     return "OK"
 
 
-async def cursor_position(timeout: int | None = None) -> ToolResult:
-    return await _send_cmd(["cursor_position"], timeout=timeout)
-
-
-async def screenshot(timeout: int | None = None) -> ToolResult:
-    return await _send_cmd(["screenshot"], timeout=timeout)
-
-
-async def mouse_move(x: int, y: int, timeout: int | None = None) -> ToolResult:
-    return await _send_cmd(
-        ["mouse_move", "--coordinate", f"{x}", f"{y}"], timeout=timeout
-    )
-
-
-async def left_click(timeout: int | None = None) -> ToolResult:
-    return await _send_cmd(["left_click"], timeout=timeout)
-
-
-async def left_click_drag(x: int, y: int, timeout: int | None = None) -> ToolResult:
-    return await _send_cmd(
-        ["left_click_drag", "--coordinate", f"{x}", f"{y}"], timeout=timeout
-    )
-
-
-async def right_click(timeout: int | None = None) -> ToolResult:
-    return await _send_cmd(["right_click"], timeout=timeout)
-
-
-async def middle_click(timeout: int | None = None) -> ToolResult:
-    return await _send_cmd(["middle_click"], timeout=timeout)
-
-
-async def double_click(timeout: int | None = None) -> ToolResult:
-    return await _send_cmd(["double_click"], timeout=timeout)
-
-
-async def press_key(key: str, timeout: int | None = None) -> ToolResult:
-    return await _send_cmd(["key", "--text", key], timeout=timeout)
-
-
-async def type(text: str, timeout: int | None = None) -> ToolResult:
-    return await _send_cmd(["type", "--text", text], timeout=timeout)
-
-
 async def computer_sandbox() -> SandboxEnvironment:
     sb = await sandbox_with("/opt/inspect/tool/computer_tool.py")
     if sb:

inspect_ai/tool/_tools/_computer/_computer.py

@@ -1,4 +1,4 @@
-from typing import Awaitable, Callable
+from typing import Awaitable, Callable, Literal, TypeVar
 
 from inspect_ai._util.content import Content, ContentImage, ContentText
 from inspect_ai.tool import Tool, ToolResult, tool
@@ -6,7 +6,7 @@ from inspect_ai.tool._tool import TOOL_INIT_MODEL_INPUT, ToolParsingError
 from inspect_ai.tool._tool_call import ToolCallModelInput
 
 from . import _common as common
-from ._common import Action
+from ._resources.tool._constants import Action
 
 ActionFunction = Callable[[str], ToolResult | Awaitable[ToolResult]]
 
@@ -26,8 +26,12 @@ def computer(max_screenshots: int | None = 1, timeout: int | None = 180) -> Tool
 
     async def execute(
         action: Action,
-        text: str | None = None,
         coordinate: list[int] | None = None,
+        duration: int | None = None,
+        scroll_amount: int | None = None,
+        scroll_direction: Literal["up", "down", "left", "right"] | None = None,
+        start_coordinate: list[int] | None = None,
+        text: str | None = None,
     ) -> ToolResult:
         """
         Use this tool to interact with a computer.
@@ -47,83 +51,93 @@ def computer(max_screenshots: int | None = 1, timeout: int | None = 180) -> Tool
                       "Shift_L", "Shift_R", "Control_L", "Control_R", "Alt_L", "Alt_R", "Scroll_Lock", "Num_Lock", "Caps_Lock", "Pause",
                       "KP_Multiply", "KP_Home", "KP_Up", "KP_Prior", "KP_Subtract", "KP_Left", "KP_Begin", "KP_Right", "KP_Add", "KP_End","KP_Down",
                       "KP_Next", "KP_Insert", "KP_Delete", "KP_Enter", "KP_Divide", "KP_Equal", "KP_Decimal",
+              - 'hold_key': Hold down a key or multiple keys for a specified duration (in seconds). Supports the same syntax as `key`.
               - `type`: Type a string of text on the keyboard. If the text contains spaces, enclose it in quotes.
                   - Example: execute(action="type", text="The crux of the biscuit is the apostrophe!")
               - `cursor_position`: Get the current (x, y) pixel coordinate of the cursor on the screen.
               - `mouse_move`: Move the cursor to a specified (x, y) pixel coordinate on the screen.
                   - Example: execute(action="mouse_move", coordinate=(100, 200))
+              - `left_mouse_down`: Press the left mouse button.
+              - `left_mouse_up`: Release the left mouse button.
               - `left_click`: Click the left mouse button.
               - `left_click_drag`: Click and drag the cursor to a specified (x, y) pixel coordinate on the screen.
                   - Example: execute(action="left_click_drag", coordinate=(150, 250))
               - `right_click`: Click the right mouse button.
               - `middle_click`: Click the middle mouse button.
               - `double_click`: Double-click the left mouse button.
+              - `triple_click`: Double-click the left mouse button.
+              - `wait`: Wait for a specified duration (in seconds).
               - `screenshot`: Take a screenshot.
+          coordinate (tuple[int, int] | None): The (x, y) pixel coordinate on the screen to which to move or drag. Required only by `action=mouse_move` and `action=left_click_drag`.
+          duration (int | None): The duration to wait or hold the key down for. Required only by `action=hold_key` and `action=wait`.
+          scroll_amount (int | None): The number of 'clicks' to scroll. Required only by `action=scroll`.
+          scroll_direction (Literal["up", "down", "left", "right] | None): The direction to scroll the screen. Required only by `action=scroll`.
+          start_coordinate (tuple[int, int] | None): The (x, y) pixel coordinate on the screen from which to initiate a drag. Required only by `action=scroll`.
           text (str | None): The text to type or the key to press. Required when action is "key" or "type".
-          coordinate (tuple[int, int] | None): The (x, y) pixel coordinate on the screen to which to move or drag. Required when action is "mouse_move" or "left_click_drag".
 
         Returns:
           The output of the command. Many commands will include a screenshot reflecting the result of the command in their output.
         """
-        if action in ("mouse_move", "left_click_drag"):
-            if coordinate is None:
-                raise ToolParsingError(f"coordinate is required for {action}")
-            if text is not None:
-                raise ToolParsingError(f"text is not accepted for {action}")
-            if not isinstance(coordinate, list) or len(coordinate) != 2:
-                raise ToolParsingError(f"{coordinate} must be a tuple of length 2")
-            if not all(isinstance(i, int) and i >= 0 for i in coordinate):
-                raise ToolParsingError(
-                    f"{coordinate} must be a tuple of non-negative ints"
+        match action:
+            case "key":
+                return await common.press_key(not_none(text, "text"), timeout=timeout)
+            case "hold_key":
+                return await common.hold_key(
+                    not_none(text, "text"),
+                    not_none(duration, "duration"),
+                    timeout=timeout,
                 )
-
-            if action == "mouse_move":
+            case "type":
+                return await common.type(not_none(text, "text"), timeout=timeout)
+            case "cursor_position":
+                return await common.cursor_position(timeout=timeout)
+            case "mouse_move":
                 return await common.mouse_move(
-                    coordinate[0], coordinate[1], timeout=timeout
+                    not_none(coordinate, "coordinate"), timeout=timeout
+                )
+            case "left_mouse_down":
+                return await common.left_mouse_down(timeout=timeout)
+            case "left_mouse_up":
+                return await common.left_mouse_up(timeout=timeout)
+            case "left_click":
+                return await common.left_click(
+                    not_none(coordinate, "coordinate"), timeout=timeout
                 )
-            elif action == "left_click_drag":
+            case "left_click_drag":
                 return await common.left_click_drag(
-                    coordinate[0], coordinate[1], timeout=timeout
+                    not_none(start_coordinate, "start_coordinate"),
+                    not_none(coordinate, "coordinate"),
+                    timeout=timeout,
                 )
-
-        if action in ("key", "type"):
-            if text is None:
-                raise ToolParsingError(f"text is required for {action}")
-            if coordinate is not None:
-                raise ToolParsingError(f"coordinate is not accepted for {action}")
-            if not isinstance(text, str):
-                raise ToolParsingError(f"{text} must be a string")
-
-            if action == "key":
-                return await common.press_key(text, timeout=timeout)
-            elif action == "type":
-                return await common.type(text, timeout=timeout)
-
-        if action in (
-            "left_click",
-            "right_click",
-            "double_click",
-            "middle_click",
-            "screenshot",
-            "cursor_position",
-        ):
-            if text is not None:
-                raise ToolParsingError(f"text is not accepted for {action}")
-            if coordinate is not None:
-                raise ToolParsingError(f"coordinate is not accepted for {action}")
-
-            if action == "screenshot":
+            case "right_click":
+                return await common.right_click(
+                    not_none(coordinate, "coordinate"), timeout=timeout
+                )
+            case "middle_click":
+                return await common.middle_click(
+                    not_none(coordinate, "coordinate"), timeout=timeout
+                )
+            case "double_click":
+                return await common.double_click(
+                    not_none(coordinate, "coordinate"), timeout=timeout
+                )
+            case "triple_click":
+                return await common.triple_click(
+                    not_none(coordinate, "coordinate"), timeout=timeout
+                )
+            case "scroll":
+                return await common.scroll(
+                    not_none(scroll_amount, "scroll_amount"),
+                    not_none(scroll_direction, "scroll_direction"),
+                    coordinate,
+                    timeout=timeout,
+                )
+            case "wait":
+                return await common.wait(
+                    not_none(duration, "duration"), timeout=timeout
+                )
+            case "screenshot":
                 return await common.screenshot(timeout=timeout)
-            elif action == "cursor_position":
-                return await common.cursor_position(timeout=timeout)
-            elif action == "left_click":
-                return await common.left_click(timeout=timeout)
-            elif action == "right_click":
-                return await common.right_click(timeout=timeout)
-            elif action == "middle_click":
-                return await common.middle_click(timeout=timeout)
-            elif action == "double_click":
-                return await common.double_click(timeout=timeout)
 
         raise ToolParsingError(f"Invalid action: {action}")
 
@@ -161,3 +175,12 @@ def _computer_model_input(max_screenshots: int) -> ToolCallModelInput:
             return input_content
 
     return model_input
+
+
+T = TypeVar("T")
+
+
+def not_none(value: T | None, name: str) -> T:
+    if value is None:
+        raise ToolParsingError(f"{name} must be provided")
+    return value

inspect_ai/tool/_tools/_computer/_resources/image_home_dir/.config/Code/User/settings.json

@@ -1,3 +1,9 @@
 {
-  "security.workspace.trust.enabled": false
+  "security.workspace.trust.enabled": false,
+  "update.showReleaseNotes": false,
+  "editor.cursorBlinking": "solid",
+  "editor.cursorWidth": 3,
+  "workbench.colorCustomizations": {
+    "editorCursor.foreground": "#FF0000"
+  }
 }

inspect_ai/tool/_tools/_computer/_resources/image_home_dir/.config/xfce4/xfconf/xfce-perchannel-xml/xfwm4.xml

@@ -0,0 +1,91 @@
+<?xml version="1.0" encoding="UTF-8"?>
+
+<channel name="xfwm4" version="1.0">
+  <property name="general" type="empty">
+    <property name="activate_action" type="string" value="bring" />
+    <property name="borderless_maximize" type="bool" value="true" />
+    <property name="box_move" type="bool" value="false" />
+    <property name="box_resize" type="bool" value="false" />
+    <property name="button_layout" type="string" value="O|SHMC" />
+    <property name="button_offset" type="int" value="0" />
+    <property name="button_spacing" type="int" value="0" />
+    <property name="click_to_focus" type="bool" value="true" />
+    <property name="cycle_apps_only" type="bool" value="false" />
+    <property name="cycle_draw_frame" type="bool" value="true" />
+    <property name="cycle_raise" type="bool" value="false" />
+    <property name="cycle_hidden" type="bool" value="true" />
+    <property name="cycle_minimum" type="bool" value="true" />
+    <property name="cycle_minimized" type="bool" value="false" />
+    <property name="cycle_preview" type="bool" value="true" />
+    <property name="cycle_tabwin_mode" type="int" value="0" />
+    <property name="cycle_workspaces" type="bool" value="false" />
+    <property name="double_click_action" type="string" value="maximize" />
+    <property name="double_click_distance" type="int" value="5" />
+    <property name="double_click_time" type="int" value="250" />
+    <property name="easy_click" type="string" value="Alt" />
+    <property name="focus_delay" type="int" value="250" />
+    <property name="focus_hint" type="bool" value="true" />
+    <property name="focus_new" type="bool" value="true" />
+    <property name="frame_opacity" type="int" value="100" />
+    <property name="frame_border_top" type="int" value="0" />
+    <property name="full_width_title" type="bool" value="true" />
+    <property name="horiz_scroll_opacity" type="bool" value="false" />
+    <property name="inactive_opacity" type="int" value="100" />
+    <property name="maximized_offset" type="int" value="0" />
+    <property name="mousewheel_rollup" type="bool" value="true" />
+    <property name="move_opacity" type="int" value="100" />
+    <property name="placement_mode" type="string" value="center" />
+    <property name="placement_ratio" type="int" value="20" />
+    <property name="popup_opacity" type="int" value="100" />
+    <property name="prevent_focus_stealing" type="bool" value="false" />
+    <property name="raise_delay" type="int" value="250" />
+    <property name="raise_on_click" type="bool" value="true" />
+    <property name="raise_on_focus" type="bool" value="false" />
+    <property name="raise_with_any_button" type="bool" value="true" />
+    <property name="repeat_urgent_blink" type="bool" value="false" />
+    <property name="resize_opacity" type="int" value="100" />
+    <property name="scroll_workspaces" type="bool" value="true" />
+    <property name="shadow_delta_height" type="int" value="0" />
+    <property name="shadow_delta_width" type="int" value="0" />
+    <property name="shadow_delta_x" type="int" value="0" />
+    <property name="shadow_delta_y" type="int" value="-3" />
+    <property name="shadow_opacity" type="int" value="50" />
+    <property name="show_app_icon" type="bool" value="false" />
+    <property name="show_dock_shadow" type="bool" value="true" />
+    <property name="show_frame_shadow" type="bool" value="true" />
+    <property name="show_popup_shadow" type="bool" value="false" />
+    <property name="snap_resist" type="bool" value="false" />
+    <property name="snap_to_border" type="bool" value="true" />
+    <property name="snap_to_windows" type="bool" value="false" />
+    <property name="snap_width" type="int" value="10" />
+    <property name="vblank_mode" type="string" value="auto" />
+    <property name="theme" type="string" value="Default" />
+    <property name="tile_on_move" type="bool" value="true" />
+    <property name="title_alignment" type="string" value="center" />
+    <property name="title_font" type="string" value="Sans Bold 9" />
+    <property name="title_horizontal_offset" type="int" value="0" />
+    <property name="titleless_maximize" type="bool" value="false" />
+    <property name="title_shadow_active" type="string" value="false" />
+    <property name="title_shadow_inactive" type="string" value="false" />
+    <property name="title_vertical_offset_active" type="int" value="0" />
+    <property name="title_vertical_offset_inactive" type="int" value="0" />
+    <property name="toggle_workspaces" type="bool" value="false" />
+    <property name="unredirect_overlays" type="bool" value="true" />
+    <property name="urgent_blink" type="bool" value="false" />
+    <property name="use_compositing" type="bool" value="true" />
+    <property name="workspace_count" type="int" value="1" />
+    <property name="wrap_cycle" type="bool" value="true" />
+    <property name="wrap_layout" type="bool" value="true" />
+    <property name="wrap_resistance" type="int" value="10" />
+    <property name="wrap_windows" type="bool" value="true" />
+    <property name="wrap_workspaces" type="bool" value="false" />
+    <property name="zoom_desktop" type="bool" value="true" />
+    <property name="zoom_pointer" type="bool" value="true" />
+    <property name="workspace_names" type="array">
+      <value type="string" value="Workspace 1" />
+      <value type="string" value="Workspace 2" />
+      <value type="string" value="Workspace 3" />
+      <value type="string" value="Workspace 4" />
+    </property>
+  </property>
+</channel>

inspect_ai/tool/_tools/_computer/_resources/tool/.pylintrc

@@ -0,0 +1,8 @@
+[MASTER]
+; R - Refactorings
+; C - Convention
+; W - Warning
+; E - Error
+enable=C,R,W,E
+disable=R0903,C0114,C0115,C0116,C0301,C0411,C1804,C1805,W0120,W0511,W0718,W1203,E0401,E1101,E0611,E1128
+score=no

inspect_ai/tool/_tools/_computer/_resources/tool/.vscode/settings.json

@@ -0,0 +1,12 @@
+{
+  "cSpell.words": [
+    "FWXGA",
+    "getmouselocation",
+    "keyup",
+    "mousedown",
+    "mousemove",
+    "mouseup",
+    "scrot",
+    "WXGA"
+  ]
+}

inspect_ai/tool/_tools/_computer/_resources/tool/_args.py

@@ -0,0 +1,78 @@
+from argparse import Action, ArgumentParser, Namespace
+from typing import Sequence
+
+
+def parse_arguments(args: Sequence[str] | None = None) -> Namespace:
+    return _create_parser().parse_args(args)
+
+
+def _create_parser() -> ArgumentParser:
+    parser = ArgumentParser(prog="computer_tool")
+    subparsers = parser.add_subparsers(dest="action", required=True)
+
+    # these take no additional arguments
+    subparsers.add_parser(
+        "screenshot",
+        aliases=["cursor_position", "left_mouse_down", "left_mouse_up"],
+    )
+
+    key_and_type = subparsers.add_parser("type", aliases=["key"])
+    _add_text(key_and_type)
+
+    hold_key = subparsers.add_parser("hold_key")
+    _add_text(hold_key)
+    _add_duration(hold_key)
+
+    mouse_move = subparsers.add_parser("mouse_move")
+    _add_coordinate(mouse_move)
+
+    click = subparsers.add_parser(
+        "left_click",
+        aliases=["right_click", "middle_click", "double_click", "triple_click"],
+    )
+    _add_coordinate(click, False)
+    _add_text(click, False)
+
+    left_click_drag = subparsers.add_parser("left_click_drag")
+    _add_start_coordinate(left_click_drag)
+    _add_coordinate(left_click_drag)
+    _add_text(left_click_drag, False)
+
+    scroll = subparsers.add_parser("scroll")
+    _add_scroll_direction(scroll)
+    _add_scroll_amount(scroll)
+    # despite what the doc says, the model doesn't always provide a coordinate
+    _add_coordinate(scroll, False)
+
+    wait = subparsers.add_parser("wait")
+    _add_duration(wait)
+
+    return parser
+
+
+def _add_scroll_direction(subparser: ArgumentParser) -> Action:
+    return subparser.add_argument(
+        "--scroll_direction", choices=["up", "down", "left", "right"], required=True
+    )
+
+
+def _add_scroll_amount(subparser: ArgumentParser) -> Action:
+    return subparser.add_argument("--scroll_amount", type=int, required=True)
+
+
+def _add_coordinate(subparser: ArgumentParser, required: bool = True) -> Action:
+    return subparser.add_argument("--coordinate", type=int, nargs=2, required=required)
+
+
+def _add_start_coordinate(subparser: ArgumentParser) -> Action:
+    return subparser.add_argument(
+        "--start_coordinate", type=int, nargs=2, required=True
+    )
+
+
+def _add_duration(subparser: ArgumentParser) -> Action:
+    return subparser.add_argument("--duration", type=int, required=True)
+
+
+def _add_text(subparser: ArgumentParser, required: bool = True) -> Action:
+    return subparser.add_argument("--text", type=str, required=required)

inspect_ai/tool/_tools/_computer/_resources/tool/_constants.py

@@ -0,0 +1,20 @@
+from typing import Literal
+
+Action = Literal[
+    "key",
+    "hold_key",
+    "type",
+    "cursor_position",
+    "mouse_move",
+    "left_mouse_down",
+    "left_mouse_up",
+    "left_click",
+    "left_click_drag",
+    "right_click",
+    "middle_click",
+    "double_click",
+    "triple_click",
+    "scroll",
+    "wait",
+    "screenshot",
+]