smooth-py 0.2.0__tar.gz → 0.2.1.dev20250911__tar.gz

{smooth_py-0.2.0 → smooth_py-0.2.1.dev20250911}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: smooth-py
-Version: 0.2.0
+Version: 0.2.1.dev20250911
 Summary: 
 Author: Luca Pinchetti
 Author-email: luca@circlemind.co

{smooth_py-0.2.0 → smooth_py-0.2.1.dev20250911}/pyproject.toml

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

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

@@ -49,8 +49,13 @@ class TaskRequest(BaseModel):
 
   task: str = Field(description="The task to run.")
   response_model: dict[str, Any] | None = Field(
-    default=None, description="If provided, the schema describing the desired output structure. Default is None"
+    default=None, description="If provided, the JSON schema describing the desired output structure. Default is None"
   )
+  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.",
+  )
+  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.")
@@ -136,7 +141,7 @@ class BaseClient:
     self.headers = {
       "apikey": self.api_key,
       "Content-Type": "application/json",
-      "User-Agent": "smooth-python-sdk/0.1.1",
+      "User-Agent": "smooth-python-sdk/0.2.0",
     }
 
   def _handle_response(self, response: requests.Response | httpx.Response) -> dict[str, Any]:
@@ -187,7 +192,11 @@ class TaskHandle:
     self._client = client
     self._task_response: TaskResponse | None = None
 
-    self.id = task_id
+    self._id = task_id
+
+  def id(self):
+    """Returns the task ID."""
+    return self._id
 
   def result(self, timeout: int | None = None, poll_interval: float = 1) -> TaskResponse:
     """Waits for the task to complete and returns the result."""
@@ -287,6 +296,8 @@ class SmoothClient(BaseClient):
     self,
     task: str,
     response_model: dict[str, Any] | Type[BaseModel] | None = None,
+    url: str | None = None,
+    metadata: dict[str, str | int | float | bool] | None = None,
     agent: Literal["smooth"] = "smooth",
     max_steps: int = 32,
     device: Literal["desktop", "mobile"] = "mobile",
@@ -305,6 +316,8 @@ class SmoothClient(BaseClient):
     Args:
         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.
         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.
@@ -324,6 +337,8 @@ class SmoothClient(BaseClient):
     payload = TaskRequest(
       task=task,
       response_model=response_model.model_json_schema() if issubclass(response_model, BaseModel) else response_model,
+      url=url,
+      metadata=metadata,
       agent=agent,
       max_steps=max_steps,
       device=device,
@@ -495,6 +510,8 @@ class SmoothAsyncClient(BaseClient):
     self,
     task: str,
     response_model: dict[str, Any] | Type[BaseModel] | None = None,
+    url: str | None = None,
+    metadata: dict[str, str | int | float | bool] | None = None,
     agent: Literal["smooth"] = "smooth",
     max_steps: int = 32,
     device: Literal["desktop", "mobile"] = "mobile",
@@ -513,6 +530,8 @@ class SmoothAsyncClient(BaseClient):
     Args:
         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.
         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.
@@ -534,6 +553,8 @@ class SmoothAsyncClient(BaseClient):
     payload = TaskRequest(
       task=task,
       response_model=response_model.model_json_schema() if issubclass(response_model, BaseModel) else response_model,
+      url=url,
+      metadata=metadata,
       agent=agent,
       max_steps=max_steps,
       device=device,