smooth-py 0.2.8.dev20251009__tar.gz → 0.3.0.dev20251013__tar.gz

{smooth_py-0.2.8.dev20251009 → smooth_py-0.3.0.dev20251013}/PKG-INFO

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

{smooth_py-0.2.8.dev20251009 → smooth_py-0.3.0.dev20251013}/pyproject.toml

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

{smooth_py-0.2.8.dev20251009 → smooth_py-0.3.0.dev20251013}/src/smooth/__init__.py

@@ -113,12 +113,12 @@ class TaskRequest(BaseModel):
     return self.profile_id
 
   @session_id.setter
-  def session_id(self, value):
+  def session_id(self, value: str | None):
     """(Deprecated) Sets the session ID."""
     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]:
+  def model_dump(self, **kwargs: Any) -> 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
@@ -151,12 +151,12 @@ class BrowserSessionRequest(BaseModel):
     return self.profile_id
 
   @session_id.setter
-  def session_id(self, value):
+  def session_id(self, value: str | None):
     """(Deprecated) Sets the session ID."""
     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]:
+  def model_dump(self, **kwargs: Any) -> 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
@@ -188,7 +188,7 @@ class BrowserSessionResponse(BaseModel):
     return self.profile_id
 
   @session_id.setter
-  def session_id(self, value):
+  def session_id(self, value: str):
     """(Deprecated) Sets the session ID."""
     warnings.warn("'session_id' is deprecated, use 'profile_id' instead", DeprecationWarning, stacklevel=2)
     self.profile_id = value
@@ -215,12 +215,12 @@ class BrowserProfilesResponse(BaseModel):
     return self.profile_ids
 
   @session_ids.setter
-  def session_ids(self, value):
+  def session_ids(self, value: list[str]):
     """(Deprecated) Sets the session IDs."""
     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]:
+  def model_dump(self, **kwargs: Any) -> 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
@@ -347,12 +347,7 @@ class TaskHandle:
 
   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
+    self._client._delete_task(self._id)
 
   def result(self, timeout: int | None = None, poll_interval: float = 1) -> TaskResponse:
     """Waits for the task to complete and returns the result."""
@@ -448,6 +443,18 @@ class SmoothClient(BaseClient):
       logger.error(f"Request failed: {e}")
       raise ApiError(status_code=0, detail=f"Request failed: {str(e)}") from None
 
+  def _delete_task(self, task_id: str):
+    """Deletes a task."""
+    if not task_id:
+      raise ValueError("Task ID cannot be empty.")
+
+    try:
+      response = self._session.delete(f"{self.base_url}/task/{task_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 run(
     self,
     task: str,
@@ -503,7 +510,7 @@ class SmoothClient(BaseClient):
     """
     payload = TaskRequest(
       task=task,
-      response_model=response_model.model_json_schema() if issubclass(response_model, BaseModel) else response_model,
+      response_model=response_model if isinstance(response_model, dict | None) else response_model.model_json_schema(),
       url=url,
       metadata=metadata,
       files=files,
@@ -656,6 +663,10 @@ class AsyncTaskHandle:
     """Returns the task ID."""
     return self._id
 
+  async def stop(self):
+    """Stops the task."""
+    await self._client._delete_task(self._id)
+
   async 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"]:
@@ -685,7 +696,7 @@ class AsyncTaskHandle:
       task_response = await self._client._get_task(self.id())
       self._task_response = task_response
       if task_response.live_url is not None:
-        return _encode_url(self._task_response.live_url, interactive=interactive, embed=embed)
+        return _encode_url(task_response.live_url, interactive=interactive, embed=embed)
       await asyncio.sleep(1)
 
     raise TimeoutError(f"Live URL not available for task {self.id()}.")
@@ -745,6 +756,18 @@ class SmoothAsyncClient(BaseClient):
       logger.error(f"Request failed: {e}")
       raise ApiError(status_code=0, detail=f"Request failed: {str(e)}") from None
 
+  async def _delete_task(self, task_id: str):
+    """Deletes a task asynchronously."""
+    if not task_id:
+      raise ValueError("Task ID cannot be empty.")
+
+    try:
+      response = await self._client.delete(f"{self.base_url}/task/{task_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 run(
     self,
     task: str,
@@ -800,7 +823,7 @@ class SmoothAsyncClient(BaseClient):
     """
     payload = TaskRequest(
       task=task,
-      response_model=response_model.model_json_schema() if issubclass(response_model, BaseModel) else response_model,
+      response_model=response_model if isinstance(response_model, dict | None) else response_model.model_json_schema(),
       url=url,
       metadata=metadata,
       files=files,