seekrai 0.5.0__tar.gz → 0.5.1__tar.gz

{seekrai-0.5.0 → seekrai-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: seekrai
-Version: 0.5.0
+Version: 0.5.1
 Summary: Python client for SeekrAI
 License: Apache-2.0
 Author: SeekrFlow

{seekrai-0.5.0 → seekrai-0.5.1}/pyproject.toml

@@ -14,7 +14,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "seekrai"
-version = "0.5.0"
+version = "0.5.1"
 authors = [
     "SeekrFlow <support@seekr.com>"
 ]

{seekrai-0.5.0 → seekrai-0.5.1}/src/seekrai/resources/chat/completions.py

@@ -147,7 +147,7 @@ class AsyncChatCompletions:
         repetition_penalty: float = 1,
         stream: bool = False,
         logprobs: bool | None = False,
-        top_logprobs: int | None = 0,
+        top_logprobs: int | None = None,
         echo: bool = False,
         n: int = 1,
         safety_model: str | None = None,

{seekrai-0.5.0 → seekrai-0.5.1}/src/seekrai/resources/files.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from pathlib import Path
-from typing import Any, Dict
+from typing import Any, Dict, Optional, Union
 
 from seekrai.abstract import api_requestor
 from seekrai.filemanager import DownloadManager, UploadManager
@@ -197,30 +197,92 @@ class Files(ResourceBase):
 
         return AlignFileMetadataValidationResp(**response.data)
 
-    def list(self) -> FileList:
+    def list(
+        self,
+        limit: Optional[int] = None,
+        offset: Optional[int] = None,
+        sort_by: Optional[str] = None,
+        sort_order: Optional[str] = None,
+        purpose: Optional[Union[FilePurpose, str]] = None,
+    ) -> FileList:
+        """
+        List files with pagination, sorting, and filtering options.
+
+        All parameter defaults are handled by the API if not specified.
+
+        Args:
+            limit: Maximum number of files to return
+            offset: Number of files to skip
+            sort_by: Field to sort by ('created_at', 'filename', 'type', or 'size')
+            sort_order: Sort order ('asc' or 'desc')
+            purpose: Filter by file purpose (FilePurpose enum or string)
+
+        Returns:
+            A FileList object containing the list of files
+
+        Raises:
+            Exception: If the API request fails
+        """
+        # Build query parameters, only including non-None values
+        params: Dict[str, Any] = {}
+        if limit is not None:
+            params["limit"] = limit
+        if offset is not None:
+            params["offset"] = offset
+        if sort_by is not None:
+            params["sort_by"] = sort_by
+        if sort_order is not None:
+            params["sort_order"] = sort_order
+
+        # Add purpose filter if provided, converting from enum if needed
+        if purpose is not None:
+            params["purpose"] = (
+                purpose.value if isinstance(purpose, FilePurpose) else purpose
+            )
+
+        # Create request and use the API requestor
+        requestor = api_requestor.APIRequestor(client=self._client)
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="GET",
+                url="flow/files",
+                params=params,
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+
+        return FileList.parse_obj(response.data)
+
+    def rename(self, id: str, new_filename: str) -> FileResponse:
+        """
+        Rename a file.
+
+        Args:
+            id (str): ID of the file to rename.
+            new_filename (str): New filename for the file.
+
+        Returns:
+            FileResponse: Object containing information about the renamed file.
+        """
         requestor = api_requestor.APIRequestor(
             client=self._client,
         )
 
         response, _, _ = requestor.request(
             options=SeekrFlowRequest(
-                method="GET",
-                url="flow/files",
+                method="PUT",
+                url=f"flow/files/{id}/rename",
+                params={"new_filename": new_filename},
             ),
             stream=False,
         )
 
         assert isinstance(response, SeekrFlowResponse)
-        files = [
-            FileResponse(
-                id=file["id"],
-                filename=file["filename"],
-                created_at=file["created_at"],
-                object="file",
-            )
-            for file in response.data["data"]
-        ]
-        return FileList(object="list", data=files)
+
+        return FileResponse(**response.data)
 
 
 class AsyncFiles(ResourceBase):
@@ -338,19 +400,89 @@ class AsyncFiles(ResourceBase):
 
         return file_responses
 
-    async def list(self) -> FileList:
+    async def list(
+        self,
+        limit: Optional[int] = None,
+        offset: Optional[int] = None,
+        sort_by: Optional[str] = None,
+        sort_order: Optional[str] = None,
+        purpose: Optional[Union[FilePurpose, str]] = None,
+    ) -> FileList:
+        """
+        List files with pagination, sorting, and filtering options.
+
+        All parameter defaults are handled by the API if not specified.
+
+        Args:
+            limit: Maximum number of files to return
+            offset: Number of files to skip
+            sort_by: Field to sort by ('created_at', 'filename', 'type', or 'size')
+            sort_order: Sort order ('asc' or 'desc')
+            purpose: Filter by file purpose (FilePurpose enum or string)
+
+        Returns:
+            A FileList object containing the list of files
+
+        Raises:
+            Exception: If the API request fails
+        """
+        # Build query parameters, only including non-None values
+        params: Dict[str, Any] = {}
+        if limit is not None:
+            params["limit"] = limit
+        if offset is not None:
+            params["offset"] = offset
+        if sort_by is not None:
+            params["sort_by"] = sort_by
+        if sort_order is not None:
+            params["sort_order"] = sort_order
+
+        # Add purpose filter if provided, converting from enum if needed
+        if purpose is not None:
+            params["purpose"] = (
+                purpose.value if isinstance(purpose, FilePurpose) else purpose
+            )
+
+        # Create request and use the API requestor
+        requestor = api_requestor.APIRequestor(client=self._client)
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="GET",
+                url="flow/files",
+                params=params,
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+
+        return FileList.parse_obj(response.data)
+
+    async def rename(self, id: str, new_filename: str) -> FileResponse:
+        """
+        Rename a file asynchronously.
+
+        Args:
+            id (str): ID of the file to rename.
+            new_filename (str): New filename for the file.
+
+        Returns:
+            FileResponse: Object containing information about the renamed file.
+        """
         requestor = api_requestor.APIRequestor(
             client=self._client,
         )
 
         response, _, _ = await requestor.arequest(
             options=SeekrFlowRequest(
-                method="GET",
-                url="flow/files",
+                method="PUT",
+                url=f"flow/files/{id}/rename",
+                params={"new_filename": new_filename},
             ),
             stream=False,
         )
 
         assert isinstance(response, SeekrFlowResponse)
 
-        return FileList(**response.data)
+        return FileResponse(**response.data)

{seekrai-0.5.0 → seekrai-0.5.1}/src/seekrai/types/files.py

@@ -14,8 +14,6 @@ class FilePurpose(str, Enum):
     FineTune = "fine-tune"
     PreTrain = "pre-train"
     Alignment = "alignment"
-    Ingestion = "ingestion"
-    VectorIngestion = "vector-ingestion"
 
 
 class TrainingFileType(str, Enum):
@@ -96,6 +94,7 @@ class FileResponse(BaseModel):
     # file byte size
     bytes: int | None = None
     created_by: str | None = None  # TODO - fix this later
+    original_file_id: str | None = None
     deleted: bool | None = None
 
 

{seekrai-0.5.0 → seekrai-0.5.1}/src/seekrai/types/vectordb.py

@@ -69,6 +69,8 @@ class VectorDatabaseFileResponse(BaseModel):
     vector_database_id: str
     filename: str
     created_at: datetime
+    status: str
+    error_message: Optional[str]
 
 
 class VectorDatabaseFileList(BaseModel):