smooth-py 0.2.8.dev20251009__py3-none-any.whl → 0.3.0__py3-none-any.whl

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,
@@ -479,7 +486,7 @@ class SmoothClient(BaseClient):
         response_model: If provided, the schema describing the desired output structure.
         url: The starting URL for the task. If not provided, the agent will infer it from the task.
         metadata: A dictionary containing variables or parameters that will be passed to the agent.
-        files: A dictionary of file names to their ids. These files will be passed to the agent.
+        files: A list of file ids to pass to the agent.
         agent: The agent to use for the task.
         max_steps: Maximum number of steps the agent can take (max 64).
         device: Device type for the task. Default is mobile.
@@ -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"]:
@@ -675,7 +686,7 @@ class AsyncTaskHandle:
       await asyncio.sleep(poll_interval)
     raise TimeoutError(f"Task {self.id()} did not complete within {timeout} seconds.")
 
-  async def live_url(self, interactive: bool = True, embed: bool = False, timeout: int | None = None):
+  async 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)
@@ -684,13 +695,13 @@ class AsyncTaskHandle:
     while timeout is None or (time.time() - start_time) < timeout:
       task_response = await self._client._get_task(self.id())
       self._task_response = task_response
-      if task_response.live_url is not None:
+      if self._task_response.live_url:
         return _encode_url(self._task_response.live_url, interactive=interactive, embed=embed)
       await asyncio.sleep(1)
 
     raise TimeoutError(f"Live URL not available for task {self.id()}.")
 
-  async def recording_url(self, timeout: int | None = None):
+  async def recording_url(self, timeout: int | None = None) -> str:
     """Returns the recording URL for the task."""
     if self._task_response and self._task_response.recording_url is not None:
       return self._task_response.recording_url
@@ -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,
@@ -776,7 +799,7 @@ class SmoothAsyncClient(BaseClient):
         response_model: If provided, the schema describing the desired output structure.
         url: The starting URL for the task. If not provided, the agent will infer it from the task.
         metadata: A dictionary containing variables or parameters that will be passed to the agent.
-        files: A dictionary of file names to their url or base64-encoded content to be used by the agent.
+        files: A list of file ids to pass to the agent.
         agent: The agent to use for the task.
         max_steps: Maximum number of steps the agent can take (max 64).
         device: Device type for the task. Default is mobile.
@@ -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,
@@ -827,8 +850,8 @@ class SmoothAsyncClient(BaseClient):
     """Opens an interactive browser instance asynchronously.
 
     Args:
+        profile_id: The profile ID to use for the session. If None, a new profile will be created.
         session_id: (Deprecated, now `profile_id`) The session ID to associate with the browser.
-        profile_id: The profile ID to associate with the browser.
         live_view: Whether to enable live view for the session.
 
     Returns:
@@ -913,11 +936,12 @@ class SmoothAsyncClient(BaseClient):
       if name is None:
         raise ValueError("File name must be provided or the file object must have a 'name' attribute.")
 
-      files = {"file": (Path(name).name, file)}
       if purpose:
         data = {"file_purpose": purpose}
       else:
         data = None
+
+      files = {"file": (Path(name).name, file)}
       response = await self._client.post(f"{self.base_url}/file", files=files, data=data)
       data = self._handle_response(response)
       return UploadFileResponse(**data["r"])

{smooth_py-0.2.8.dev20251009.dist-info → smooth_py-0.3.0.dist-info}/METADATA

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

smooth_py-0.3.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+smooth/__init__.py,sha256=yhlG6m00uOw8umRC7ONImCMCiCkWTV5khj050qfD_V8,38411
+smooth/mcp/__init__.py,sha256=0aJVFi2a8Ah3-5xtgyZ5UMbaaJsBWu2T8QLWoFQITk8,219
+smooth/mcp/server.py,sha256=9SymTD4NOGTMN8P-LNGlvYNvv81yCIZfZeeuhEcAc6s,20068
+smooth_py-0.3.0.dist-info/METADATA,sha256=OJjBn5D0m9N78qyFfiVwN8nd3pSrMNG4V0ubVkMuyfM,7517
+smooth_py-0.3.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+smooth_py-0.3.0.dist-info/RECORD,,

smooth_py-0.2.8.dev20251009.dist-info/RECORD

@@ -1,6 +0,0 @@
-smooth/__init__.py,sha256=4jYb61Mn-G6GIjIi4rXvzXQY7yWHfIOVvik8RNNJWcU,37701
-smooth/mcp/__init__.py,sha256=0aJVFi2a8Ah3-5xtgyZ5UMbaaJsBWu2T8QLWoFQITk8,219
-smooth/mcp/server.py,sha256=9SymTD4NOGTMN8P-LNGlvYNvv81yCIZfZeeuhEcAc6s,20068
-smooth_py-0.2.8.dev20251009.dist-info/METADATA,sha256=pyfY6EygBMxQnAoLN8NxgbqwD28XFzCJe24HXCcuaaY,7529
-smooth_py-0.2.8.dev20251009.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-smooth_py-0.2.8.dev20251009.dist-info/RECORD,,