smooth-py 0.2.7.dev20251003__tar.gz → 0.2.8.dev20251006__tar.gz

{smooth_py-0.2.7.dev20251003 → smooth_py-0.2.8.dev20251006}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: smooth-py
-Version: 0.2.7.dev20251003
+Version: 0.2.8.dev20251006
 Summary: 
 Author: Luca Pinchetti
 Author-email: luca@circlemind.co

{smooth_py-0.2.7.dev20251003 → smooth_py-0.2.8.dev20251006}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "smooth-py"
-version = "0.2.7.dev20251003"
+version = "0.2.8.dev20251006"
 description = ""
 authors = [
     {name = "Luca Pinchetti",email = "luca@circlemind.co"}

{smooth_py-0.2.7.dev20251003 → smooth_py-0.2.8.dev20251006}/src/smooth/__init__.py

@@ -13,7 +13,7 @@ from typing import Any, Literal, Type
 import httpx
 import requests
 from deprecated import deprecated
-from pydantic import BaseModel, Field, model_validator
+from pydantic import BaseModel, ConfigDict, Field, model_validator
 
 # Configure logging
 logger = logging.getLogger("smooth")
@@ -39,8 +39,10 @@ def _encode_url(url: str, interactive: bool = True, embed: bool = False) -> str:
 class TaskResponse(BaseModel):
   """Task response model."""
 
+  model_config = ConfigDict(extra="allow")
+
   id: str = Field(description="The ID of the task.")
-  status: Literal["waiting", "running", "done", "failed"] = Field(description="The status of the task.")
+  status: Literal["waiting", "running", "done", "failed", "cancelled"] = Field(description="The status of the task.")
   output: Any | None = Field(default=None, description="The output of the task.")
   credits_used: int | None = Field(default=None, description="The amount of credits used to perform the task.")
   device: Literal["desktop", "mobile"] | None = Field(default=None, description="The device type used for the task.")
@@ -89,7 +91,9 @@ class TaskRequest(BaseModel):
   )
   proxy_username: str | None = Field(default=None, description="Proxy server username.")
   proxy_password: str | None = Field(default=None, description="Proxy server password.")
-  experimental_features: dict[str, Any] | None = Field(default=None, description="Experimental features to enable for the task.")
+  experimental_features: dict[str, Any] | None = Field(
+    default=None, description="Experimental features to enable for the task."
+  )
 
   @model_validator(mode="before")
   @classmethod
@@ -145,6 +149,7 @@ class BrowserSessionResponse(BaseModel):
   """Browser session response model."""
 
   profile_id: str = Field(description="The ID of the browser profile associated with the opened browser instance.")
+  live_id: str | None = Field(description="The ID of the live browser session.")
   live_url: str | None = Field(default=None, description="The live URL to interact with the browser session.")
 
   @model_validator(mode="before")
@@ -196,6 +201,7 @@ class BrowserProfilesResponse(BaseModel):
 
 class BrowserSessionsResponse(BrowserProfilesResponse):
   """Response model for listing browser profiles."""
+
   pass
 
 
@@ -309,6 +315,15 @@ class TaskHandle:
     """Returns the task ID."""
     return self._id
 
+  def stop(self):
+    """Stops the task."""
+    try:
+      response = self._client._client.delete(f"{self._client.base_url}/task/{self._id}")
+      self._handle_response(response)
+    except requests.exceptions.RequestException as e:
+      logger.error(f"Request failed: {e}")
+      raise ApiError(status_code=0, detail=f"Request failed: {str(e)}") from None
+
   def result(self, timeout: int | None = None, poll_interval: float = 1) -> TaskResponse:
     """Waits for the task to complete and returns the result."""
     if self._task_response and self._task_response.status not in ["running", "waiting"]:
@@ -417,11 +432,12 @@ class SmoothClient(BaseClient):
     enable_recording: bool = False,
     session_id: str | None = None,
     profile_id: str | None = None,
-    profile_read_only: bool | None = None,
+    profile_read_only: bool = False,
     stealth_mode: bool = False,
     proxy_server: str | None = None,
     proxy_username: str | None = None,
     proxy_password: str | None = None,
+    experimental_features: dict[str, Any] | None = None,
   ) -> TaskHandle:
     """Runs a task and returns a handle to the task.
 
@@ -447,6 +463,7 @@ class SmoothClient(BaseClient):
         proxy_server: Proxy server url to route browser traffic through.
         proxy_username: Proxy server username.
         proxy_password: Proxy server password.
+        experimental_features: Experimental features to enable for the task.
 
     Returns:
         A handle to the running task.
@@ -471,6 +488,7 @@ class SmoothClient(BaseClient):
       proxy_server=proxy_server,
       proxy_username=proxy_username,
       proxy_password=proxy_password,
+      experimental_features=experimental_features,
     )
     initial_response = self._submit_task(payload)
 
@@ -503,6 +521,15 @@ class SmoothClient(BaseClient):
       logger.error(f"Request failed: {e}")
       raise ApiError(status_code=0, detail=f"Request failed: {str(e)}") from None
 
+  def close_session(self, live_id: str):
+    """Closes a browser session."""
+    try:
+      response = self._session.delete(f"{self.base_url}/browser/session/{live_id}")
+      self._handle_response(response)
+    except requests.exceptions.RequestException as e:
+      logger.error(f"Request failed: {e}")
+      raise ApiError(status_code=0, detail=f"Request failed: {str(e)}") from None
+
   def list_profiles(self):
     """Lists all browser profiles for the user.
 
@@ -702,11 +729,12 @@ class SmoothAsyncClient(BaseClient):
     enable_recording: bool = False,
     session_id: str | None = None,
     profile_id: str | None = None,
-    profile_read_only: bool | None = None,
+    profile_read_only: bool = False,
     stealth_mode: bool = False,
     proxy_server: str | None = None,
     proxy_username: str | None = None,
     proxy_password: str | None = None,
+    experimental_features: dict[str, Any] | None = None,
   ) -> AsyncTaskHandle:
     """Runs a task and returns a handle to the task asynchronously.
 
@@ -732,8 +760,7 @@ class SmoothAsyncClient(BaseClient):
         proxy_server: Proxy server url to route browser traffic through.
         proxy_username: Proxy server username.
         proxy_password: Proxy server password.
-        poll_interval: The time in seconds to wait between polling for status.
-        timeout: The maximum time in seconds to wait for the task to complete.
+        experimental_features: Experimental features to enable for the task.
 
     Returns:
         A handle to the running task.
@@ -758,6 +785,7 @@ class SmoothAsyncClient(BaseClient):
       proxy_server=proxy_server,
       proxy_username=proxy_username,
       proxy_password=proxy_password,
+      experimental_features=experimental_features,
     )
 
     initial_response = await self._submit_task(payload)
@@ -790,6 +818,15 @@ class SmoothAsyncClient(BaseClient):
       logger.error(f"Request failed: {e}")
       raise ApiError(status_code=0, detail=f"Request failed: {str(e)}") from None
 
+  async def close_session(self, live_id: str):
+    """Closes a browser session."""
+    try:
+      response = await self._client.delete(f"{self.base_url}/browser/session/{live_id}")
+      self._handle_response(response)
+    except httpx.RequestError as e:
+      logger.error(f"Request failed: {e}")
+      raise ApiError(status_code=0, detail=f"Request failed: {str(e)}") from None
+
   async def list_profiles(self):
     """Lists all browser profiles for the user.