smooth-py 0.3.0.dev20251009__tar.gz → 0.3.0.dev20251014__tar.gz

{smooth_py-0.3.0.dev20251009 → smooth_py-0.3.0.dev20251014}/PKG-INFO

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

{smooth_py-0.3.0.dev20251009 → smooth_py-0.3.0.dev20251014}/pyproject.toml

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

{smooth_py-0.3.0.dev20251009 → smooth_py-0.3.0.dev20251014}/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
@@ -486,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.
@@ -510,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,
@@ -686,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)
@@ -695,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
@@ -799,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.
@@ -823,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,
@@ -850,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:
@@ -936,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"])