seekrai 0.5.11__tar.gz → 0.5.13__tar.gz

{seekrai-0.5.11 → seekrai-0.5.13}/PKG-INFO

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

{seekrai-0.5.11 → seekrai-0.5.13}/pyproject.toml

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

{seekrai-0.5.11 → seekrai-0.5.13}/src/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-0.5.11 → seekrai-0.5.13}/src/seekrai/resources/agents/agent_inference.py

@@ -3,6 +3,7 @@ from typing import Any, AsyncGenerator, Iterator, Optional, Union
 from seekrai.abstract import api_requestor
 from seekrai.seekrflow_response import SeekrFlowResponse
 from seekrai.types import ModelSettings, Run, RunRequest, RunResponse, SeekrFlowRequest
+from seekrai.types.agents.runs import ResponseFormat
 
 
 class AgentInference:
@@ -16,7 +17,8 @@ class AgentInference:
         thread_id: str,
         *,
         stream: bool = False,
-        model_settings: Optional[ModelSettings] = None,
+        model_settings: ModelSettings = ModelSettings(),
+        response_format: Optional[Any] = None,
     ) -> Union[RunResponse, Iterator[Any]]:
         """
         Run an inference call on a deployed agent.
@@ -26,13 +28,18 @@ class AgentInference:
             thread_id (str): A thread identifier.
             stream (bool, optional): Whether to stream the response. Defaults to False.
             model_settings (optional): Additional parameters (such as temperature, max_tokens, etc).
+            response_format: Optional structured output specification. If provided, the LLM will be constrained to return JSON matching this schema.
 
         Returns:
             A dictionary with the response (if non-streaming) or an iterator over response chunks.
         """
-        payload = RunRequest(agent_id=agent_id).model_dump()
-        if model_settings is not None:
-            payload["model_settings"] = model_settings.model_dump()
+        payload = RunRequest(
+            agent_id=agent_id,
+            model_settings=model_settings,
+            response_format=ResponseFormat.from_value(response_format)
+            if response_format
+            else None,
+        ).model_dump()
         endpoint = f"threads/{thread_id}/runs"
         if stream:
             endpoint += "/stream"
@@ -147,7 +154,8 @@ class AsyncAgentInference:
         thread_id: str,
         *,
         stream: bool = False,
-        model_settings: Optional[ModelSettings] = None,
+        model_settings: ModelSettings = ModelSettings(),
+        response_format: Optional[Any] = None,
     ) -> Union[RunResponse, AsyncGenerator[Any, None]]:
         """
         Run an inference call on a deployed agent.
@@ -157,13 +165,18 @@ class AsyncAgentInference:
             thread_id (str): A thread identifier.
             stream (bool, optional): Whether to stream the response. Defaults to False.
             model_settings (optional): Additional parameters (such as temperature, max_tokens, etc).
+            response_format: Optional structured output specification. If provided, the LLM will be constrained to return JSON matching this schema.
 
         Returns:
             A dictionary with the response (if non-streaming) or an iterator over response chunks.
         """
-        payload = RunRequest(agent_id=agent_id).model_dump()
-        if model_settings is not None:
-            payload["model_settings"] = model_settings.model_dump()
+        payload = RunRequest(
+            agent_id=agent_id,
+            model_settings=model_settings,
+            response_format=ResponseFormat.from_value(response_format)
+            if response_format
+            else None,
+        ).model_dump()
         endpoint = f"threads/{thread_id}/runs"
         if stream:
             endpoint += "/stream"

{seekrai-0.5.11 → seekrai-0.5.13}/src/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-0.5.13/src/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-0.5.11 → seekrai-0.5.13}/src/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-0.5.13/src/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-0.5.11 → seekrai-0.5.13}/src/seekrai/types/agents/runs.py

@@ -1,16 +1,63 @@
 import datetime
 from enum import Enum
-from typing import Any, Optional, Union
+from typing import Any, Dict, Optional, Union
 
+import pydantic
 from pydantic import Field
 
 from seekrai.types.abstract import BaseModel
 
 
+class ModelSettings(BaseModel):
+    """Settings to use when calling an LLM.
+
+    This class holds optional model configuration parameters (e.g. temperature,
+    top_p, penalties, truncation, etc.).
+
+    Not all models/providers support all of these parameters, so please check the API documentation
+    for the specific model and provider you are using.
+    """
+
+    temperature: float = Field(default=1.0, ge=0.0, le=2.0)
+    top_p: float = Field(default=1.0, ge=0.0, le=1.0)
+    frequency_penalty: float = Field(default=0.0, ge=-2.0, le=2.0)
+    presence_penalty: float = Field(default=0.0, ge=-2.0, le=2.0)
+    max_tokens: Optional[int] = None
+
+
+class ResponseFormat(BaseModel):
+    """Specifies a JSON schema for the response format.
+
+    When provided, the LLM will be constrained to return a JSON response
+    that matches the specified schema.
+
+    Can be instantiated with:
+    - A JSON schema dictionary
+    - A Pydantic model class
+    - An existing ResponseFormat instance
+    """
+
+    json_schema: Dict[str, Any]
+
+    @classmethod
+    def from_value(cls, value: Any) -> "ResponseFormat":
+        if isinstance(value, cls):
+            return value
+        if isinstance(value, dict):
+            return cls(json_schema=value)
+        if isinstance(value, type) and issubclass(value, pydantic.BaseModel):
+            return cls(json_schema=value.model_json_schema())
+        raise ValueError(
+            "ResponseFormat configuration is invalid. Expected ResponseFormat, a valid schema or a Pydantic BaseModel."
+        )
+
+
 class RunRequest(BaseModel):
     """Request model for creating a run."""
 
     agent_id: str = Field(default="default_agent")
+    model_settings: ModelSettings = ModelSettings()
+    response_format: Optional[Union[ResponseFormat, Dict[str, Any], type]] = None
 
 
 class RunResponse(BaseModel):
@@ -115,20 +162,3 @@ class RunStep(BaseModel):
     completed_at: Optional[datetime.datetime] = None
     meta_data: dict[str, Any] = Field(default_factory=dict)
     usage: Optional[RunStepUsage] = None
-
-
-class ModelSettings(BaseModel):
-    """Settings to use when calling an LLM.
-
-    This class holds optional model configuration parameters (e.g. temperature,
-    top_p, penalties, truncation, etc.).
-
-    Not all models/providers support all of these parameters, so please check the API documentation
-    for the specific model and provider you are using.
-    """
-
-    temperature: float = Field(default=1.0, ge=0.0, le=2.0)
-    top_p: float = Field(default=1.0, ge=0.0, le=1.0)
-    frequency_penalty: float = Field(default=0.0, ge=-2.0, le=2.0)
-    presence_penalty: float = Field(default=0.0, ge=-2.0, le=2.0)
-    max_tokens: Optional[int] = None

{seekrai-0.5.11 → seekrai-0.5.13}/src/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.11 → seekrai-0.5.13}/src/seekrai/types/files.py

@@ -11,6 +11,7 @@ from seekrai.types.common import (
 
 
 class FilePurpose(str, Enum):
+    ReinforcementFineTune = "reinforcement-fine-tune"
     FineTune = "fine-tune"
     PreTrain = "pre-train"
     Alignment = "alignment"