agentrun-inner-test 0.0.46__py3-none-any.whl

agentrun/sandbox/api/playwright_sync.py

@@ -0,0 +1,492 @@
+"""Playwright同步API封装 / Playwright Sync API Wrapper
+
+提供Playwright的同步API封装,用于浏览器沙箱操作。
+Provides sync API wrapper for Playwright, used for browser sandbox operations.
+"""
+
+# THIS FILE IS GENERATED BY playwright_async.py, DO NOT MODIFY THIS FILE
+
+from pathlib import Path
+from typing import Any, Dict, List, Literal, Optional, Sequence, Union
+
+from playwright.sync_api import (
+    Browser,
+    BrowserContext,
+    FloatRect,
+    Locator,
+    Page,
+    Playwright,
+    Position,
+    Response,
+    sync_playwright,
+)
+
+
+class BrowserPlaywrightSync:
+    """A small helper wrapper around Playwright's Sync API.
+
+    This class connects to an existing Chromium instance over CDP and exposes a set
+    of common page operations. Internally it lazily ensures a `Browser`,
+    `BrowserContext`, and `Page` and keeps references to reuse them across calls.
+
+    Notes
+    -----
+    - Connection is established via CDP using the given `url`.
+    - If `auto_close_browser`/`auto_close_page` are enabled, `close()` will attempt
+      to close the browser/page respectively.
+    - Methods that act on the page automatically bring the page to the front.
+    """
+
+    def __init__(
+        self,
+        url: str,
+        browser_type: str = "chrome",
+        auto_close_browser: bool = False,
+        auto_close_page: bool = False,
+        headers: Optional[Dict[str, str]] = None,
+    ):
+        self.url = url
+        self.browser_type = browser_type
+        self.auto_close_browser = auto_close_browser
+        self.auto_close_page = auto_close_page
+        self.headers = headers
+
+        self._playwright = sync_playwright()
+        self._playwright_instance: Optional[Playwright] = None
+        self._browser: Optional[Browser] = None
+        self._context: Optional[BrowserContext] = None
+        self._page: Optional[Page] = None
+
+    def open(self) -> "BrowserPlaywrightSync":
+        """Establish a connection to the remote browser if not already connected.
+
+        Returns
+        -------
+        BrowserPlaywrightSync
+            The current instance for fluent chaining.
+        """
+        if self._browser:
+            return self
+
+        self._playwright_instance = self._playwright.start()
+        self._browser = self._playwright_instance.chromium.connect_over_cdp(
+            self.url, headers=self.headers
+        )
+
+        return self
+
+    def close(self) -> None:
+        """Close held resources according to the auto-close flags.
+
+        Closes page and/or browser if the corresponding auto-close flags are set
+        and stops the Playwright driver when present.
+        """
+        if self.auto_close_page and self._page:
+            self._page.close()
+
+        if self.auto_close_browser and self._browser:
+            self._browser.close()
+
+        if self._playwright_instance:
+            self._playwright_instance.stop()
+
+    def __enter__(self) -> "BrowserPlaywrightSync":
+        """Enter context by ensuring the connection is open.
+
+        Returns
+        -------
+        BrowserPlaywrightSync
+            The current instance.
+        """
+        return self.open()
+
+    def __exit__(self, *args: Any) -> None:
+        """Exit context by closing resources based on auto-close flags."""
+        self.close()
+
+    def ensure_browser(self) -> Browser:
+        """Ensure a `Browser` instance is available.
+
+        Returns
+        -------
+        Browser
+            A connected Playwright `Browser` instance.
+        """
+        if self._browser:
+            return self._browser
+
+        self.open()
+
+        assert self._browser is not None
+
+        return self._browser
+
+    def ensure_context(self) -> BrowserContext:
+        """Ensure a `BrowserContext` is available, creating one if necessary.
+
+        Returns
+        -------
+        BrowserContext
+            The ensured `BrowserContext`.
+        """
+        browser = self.ensure_browser()
+
+        if self._context:
+            return self._context
+
+        if len(browser.contexts) > 0:
+            self._context = browser.contexts[0]
+        else:
+            self._context = browser.new_context()
+
+        return self._context
+
+    def ensure_page(self) -> Page:
+        """Ensure a `Page` is available in the current context.
+
+        Returns
+        -------
+        Page
+            The ensured `Page` which is brought to the front.
+        """
+        ctx = self.ensure_context()
+
+        if self._page:
+            self._page.bring_to_front()
+            return self._page
+
+        if len(ctx.pages) > 0:
+            self._page = ctx.pages[0]
+        else:
+            self._page = ctx.new_page()
+
+        self._page.bring_to_front()
+        return self._page
+
+    def _use_page(self, page: Page) -> Page:
+        """Set the active page and context.
+
+        Parameters
+        ----------
+        page : Page
+            The page to make active.
+
+        Returns
+        -------
+        Page
+            The provided page.
+        """
+        page.bring_to_front()
+        self._page = page
+        self._context = page.context
+        return self._page
+
+    def list_pages(self) -> List[Page]:
+        """List all pages across all contexts in the connected browser."""
+        pages: List[Page] = []
+
+        browser = self.ensure_browser()
+        for context in browser.contexts:
+            for page in context.pages:
+                pages.append(page)
+
+        return pages
+
+    def new_page(self) -> Page:
+        """Create and switch to a new page in the ensured context."""
+        context = self.ensure_context()
+        page = context.new_page()
+        return self._use_page(page)
+
+    def select_tab(self, index: int) -> Page:
+        """Select a page by index across all open tabs.
+
+        Parameters
+        ----------
+        index : int
+            Zero-based page index.
+
+        Returns
+        -------
+        Page
+            The selected page.
+
+        Raises
+        ------
+        IndexError
+            If the index is out of bounds.
+        """
+        pages = self.list_pages()
+        if 0 <= index < len(pages):
+            return self._use_page(pages[index])
+        else:
+            raise IndexError("Tab index out of range")
+
+    def goto(
+        self,
+        url: str,
+        timeout: Optional[float] = None,
+        wait_until: Optional[
+            Literal["commit", "domcontentloaded", "load", "networkidle"]
+        ] = None,
+        referer: Optional[str] = None,
+    ) -> Optional[Response]:
+        """Navigate to a URL on the active page.
+
+        Returns
+        -------
+        Optional[Response]
+            The main resource response if available; otherwise `None`.
+        """
+        page = self.ensure_page()
+        return page.goto(
+            url, timeout=timeout, wait_until=wait_until, referer=referer
+        )
+
+    def click(
+        self,
+        selector: str,
+        modifiers: Optional[
+            Sequence[
+                Literal["Alt", "Control", "ControlOrMeta", "Meta", "Shift"]
+            ]
+        ] = None,
+        position: Optional[Position] = None,
+        delay: Optional[float] = None,
+        button: Optional[Literal["left", "middle", "right"]] = None,
+        click_count: Optional[int] = None,
+        timeout: Optional[float] = None,
+        force: Optional[bool] = None,
+        no_wait_after: Optional[bool] = None,
+        trial: Optional[bool] = None,
+        strict: Optional[bool] = None,
+    ) -> None:
+        """Click an element matching the selector on the active page."""
+        page = self.ensure_page()
+        return page.click(
+            selector,
+            modifiers=modifiers,
+            position=position,
+            delay=delay,
+            button=button,
+            click_count=click_count,
+            timeout=timeout,
+            force=force,
+            no_wait_after=no_wait_after,
+            trial=trial,
+            strict=strict,
+        )
+
+    def drag_and_drop(
+        self,
+        source: str,
+        target: str,
+        source_position: Optional[Position] = None,
+        target_position: Optional[Position] = None,
+        force: Optional[bool] = None,
+        no_wait_after: Optional[bool] = None,
+        timeout: Optional[float] = None,
+        strict: Optional[bool] = None,
+        trial: Optional[bool] = None,
+    ) -> None:
+        """Alias for `drag` using Playwright's `drag_and_drop` under the hood."""
+        page = self.ensure_page()
+        return page.drag_and_drop(
+            source=source,
+            target=target,
+            source_position=source_position,
+            target_position=target_position,
+            force=force,
+            no_wait_after=no_wait_after,
+            timeout=timeout,
+            strict=strict,
+            trial=trial,
+        )
+
+    def dblclick(
+        self,
+        selector: str,
+        modifiers: Optional[
+            Sequence[
+                Literal["Alt", "Control", "ControlOrMeta", "Meta", "Shift"]
+            ]
+        ] = None,
+        position: Optional[Position] = None,
+        delay: Optional[float] = None,
+        button: Optional[Literal["left", "middle", "right"]] = None,
+        timeout: Optional[float] = None,
+        force: Optional[bool] = None,
+        no_wait_after: Optional[bool] = None,
+        strict: Optional[bool] = None,
+        trial: Optional[bool] = None,
+    ) -> None:
+        """Double-click an element matching the selector on the active page."""
+        page = self.ensure_page()
+        return page.dblclick(
+            selector=selector,
+            modifiers=modifiers,
+            position=position,
+            delay=delay,
+            button=button,
+            timeout=timeout,
+            force=force,
+            no_wait_after=no_wait_after,
+            strict=strict,
+            trial=trial,
+        )
+
+    def fill(
+        self,
+        selector: str,
+        value: str,
+        timeout: Optional[float] = None,
+        no_wait_after: Optional[bool] = None,
+        strict: Optional[bool] = None,
+        force: Optional[bool] = None,
+    ) -> None:
+        """Fill an input/textarea matched by selector with the provided value."""
+        page = self.ensure_page()
+        return page.fill(
+            selector=selector,
+            value=value,
+            timeout=timeout,
+            no_wait_after=no_wait_after,
+            strict=strict,
+            force=force,
+        )
+
+    def hover(
+        self,
+        selector: str,
+        modifiers: Optional[
+            Sequence[
+                Literal["Alt", "Control", "ControlOrMeta", "Meta", "Shift"]
+            ]
+        ] = None,
+        position: Optional[Position] = None,
+        timeout: Optional[float] = None,
+        no_wait_after: Optional[bool] = None,
+        force: Optional[bool] = None,
+        strict: Optional[bool] = None,
+        trial: Optional[bool] = None,
+    ) -> None:
+        """Hover over the element matched by the selector."""
+        page = self.ensure_page()
+        return page.hover(
+            selector=selector,
+            modifiers=modifiers,
+            position=position,
+            timeout=timeout,
+            no_wait_after=no_wait_after,
+            force=force,
+            strict=strict,
+            trial=trial,
+        )
+
+    def type(
+        self,
+        selector: str,
+        text: str,
+        delay: Optional[float] = None,
+        timeout: Optional[float] = None,
+        no_wait_after: Optional[bool] = None,
+        strict: Optional[bool] = None,
+    ) -> None:
+        """Type text into an element matched by the selector."""
+        page = self.ensure_page()
+        return page.type(
+            selector=selector,
+            text=text,
+            delay=delay,
+            timeout=timeout,
+            no_wait_after=no_wait_after,
+            strict=strict,
+        )
+
+    def go_forward(
+        self,
+        timeout: Optional[float] = None,
+        wait_until: Optional[
+            Literal["commit", "domcontentloaded", "load", "networkidle"]
+        ] = None,
+    ) -> Optional[Response]:
+        """Go forward in the page history if possible."""
+        page = self.ensure_page()
+        return page.go_forward(timeout=timeout, wait_until=wait_until)
+
+    def go_back(
+        self,
+        timeout: Optional[float] = None,
+        wait_until: Optional[
+            Literal["commit", "domcontentloaded", "load", "networkidle"]
+        ] = None,
+    ) -> Optional[Response]:
+        """Go back in the page history if possible."""
+        page = self.ensure_page()
+        return page.go_back(timeout=timeout, wait_until=wait_until)
+
+    def evaluate(
+        self,
+        expression: str,
+        arg: Optional[Any] = None,
+    ) -> Any:
+        """Evaluate a JavaScript expression in the page context."""
+
+        page = self.ensure_page()
+        return page.evaluate(expression=expression, arg=arg)
+
+    def wait(self, timeout: float) -> None:
+        """Wait for the given timeout in milliseconds."""
+        page = self.ensure_page()
+        return page.wait_for_timeout(timeout=timeout)
+
+    def html_content(
+        self,
+    ) -> str:
+        """Get the current page's HTML content as a string."""
+        page = self.ensure_page()
+        return page.content()
+
+    def screenshot(
+        self,
+        timeout: Optional[float] = None,
+        type: Optional[Literal["jpeg", "png"]] = None,
+        path: Union[Path, str, None] = None,
+        quality: Optional[int] = None,
+        omit_background: Optional[bool] = None,
+        full_page: Optional[bool] = None,
+        clip: Optional[FloatRect] = None,
+        animations: Optional[Literal["allow", "disabled"]] = None,
+        caret: Optional[Literal["hide", "initial"]] = None,
+        scale: Optional[Literal["css", "device"]] = None,
+        mask: Optional[Sequence[Locator]] = None,
+        mask_color: Optional[str] = None,
+        style: Optional[str] = None,
+    ) -> bytes:
+        """Capture a screenshot of the page.
+
+        Returns
+        -------
+        bytes
+            The image bytes of the screenshot.
+        """
+        page = self.ensure_page()
+        return page.screenshot(
+            timeout=timeout,
+            type=type,
+            path=path,
+            quality=quality,
+            omit_background=omit_background,
+            full_page=full_page,
+            clip=clip,
+            animations=animations,
+            caret=caret,
+            scale=scale,
+            mask=mask,
+            mask_color=mask_color,
+            style=style,
+        )
+
+    def title(self) -> str:
+        page = self.ensure_page()
+        return page.title()

agentrun/sandbox/api/sandbox_data.py

@@ -0,0 +1,154 @@
+"""
+This file is auto generated by the code generation script.
+Do not modify this file manually.
+Use the `make codegen` command to regenerate.
+
+当前文件为自动生成的控制 API 客户端代码。请勿手动修改此文件。
+使用 `make codegen` 命令重新生成。
+
+source: agentrun/sandbox/api/__sandbox_data_async_template.py
+
+Sandbox数据API模板 / Sandbox Data API Template
+
+此模板用于生成沙箱数据API代码。
+This template is used to generate sandbox data API code.
+"""
+
+from typing import Any, Dict, Optional
+
+from agentrun.utils.config import Config
+from agentrun.utils.data_api import DataAPI, ResourceType
+
+
+class SandboxDataAPI(DataAPI):
+
+    def __init__(
+        self,
+        *,
+        sandbox_id: Optional[str] = None,
+        template_name: Optional[str] = None,
+        config: Optional[Config] = None,
+    ):
+
+        super().__init__(
+            resource_name="",
+            resource_type=ResourceType.Template,
+            namespace="sandboxes",
+            config=config,
+        )
+        self.access_token_map = {}
+
+        if sandbox_id or template_name:
+            self.__refresh_access_token(
+                sandbox_id=sandbox_id,
+                template_name=template_name,
+                config=config,
+            )
+
+    def __refresh_access_token(
+        self,
+        *,
+        sandbox_id: Optional[str] = None,
+        template_name: Optional[str] = None,
+        config: Optional[Config] = None,
+    ):
+        cfg = Config.with_configs(config, self.config)
+        token = self.access_token_map.get(sandbox_id or template_name)
+        if sandbox_id:
+            self.resource_name = sandbox_id
+            self.resource_type = ResourceType.Sandbox
+            self.namespace = f"sandboxes/{sandbox_id}"
+        else:
+            self.resource_name = template_name
+            self.resource_type = ResourceType.Template
+            self.namespace = "sandboxes"
+
+        if token:
+            self.access_token = token
+            return
+
+        # 没有缓存过的 token
+
+        self.access_token = None
+        self.auth(config=cfg)
+        self.access_token_map[sandbox_id or template_name] = self.access_token
+
+    async def check_health_async(self):
+        return await self.get_async("/health")
+
+    def check_health(self):
+        return self.get("/health")
+
+    async def create_sandbox_async(
+        self,
+        template_name: str,
+        sandbox_idle_timeout_seconds: Optional[int] = 600,
+        nas_config: Optional[Dict[str, Any]] = None,
+        oss_mount_config: Optional[Dict[str, Any]] = None,
+        polar_fs_config: Optional[Dict[str, Any]] = None,
+        config: Optional[Config] = None,
+    ):
+        self.__refresh_access_token(template_name=template_name, config=config)
+        data: Dict[str, Any] = {
+            "templateName": template_name,
+            "sandboxIdleTimeoutSeconds": sandbox_idle_timeout_seconds,
+        }
+        if nas_config is not None:
+            data["nasConfig"] = nas_config
+        if oss_mount_config is not None:
+            data["ossMountConfig"] = oss_mount_config
+        if polar_fs_config is not None:
+            data["polarFsConfig"] = polar_fs_config
+        return await self.post_async("/", data=data)
+
+    def create_sandbox(
+        self,
+        template_name: str,
+        sandbox_idle_timeout_seconds: Optional[int] = 600,
+        nas_config: Optional[Dict[str, Any]] = None,
+        oss_mount_config: Optional[Dict[str, Any]] = None,
+        polar_fs_config: Optional[Dict[str, Any]] = None,
+        config: Optional[Config] = None,
+    ):
+        self.__refresh_access_token(template_name=template_name, config=config)
+        data: Dict[str, Any] = {
+            "templateName": template_name,
+            "sandboxIdleTimeoutSeconds": sandbox_idle_timeout_seconds,
+        }
+        if nas_config is not None:
+            data["nasConfig"] = nas_config
+        if oss_mount_config is not None:
+            data["ossMountConfig"] = oss_mount_config
+        if polar_fs_config is not None:
+            data["polarFsConfig"] = polar_fs_config
+        return self.post("/", data=data)
+
+    async def delete_sandbox_async(
+        self, sandbox_id: str, config: Optional[Config] = None
+    ):
+        self.__refresh_access_token(sandbox_id=sandbox_id, config=config)
+        return await self.delete_async("/")
+
+    def delete_sandbox(self, sandbox_id: str, config: Optional[Config] = None):
+        self.__refresh_access_token(sandbox_id=sandbox_id, config=config)
+        return self.delete("/")
+
+    async def stop_sandbox_async(
+        self, sandbox_id: str, config: Optional[Config] = None
+    ):
+        self.__refresh_access_token(sandbox_id=sandbox_id, config=config)
+        return await self.post_async("/stop")
+
+    def stop_sandbox(self, sandbox_id: str, config: Optional[Config] = None):
+        self.__refresh_access_token(sandbox_id=sandbox_id, config=config)
+        return self.post("/stop")
+
+    async def get_sandbox_async(
+        self, sandbox_id: str, config: Optional[Config] = None
+    ):
+        self.__refresh_access_token(sandbox_id=sandbox_id, config=config)
+        return await self.get_async("/")
+
+    def get_sandbox(self, sandbox_id: str, config: Optional[Config] = None):
+        self.__refresh_access_token(sandbox_id=sandbox_id, config=config)
+        return self.get("/")