smooth-py 0.2.8.dev20251003__tar.gz → 0.2.8.dev20251008__tar.gz

{smooth_py-0.2.8.dev20251003 → smooth_py-0.2.8.dev20251008}/PKG-INFO

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

{smooth_py-0.2.8.dev20251003 → smooth_py-0.2.8.dev20251008}/pyproject.toml

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

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

@@ -38,10 +38,11 @@ def _encode_url(url: str, interactive: bool = True, embed: bool = False) -> str:
 
 class TaskResponse(BaseModel):
   """Task response model."""
-  model_config = ConfigDict(extra='allow')
+
+  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.")
@@ -52,6 +53,8 @@ class TaskResponse(BaseModel):
 class TaskRequest(BaseModel):
   """Run task request model."""
 
+  model_config = ConfigDict(extra="allow")
+
   task: str = Field(description="The task to run.")
   response_model: dict[str, Any] | None = Field(
     default=None, description="If provided, the JSON schema describing the desired output structure. Default is None"
@@ -64,7 +67,7 @@ class TaskRequest(BaseModel):
     default=None, description="A dictionary containing variables or parameters that will be passed to the agent."
   )
   files: list[str] | None = Field(default=None, description="A list of file ids to pass to the agent.")
-  agent: Literal["smooth"] = Field(default="smooth", description="The agent to use for the task.")
+  agent: Literal["smooth", "smooth-lite"] = Field(default="smooth", description="The agent to use for the task.")
   max_steps: int = Field(default=32, ge=2, le=128, description="Maximum number of steps the agent can take (min 2, max 128).")
   device: Literal["desktop", "mobile"] = Field(default="mobile", description="Device type for the task. Default is mobile.")
   allowed_urls: list[str] | None = Field(
@@ -114,6 +117,14 @@ class TaskRequest(BaseModel):
     warnings.warn("'session_id' is deprecated, use 'profile_id' instead", DeprecationWarning, stacklevel=2)
     self.profile_id = value
 
+  def model_dump(self, **kwargs) -> dict[str, Any]:
+    """Dump model to dict, including deprecated session_id for retrocompatibility."""
+    data = super().model_dump(**kwargs)
+    # Add deprecated session_id field for retrocompatibility
+    if "profile_id" in data:
+      data["session_id"] = data["profile_id"]
+    return data
+
 
 class BrowserSessionRequest(BaseModel):
   """Request model for creating a browser session."""
@@ -143,11 +154,20 @@ class BrowserSessionRequest(BaseModel):
     warnings.warn("'session_id' is deprecated, use 'profile_id' instead", DeprecationWarning, stacklevel=2)
     self.profile_id = value
 
+  def model_dump(self, **kwargs) -> dict[str, Any]:
+    """Dump model to dict, including deprecated session_id for retrocompatibility."""
+    data = super().model_dump(**kwargs)
+    # Add deprecated session_id field for retrocompatibility
+    if "profile_id" in data:
+      data["session_id"] = data["profile_id"]
+    return data
+
 
 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")
@@ -170,6 +190,14 @@ class BrowserSessionResponse(BaseModel):
     warnings.warn("'session_id' is deprecated, use 'profile_id' instead", DeprecationWarning, stacklevel=2)
     self.profile_id = value
 
+  def model_dump(self, **kwargs) -> dict[str, Any]:
+    """Dump model to dict, including deprecated session_id for retrocompatibility."""
+    data = super().model_dump(**kwargs)
+    # Add deprecated session_id field for retrocompatibility
+    if "profile_id" in data:
+      data["session_id"] = data["profile_id"]
+    return data
+
 
 class BrowserProfilesResponse(BaseModel):
   """Response model for listing browser profiles."""
@@ -196,9 +224,18 @@ class BrowserProfilesResponse(BaseModel):
     warnings.warn("'session_ids' is deprecated, use 'profile_ids' instead", DeprecationWarning, stacklevel=2)
     self.profile_ids = value
 
+  def model_dump(self, **kwargs) -> dict[str, Any]:
+    """Dump model to dict, including deprecated session_ids for retrocompatibility."""
+    data = super().model_dump(**kwargs)
+    # Add deprecated session_ids field for retrocompatibility
+    if "profile_ids" in data:
+      data["session_ids"] = data["profile_ids"]
+    return data
+
 
 class BrowserSessionsResponse(BrowserProfilesResponse):
   """Response model for listing browser profiles."""
+
   pass
 
 
@@ -312,6 +349,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"]:
@@ -331,7 +377,7 @@ class TaskHandle:
       time.sleep(poll_interval)
     raise TimeoutError(f"Task {self.id()} did not complete within {timeout} seconds.")
 
-  def live_url(self, interactive: bool = True, embed: bool = False, timeout: int | None = None):
+  def live_url(self, interactive: bool = False, embed: bool = False, timeout: int | None = None):
     """Returns the live URL for the task."""
     if self._task_response and self._task_response.live_url:
       return _encode_url(self._task_response.live_url, interactive=interactive, embed=embed)
@@ -509,6 +555,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.
 
@@ -797,6 +852,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.