smooth-py 0.2.8.dev20251008__py3-none-any.whl → 0.3.0.dev20251009__py3-none-any.whl

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 ---
@@ -100,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."""
@@ -137,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."""
@@ -173,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."""
@@ -190,14 +193,6 @@ 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."""
@@ -207,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."""
@@ -351,12 +347,7 @@ class TaskHandle:
 
   def stop(self):
     """Stops the task."""
-    try:
-      response = self._client._client.delete(f"{self._client.base_url}/task/{self._id}")
-      self._handle_response(response)
-    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
+    self._client._delete_task(self._id)
 
   def result(self, timeout: int | None = None, poll_interval: float = 1) -> TaskResponse:
     """Waits for the task to complete and returns the result."""
@@ -452,6 +443,18 @@ class SmoothClient(BaseClient):
       logger.error(f"Request failed: {e}")
       raise ApiError(status_code=0, detail=f"Request failed: {str(e)}") from None
 
+  def _delete_task(self, task_id: str):
+    """Deletes a task."""
+    if not task_id:
+      raise ValueError("Task ID cannot be empty.")
+
+    try:
+      response = self._session.delete(f"{self.base_url}/task/{task_id}")
+      self._handle_response(response)
+    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 run(
     self,
     task: str,
@@ -660,6 +663,10 @@ class AsyncTaskHandle:
     """Returns the task ID."""
     return self._id
 
+  async def stop(self):
+    """Stops the task."""
+    await self._client._delete_task(self._id)
+
   async def result(self, timeout: int | None = None, poll_interval: float = 1) -> TaskResponse:
     """Waits for the task to complete and returns the result."""
     if self._task_response and self._task_response.status not in ["running", "waiting"]:
@@ -749,6 +756,18 @@ class SmoothAsyncClient(BaseClient):
       logger.error(f"Request failed: {e}")
       raise ApiError(status_code=0, detail=f"Request failed: {str(e)}") from None
 
+  async def _delete_task(self, task_id: str):
+    """Deletes a task asynchronously."""
+    if not task_id:
+      raise ValueError("Task ID cannot be empty.")
+
+    try:
+      response = await self._client.delete(f"{self.base_url}/task/{task_id}")
+      self._handle_response(response)
+    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 run(
     self,
     task: str,

{smooth_py-0.2.8.dev20251008.dist-info → smooth_py-0.3.0.dev20251009.dist-info}/METADATA

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

smooth_py-0.3.0.dev20251009.dist-info/RECORD

@@ -0,0 +1,6 @@
+smooth/__init__.py,sha256=0dn5X327eZwSdXklCCV4KN9L9-353GtmzaqjNs9_HeQ,38407
+smooth/mcp/__init__.py,sha256=0aJVFi2a8Ah3-5xtgyZ5UMbaaJsBWu2T8QLWoFQITk8,219
+smooth/mcp/server.py,sha256=9SymTD4NOGTMN8P-LNGlvYNvv81yCIZfZeeuhEcAc6s,20068
+smooth_py-0.3.0.dev20251009.dist-info/METADATA,sha256=cQ0TzuWCfXke3Rf7Zrp01lPUypKpOc2AB2tSQAx-wf8,7529
+smooth_py-0.3.0.dev20251009.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+smooth_py-0.3.0.dev20251009.dist-info/RECORD,,

smooth_py-0.2.8.dev20251008.dist-info/RECORD

@@ -1,6 +0,0 @@
-smooth/__init__.py,sha256=UFvVVla_T5EiOYimpxKQs0D24-F0ZZyRnl1hpJ8YohA,37737
-smooth/mcp/__init__.py,sha256=0aJVFi2a8Ah3-5xtgyZ5UMbaaJsBWu2T8QLWoFQITk8,219
-smooth/mcp/server.py,sha256=9SymTD4NOGTMN8P-LNGlvYNvv81yCIZfZeeuhEcAc6s,20068
-smooth_py-0.2.8.dev20251008.dist-info/METADATA,sha256=Ya-GED2N4TX7XU3MF0Cmy4PaWpwEF45-agZY9daW5F0,7529
-smooth_py-0.2.8.dev20251008.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-smooth_py-0.2.8.dev20251008.dist-info/RECORD,,