smooth-py 0.2.4__tar.gz → 0.2.5.dev20250922__tar.gz

{smooth_py-0.2.4 → smooth_py-0.2.5.dev20250922}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: smooth-py
-Version: 0.2.4
+Version: 0.2.5.dev20250922
 Summary: 
 Author: Luca Pinchetti
 Author-email: luca@circlemind.co

{smooth_py-0.2.4 → smooth_py-0.2.5.dev20250922}/pyproject.toml

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

{smooth_py-0.2.4 → smooth_py-0.2.5.dev20250922}/src/smooth/__init__.py

@@ -6,6 +6,7 @@ import logging
 import os
 import time
 import urllib.parse
+from pathlib import Path
 from typing import Any, Literal, Type
 
 import httpx
@@ -60,7 +61,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 dictionary of file ids to pass to the agent."
+    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.")
   max_steps: int = Field(default=32, ge=2, le=128, description="Maximum number of steps the agent can take (min 2, max 128).")
@@ -151,7 +152,7 @@ class BaseClient:
     self.base_url = f"{base_url.rstrip('/')}/{api_version}"
     self.headers = {
       "apikey": self.api_key,
-      "User-Agent": "smooth-python-sdk/0.2.4",
+      "User-Agent": "smooth-python-sdk/0.2.5",
     }
 
   def _handle_response(self, response: requests.Response | httpx.Response) -> dict[str, Any]:
@@ -416,12 +417,13 @@ class SmoothClient(BaseClient):
       logger.error(f"Request failed: {e}")
       raise ApiError(status_code=0, detail=f"Request failed: {str(e)}") from None
 
-  def upload_file(self, file: io.IOBase, name: str) -> UploadFileResponse:
+  def upload_file(self, file: io.IOBase, name: str | None = None, purpose: str | None = None) -> UploadFileResponse:
     """Upload a file and return the file ID.
 
     Args:
         file: File object to be uploaded.
-        name: The name to assign to the uploaded file.
+        name: A custom name to assign to the uploaded file.
+        purpose: An optional short description of the file to describe its purpose.
 
     Returns:
         The file ID assigned to the uploaded file.
@@ -431,10 +433,21 @@ class SmoothClient(BaseClient):
         ApiError: If the API request fails.
     """
     try:
+      name = name or getattr(file, "name", None)
+      if name is None:
+        raise ValueError("File name must be provided or the file object must have a 'name' attribute.")
+
+      if purpose:
+        data = {
+          "file_purpose": purpose
+        }
+      else:
+        data = None
+
       files = {
-        "file": (name, file)
+        "file": (Path(name).name, file)
       }
-      response = self._session.post(f"{self.base_url}/file", files=files)
+      response = self._session.post(f"{self.base_url}/file", files=files, data=data)
       data = self._handle_response(response)
       return UploadFileResponse(**data["r"])
     except requests.exceptions.RequestException as e:
@@ -673,12 +686,13 @@ class SmoothAsyncClient(BaseClient):
       logger.error(f"Request failed: {e}")
       raise ApiError(status_code=0, detail=f"Request failed: {str(e)}") from None
 
-  async def upload_file(self, file: io.IOBase, name: str) -> UploadFileResponse:
+  async def upload_file(self, file: io.IOBase, name: str | None = None, purpose: str | None = None) -> UploadFileResponse:
     """Upload a file and return the file ID.
 
     Args:
         file: File object to be uploaded.
-        name: The name to assign to the uploaded file.
+        name: A custom name to assign to the uploaded file.
+        purpose: An optional short description of the file to describe its purpose.
 
     Returns:
         The file ID assigned to the uploaded file.
@@ -688,10 +702,20 @@ class SmoothAsyncClient(BaseClient):
         ApiError: If the API request fails.
     """
     try:
+      name = name or getattr(file, "name", None)
+      if name is None:
+        raise ValueError("File name must be provided or the file object must have a 'name' attribute.")
+
       files = {
-        "file": (name, file)
+        "file": (Path(name).name, file)
       }
-      response = await self._client.post(f"{self.base_url}/file", files=files)
+      if purpose:
+        data = {
+          "file_purpose": purpose
+        }
+      else:
+        data = None
+      response = await self._client.post(f"{self.base_url}/file", files=files, data=data)
       data = self._handle_response(response)
       return UploadFileResponse(**data["r"])
     except httpx.RequestError as e: