smooth-py 0.2.8.dev20251006__tar.gz → 0.2.8.dev20251008__tar.gz

{smooth_py-0.2.8.dev20251006 → smooth_py-0.2.8.dev20251008}/PKG-INFO

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

{smooth_py-0.2.8.dev20251006 → smooth_py-0.2.8.dev20251008}/pyproject.toml

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

{smooth_py-0.2.8.dev20251006 → smooth_py-0.2.8.dev20251008}/src/smooth/__init__.py

@@ -53,6 +53,8 @@ class TaskResponse(BaseModel):
 class TaskRequest(BaseModel):
   """Run task request model."""
 
+  model_config = ConfigDict(extra="allow")
+
   task: str = Field(description="The task to run.")
   response_model: dict[str, Any] | None = Field(
     default=None, description="If provided, the JSON schema describing the desired output structure. Default is None"
@@ -65,7 +67,7 @@ class TaskRequest(BaseModel):
     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 list of file ids to pass to the agent.")
-  agent: Literal["smooth"] = Field(default="smooth", description="The agent to use for the task.")
+  agent: Literal["smooth", "smooth-lite"] = 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.")
   allowed_urls: list[str] | None = Field(
@@ -115,6 +117,14 @@ class TaskRequest(BaseModel):
     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]:
+    """Dump model to dict, including deprecated session_id for retrocompatibility."""
+    data = super().model_dump(**kwargs)
+    # Add deprecated session_id field for retrocompatibility
+    if "profile_id" in data:
+      data["session_id"] = data["profile_id"]
+    return data
+
 
 class BrowserSessionRequest(BaseModel):
   """Request model for creating a browser session."""
@@ -144,6 +154,14 @@ class BrowserSessionRequest(BaseModel):
     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]:
+    """Dump model to dict, including deprecated session_id for retrocompatibility."""
+    data = super().model_dump(**kwargs)
+    # Add deprecated session_id field for retrocompatibility
+    if "profile_id" in data:
+      data["session_id"] = data["profile_id"]
+    return data
+
 
 class BrowserSessionResponse(BaseModel):
   """Browser session response model."""
@@ -172,6 +190,14 @@ class BrowserSessionResponse(BaseModel):
     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]:
+    """Dump model to dict, including deprecated session_id for retrocompatibility."""
+    data = super().model_dump(**kwargs)
+    # Add deprecated session_id field for retrocompatibility
+    if "profile_id" in data:
+      data["session_id"] = data["profile_id"]
+    return data
+
 
 class BrowserProfilesResponse(BaseModel):
   """Response model for listing browser profiles."""
@@ -198,6 +224,14 @@ class BrowserProfilesResponse(BaseModel):
     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]:
+    """Dump model to dict, including deprecated session_ids for retrocompatibility."""
+    data = super().model_dump(**kwargs)
+    # Add deprecated session_ids field for retrocompatibility
+    if "profile_ids" in data:
+      data["session_ids"] = data["profile_ids"]
+    return data
+
 
 class BrowserSessionsResponse(BrowserProfilesResponse):
   """Response model for listing browser profiles."""
@@ -343,7 +377,7 @@ class TaskHandle:
       time.sleep(poll_interval)
     raise TimeoutError(f"Task {self.id()} did not complete within {timeout} seconds.")
 
-  def live_url(self, interactive: bool = True, embed: bool = False, timeout: int | None = None):
+  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)