hyperbrowser 0.34.0__tar.gz → 0.36.0__tar.gz

{hyperbrowser-0.34.0 → hyperbrowser-0.36.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: hyperbrowser
-Version: 0.34.0
+Version: 0.36.0
 Summary: Python SDK for hyperbrowser
 License: MIT
 Author: Nikhil Shahi

{hyperbrowser-0.34.0 → hyperbrowser-0.36.0}/hyperbrowser/tools/__init__.py

@@ -1,5 +1,5 @@
 import json
-from hyperbrowser.models.beta.agents.browser_use import StartBrowserUseTaskParams
+from hyperbrowser.models.agents.browser_use import StartBrowserUseTaskParams
 from hyperbrowser.models.crawl import StartCrawlJobParams
 from hyperbrowser.models.extract import StartExtractJobParams
 from hyperbrowser.models.scrape import StartScrapeJobParams
@@ -9,12 +9,14 @@ from .openai import (
     BROWSER_USE_TOOL_OPENAI,
     EXTRACT_TOOL_OPENAI,
     SCRAPE_TOOL_OPENAI,
+    SCREENSHOT_TOOL_OPENAI,
     CRAWL_TOOL_OPENAI,
 )
 from .anthropic import (
     BROWSER_USE_TOOL_ANTHROPIC,
     EXTRACT_TOOL_ANTHROPIC,
     SCRAPE_TOOL_ANTHROPIC,
+    SCREENSHOT_TOOL_ANTHROPIC,
     CRAWL_TOOL_ANTHROPIC,
 )
 
@@ -34,6 +36,21 @@ class WebsiteScrapeTool:
         return resp.data.markdown if resp.data and resp.data.markdown else ""
 
 
+class WebsiteScreenshotTool:
+    openai_tool_definition = SCREENSHOT_TOOL_OPENAI
+    anthropic_tool_definition = SCREENSHOT_TOOL_ANTHROPIC
+
+    @staticmethod
+    def runnable(hb: Hyperbrowser, params: dict) -> str:
+        resp = hb.scrape.start_and_wait(params=StartScrapeJobParams(**params))
+        return resp.data.screenshot if resp.data and resp.data.screenshot else ""
+
+    @staticmethod
+    async def async_runnable(hb: AsyncHyperbrowser, params: dict) -> str:
+        resp = await hb.scrape.start_and_wait(params=StartScrapeJobParams(**params))
+        return resp.data.screenshot if resp.data and resp.data.screenshot else ""
+
+
 class WebsiteCrawlTool:
     openai_tool_definition = CRAWL_TOOL_OPENAI
     anthropic_tool_definition = CRAWL_TOOL_ANTHROPIC
@@ -88,14 +105,14 @@ class BrowserUseTool:
 
     @staticmethod
     def runnable(hb: Hyperbrowser, params: dict) -> str:
-        resp = hb.beta.agents.browser_use.start_and_wait(
+        resp = hb.agents.browser_use.start_and_wait(
             params=StartBrowserUseTaskParams(**params)
         )
         return resp.data.final_result if resp.data and resp.data.final_result else ""
 
     @staticmethod
     async def async_runnable(hb: AsyncHyperbrowser, params: dict) -> str:
-        resp = await hb.beta.agents.browser_use.start_and_wait(
+        resp = await hb.agents.browser_use.start_and_wait(
             params=StartBrowserUseTaskParams(**params)
         )
         return resp.data.final_result if resp.data and resp.data.final_result else ""

{hyperbrowser-0.34.0 → hyperbrowser-0.36.0}/hyperbrowser/tools/anthropic.py

@@ -6,6 +6,7 @@ from hyperbrowser.tools.schema import (
     CRAWL_SCHEMA,
     EXTRACT_SCHEMA,
     SCRAPE_SCHEMA,
+    SCREENSHOT_SCHEMA,
 )
 
 
@@ -54,6 +55,12 @@ SCRAPE_TOOL_ANTHROPIC: ToolParam = {
     "description": "Scrape content from a webpage and return the content in markdown format",
 }
 
+SCREENSHOT_TOOL_ANTHROPIC: ToolParam = {
+    "input_schema": SCREENSHOT_SCHEMA,
+    "name": "screenshot_webpage",
+    "description": "Scrape content from a webpage and return the content in screenshot format",
+}
+
 CRAWL_TOOL_ANTHROPIC: ToolParam = {
     "input_schema": CRAWL_SCHEMA,
     "name": "crawl_website",

{hyperbrowser-0.34.0 → hyperbrowser-0.36.0}/hyperbrowser/tools/openai.py

@@ -6,6 +6,7 @@ from hyperbrowser.tools.schema import (
     CRAWL_SCHEMA,
     EXTRACT_SCHEMA,
     SCRAPE_SCHEMA,
+    SCREENSHOT_SCHEMA,
 )
 
 FunctionParameters: TypeAlias = Dict[str, object]
@@ -63,6 +64,16 @@ SCRAPE_TOOL_OPENAI: ChatCompletionToolParam = {
     },
 }
 
+SCREENSHOT_TOOL_OPENAI: ChatCompletionToolParam = {
+    "type": "function",
+    "function": {
+        "name": "screenshot_webpage",
+        "description": "Scrape content from a webpage and return the content in screenshot format",
+        "parameters": SCREENSHOT_SCHEMA,
+        "strict": True,
+    },
+}
+
 CRAWL_TOOL_OPENAI: ChatCompletionToolParam = {
     "type": "function",
     "function": {

{hyperbrowser-0.34.0 → hyperbrowser-0.36.0}/hyperbrowser/tools/schema.py

@@ -1,38 +1,67 @@
-SCRAPE_OPTIONS = {
-    "type": "object",
-    "description": "The options for the scrape",
-    "properties": {
-        "include_tags": {
-            "type": "array",
-            "items": {
-                "type": "string",
+from typing import Literal, List
+
+scrape_types = Literal["markdown", "screenshot"]
+
+
+def get_scrape_options(formats: List[scrape_types] = ["markdown"]):
+    return {
+        "type": "object",
+        "description": "The options for the scrape",
+        "properties": {
+            "formats": {
+                "type": "array",
+                "description": "The format of the content to scrape",
+                "items": {
+                    "type": "string",
+                    "enum": formats,
+                },
             },
-            "description": "An array of HTML tags, classes, or IDs to include in the scraped content. Only elements matching these selectors will be returned.",
-        },
-        "exclude_tags": {
-            "type": "array",
-            "items": {
-                "type": "string",
+            "include_tags": {
+                "type": "array",
+                "items": {"type": "string"},
+                "description": "An array of HTML tags, classes, or IDs to include in the scraped content. Only elements matching these selectors will be returned.",
+            },
+            "exclude_tags": {
+                "type": "array",
+                "items": {"type": "string"},
+                "description": "An array of HTML tags, classes, or IDs to exclude from the scraped content. Elements matching these selectors will be omitted from the response.",
+            },
+            "only_main_content": {
+                "type": "boolean",
+                "description": "Whether to only return the main content of the page. If true, only the main content of the page will be returned, excluding any headers, navigation menus,footers, or other non-main content.",
             },
-            "description": "An array of HTML tags, classes, or IDs to exclude from the scraped content. Elements matching these selectors will be omitted from the response.",
         },
-        "only_main_content": {
-            "type": "boolean",
-            "description": "Whether to only return the main content of the page. If true, only the main content of the page will be returned, excluding any headers, navigation menus,footers, or other non-main content.",
+        "required": [
+            "include_tags",
+            "exclude_tags",
+            "only_main_content",
+            "formats",
+        ],
+        "additionalProperties": False,
+    }
+
+
+SCRAPE_SCHEMA = {
+    "type": "object",
+    "properties": {
+        "url": {
+            "type": "string",
+            "description": "The URL of the website to scrape",
         },
+        "scrape_options": get_scrape_options(),
     },
-    "required": ["include_tags", "exclude_tags", "only_main_content"],
+    "required": ["url", "scrape_options"],
     "additionalProperties": False,
 }
 
-SCRAPE_SCHEMA = {
+SCREENSHOT_SCHEMA = {
     "type": "object",
     "properties": {
         "url": {
             "type": "string",
             "description": "The URL of the website to scrape",
         },
-        "scrape_options": SCRAPE_OPTIONS,
+        "scrape_options": get_scrape_options(["screenshot"]),
     },
     "required": ["url", "scrape_options"],
     "additionalProperties": False,
@@ -71,7 +100,7 @@ CRAWL_SCHEMA = {
             },
             "description": "An array of regular expressions or wildcard patterns specifying which URLs should be included in the crawl. Only pages whose URLs' path match one of these path patterns will be visited. Example: ['/admin', '/careers/*']",
         },
-        "scrape_options": SCRAPE_OPTIONS,
+        "scrape_options": get_scrape_options(),
     },
     "required": [
         "url",

{hyperbrowser-0.34.0 → hyperbrowser-0.36.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "hyperbrowser"
-version = "0.34.0"
+version = "0.36.0"
 description = "Python SDK for hyperbrowser"
 authors = ["Nikhil Shahi <nshahi1998@gmail.com>"]
 license = "MIT"