diff --git a/pyproject.toml b/pyproject.toml index 94524cb..aba1232 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta" [project] name = "stagehand" -version = "0.0.5" +version = "0.0.6" description = "Python SDK for Stagehand" readme = "README.md" license = {text = "MIT"} diff --git a/stagehand/page.py b/stagehand/page.py index 2b67d35..c0b83d4 100644 --- a/stagehand/page.py +++ b/stagehand/page.py @@ -374,33 +374,6 @@ async def extract( ) return result_dict - async def screenshot(self, options: Optional[dict] = None) -> str: - """ - Take a screenshot of the current page via the Stagehand server. - - Args: - options (Optional[dict]): Optional screenshot options. - May include: - - type: "png" or "jpeg" (default: "png") - - fullPage: whether to take a full page screenshot (default: False) - - quality: for jpeg only, 0-100 (default: 80) - - clip: viewport clip rectangle - - omitBackground: whether to hide default white background (default: False) - - Returns: - str: Base64-encoded screenshot data. - """ - payload = options or {} - - if self._stagehand.use_api: - lock = self._stagehand._get_lock_for_session() - async with lock: - result = await self._stagehand._execute("screenshot", payload) - - return result - else: - return await self._page.screenshot(**payload) - # Method to get or initialize the persistent CDP client async def get_cdp_client(self) -> CDPSession: """Gets the persistent CDP client, initializing it if necessary."""