smooth-py 0.1.0__tar.gz → 0.1.1.post0__tar.gz

{smooth_py-0.1.0 → smooth_py-0.1.1.post0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.3
 Name: smooth-py
-Version: 0.1.0
+Version: 0.1.1.post0
 Summary: 
 Author: Luca Pinchetti
-Author-email: pincoluca1@gmail.com
+Author-email: luca@circlemind.co
 Requires-Python: >=3.10
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
@@ -46,10 +46,10 @@ The SDK requires an API key for authentication. You can provide the API key in t
 
 2.  **As an environment variable**:
 
-    Set the `SMOOTH_API_KEY` environment variable, and the client will automatically use it.
+    Set the `CIRCLEMIND_API_KEY` environment variable, and the client will automatically use it.
 
     ```bash
-    export SMOOTH_API_KEY="YOUR_API_KEY"
+    export CIRCLEMIND_API_KEY="YOUR_API_KEY"
     ```
 
     ```python

{smooth_py-0.1.0 → smooth_py-0.1.1.post0}/README.md

@@ -30,10 +30,10 @@ The SDK requires an API key for authentication. You can provide the API key in t
 
 2.  **As an environment variable**:
 
-    Set the `SMOOTH_API_KEY` environment variable, and the client will automatically use it.
+    Set the `CIRCLEMIND_API_KEY` environment variable, and the client will automatically use it.
 
     ```bash
-    export SMOOTH_API_KEY="YOUR_API_KEY"
+    export CIRCLEMIND_API_KEY="YOUR_API_KEY"
     ```
 
     ```python

{smooth_py-0.1.0 → smooth_py-0.1.1.post0}/pyproject.toml

@@ -1,9 +1,9 @@
 [project]
 name = "smooth-py"
-version = "0.1.0"
+version = "0.1.1.post0"
 description = ""
 authors = [
-    {name = "Luca Pinchetti",email = "pincoluca1@gmail.com"}
+    {name = "Luca Pinchetti",email = "luca@circlemind.co"}
 ]
 readme = "README.md"
 requires-python = ">=3.10"

{smooth_py-0.1.0 → smooth_py-0.1.1.post0}/src/smooth/__init__.py

@@ -20,23 +20,17 @@ BASE_URL = "https://api2.circlemind.co/api/"
 # These models define the data structures for API requests and responses.
 
 
-class TaskData(BaseModel):
-  """Task data model."""
-
-  result: Any | None = Field(default=None, description="The result of the task if successful.")
-  error: str | None = Field(default=None, description="Error message if the task failed.")
-  credits_used: int | None = Field(default=None, description="The amount of credits used to perform the task.")
-  src: str | None = Field(default=None, description="")
-
-
 class TaskResponse(BaseModel):
   """Task response model."""
 
   model_config = ConfigDict(extra="forbid")
 
   id: str = Field(description="The ID of the task.")
-  status: str = Field(default="RUNNING", description="The status of the task.")
-  data: TaskData = Field(default_factory=lambda: TaskData(), description="The data associated with the task.")
+  status: Literal["waiting", "running", "done", "failed"] = Field(description="The status of the task.")
+  result: Any | None = Field(default=None, description="The result of the task if successful.")
+  error: str | None = Field(default=None, description="Error message if the task failed.")
+  credits_used: int | None = Field(default=None, description="The amount of credits used to perform the task.")
+  src: str | None = Field(default=None, description="")
 
 
 class TaskRequest(BaseModel):
@@ -53,7 +47,13 @@ class TaskRequest(BaseModel):
     description="(optional) Browser session ID to use. Each session maintains its own state, such as login credentials.",
   )
   stealth_mode: bool = Field(default=False, description="(optional) Run the browser in stealth mode.")
-  proxy_server: Optional[str] = Field(default=None, description="(optional) Proxy server URL.")
+  proxy_server: Optional[str] = Field(
+    default=None,
+    description=(
+      "(optional) Proxy server url to route browser traffic through."
+      " Must include the protocol to use (e.g. http:// or https://)"
+    ),
+  )
   proxy_username: Optional[str] = Field(default=None, description="(optional) Proxy server username.")
   proxy_password: Optional[str] = Field(default=None, description="(optional) Proxy server password.")
 
@@ -109,10 +109,10 @@ class BaseClient:
     """Initializes the base client."""
     # Try to get API key from environment if not provided
     if not api_key:
-      api_key = os.getenv("SMOOTH_API_KEY")
+      api_key = os.getenv("CIRCLEMIND_API_KEY")
 
     if not api_key:
-      raise ValueError("API key is required. Provide it directly or set SMOOTH_API_KEY environment variable.")
+      raise ValueError("API key is required. Provide it directly or set CIRCLEMIND_API_KEY environment variable.")
 
     if not base_url:
       raise ValueError("Base URL cannot be empty.")
@@ -120,12 +120,12 @@ class BaseClient:
     self.api_key = api_key
     self.base_url = f"{base_url.rstrip('/')}/{api_version}"
     self.headers = {
-      "Authorization": f"Bearer {self.api_key}",
+      "apikey": self.api_key,
       "Content-Type": "application/json",
       "User-Agent": "smooth-python-sdk/0.1.0",
     }
 
-  def _handle_response(self, response: Union[requests.Response, httpx.Response]) -> dict:
+  def _handle_response(self, response: Union[requests.Response, httpx.Response]) -> dict[str, Any]:
     """Handles HTTP responses and raises exceptions for errors."""
     if 200 <= response.status_code < 300:
       try:
@@ -135,6 +135,7 @@ class BaseClient:
         raise ApiError(status_code=response.status_code, detail="Invalid JSON response from server") from None
 
     # Handle error responses
+    error_data = None
     try:
       error_data = response.json()
       detail = error_data.get("detail", response.text)
@@ -143,7 +144,7 @@ class BaseClient:
 
     logger.error(f"API error: {response.status_code} - {detail}")
     raise ApiError(
-      status_code=response.status_code, detail=detail, response_data=error_data if "error_data" in locals() else None
+      status_code=response.status_code, detail=detail, response_data=error_data
     )
 
 
@@ -163,7 +164,7 @@ class SyncClient(BaseClient):
     """Enters the synchronous context manager."""
     return self
 
-  def __exit__(self, exc_type, exc_val, exc_tb):
+  def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any):
     """Exits the synchronous context manager."""
     self.close()
 
@@ -267,7 +268,7 @@ class SyncClient(BaseClient):
     Raises:
         ApiException: If the API request fails.
     """
-    params = {}
+    params: dict[str, Any] = {}
     if session_id:
       params["session_id"] = session_id
     if session_name:
@@ -314,7 +315,7 @@ class AsyncClient(BaseClient):
     """Enters the asynchronous context manager."""
     return self
 
-  async def __aexit__(self, exc_type, exc_val, exc_tb):
+  async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any):
     """Exits the asynchronous context manager."""
     await self.close()
 
@@ -416,7 +417,7 @@ class AsyncClient(BaseClient):
     Raises:
         ApiException: If the API request fails.
     """
-    params = {}
+    params: dict[str, Any] = {}
     if session_id:
       params["session_id"] = session_id
     if session_name:
@@ -430,6 +431,23 @@ class AsyncClient(BaseClient):
       logger.error(f"Request failed: {e}")
       raise ApiError(status_code=0, detail=f"Request failed: {str(e)}") from None
 
+  async def list_sessions(self) -> BrowserSessionsResponse:
+    """Lists all browser sessions for the user.
+
+    Returns:
+        A list of existing browser sessions.
+
+    Raises:
+        ApiException: If the API request fails.
+    """
+    try:
+      response = await self._client.get(f"{self.base_url}/browser/session")
+      data = self._handle_response(response)
+      return BrowserSessionsResponse(**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 close(self):
     """Closes the async client session."""
     await self._client.aclose()