seekrai 0.5.0__tar.gz → 0.5.2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: seekrai
-Version: 0.5.0
+Version: 0.5.2
 Summary: Python client for SeekrAI
 License: Apache-2.0
 Author: SeekrFlow
@@ -70,7 +70,7 @@ from seekrai import SeekrFlow
 client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
 
 response = client.chat.completions.create(
-    model="meta-llama-3-8b-instruct",
+    model="meta-llama/Llama-3.1-8B-Instruct",
     messages=[{"role": "user", "content": "tell me about new york"}],
 )
 print(response.choices[0].message.content)
@@ -84,7 +84,7 @@ from seekrai import SeekrFlow
 
 client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
 stream = client.chat.completions.create(
-    model="meta-llama-3-8b-instruct",
+    model="meta-llama/Llama-3.1-8B-Instruct",
     messages=[{"role": "user", "content": "tell me about new york"}],
     stream=True,
 )
@@ -110,7 +110,7 @@ async def async_chat_completion(messages):
     async_client = AsyncSeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
     tasks = [
         async_client.chat.completions.create(
-            model="meta-llama-3-8b-instruct",
+            model="meta-llama/Llama-3.1-8B-Instruct",
             messages=[{"role": "user", "content": message}],
         )
         for message in messages
@@ -151,7 +151,7 @@ client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
 
 client.fine_tuning.create(
     training_file='file-d0d318cb-b7d9-493a-bd70-1cfe089d3815',
-    model='meta-llama-3-8b-instruct',
+    model='meta-llama/Llama-3.1-8B-Instruct',
     n_epochs=3,
     n_checkpoints=1,
     batch_size=4,

{seekrai-0.5.0 → seekrai-0.5.2}/README.md

@@ -37,7 +37,7 @@ from seekrai import SeekrFlow
 client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
 
 response = client.chat.completions.create(
-    model="meta-llama-3-8b-instruct",
+    model="meta-llama/Llama-3.1-8B-Instruct",
     messages=[{"role": "user", "content": "tell me about new york"}],
 )
 print(response.choices[0].message.content)
@@ -51,7 +51,7 @@ from seekrai import SeekrFlow
 
 client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
 stream = client.chat.completions.create(
-    model="meta-llama-3-8b-instruct",
+    model="meta-llama/Llama-3.1-8B-Instruct",
     messages=[{"role": "user", "content": "tell me about new york"}],
     stream=True,
 )
@@ -77,7 +77,7 @@ async def async_chat_completion(messages):
     async_client = AsyncSeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
     tasks = [
         async_client.chat.completions.create(
-            model="meta-llama-3-8b-instruct",
+            model="meta-llama/Llama-3.1-8B-Instruct",
             messages=[{"role": "user", "content": message}],
         )
         for message in messages
@@ -118,7 +118,7 @@ client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
 
 client.fine_tuning.create(
     training_file='file-d0d318cb-b7d9-493a-bd70-1cfe089d3815',
-    model='meta-llama-3-8b-instruct',
+    model='meta-llama/Llama-3.1-8B-Instruct',
     n_epochs=3,
     n_checkpoints=1,
     batch_size=4,

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

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

{seekrai-0.5.0 → seekrai-0.5.2}/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.2}/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.2/src/seekrai/types/agents/tools/schemas/file_search_env.py

@@ -0,0 +1,16 @@
+from pydantic import Field
+
+from seekrai.types.agents.tools.env_model_config import EnvConfig
+
+
+# TODO: figure out better way of creating tool environment models (within tool ideally), but retaining separate model_configs
+class FileSearchEnv(EnvConfig):
+    file_search_index: str = Field(min_length=1)
+    document_tool_desc: str = Field(min_length=1)
+    top_k: int = Field(
+        default=10, ge=1, le=100, description="Top K must be >= 1 and <= 100"
+    )
+    score_threshold: float = Field(
+        default=0, ge=0, lt=1.0, description="Score must be ≥ 0.0 and < 1.0"
+    )
+

{seekrai-0.5.0 → seekrai-0.5.2}/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.2}/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):

seekrai-0.5.0/src/seekrai/types/agents/tools/schemas/file_search_env.py

@@ -1,11 +0,0 @@
-from pydantic import Field
-
-from seekrai.types.agents.tools.env_model_config import EnvConfig
-
-
-# TODO: figure out better way of creating tool environment models (within tool ideally), but retaining separate model_configs
-class FileSearchEnv(EnvConfig):
-    file_search_index: str
-    document_tool_desc: str
-    top_k: int = Field(default=10)
-    score_threshold: int = Field(default=0)