smooth-py 0.2.1.post0.dev20250911__tar.gz → 0.2.3__tar.gz

{smooth_py-0.2.1.post0.dev20250911 → smooth_py-0.2.3}/PKG-INFO

@@ -1,19 +1,20 @@
 Metadata-Version: 2.3
 Name: smooth-py
-Version: 0.2.1.post0.dev20250911
+Version: 0.2.3
 Summary: 
 Author: Luca Pinchetti
 Author-email: luca@circlemind.co
-Requires-Python: >=3.10
+Requires-Python: >=3.10,<4.0
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Provides-Extra: mcp
-Requires-Dist: fastmcp (>=2.0.0,<3.0.0) ; extra == "mcp"
+Requires-Dist: fastmcp (>=2.12.0,<3.0.0) ; extra == "mcp"
 Requires-Dist: httpx (>=0.28.1,<0.29.0)
 Requires-Dist: pydantic (>=2.11.7,<3.0.0)
+Requires-Dist: requests (>=2.32.5,<3.0.0)
 Description-Content-Type: text/markdown
 
 # Smooth Python SDK

{smooth_py-0.2.1.post0.dev20250911 → smooth_py-0.2.3}/pyproject.toml

@@ -1,20 +1,21 @@
 [project]
 name = "smooth-py"
-version = "0.2.1.post0.dev20250911"
+version = "0.2.3"
 description = ""
 authors = [
     {name = "Luca Pinchetti",email = "luca@circlemind.co"}
 ]
 readme = "README.md"
-requires-python = ">=3.10"
+requires-python = ">=3.10,<4.0"
 dependencies = [
     "pydantic (>=2.11.7,<3.0.0)",
-    "httpx (>=0.28.1,<0.29.0)"
+    "httpx (>=0.28.1,<0.29.0)",
+    "requests (>=2.32.5,<3.0.0)"
 ]
 
 [project.optional-dependencies]
 mcp = [
-    "fastmcp (>=2.0.0,<3.0.0)"
+    "fastmcp (>=2.12.0,<3.0.0)"
 ]
 
 [tool.poetry]

{smooth_py-0.2.1.post0.dev20250911 → smooth_py-0.2.3}/src/smooth/__init__.py

@@ -1,6 +1,7 @@
 """Smooth python SDK."""
 
 import asyncio
+import io
 import logging
 import os
 import time
@@ -53,9 +54,14 @@ class TaskRequest(BaseModel):
   )
   url: str | None = Field(
     default=None,
-    description="(Optional) The starting URL for the task. If not provided, the agent will infer it from the task.",
+    description="The starting URL for the task. If not provided, the agent will infer it from the task.",
+  )
+  metadata: dict[str, str | int | float | bool] | None = Field(
+    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 dictionary of file ids to pass to the agent."
   )
-  metadata: dict[str, str | int | float | bool] | None = Field(default=None, description="Optional metadata for the task.")
   agent: Literal["smooth"] = 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.")
@@ -98,6 +104,11 @@ class BrowserSessionsResponse(BaseModel):
   session_ids: list[str] = Field(description="The IDs of the browser sessions.")
 
 
+class UploadFileResponse(BaseModel):
+  """Response model for listing browser sessions."""
+  id: str = Field(description="The ID assigned to the uploaded file.")
+
+
 # --- Exception Handling ---
 
 
@@ -140,7 +151,6 @@ class BaseClient:
     self.base_url = f"{base_url.rstrip('/')}/{api_version}"
     self.headers = {
       "apikey": self.api_key,
-      "Content-Type": "application/json",
       "User-Agent": "smooth-python-sdk/0.2.0",
     }
 
@@ -279,6 +289,19 @@ class SmoothClient(BaseClient):
       logger.error(f"Request failed: {e}")
       raise ApiError(status_code=0, detail=f"Request failed: {str(e)}") from None
 
+  def _upload_file(self, file: io.IOBase, name: str) -> UploadFileResponse:
+    """Uploads a file and returns the assigned file id."""
+    try:
+      files = {
+        "file": (name, file)
+      }
+      response = self._session.post(f"{self.base_url}/upload_file", files=files)
+      data = self._handle_response(response)
+      return UploadFileResponse(**data["r"])
+    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 _get_task(self, task_id: str) -> TaskResponse:
     """Retrieves the status and result of a task."""
     if not task_id:
@@ -298,6 +321,7 @@ class SmoothClient(BaseClient):
     response_model: dict[str, Any] | Type[BaseModel] | None = None,
     url: str | None = None,
     metadata: dict[str, str | int | float | bool] | None = None,
+    files: list[str] | None = None,
     agent: Literal["smooth"] = "smooth",
     max_steps: int = 32,
     device: Literal["desktop", "mobile"] = "mobile",
@@ -317,7 +341,8 @@ class SmoothClient(BaseClient):
         task: The task to run.
         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: Optional metadata for 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.
         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.
@@ -339,6 +364,7 @@ class SmoothClient(BaseClient):
       response_model=response_model.model_json_schema() if issubclass(response_model, BaseModel) else response_model,
       url=url,
       metadata=metadata,
+      files=files,
       agent=agent,
       max_steps=max_steps,
       device=device,
@@ -399,10 +425,26 @@ class SmoothClient(BaseClient):
     try:
       response = self._session.delete(f"{self.base_url}/browser/session/{session_id}")
       self._handle_response(response)
-    except httpx.RequestError as e:
+    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 upload_file(self, file: io.IOBase, name: str) -> UploadFileResponse:
+    """Upload a file and return the file ID.
+
+    Args:
+        file: File object to be uploaded.
+        name: The name to assign to the uploaded file.
+
+    Returns:
+        The file ID assigned to the uploaded file.
+
+    Raises:
+        ValueError: If the file doesn't exist or can't be read.
+        ApiError: If the API request fails.
+    """
+    return self._upload_file(file=file, name=name)
+
 
 # --- Asynchronous Client ---
 
@@ -497,6 +539,19 @@ class SmoothAsyncClient(BaseClient):
       logger.error(f"Request failed: {e}")
       raise ApiError(status_code=0, detail=f"Request failed: {str(e)}") from None
 
+  async def _upload_file(self, file: io.IOBase, name: str) -> UploadFileResponse:
+    """Uploads a file and returns the assigned file id."""
+    try:
+      files = {
+        "file": (name, file)
+      }
+      response = await self._client.post(f"{self.base_url}/upload_file", files=files)
+      data = self._handle_response(response)
+      return UploadFileResponse(**data["r"])
+    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 _get_task(self, task_id: str) -> TaskResponse:
     """Retrieves the status and result of a task asynchronously."""
     if not task_id:
@@ -516,6 +571,7 @@ class SmoothAsyncClient(BaseClient):
     response_model: dict[str, Any] | Type[BaseModel] | None = None,
     url: str | None = None,
     metadata: dict[str, str | int | float | bool] | None = None,
+    files: list[str] | None = None,
     agent: Literal["smooth"] = "smooth",
     max_steps: int = 32,
     device: Literal["desktop", "mobile"] = "mobile",
@@ -535,7 +591,8 @@ class SmoothAsyncClient(BaseClient):
         task: The task to run.
         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: Optional metadata for 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.
         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.
@@ -559,6 +616,7 @@ class SmoothAsyncClient(BaseClient):
       response_model=response_model.model_json_schema() if issubclass(response_model, BaseModel) else response_model,
       url=url,
       metadata=metadata,
+      files=files,
       agent=agent,
       max_steps=max_steps,
       device=device,
@@ -623,6 +681,22 @@ class SmoothAsyncClient(BaseClient):
       logger.error(f"Request failed: {e}")
       raise ApiError(status_code=0, detail=f"Request failed: {str(e)}") from None
 
+  async def upload_file(self, file: io.IOBase, name: str) -> UploadFileResponse:
+    """Upload a file and return the file ID.
+
+    Args:
+        file: File object to be uploaded.
+        name: The name to assign to the uploaded file.
+
+    Returns:
+        The file ID assigned to the uploaded file.
+
+    Raises:
+        ValueError: If the file doesn't exist or can't be read.
+        ApiError: If the API request fails.
+    """
+    return await self._upload_file(file=file, name=name)
+
   async def close(self):
     """Closes the async client session."""
     await self._client.aclose()
@@ -640,6 +714,7 @@ __all__ = [
   "BrowserSessionRequest",
   "BrowserSessionResponse",
   "BrowserSessionsResponse",
+  "UploadFileResponse",
   "ApiError",
   "TimeoutError",
 ]