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

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

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

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

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

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

@@ -13,13 +13,13 @@ from typing import Any, Literal, Type
 import httpx
 import requests
 from deprecated import deprecated
-from pydantic import BaseModel, ConfigDict, Field, model_validator
+from pydantic import BaseModel, ConfigDict, Field, computed_field, model_validator
 
 # Configure logging
 logger = logging.getLogger("smooth")
 
 
-BASE_URL = "https://api2.circlemind.co/api/"
+BASE_URL = "https://api.smooth.sh/api/"
 
 
 # --- Utils ---
@@ -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(
@@ -98,11 +100,12 @@ class TaskRequest(BaseModel):
   @model_validator(mode="before")
   @classmethod
   def _handle_deprecated_session_id(cls, data: Any) -> Any:
-    if isinstance(data, dict) and "session_id" in data:
+    if isinstance(data, dict) and "session_id" in data and "profile_id" not in data:
       warnings.warn("'session_id' is deprecated, use 'profile_id' instead", DeprecationWarning, stacklevel=2)
       data["profile_id"] = data.pop("session_id")
     return data
 
+  @computed_field(return_type=str | None)
   @property
   def session_id(self):
     """(Deprecated) Returns the session ID."""
@@ -115,6 +118,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."""
@@ -127,11 +138,12 @@ class BrowserSessionRequest(BaseModel):
   @model_validator(mode="before")
   @classmethod
   def _handle_deprecated_session_id(cls, data: Any) -> Any:
-    if isinstance(data, dict) and "session_id" in data:
+    if isinstance(data, dict) and "session_id" in data and "profile_id" not in data:
       warnings.warn("'session_id' is deprecated, use 'profile_id' instead", DeprecationWarning, stacklevel=2)
       data["profile_id"] = data.pop("session_id")
     return data
 
+  @computed_field(return_type=str | None)
   @property
   def session_id(self):
     """(Deprecated) Returns the session ID."""
@@ -144,6 +156,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."""
@@ -155,11 +175,12 @@ class BrowserSessionResponse(BaseModel):
   @model_validator(mode="before")
   @classmethod
   def _handle_deprecated_session_id(cls, data: Any) -> Any:
-    if isinstance(data, dict) and "session_id" in data:
+    if isinstance(data, dict) and "session_id" in data and "profile_id" not in data:
       warnings.warn("'session_id' is deprecated, use 'profile_id' instead", DeprecationWarning, stacklevel=2)
       data["profile_id"] = data.pop("session_id")
     return data
 
+  @computed_field(return_type=str | None)
   @property
   def session_id(self):
     """(Deprecated) Returns the session ID."""
@@ -181,11 +202,12 @@ class BrowserProfilesResponse(BaseModel):
   @model_validator(mode="before")
   @classmethod
   def _handle_deprecated_session_ids(cls, data: Any) -> Any:
-    if isinstance(data, dict) and "session_ids" in data:
+    if isinstance(data, dict) and "session_ids" in data and "profile_ids" not in data:
       warnings.warn("'session_ids' is deprecated, use 'profile_ids' instead", DeprecationWarning, stacklevel=2)
       data["profile_ids"] = data.pop("session_ids")
     return data
 
+  @computed_field(return_type=list[str])
   @property
   def session_ids(self):
     """(Deprecated) Returns the session IDs."""
@@ -198,6 +220,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 +373,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)