seekrai 0.5.12__py3-none-any.whl → 0.5.14__py3-none-any.whl

seekrai/resources/agents/__init__.py

@@ -1,5 +1,9 @@
 from seekrai.resources.agents.agent_inference import AgentInference, AsyncAgentInference
 from seekrai.resources.agents.agents import Agents, AsyncAgents
+from seekrai.resources.agents.python_functions import (
+    AsyncCustomFunctions,
+    CustomFunctions,
+)
 from seekrai.resources.agents.threads import AgentThreads, AsyncAgentThreads
 
 
@@ -10,4 +14,6 @@ __all__ = [
     "AsyncAgents",
     "AgentThreads",
     "AsyncAgentThreads",
+    "CustomFunctions",
+    "AsyncCustomFunctions",
 ]

seekrai/resources/agents/agents.py

@@ -1,5 +1,9 @@
 from seekrai.abstract import api_requestor
 from seekrai.resources.agents.agent_inference import AgentInference, AsyncAgentInference
+from seekrai.resources.agents.python_functions import (
+    AsyncCustomFunctions,
+    CustomFunctions,
+)
 from seekrai.resources.agents.threads import AgentThreads, AsyncAgentThreads
 from seekrai.seekrflow_response import SeekrFlowResponse
 from seekrai.types import SeekrFlowClient, SeekrFlowRequest
@@ -19,6 +23,7 @@ class Agents:
         )
         self.runs = AgentInference(client)
         self.threads = AgentThreads(client)
+        self.custom_functions = CustomFunctions(client)
 
     def retrieve(self, agent_id: str) -> Agent:
         """
@@ -172,6 +177,7 @@ class AsyncAgents:
         )
         self.runs = AsyncAgentInference(client)
         self.threads = AsyncAgentThreads(client)
+        self.custom_functions = AsyncCustomFunctions(client)
 
     async def retrieve(self, agent_id: str) -> Agent:
         """

seekrai/resources/agents/python_functions.py

@@ -0,0 +1,330 @@
+from pathlib import Path
+from typing import Union
+
+from seekrai.abstract import api_requestor
+from seekrai.seekrflow_response import SeekrFlowResponse
+from seekrai.types import SeekrFlowClient, SeekrFlowRequest
+from seekrai.types.agents.python_functions import (
+    DeletePythonFunctionResponse,
+    PythonFunctionResponse,
+)
+
+
+class CustomFunctions:
+    def __init__(self, client: SeekrFlowClient) -> None:
+        self._client = client
+        self._requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+    def create(
+        self, file_path: Union[str, Path], description: Union[str, None] = None
+    ) -> PythonFunctionResponse:
+        """
+        Upload a new Python function for the user.
+
+        Args:
+            file_path: Path to the Python function file to upload (can be relative or absolute).
+            description: Optional description for the function.
+
+        Returns:
+            The newly created Python function.
+        """
+        # Convert string to Path if needed
+        if isinstance(file_path, str):
+            file_path = Path(file_path)
+
+        # Read the file contents
+        with file_path.open("rb") as f:
+            file_content = f.read()
+
+        # Prepare multipart form data
+        files = {"file": (file_path.name, file_content, "text/plain")}
+        params = {}
+        if description:
+            params["description"] = description
+
+        response, _, _ = self._requestor.request(
+            options=SeekrFlowRequest(
+                method="POST",
+                url="functions/",
+                files=files,
+                params=params,
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return PythonFunctionResponse(**response.data)
+
+    def retrieve(self, function_id: str) -> PythonFunctionResponse:
+        """
+        Retrieve a Python function by its ID.
+
+        Args:
+            function_id: The ID of the Python function to retrieve.
+
+        Returns:
+            The Python function.
+        """
+        response, _, _ = self._requestor.request(
+            options=SeekrFlowRequest(
+                method="GET",
+                url=f"functions/{function_id}",
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return PythonFunctionResponse(**response.data)
+
+    def list_functions(
+        self, limit: int = 20, offset: int = 0, order: str = "desc"
+    ) -> list[PythonFunctionResponse]:
+        """
+        List all Python functions for the user.
+
+        Args:
+            limit: Maximum number of functions to return (default: 20).
+            offset: Number of functions to skip (default: 0).
+            order: Sort order, 'asc' or 'desc' (default: 'desc').
+
+        Returns:
+            A list of Python functions.
+        """
+        response, _, _ = self._requestor.request(
+            options=SeekrFlowRequest(
+                method="GET",
+                url="functions/",
+                params={"limit": limit, "offset": offset, "order": order},
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        functions = [PythonFunctionResponse(**func) for func in response.data]  # type: ignore
+        return functions
+
+    def update(
+        self,
+        function_id: str,
+        file_path: Union[str, Path, None] = None,
+        description: Union[str, None] = None,
+    ) -> PythonFunctionResponse:
+        """
+        Update an existing Python function.
+
+        Args:
+            function_id: The ID of the Python function to update.
+            file_path: Optional path to a new Python function file (can be relative or absolute).
+            description: Optional new description for the function.
+
+        Returns:
+            The updated Python function.
+        """
+        files = None
+        params = {}
+
+        if file_path:
+            # Convert string to Path if needed
+            if isinstance(file_path, str):
+                file_path = Path(file_path)
+
+            # Read the file contents
+            with file_path.open("rb") as f:
+                file_content = f.read()
+
+            # Prepare multipart form data
+            files = {"file": (file_path.name, file_content, "text/plain")}
+
+        if description:
+            params["description"] = description
+
+        response, _, _ = self._requestor.request(
+            options=SeekrFlowRequest(
+                method="PATCH",
+                url=f"functions/{function_id}",
+                files=files,
+                params=params,
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return PythonFunctionResponse(**response.data)
+
+    def delete(self, function_id: str) -> DeletePythonFunctionResponse:
+        """
+        Delete a Python function by its ID.
+
+        Args:
+            function_id: The ID of the Python function to delete.
+
+        Returns:
+            A response indicating whether the delete operation was successful.
+        """
+        response, _, _ = self._requestor.request(
+            options=SeekrFlowRequest(
+                method="DELETE",
+                url=f"functions/{function_id}",
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return DeletePythonFunctionResponse(**response.data)
+
+
+class AsyncCustomFunctions:
+    def __init__(self, client: SeekrFlowClient) -> None:
+        self._client = client
+        self._requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+    async def create(
+        self, file_path: Union[str, Path], description: Union[str, None] = None
+    ) -> PythonFunctionResponse:
+        """
+        Upload a new Python function for the user.
+
+        Args:
+            file_path: Path to the Python function file to upload (can be relative or absolute).
+            description: Optional description for the function.
+
+        Returns:
+            The newly created Python function.
+        """
+        # Convert string to Path if needed
+        if isinstance(file_path, str):
+            file_path = Path(file_path)
+
+        # Read the file contents
+        with file_path.open("rb") as f:
+            file_content = f.read()
+
+        # Prepare multipart form data
+        files = {"file": (file_path.name, file_content, "text/plain")}
+        params = {}
+        if description:
+            params["description"] = description
+
+        response, _, _ = await self._requestor.arequest(
+            options=SeekrFlowRequest(
+                method="POST",
+                url="functions/",
+                files=files,
+                params=params,
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return PythonFunctionResponse(**response.data)
+
+    async def retrieve(self, function_id: str) -> PythonFunctionResponse:
+        """
+        Retrieve a Python function by its ID.
+
+        Args:
+            function_id: The ID of the Python function to retrieve.
+
+        Returns:
+            The Python function.
+        """
+        response, _, _ = await self._requestor.arequest(
+            options=SeekrFlowRequest(
+                method="GET",
+                url=f"functions/{function_id}",
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return PythonFunctionResponse(**response.data)
+
+    async def list_functions(
+        self, limit: int = 20, offset: int = 0, order: str = "desc"
+    ) -> list[PythonFunctionResponse]:
+        """
+        List all Python functions for the user.
+
+        Args:
+            limit: Maximum number of functions to return (default: 20).
+            offset: Number of functions to skip (default: 0).
+            order: Sort order, 'asc' or 'desc' (default: 'desc').
+
+        Returns:
+            A list of Python functions.
+        """
+        response, _, _ = await self._requestor.arequest(
+            options=SeekrFlowRequest(
+                method="GET",
+                url="functions/",
+                params={"limit": limit, "offset": offset, "order": order},
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        functions = [PythonFunctionResponse(**func) for func in response.data]  # type: ignore
+        return functions
+
+    async def update(
+        self,
+        function_id: str,
+        file_path: Union[str, Path, None] = None,
+        description: Union[str, None] = None,
+    ) -> PythonFunctionResponse:
+        """
+        Update an existing Python function.
+
+        Args:
+            function_id: The ID of the Python function to update.
+            file_path: Optional path to a new Python function file (can be relative or absolute).
+            description: Optional new description for the function.
+
+        Returns:
+            The updated Python function.
+        """
+        files = None
+        params = {}
+
+        if file_path:
+            # Convert string to Path if needed
+            if isinstance(file_path, str):
+                file_path = Path(file_path)
+
+            # Read the file contents
+            with file_path.open("rb") as f:
+                file_content = f.read()
+
+            # Prepare multipart form data
+            files = {"file": (file_path.name, file_content, "text/plain")}
+
+        if description:
+            params["description"] = description
+
+        response, _, _ = await self._requestor.arequest(
+            options=SeekrFlowRequest(
+                method="PATCH",
+                url=f"functions/{function_id}",
+                files=files,
+                params=params,
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return PythonFunctionResponse(**response.data)
+
+    async def delete(self, function_id: str) -> DeletePythonFunctionResponse:
+        """
+        Delete a Python function by its ID.
+
+        Args:
+            function_id: The ID of the Python function to delete.
+
+        Returns:
+            A response indicating whether the delete operation was successful.
+        """
+        response, _, _ = await self._requestor.arequest(
+            options=SeekrFlowRequest(
+                method="DELETE",
+                url=f"functions/{function_id}",
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return DeletePythonFunctionResponse(**response.data)

seekrai/resources/models.py

@@ -38,7 +38,6 @@ class Models(ResourceBase):
                 object=ObjectType.Model,
                 name=model["name"],
                 bytes=model["size"],
-                created_at=parse_timestamp(model["created_at"]),
                 model_type=model["model_type"],
             )
             for model in response.data["data"]
@@ -79,7 +78,6 @@ class AsyncModels(ResourceBase):
                 object=ObjectType.Model,
                 name=model["name"],
                 bytes=model["size"],
-                created_at=parse_timestamp(model["created_at"]),
                 model_type=model["model_type"],
             )
             for model in response.data["data"]

seekrai/types/agents/__init__.py

@@ -5,6 +5,11 @@ from seekrai.types.agents.agent import (
     CreateAgentRequest,
     ReasoningEffort,
 )
+from seekrai.types.agents.python_functions import (
+    DeletePythonFunctionResponse,
+    PythonFunctionBase,
+    PythonFunctionResponse,
+)
 from seekrai.types.agents.runs import (
     ModelSettings,
     Run,
@@ -100,4 +105,7 @@ __all__ = [
     "RunPythonEnv",
     "WebSearch",
     "WebSearchEnv",
+    "PythonFunctionBase",
+    "PythonFunctionResponse",
+    "DeletePythonFunctionResponse",
 ]

seekrai/types/agents/python_functions.py

@@ -0,0 +1,29 @@
+from datetime import datetime
+
+from pydantic import BaseModel, ConfigDict
+
+
+class PythonFunctionBase(BaseModel):
+    """Base model for a Python function, including metadata fields."""
+
+    model_config = ConfigDict(from_attributes=True)
+    id: str
+    version: int
+    name: str
+    description: str
+    active: bool
+
+
+class PythonFunctionResponse(PythonFunctionBase):
+    """Response model for a Python function, including code and user info."""
+
+    code: str
+    user_id: str
+    created_at: datetime
+    updated_at: datetime
+
+
+class DeletePythonFunctionResponse(BaseModel):
+    """Response model for Python function deletion."""
+
+    deleted: bool

seekrai/types/agents/tools/schemas/run_python_env.py

@@ -5,3 +5,4 @@ from seekrai.types.agents.tools.env_model_config import EnvConfig
 
 class RunPythonEnv(EnvConfig):
     run_python_tool_desc: Optional[str] = None
+    function_ids: Optional[list[str]] = None

{seekrai-0.5.12.dist-info → seekrai-0.5.14.dist-info}/METADATA

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

{seekrai-0.5.12.dist-info → seekrai-0.5.14.dist-info}/RECORD

@@ -7,9 +7,10 @@ seekrai/constants.py,sha256=hoR2iF5te5Ydjt_lxIOSGID4vESIakG4F-3xAWdwxaU,1854
 seekrai/error.py,sha256=rAYL8qEd8INwYMMKvhS-HKeC3QkWL4Wq-zfazFU-zBg,4861
 seekrai/filemanager.py,sha256=bO2OvjZ9Cx5r2vr3Ocpd_0qVc3owRDT2LCU4Zmp2uDY,15489
 seekrai/resources/__init__.py,sha256=EPMOqI3mvpzMjNJIr2Da2tkDqSq_lntMph2O-j4pyIA,1501
-seekrai/resources/agents/__init__.py,sha256=qPdo3vMZUaGZPdZCNYL0hjtX-T6yAlnpE_zc5otkjak,373
+seekrai/resources/agents/__init__.py,sha256=LJIOKMZaQq2wuK-32wgKRKOMq06svWyia1SsHjEJsc4,529
 seekrai/resources/agents/agent_inference.py,sha256=4dZNeKBHWhv22J4ADSfr_0HXR6B4Vbz3syGnatmLkPM,9834
-seekrai/resources/agents/agents.py,sha256=d3C4VaXq4q5WlUr0f8txdRD3hvEKKvz4UQA-a_sLgso,9385
+seekrai/resources/agents/agents.py,sha256=frmLHvrqeUxnMkE9Xn4w1ugChM14S9ik4MwWyklPMWI,9607
+seekrai/resources/agents/python_functions.py,sha256=2GSiWhKTto8ixE5OG6WsN-GWEwyf1wpdhG3_EziACsg,10367
 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
@@ -22,15 +23,16 @@ seekrai/resources/files.py,sha256=bEn4jfYWfsI2OqKRGCUpnefIN-udNnafItgT2A7m-e4,15
 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
-seekrai/resources/models.py,sha256=pj7dmtQDEFwkkeMBdK1oLjq7-fdk7LBB1SMyWENS9ic,2418
+seekrai/resources/models.py,sha256=I1JhvU6rEo_qZQ8vDpkLSE7M-zsXl_qdzCz0NaKk0d4,2288
 seekrai/resources/projects.py,sha256=Nmuh0_BwWoAO89r-p0ZEM8p4NHIH1EUeP83ivRoW5hw,3682
 seekrai/resources/resource_base.py,sha256=rFIHFeqKPiAEbMYcMiIGHIym7qxwmh-EGsWiZcMDHdk,224
 seekrai/resources/vectordb.py,sha256=RRULyuldM4A0RlveBNZWFrav7l405wm89ua3k3Bqkgc,14527
 seekrai/seekrflow_response.py,sha256=5RFEQzamDy7sTSDkxSsZQThZ3biNmeCPeHWdrFId5Go,1320
 seekrai/types/__init__.py,sha256=N72VsnmOdvLxnhBFLgfugIkT0tqr-M7WVFpAFtXEjLI,4665
 seekrai/types/abstract.py,sha256=TqWFQV_6bPblywfCH-r8FCkXWvPkc9KlJ4QVgyrnaMc,642
-seekrai/types/agents/__init__.py,sha256=akuEDZDs4-OzvPIt0nupF63vBmI_CPfSLKsQUHTl1s8,2163
+seekrai/types/agents/__init__.py,sha256=STRQlgnapxFT5egAP_i3yZSh9gauGmJBHTzdvH8tbdk,2395
 seekrai/types/agents/agent.py,sha256=85D4GeHF-bYYnPirJSi1MbFg_2uFE2fSEmAHV9LxZfQ,1132
+seekrai/types/agents/python_functions.py,sha256=31Jm46gryHsgNsC7nlivuyY0TSko58y2YVxsu_7bEAg,653
 seekrai/types/agents/runs.py,sha256=hXH8F6BjF6MoKPWiZY-QtgNTjr6Y256peaW5ASXLWxA,4823
 seekrai/types/agents/threads.py,sha256=TinCMKv1bi5LzboDyCx1XI4Zzd8UzUZos4VOrTNhmEc,6835
 seekrai/types/agents/tools/__init__.py,sha256=4MmlL13JLhWgMUlL3TKfegiA-IXGG06YellZTSTVFC8,537
@@ -39,7 +41,7 @@ seekrai/types/agents/tools/schemas/__init__.py,sha256=_HQoqgBk4T1IvlD0Yt3REz0_x_
 seekrai/types/agents/tools/schemas/file_search.py,sha256=7reLq-S3jMnHlGgP0ihZrU2S4Mcaizoz4IXm3SqCj4A,304
 seekrai/types/agents/tools/schemas/file_search_env.py,sha256=3iS69R8DpLPTJY2WhiCbOMV47-vXvMjBA7csgfIxKeU,582
 seekrai/types/agents/tools/schemas/run_python.py,sha256=CMw9C0T9UBWfHSW6mRhNDIIe_uGfxjgRkPIbqGlQxjo,297
-seekrai/types/agents/tools/schemas/run_python_env.py,sha256=lvmGSg-u6CoGSEGhAHHkmDffFj6kpFDERIwAhggzgwI,175
+seekrai/types/agents/tools/schemas/run_python_env.py,sha256=q2lhxWvmOEqdj3c1DktL7CJrKqXp8s1wZuQvh7eReb4,220
 seekrai/types/agents/tools/schemas/web_search.py,sha256=HrcVIC-SQGlBIXvJ0MxXYVQe6JnlgacleViXIo5OrGw,297
 seekrai/types/agents/tools/schemas/web_search_env.py,sha256=aHaricIcXaWz9N4G_gEipstANPbJPf41F9VZC5LwPog,182
 seekrai/types/agents/tools/tool.py,sha256=rX-SoN-XgweiHQIKRhNxBWK4ClNjEiXv-YaLdejpgmc,397
@@ -65,8 +67,8 @@ seekrai/utils/api_helpers.py,sha256=0Y8BblNIr9h_R12zdmhkxgTlxgoRkbq84QNi4nNWGu8,
 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.12.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-seekrai-0.5.12.dist-info/METADATA,sha256=7n2YKQjR7WVH9rShRYO5ZcdxW4kN9moWX0siQMmFp1o,4781
-seekrai-0.5.12.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-seekrai-0.5.12.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
-seekrai-0.5.12.dist-info/RECORD,,
+seekrai-0.5.14.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+seekrai-0.5.14.dist-info/METADATA,sha256=pLMn2jpTZxyuoiTWSzYM9ooGKMlJSPrQlQws6tNwgZQ,4781
+seekrai-0.5.14.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+seekrai-0.5.14.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
+seekrai-0.5.14.dist-info/RECORD,,