hyperbrowser 0.10.0__tar.gz → 0.12.0__tar.gz

{hyperbrowser-0.10.0 → hyperbrowser-0.12.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hyperbrowser
-Version: 0.10.0
+Version: 0.12.0
 Summary: Python SDK for hyperbrowser
 Home-page: https://github.com/hyperbrowserai/python-sdk
 License: MIT
@@ -19,6 +19,10 @@ Requires-Dist: pydantic (>=2.10.0,<3.0.0)
 Project-URL: Repository, https://github.com/hyperbrowserai/python-sdk
 Description-Content-Type: text/markdown
 
+# Hyperbrowser Python SDK
+
+Checkout the full documentation [here](https://docs.hyperbrowser.ai/)
+
 ## Installation
 
 Currently Hyperbrowser supports creating a browser session in two ways:

{hyperbrowser-0.10.0 → hyperbrowser-0.12.0}/README.md

@@ -1,3 +1,7 @@
+# Hyperbrowser Python SDK
+
+Checkout the full documentation [here](https://docs.hyperbrowser.ai/)
+
 ## Installation
 
 Currently Hyperbrowser supports creating a browser session in two ways:

{hyperbrowser-0.10.0 → hyperbrowser-0.12.0}/hyperbrowser/client/async_client.py

@@ -1,7 +1,10 @@
 from typing import Optional
+
+from .managers.async_manager.profile import ProfileManager
 from .managers.async_manager.session import SessionManager
 from .managers.async_manager.scrape import ScrapeManager
 from .managers.async_manager.crawl import CrawlManager
+from .managers.async_manager.extension import ExtensionManager
 from .base import HyperbrowserBase
 from ..transport.async_transport import AsyncTransport
 from ..config import ClientConfig
@@ -15,11 +18,15 @@ class AsyncHyperbrowser(HyperbrowserBase):
         config: Optional[ClientConfig] = None,
         api_key: Optional[str] = None,
         base_url: Optional[str] = None,
+        timeout: Optional[int] = 30,
     ):
         super().__init__(AsyncTransport, config, api_key, base_url)
+        self.transport.client.timeout = timeout
         self.sessions = SessionManager(self)
         self.scrape = ScrapeManager(self)
         self.crawl = CrawlManager(self)
+        self.profiles = ProfileManager(self)
+        self.extensions = ExtensionManager(self)
 
     async def close(self) -> None:
         await self.transport.close()

hyperbrowser-0.12.0/hyperbrowser/client/managers/async_manager/extension.py

@@ -0,0 +1,40 @@
+import os
+from typing import List
+
+from hyperbrowser.exceptions import HyperbrowserError
+from hyperbrowser.models.extension import CreateExtensionParams, ExtensionResponse
+
+
+class ExtensionManager:
+    def __init__(self, client):
+        self._client = client
+
+    async def create(self, params: CreateExtensionParams) -> ExtensionResponse:
+        file_path = params.file_path
+        params.file_path = None
+
+        # Check if file exists before trying to open it
+        if not os.path.exists(file_path):
+            raise FileNotFoundError(f"Extension file not found at path: {file_path}")
+
+        response = await self._client.transport.post(
+            self._client._build_url("/extensions/add"),
+            data=(
+                {}
+                if params is None
+                else params.model_dump(exclude_none=True, by_alias=True)
+            ),
+            files={"file": open(file_path, "rb")},
+        )
+        return ExtensionResponse(**response.data)
+
+    async def list(self) -> List[ExtensionResponse]:
+        response = await self._client.transport.get(
+            self._client._build_url("/extensions/list"),
+        )
+        if not isinstance(response.data, list):
+            raise HyperbrowserError(
+                f"Expected list response but got {type(response.data)}",
+                original_error=None,
+            )
+        return [ExtensionResponse(**extension) for extension in response.data]

hyperbrowser-0.12.0/hyperbrowser/client/managers/async_manager/profile.py

@@ -0,0 +1,25 @@
+from hyperbrowser.models.profile import ProfileResponse, CreateProfileResponse
+from hyperbrowser.models.session import BasicResponse
+
+
+class ProfileManager:
+    def __init__(self, client):
+        self._client = client
+
+    async def create(self) -> CreateProfileResponse:
+        response = await self._client.transport.post(
+            self._client._build_url("/profile"),
+        )
+        return CreateProfileResponse(**response.data)
+
+    async def get(self, id: str) -> ProfileResponse:
+        response = await self._client.transport.get(
+            self._client._build_url(f"/profile/{id}"),
+        )
+        return ProfileResponse(**response.data)
+
+    async def delete(self, id: str) -> BasicResponse:
+        response = await self._client.transport.delete(
+            self._client._build_url(f"/profile/{id}"),
+        )
+        return BasicResponse(**response.data)

{hyperbrowser-0.10.0 → hyperbrowser-0.12.0}/hyperbrowser/client/managers/async_manager/session.py

@@ -16,7 +16,11 @@ class SessionManager:
     async def create(self, params: CreateSessionParams = None) -> SessionDetail:
         response = await self._client.transport.post(
             self._client._build_url("/session"),
-            data={} if params is None else params.model_dump(exclude_none=True, by_alias=True),
+            data=(
+                {}
+                if params is None
+                else params.model_dump(exclude_none=True, by_alias=True)
+            ),
         )
         return SessionDetail(**response.data)
 

hyperbrowser-0.12.0/hyperbrowser/client/managers/sync_manager/extension.py

@@ -0,0 +1,40 @@
+import os
+from typing import List
+
+from hyperbrowser.exceptions import HyperbrowserError
+from hyperbrowser.models.extension import CreateExtensionParams, ExtensionResponse
+
+
+class ExtensionManager:
+    def __init__(self, client):
+        self._client = client
+
+    def create(self, params: CreateExtensionParams) -> ExtensionResponse:
+        file_path = params.file_path
+        params.file_path = None
+
+        # Check if file exists before trying to open it
+        if not os.path.exists(file_path):
+            raise FileNotFoundError(f"Extension file not found at path: {file_path}")
+
+        response = self._client.transport.post(
+            self._client._build_url("/extensions/add"),
+            data=(
+                {}
+                if params is None
+                else params.model_dump(exclude_none=True, by_alias=True)
+            ),
+            files={"file": open(file_path, "rb")},
+        )
+        return ExtensionResponse(**response.data)
+
+    def list(self) -> List[ExtensionResponse]:
+        response = self._client.transport.get(
+            self._client._build_url("/extensions/list"),
+        )
+        if not isinstance(response.data, list):
+            raise HyperbrowserError(
+                f"Expected list response but got {type(response.data)}",
+                original_error=None,
+            )
+        return [ExtensionResponse(**extension) for extension in response.data]

hyperbrowser-0.12.0/hyperbrowser/client/managers/sync_manager/profile.py

@@ -0,0 +1,25 @@
+from hyperbrowser.models.profile import ProfileResponse, CreateProfileResponse
+from hyperbrowser.models.session import BasicResponse
+
+
+class ProfileManager:
+    def __init__(self, client):
+        self._client = client
+
+    def create(self) -> CreateProfileResponse:
+        response = self._client.transport.post(
+            self._client._build_url("/profile"),
+        )
+        return CreateProfileResponse(**response.data)
+
+    def get(self, id: str) -> ProfileResponse:
+        response = self._client.transport.get(
+            self._client._build_url(f"/profile/{id}"),
+        )
+        return ProfileResponse(**response.data)
+
+    def delete(self, id: str) -> BasicResponse:
+        response = self._client.transport.delete(
+            self._client._build_url(f"/profile/{id}"),
+        )
+        return BasicResponse(**response.data)

{hyperbrowser-0.10.0 → hyperbrowser-0.12.0}/hyperbrowser/client/managers/sync_manager/session.py

@@ -16,7 +16,11 @@ class SessionManager:
     def create(self, params: CreateSessionParams = None) -> SessionDetail:
         response = self._client.transport.post(
             self._client._build_url("/session"),
-            data={} if params is None else params.model_dump(exclude_none=True, by_alias=True),
+            data=(
+                {}
+                if params is None
+                else params.model_dump(exclude_none=True, by_alias=True)
+            ),
         )
         return SessionDetail(**response.data)
 

{hyperbrowser-0.10.0 → hyperbrowser-0.12.0}/hyperbrowser/client/sync.py

@@ -1,7 +1,9 @@
 from typing import Optional
+from .managers.sync_manager.profile import ProfileManager
 from .managers.sync_manager.session import SessionManager
 from .managers.sync_manager.scrape import ScrapeManager
 from .managers.sync_manager.crawl import CrawlManager
+from .managers.sync_manager.extension import ExtensionManager
 from .base import HyperbrowserBase
 from ..transport.sync import SyncTransport
 from ..config import ClientConfig
@@ -15,11 +17,15 @@ class Hyperbrowser(HyperbrowserBase):
         config: Optional[ClientConfig] = None,
         api_key: Optional[str] = None,
         base_url: Optional[str] = None,
+        timeout: Optional[int] = 30,
     ):
         super().__init__(SyncTransport, config, api_key, base_url)
+        self.transport.client.timeout = timeout
         self.sessions = SessionManager(self)
         self.scrape = ScrapeManager(self)
         self.crawl = CrawlManager(self)
+        self.profiles = ProfileManager(self)
+        self.extensions = ExtensionManager(self)
 
     def close(self) -> None:
         self.transport.close()

{hyperbrowser-0.10.0 → hyperbrowser-0.12.0}/hyperbrowser/models/consts.py

@@ -1,6 +1,6 @@
 from typing import Literal
 
-ScrapeFormat = Literal["markdown", "html", "links"]
+ScrapeFormat = Literal["markdown", "html", "links", "screenshot"]
 
 Country = Literal[
     "AD",

hyperbrowser-0.12.0/hyperbrowser/models/extension.py

@@ -0,0 +1,27 @@
+from datetime import datetime
+from typing import List, Literal, Optional
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class CreateExtensionParams(BaseModel):
+    """
+    Parameters for creating a new extension.
+    """
+
+    model_config = ConfigDict(
+        populate_by_alias=True,
+    )
+
+    name: Optional[str] = Field(default=None, serialization_alias="name")
+    file_path: str = Field(serialization_alias="filePath")
+
+
+class ExtensionResponse(BaseModel):
+    model_config = ConfigDict(
+        populate_by_alias=True,
+    )
+
+    id: str = Field(serialization_alias="id")
+    name: str = Field(serialization_alias="name")
+    created_at: datetime = Field(serialization_alias="createdAt", alias="createdAt")
+    updated_at: datetime = Field(serialization_alias="updatedAt", alias="updatedAt")

hyperbrowser-0.12.0/hyperbrowser/models/profile.py

@@ -0,0 +1,17 @@
+from pydantic import BaseModel, ConfigDict, Field
+from datetime import datetime
+
+
+class CreateProfileResponse(BaseModel):
+    id: str
+
+
+class ProfileResponse(BaseModel):
+    model_config = ConfigDict(
+        populate_by_alias=True,
+    )
+
+    id: str
+    team_id: str = Field(alias="teamId")
+    created_at: datetime = Field(alias="createdAt")
+    updated_at: datetime = Field(alias="updatedAt")

{hyperbrowser-0.10.0 → hyperbrowser-0.12.0}/hyperbrowser/models/session.py

@@ -101,6 +101,17 @@ class ScreenConfig(BaseModel):
     height: int = Field(default=720, serialization_alias="height")
 
 
+class CreateSessionProfile(BaseModel):
+    """
+    Profile configuration parameters for browser session.
+    """
+
+    id: Optional[str] = Field(default=None, serialization_alias="id")
+    persist_changes: Optional[bool] = Field(
+        default=None, serialization_alias="persistChanges"
+    )
+
+
 class CreateSessionParams(BaseModel):
     """
     Parameters for creating a new browser session.
@@ -134,7 +145,11 @@ class CreateSessionParams(BaseModel):
     trackers: bool = Field(default=False, serialization_alias="trackers")
     annoyances: bool = Field(default=False, serialization_alias="annoyances")
     enable_web_recording: Optional[bool] = Field(
-        default=False, serialization_alias="enableWebRecording"
+        default=True, serialization_alias="enableWebRecording"
+    )
+    profile: Optional[CreateSessionProfile] = Field(default=None)
+    extension_ids: Optional[List[str]] = Field(
+        default=None, serialization_alias="extensionIds"
     )
 
 

{hyperbrowser-0.10.0 → hyperbrowser-0.12.0}/hyperbrowser/transport/async_transport.py

@@ -66,9 +66,14 @@ class AsyncTransport(TransportStrategy):
         except httpx.RequestError as e:
             raise HyperbrowserError("Request failed", original_error=e)
 
-    async def post(self, url: str, data: Optional[dict] = None) -> APIResponse:
+    async def post(
+        self, url: str, data: Optional[dict] = None, files: Optional[dict] = None
+    ) -> APIResponse:
         try:
-            response = await self.client.post(url, json=data)
+            if files:
+                response = await self.client.post(url, data=data, files=files)
+            else:
+                response = await self.client.post(url, json=data)
             return await self._handle_response(response)
         except HyperbrowserError:
             raise
@@ -94,3 +99,12 @@ class AsyncTransport(TransportStrategy):
             raise
         except Exception as e:
             raise HyperbrowserError("Put request failed", original_error=e)
+
+    async def delete(self, url: str) -> APIResponse:
+        try:
+            response = await self.client.delete(url)
+            return await self._handle_response(response)
+        except HyperbrowserError:
+            raise
+        except Exception as e:
+            raise HyperbrowserError("Delete request failed", original_error=e)

{hyperbrowser-0.10.0 → hyperbrowser-0.12.0}/hyperbrowser/transport/base.py

@@ -55,3 +55,7 @@ class TransportStrategy(ABC):
     @abstractmethod
     def put(self, url: str) -> APIResponse:
         pass
+
+    @abstractmethod
+    def delete(self, url: str) -> APIResponse:
+        pass

{hyperbrowser-0.10.0 → hyperbrowser-0.12.0}/hyperbrowser/transport/sync.py

@@ -45,9 +45,14 @@ class SyncTransport(TransportStrategy):
     def close(self) -> None:
         self.client.close()
 
-    def post(self, url: str, data: Optional[dict] = None) -> APIResponse:
+    def post(
+        self, url: str, data: Optional[dict] = None, files: Optional[dict] = None
+    ) -> APIResponse:
         try:
-            response = self.client.post(url, json=data)
+            if files:
+                response = self.client.post(url, data=data, files=files)
+            else:
+                response = self.client.post(url, json=data)
             return self._handle_response(response)
         except HyperbrowserError:
             raise
@@ -73,3 +78,12 @@ class SyncTransport(TransportStrategy):
             raise
         except Exception as e:
             raise HyperbrowserError("Put request failed", original_error=e)
+
+    def delete(self, url: str) -> APIResponse:
+        try:
+            response = self.client.delete(url)
+            return self._handle_response(response)
+        except HyperbrowserError:
+            raise
+        except Exception as e:
+            raise HyperbrowserError("Delete request failed", original_error=e)

{hyperbrowser-0.10.0 → hyperbrowser-0.12.0}/pyproject.toml

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