seekrai 0.5.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

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/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/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/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):

{seekrai-0.5.0.dist-info → seekrai-0.5.1.dist-info}/METADATA

@@ -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.dist-info → seekrai-0.5.1.dist-info}/RECORD

@@ -13,11 +13,11 @@ seekrai/resources/agents/agents.py,sha256=Cs184v2jYQRlk5MmsAH_hRI93D7qS6mJpuCLGI
 seekrai/resources/agents/threads.py,sha256=BwZ2_6wlezsb12PQjEw1fgdJh5S83SPgD6qZQoGvyIM,14544
 seekrai/resources/alignment.py,sha256=IOKlKK2I9_NhS9pwcrsd9-5OO7lVT8Uw0y_wuGHOnyA,5839
 seekrai/resources/chat/__init__.py,sha256=KmtPupgECtEN80NyvcnSmieTAFXhwmVxhMHP0qhspA4,618
-seekrai/resources/chat/completions.py,sha256=U75hKIDBAX8h5-4aFe20DowO1adqBD6ZRHjPK9EdqqM,11653
+seekrai/resources/chat/completions.py,sha256=9KP1eiQ7uHuf8A4BPOKOKwgnsn6kkp2swT6Q5g_4xoI,11656
 seekrai/resources/completions.py,sha256=JhTN_lW2mblfHHONFmPC7QZei3wo5vx6GliMs9FkbOY,8452
 seekrai/resources/deployments.py,sha256=DY7IN7QgqDduCHGNuHENSVwrE5PXFL88jWgh8SES7Qk,5970
 seekrai/resources/embeddings.py,sha256=UMVb6LIJn8KFACMajVRL3SY88bbZl1aQ2Sx5MdT1sQQ,2593
-seekrai/resources/files.py,sha256=AJRTXbTxV3lYWML0eGFrhbBcGoFQFAQUA2Dc287E250,10875
+seekrai/resources/files.py,sha256=bEn4jfYWfsI2OqKRGCUpnefIN-udNnafItgT2A7m-e4,15329
 seekrai/resources/finetune.py,sha256=Aw8lU9TohZdJGOstex12gg2t-RDppOponnWg203Bn-Q,11304
 seekrai/resources/images.py,sha256=VjZiU2cxq2uNrJzm-EwNpOW3rBIgFyRHss8_DF1OuVE,4799
 seekrai/resources/ingestion.py,sha256=tSJhX6kMzSdNxhHGBdzUKg9gsMP_aOFBz4EBoIFlGUM,4854
@@ -47,21 +47,21 @@ seekrai/types/completions.py,sha256=lm9AFdZR3Xg5AHPkV-qETHikkwMJmkHrLGr5GG-YR-M,
 seekrai/types/deployments.py,sha256=GdZPDaQgzmk9W1aXxZr9CDxqJRNv7NP0pNvqRV3E4xM,1760
 seekrai/types/embeddings.py,sha256=OANoLNOs0aceS8NppVvvcNYQbF7-pAOAmcr30pw64OU,749
 seekrai/types/error.py,sha256=uTKISs9aRC4_6zwirtNkanxepN8KY-SqCq0kNbfZylQ,370
-seekrai/types/files.py,sha256=aG8pTMiUJh11wpNqvJBHaDCfr45clcXBpSPa5Hv2V4Y,2737
+seekrai/types/files.py,sha256=yjJYT8twY-cNh9AY9qlcN_moTeCfR0tJSSCQsOVB02Y,2708
 seekrai/types/finetune.py,sha256=e1TexW2PdNcRAGdHeNmRSWGKE2K9NLhbdcce0XYvWio,6162
 seekrai/types/images.py,sha256=Fusj8OhVYFsT8kz636lRGGivLbPXo_ZNgakKwmzJi3U,914
 seekrai/types/ingestion.py,sha256=uUdKOR4xqSfAXWQOR1UOltSlOnuyAwKVA1Q2a6Yslk8,919
 seekrai/types/models.py,sha256=9Z0nvLdlAfpF8mNRW5-IqBdDHoE-3qQ5przmIDJgwLo,1345
 seekrai/types/projects.py,sha256=JFgpZdovia8Orcnhp6QkIEAXzyPCfKT_bUiwjxUaHHQ,670
-seekrai/types/vectordb.py,sha256=JzVi6na1sHx0SfdNZp_1OKIKN9wO-0_iMoZHtcC-vJ8,2219
+seekrai/types/vectordb.py,sha256=zWtB9OEfTpFf0hABko8WqhyZ0SH3os6ZRWXxkvg88-Q,2268
 seekrai/utils/__init__.py,sha256=dfbiYEc47EBVRkq6C4O9y6tTGuPuV3LbV3__v01Mbds,658
 seekrai/utils/_log.py,sha256=Cayw5B394H2WGVTXPXS2AN8znQdxsgrLqADXgqmokvU,1649
 seekrai/utils/api_helpers.py,sha256=0Y8BblNIr9h_R12zdmhkxgTlxgoRkbq84QNi4nNWGu8,2385
 seekrai/utils/files.py,sha256=7ixn_hgV-6pEhYqLyOp-EN0o8c1CzUwJzX9n3PQ5oqo,7164
 seekrai/utils/tools.py,sha256=jgJTL-dOIouDbEJLdQpQfpXhqaz_poQYS52adyUtBjo,1781
 seekrai/version.py,sha256=q6iGQVFor8zXiPP5F-3vy9TndOxKv5JXbaNJ2kdOQws,125
-seekrai-0.5.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-seekrai-0.5.0.dist-info/METADATA,sha256=woZebLMxAmACmHj2a5FmVa-fvVgas0TJgBCVkG2PfZc,4748
-seekrai-0.5.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-seekrai-0.5.0.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
-seekrai-0.5.0.dist-info/RECORD,,
+seekrai-0.5.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+seekrai-0.5.1.dist-info/METADATA,sha256=WE8Hbfl-J_XWofa47b1BBFjItV4ny3t8AOBsG8zbwQw,4748
+seekrai-0.5.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+seekrai-0.5.1.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
+seekrai-0.5.1.dist-info/RECORD,,