ConradLinus's picture
Upload folder using huggingface_hub
d631808 verified
import abc
from typing import Literal
Environment = Literal["mac", "windows", "ubuntu", "browser"]
Button = Literal["left", "right", "wheel", "back", "forward"]
class Computer(abc.ABC):
"""A computer implemented with sync operations. The Computer interface abstracts the
operations needed to control a computer or browser."""
@property
@abc.abstractmethod
def environment(self) -> Environment:
pass
@property
@abc.abstractmethod
def dimensions(self) -> tuple[int, int]:
pass
@abc.abstractmethod
def screenshot(self) -> str:
pass
@abc.abstractmethod
def click(self, x: int, y: int, button: Button) -> None:
pass
@abc.abstractmethod
def double_click(self, x: int, y: int) -> None:
pass
@abc.abstractmethod
def scroll(self, x: int, y: int, scroll_x: int, scroll_y: int) -> None:
pass
@abc.abstractmethod
def type(self, text: str) -> None:
pass
@abc.abstractmethod
def wait(self) -> None:
pass
@abc.abstractmethod
def move(self, x: int, y: int) -> None:
pass
@abc.abstractmethod
def keypress(self, keys: list[str]) -> None:
pass
@abc.abstractmethod
def drag(self, path: list[tuple[int, int]]) -> None:
pass
class AsyncComputer(abc.ABC):
"""A computer implemented with async operations. The Computer interface abstracts the
operations needed to control a computer or browser."""
@property
@abc.abstractmethod
def environment(self) -> Environment:
pass
@property
@abc.abstractmethod
def dimensions(self) -> tuple[int, int]:
pass
@abc.abstractmethod
async def screenshot(self) -> str:
pass
@abc.abstractmethod
async def click(self, x: int, y: int, button: Button) -> None:
pass
@abc.abstractmethod
async def double_click(self, x: int, y: int) -> None:
pass
@abc.abstractmethod
async def scroll(self, x: int, y: int, scroll_x: int, scroll_y: int) -> None:
pass
@abc.abstractmethod
async def type(self, text: str) -> None:
pass
@abc.abstractmethod
async def wait(self) -> None:
pass
@abc.abstractmethod
async def move(self, x: int, y: int) -> None:
pass
@abc.abstractmethod
async def keypress(self, keys: list[str]) -> None:
pass
@abc.abstractmethod
async def drag(self, path: list[tuple[int, int]]) -> None:
pass