hud-python 0.4.1__py3-none-any.whl → 0.4.3__py3-none-any.whl

hud/tools/computer/openai.py

@@ -1,295 +1,295 @@
-# flake8: noqa: B008
-from __future__ import annotations
-
-import logging
-from typing import TYPE_CHECKING, Any, Literal, cast
-
-from mcp import ErrorData, McpError
-from mcp.types import INTERNAL_ERROR, INVALID_PARAMS, ContentBlock, TextContent
-from pydantic import Field
-
-from hud.tools.computer.settings import computer_settings
-from hud.tools.types import ContentResult
-
-from .hud import HudComputerTool
-
-if TYPE_CHECKING:
-    from hud.tools.executors.base import BaseExecutor
-
-logger = logging.getLogger(__name__)
-
-# Map OpenAI key names to CLA standard keys
-OPENAI_TO_CLA_KEYS = {
-    # Common variations
-    "return": "enter",
-    "escape": "escape",
-    "arrowup": "up",
-    "arrowdown": "down",
-    "arrowleft": "left",
-    "arrowright": "right",
-    "backspace": "backspace",
-    "delete": "delete",
-    "tab": "tab",
-    "space": "space",
-    "control": "ctrl",
-    "alt": "alt",
-    "shift": "shift",
-    "meta": "win",
-    "cmd": "cmd",
-    "command": "cmd",
-    "super": "win",
-    "pageup": "pageup",
-    "pagedown": "pagedown",
-    "home": "home",
-    "end": "end",
-    "insert": "insert",
-}
-
-
-class OpenAIComputerTool(HudComputerTool):
-    """
-    OpenAI Computer Use tool for interacting with the computer.
-    """
-
-    def __init__(
-        self,
-        # Define within environment based on platform
-        executor: BaseExecutor | None = None,
-        platform_type: Literal["auto", "xdo", "pyautogui"] = "auto",
-        display_num: int | None = None,
-        # Overrides for what dimensions the agent thinks it operates in
-        width: int = computer_settings.OPENAI_COMPUTER_WIDTH,
-        height: int = computer_settings.OPENAI_COMPUTER_HEIGHT,
-        rescale_images: bool = computer_settings.OPENAI_RESCALE_IMAGES,
-        name: str | None = None,
-        title: str | None = None,
-        description: str | None = None,
-        **kwargs: Any,
-    ) -> None:
-        """
-        Initialize with OpenAI's default dimensions.
-
-        Args:
-            width: Target width for rescaling (default: 1024 for OpenAI)
-            height: Target height for rescaling (default: 768 for OpenAI)
-            rescale_images: If True, rescale screenshots. If False, only rescale action coordinates
-            name: Tool name for MCP registration (auto-generated from class name if not provided)
-            title: Human-readable display name for the tool (auto-generated from class name)
-            description: Tool description (auto-generated from docstring if not provided)
-        """
-        super().__init__(
-            executor=executor,
-            platform_type=platform_type,
-            display_num=display_num,
-            width=width,
-            height=height,
-            rescale_images=rescale_images,
-            name=name or "openai_computer",
-            title=title or "OpenAI Computer Tool",
-            description=description or "Control computer with mouse, keyboard, and screenshots",
-            **kwargs,
-        )
-
-    def _map_openai_key_to_cla(self, key: str) -> str:
-        """Map OpenAI key name to CLA standard key."""
-        # OpenAI uses lowercase key names
-        return OPENAI_TO_CLA_KEYS.get(key.lower(), key.lower())
-
-    async def __call__(
-        self,
-        type: str = Field(..., description="The action type to perform"),
-        # Coordinate parameters
-        x: int | None = Field(None, description="X coordinate for click/move/scroll actions"),
-        y: int | None = Field(None, description="Y coordinate for click/move/scroll actions"),
-        # Button parameter
-        button: str | None = Field(
-            None, description="Mouse button for click actions (left, right, middle, wheel)"
-        ),
-        # Text parameter
-        text: str | None = Field(None, description="Text to type or response text"),
-        # Scroll parameters
-        scroll_x: int | None = Field(None, description="Horizontal scroll amount"),
-        scroll_y: int | None = Field(None, description="Vertical scroll amount"),
-        # Wait parameter
-        ms: int | None = Field(None, description="Time to wait in milliseconds"),
-        # Key press parameter
-        keys: list[str] | None = Field(None, description="Keys to press"),
-        # Drag parameter
-        path: list[dict[str, int]] | None = Field(
-            None, description="Path for drag actions as list of {x, y} dicts"
-        ),
-        # Custom action parameter
-        action: str | None = Field(None, description="Custom action name"),
-    ) -> list[ContentBlock]:
-        """
-        Handle OpenAI Computer Use API calls.
-
-        This converts OpenAI's action format (based on OperatorAdapter) to HudComputerTool's format.
-
-        Returns:
-            List of MCP content blocks
-        """
-        logger.info("OpenAIComputerTool received type: %s", type)
-
-        # Map button names
-        button_map = {"wheel": "middle"}
-        if button:
-            button = button_map.get(button, button)
-
-        # Process based on action type
-        if type == "screenshot":
-            screenshot_base64 = await self.executor.screenshot()
-            if screenshot_base64:
-                # Rescale screenshot if requested
-                result = ContentResult(base64_image=screenshot_base64)
-            else:
-                result = ContentResult(error="Failed to take screenshot")
-
-        elif type == "click":
-            if x is not None and y is not None:
-                # Cast button to proper literal type
-                button_literal = cast(
-                    "Literal['left', 'right', 'middle', 'back', 'forward']", button or "left"
-                )
-                scaled_x, scaled_y = self._scale_coordinates(x, y)
-                logger.info("Scaled coordinates: %s, %s", scaled_x, scaled_y)
-                result = await self.executor.click(x=scaled_x, y=scaled_y, button=button_literal)
-            else:
-                raise McpError(
-                    ErrorData(code=INVALID_PARAMS, message="x and y coordinates required for click")
-                )
-
-        elif type == "double_click":
-            if x is not None and y is not None:
-                # Use pattern for double-click
-                scaled_x, scaled_y = self._scale_coordinates(x, y)
-                result = await self.executor.click(
-                    x=scaled_x, y=scaled_y, button="left", pattern=[100]
-                )
-            else:
-                raise McpError(
-                    ErrorData(
-                        code=INVALID_PARAMS, message="x and y coordinates required for double_click"
-                    )
-                )
-
-        elif type == "scroll":
-            if x is None or y is None:
-                raise McpError(
-                    ErrorData(
-                        code=INVALID_PARAMS, message="x and y coordinates required for scroll"
-                    )
-                )
-
-            # scroll_x and scroll_y default to 0 if not provided
-            scaled_x, scaled_y = self._scale_coordinates(x, y)
-            result = await self.executor.scroll(
-                x=scaled_x, y=scaled_y, scroll_x=scroll_x or 0, scroll_y=scroll_y or 0
-            )
-
-        elif type == "type":
-            if text is None:
-                raise McpError(ErrorData(code=INVALID_PARAMS, message="text is required for type"))
-            result = await self.executor.write(text=text, enter_after=False)
-
-        elif type == "wait":
-            wait_time = ms or 1000  # Default to 1 second
-            result = await self.executor.wait(time=wait_time)
-
-        elif type == "move":
-            if x is not None and y is not None:
-                scaled_x, scaled_y = self._scale_coordinates(x, y)
-                result = await self.executor.move(x=scaled_x, y=scaled_y)
-            else:
-                raise McpError(
-                    ErrorData(code=INVALID_PARAMS, message="x and y coordinates required for move")
-                )
-
-        elif type == "keypress":
-            if keys is None or len(keys) == 0:
-                raise McpError(
-                    ErrorData(code=INVALID_PARAMS, message="keys is required for keypress")
-                )
-
-            # Map OpenAI keys to CLA standard
-            cla_keys = []
-            for key in keys:
-                cla_key = self._map_openai_key_to_cla(key)
-                cla_keys.append(cla_key)
-
-            result = await self.executor.press(keys=cla_keys)
-
-        elif type == "drag":
-            if path is None or len(path) < 2:
-                raise McpError(
-                    ErrorData(
-                        code=INVALID_PARAMS, message="path with at least 2 points required for drag"
-                    )
-                )
-
-            # Convert path from list of dicts to list of tuples
-            drag_path = []
-            for point in path:
-                if "x" in point and "y" in point:
-                    drag_path.append((point["x"], point["y"]))
-                else:
-                    raise McpError(
-                        ErrorData(
-                            code=INVALID_PARAMS, message="Each point in path must have x and y"
-                        )
-                    )
-
-            scaled_path = self._scale_path(drag_path)
-            result = await self.executor.drag(path=scaled_path)
-
-        elif type == "response":
-            if text is None:
-                raise McpError(
-                    ErrorData(code=INVALID_PARAMS, message="text is required for response")
-                )
-            # Response returns content blocks directly
-            return [TextContent(text=text, type="text")]
-
-        elif type == "custom":
-            # For custom actions, we just return an error since HudComputerTool doesn't support them
-            raise McpError(
-                ErrorData(code=INVALID_PARAMS, message=f"Custom action not supported: {action}")
-            )
-
-        else:
-            raise McpError(ErrorData(code=INTERNAL_ERROR, message=f"Invalid action type: {type}"))
-
-        # Rescale screenshot in result if present
-        if isinstance(result, ContentResult) and result.base64_image and self.rescale_images:
-            rescaled_image = await self._rescale_screenshot(result.base64_image)
-            result.base64_image = rescaled_image
-
-        # Handle screenshot for actions that need it
-        screenshot_actions = {
-            "screenshot",
-            "click",
-            "double_click",
-            "scroll",
-            "type",
-            "move",
-            "keypress",
-            "drag",
-            "wait",
-        }
-
-        if (
-            type in screenshot_actions
-            and type != "screenshot"
-            and isinstance(result, ContentResult)
-            and not result.base64_image
-        ):
-            screenshot_base64 = await self.executor.screenshot()
-            if screenshot_base64:
-                # Rescale screenshot if requested
-                screenshot_base64 = await self._rescale_screenshot(screenshot_base64)
-                result = ContentResult(
-                    output=result.output, error=result.error, base64_image=screenshot_base64
-                )
-
-        # Convert to content blocks
-        return result.to_content_blocks()
+# flake8: noqa: B008
+from __future__ import annotations
+
+import logging
+from typing import TYPE_CHECKING, Any, Literal, cast
+
+from mcp import ErrorData, McpError
+from mcp.types import INTERNAL_ERROR, INVALID_PARAMS, ContentBlock, TextContent
+from pydantic import Field
+
+from hud.tools.computer.settings import computer_settings
+from hud.tools.types import ContentResult
+
+from .hud import HudComputerTool
+
+if TYPE_CHECKING:
+    from hud.tools.executors.base import BaseExecutor
+
+logger = logging.getLogger(__name__)
+
+# Map OpenAI key names to CLA standard keys
+OPENAI_TO_CLA_KEYS = {
+    # Common variations
+    "return": "enter",
+    "escape": "escape",
+    "arrowup": "up",
+    "arrowdown": "down",
+    "arrowleft": "left",
+    "arrowright": "right",
+    "backspace": "backspace",
+    "delete": "delete",
+    "tab": "tab",
+    "space": "space",
+    "control": "ctrl",
+    "alt": "alt",
+    "shift": "shift",
+    "meta": "win",
+    "cmd": "cmd",
+    "command": "cmd",
+    "super": "win",
+    "pageup": "pageup",
+    "pagedown": "pagedown",
+    "home": "home",
+    "end": "end",
+    "insert": "insert",
+}
+
+
+class OpenAIComputerTool(HudComputerTool):
+    """
+    OpenAI Computer Use tool for interacting with the computer.
+    """
+
+    def __init__(
+        self,
+        # Define within environment based on platform
+        executor: BaseExecutor | None = None,
+        platform_type: Literal["auto", "xdo", "pyautogui"] = "auto",
+        display_num: int | None = None,
+        # Overrides for what dimensions the agent thinks it operates in
+        width: int = computer_settings.OPENAI_COMPUTER_WIDTH,
+        height: int = computer_settings.OPENAI_COMPUTER_HEIGHT,
+        rescale_images: bool = computer_settings.OPENAI_RESCALE_IMAGES,
+        name: str | None = None,
+        title: str | None = None,
+        description: str | None = None,
+        **kwargs: Any,
+    ) -> None:
+        """
+        Initialize with OpenAI's default dimensions.
+
+        Args:
+            width: Target width for rescaling (default: 1024 for OpenAI)
+            height: Target height for rescaling (default: 768 for OpenAI)
+            rescale_images: If True, rescale screenshots. If False, only rescale action coordinates
+            name: Tool name for MCP registration (auto-generated from class name if not provided)
+            title: Human-readable display name for the tool (auto-generated from class name)
+            description: Tool description (auto-generated from docstring if not provided)
+        """
+        super().__init__(
+            executor=executor,
+            platform_type=platform_type,
+            display_num=display_num,
+            width=width,
+            height=height,
+            rescale_images=rescale_images,
+            name=name or "openai_computer",
+            title=title or "OpenAI Computer Tool",
+            description=description or "Control computer with mouse, keyboard, and screenshots",
+            **kwargs,
+        )
+
+    def _map_openai_key_to_cla(self, key: str) -> str:
+        """Map OpenAI key name to CLA standard key."""
+        # OpenAI uses lowercase key names
+        return OPENAI_TO_CLA_KEYS.get(key.lower(), key.lower())
+
+    async def __call__(
+        self,
+        type: str = Field(..., description="The action type to perform"),
+        # Coordinate parameters
+        x: int | None = Field(None, description="X coordinate for click/move/scroll actions"),
+        y: int | None = Field(None, description="Y coordinate for click/move/scroll actions"),
+        # Button parameter
+        button: str | None = Field(
+            None, description="Mouse button for click actions (left, right, middle, wheel)"
+        ),
+        # Text parameter
+        text: str | None = Field(None, description="Text to type or response text"),
+        # Scroll parameters
+        scroll_x: int | None = Field(None, description="Horizontal scroll amount"),
+        scroll_y: int | None = Field(None, description="Vertical scroll amount"),
+        # Wait parameter
+        ms: int | None = Field(None, description="Time to wait in milliseconds"),
+        # Key press parameter
+        keys: list[str] | None = Field(None, description="Keys to press"),
+        # Drag parameter
+        path: list[dict[str, int]] | None = Field(
+            None, description="Path for drag actions as list of {x, y} dicts"
+        ),
+        # Custom action parameter
+        action: str | None = Field(None, description="Custom action name"),
+    ) -> list[ContentBlock]:
+        """
+        Handle OpenAI Computer Use API calls.
+
+        This converts OpenAI's action format (based on OperatorAdapter) to HudComputerTool's format.
+
+        Returns:
+            List of MCP content blocks
+        """
+        logger.info("OpenAIComputerTool received type: %s", type)
+
+        # Map button names
+        button_map = {"wheel": "middle"}
+        if button:
+            button = button_map.get(button, button)
+
+        # Process based on action type
+        if type == "screenshot":
+            screenshot_base64 = await self.executor.screenshot()
+            if screenshot_base64:
+                # Rescale screenshot if requested
+                result = ContentResult(base64_image=screenshot_base64)
+            else:
+                result = ContentResult(error="Failed to take screenshot")
+
+        elif type == "click":
+            if x is not None and y is not None:
+                # Cast button to proper literal type
+                button_literal = cast(
+                    "Literal['left', 'right', 'middle', 'back', 'forward']", button or "left"
+                )
+                scaled_x, scaled_y = self._scale_coordinates(x, y)
+                logger.info("Scaled coordinates: %s, %s", scaled_x, scaled_y)
+                result = await self.executor.click(x=scaled_x, y=scaled_y, button=button_literal)
+            else:
+                raise McpError(
+                    ErrorData(code=INVALID_PARAMS, message="x and y coordinates required for click")
+                )
+
+        elif type == "double_click":
+            if x is not None and y is not None:
+                # Use pattern for double-click
+                scaled_x, scaled_y = self._scale_coordinates(x, y)
+                result = await self.executor.click(
+                    x=scaled_x, y=scaled_y, button="left", pattern=[100]
+                )
+            else:
+                raise McpError(
+                    ErrorData(
+                        code=INVALID_PARAMS, message="x and y coordinates required for double_click"
+                    )
+                )
+
+        elif type == "scroll":
+            if x is None or y is None:
+                raise McpError(
+                    ErrorData(
+                        code=INVALID_PARAMS, message="x and y coordinates required for scroll"
+                    )
+                )
+
+            # scroll_x and scroll_y default to 0 if not provided
+            scaled_x, scaled_y = self._scale_coordinates(x, y)
+            result = await self.executor.scroll(
+                x=scaled_x, y=scaled_y, scroll_x=scroll_x or 0, scroll_y=scroll_y or 0
+            )
+
+        elif type == "type":
+            if text is None:
+                raise McpError(ErrorData(code=INVALID_PARAMS, message="text is required for type"))
+            result = await self.executor.write(text=text, enter_after=False)
+
+        elif type == "wait":
+            wait_time = ms or 1000  # Default to 1 second
+            result = await self.executor.wait(time=wait_time)
+
+        elif type == "move":
+            if x is not None and y is not None:
+                scaled_x, scaled_y = self._scale_coordinates(x, y)
+                result = await self.executor.move(x=scaled_x, y=scaled_y)
+            else:
+                raise McpError(
+                    ErrorData(code=INVALID_PARAMS, message="x and y coordinates required for move")
+                )
+
+        elif type == "keypress":
+            if keys is None or len(keys) == 0:
+                raise McpError(
+                    ErrorData(code=INVALID_PARAMS, message="keys is required for keypress")
+                )
+
+            # Map OpenAI keys to CLA standard
+            cla_keys = []
+            for key in keys:
+                cla_key = self._map_openai_key_to_cla(key)
+                cla_keys.append(cla_key)
+
+            result = await self.executor.press(keys=cla_keys)
+
+        elif type == "drag":
+            if path is None or len(path) < 2:
+                raise McpError(
+                    ErrorData(
+                        code=INVALID_PARAMS, message="path with at least 2 points required for drag"
+                    )
+                )
+
+            # Convert path from list of dicts to list of tuples
+            drag_path = []
+            for point in path:
+                if "x" in point and "y" in point:
+                    drag_path.append((point["x"], point["y"]))
+                else:
+                    raise McpError(
+                        ErrorData(
+                            code=INVALID_PARAMS, message="Each point in path must have x and y"
+                        )
+                    )
+
+            scaled_path = self._scale_path(drag_path)
+            result = await self.executor.drag(path=scaled_path)
+
+        elif type == "response":
+            if text is None:
+                raise McpError(
+                    ErrorData(code=INVALID_PARAMS, message="text is required for response")
+                )
+            # Response returns content blocks directly
+            return [TextContent(text=text, type="text")]
+
+        elif type == "custom":
+            # For custom actions, we just return an error since HudComputerTool doesn't support them
+            raise McpError(
+                ErrorData(code=INVALID_PARAMS, message=f"Custom action not supported: {action}")
+            )
+
+        else:
+            raise McpError(ErrorData(code=INTERNAL_ERROR, message=f"Invalid action type: {type}"))
+
+        # Rescale screenshot in result if present
+        if isinstance(result, ContentResult) and result.base64_image and self.rescale_images:
+            rescaled_image = await self._rescale_screenshot(result.base64_image)
+            result.base64_image = rescaled_image
+
+        # Handle screenshot for actions that need it
+        screenshot_actions = {
+            "screenshot",
+            "click",
+            "double_click",
+            "scroll",
+            "type",
+            "move",
+            "keypress",
+            "drag",
+            "wait",
+        }
+
+        if (
+            type in screenshot_actions
+            and type != "screenshot"
+            and isinstance(result, ContentResult)
+            and not result.base64_image
+        ):
+            screenshot_base64 = await self.executor.screenshot()
+            if screenshot_base64:
+                # Rescale screenshot if requested
+                screenshot_base64 = await self._rescale_screenshot(screenshot_base64)
+                result = ContentResult(
+                    output=result.output, error=result.error, base64_image=screenshot_base64
+                )
+
+        # Convert to content blocks
+        return result.to_content_blocks()