Add proxy update (#61)

Author: AHarmlessPyro

hyperbrowser/client/managers/async_manager/session.py

@@ -15,6 +15,7 @@
     SessionEventLogListResponse,
     SessionEventLog,
     UpdateSessionProfileParams,
+    UpdateSessionProxyParams,
     SessionGetParams,
 )
 
@@ -183,6 +184,20 @@ async def update_profile_params(
         )
         return BasicResponse(**response.data)
 
+    async def update_proxy_params(
+        self,
+        id: str,
+        params: UpdateSessionProxyParams,
+    ) -> BasicResponse:
+        response = await self._client.transport.put(
+            self._client._build_url(f"/session/{id}/update"),
+            data={
+                "type": "proxy",
+                "params": params.model_dump(exclude_none=True, by_alias=True),
+            },
+        )
+        return BasicResponse(**response.data)
+
     def _warn_update_profile_params_boolean_deprecated(self) -> None:
         if SessionManager._has_warned_update_profile_params_boolean_deprecated:
             return

hyperbrowser/client/managers/sync_manager/session.py

@@ -13,8 +13,8 @@
     UploadFileResponse,
     SessionEventLogListParams,
     SessionEventLogListResponse,
-    SessionEventLog,
     UpdateSessionProfileParams,
+    UpdateSessionProxyParams,
     SessionGetParams,
 )
 
@@ -179,6 +179,20 @@ def update_profile_params(
         )
         return BasicResponse(**response.data)
 
+    def update_proxy_params(
+        self,
+        id: str,
+        params: UpdateSessionProxyParams,
+    ) -> BasicResponse:
+        response = self._client.transport.put(
+            self._client._build_url(f"/session/{id}/update"),
+            data={
+                "type": "proxy",
+                "params": params.model_dump(exclude_none=True, by_alias=True),
+            },
+        )
+        return BasicResponse(**response.data)
+
     def _warn_update_profile_params_boolean_deprecated(self) -> None:
         if SessionManager._has_warned_update_profile_params_boolean_deprecated:
             return

hyperbrowser/models/__init__.py

@@ -234,6 +234,8 @@
     UploadFileResponse,
     ImageCaptchaParam,
     UpdateSessionProfileParams,
+    UpdateSessionProxyLocationParams,
+    UpdateSessionProxyParams,
 )
 from .sandbox import (
     SandboxStatus,
@@ -485,6 +487,8 @@
     "UploadFileResponse",
     "ImageCaptchaParam",
     "UpdateSessionProfileParams",
+    "UpdateSessionProxyLocationParams",
+    "UpdateSessionProxyParams",
     # sandbox
     "SandboxStatus",
     "SandboxRegion",

hyperbrowser/models/session.py

@@ -1,6 +1,5 @@
 from datetime import datetime
 from typing import Any, List, Literal, Optional, Union, Dict
-from .computer_action import ComputerActionParams, ComputerActionResponse
 
 from pydantic import BaseModel, ConfigDict, Field, field_validator
 
@@ -70,6 +69,37 @@ class UpdateSessionProfileParams(BaseModel):
     )
 
 
+class UpdateSessionProxyLocationParams(BaseModel):
+    """
+    Managed proxy geolocation overrides for a running session.
+    """
+
+    model_config = ConfigDict(
+        populate_by_alias=True,
+    )
+
+    country: Optional[Country] = Field(default=None, serialization_alias="country")
+    state: Optional[State] = Field(default=None, serialization_alias="state")
+    city: Optional[str] = Field(default=None, serialization_alias="city")
+
+
+class UpdateSessionProxyParams(BaseModel):
+    """
+    Parameters for enabling, disabling, or reconfiguring a session proxy.
+    """
+
+    model_config = ConfigDict(
+        populate_by_alias=True,
+    )
+
+    enabled: bool = Field(serialization_alias="enabled")
+    static_ip_id: Optional[str] = Field(default=None, serialization_alias="staticIpId")
+    location: Optional[UpdateSessionProxyLocationParams] = Field(
+        default=None,
+        serialization_alias="location",
+    )
+
+
 class SessionLaunchState(BaseModel):
     model_config = ConfigDict(
         populate_by_alias=True,

pyproject.toml

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