seekrai 0.5.24__tar.gz → 0.5.26__tar.gz

{seekrai-0.5.24 → seekrai-0.5.26}/PKG-INFO

@@ -1,9 +1,9 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: seekrai
-Version: 0.5.24
+Version: 0.5.26
 Summary: Python client for SeekrAI
-Home-page: https://www.seekr.com
 License: Apache-2.0
+License-File: LICENSE
 Author: SeekrFlow
 Author-email: support@seekr.com
 Requires-Python: >=3.9,<4.0
@@ -13,6 +13,9 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Requires-Dist: click (>=8.1.7,<9.0.0)
 Requires-Dist: eval-type-backport (>=0.1.3,<0.3.0)
 Requires-Dist: filelock (>=3.13.1,<4.0.0)
@@ -26,6 +29,7 @@ Requires-Dist: requests (>=2.31.0,<3.0.0)
 Requires-Dist: tabulate (>=0.9.0,<0.10.0)
 Requires-Dist: tqdm (>=4.66.2,<5.0.0)
 Requires-Dist: typer (>=0.9,<0.13)
+Project-URL: Homepage, https://www.seekr.com
 Description-Content-Type: text/markdown
 
 The Seekr Python Library is the official Python client for SeekrFlow's API platform, providing a convenient way for interacting with the REST APIs and enables easy integrations with Python 3.9+ applications with easy to use synchronous and asynchronous clients.

{seekrai-0.5.24 → seekrai-0.5.26}/pyproject.toml

@@ -13,7 +13,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "seekrai"
-version = "0.5.24"
+version = "0.5.26"
 authors = [
     "SeekrFlow <support@seekr.com>"
 ]
@@ -65,6 +65,7 @@ pytest = ">=7.4.2,<9.0.0"
 pytest-watch = "^4.2.0"
 tox = "^4.14.1"
 pytest-asyncio = "^0.26.0"
+tenacity = "^8.5.0"
 
 [tool.poetry.group.dev.dependencies]
 ipdb = "^0.13.13"

{seekrai-0.5.24 → seekrai-0.5.26}/src/seekrai/client.py

@@ -91,6 +91,7 @@ class SeekrFlow:
         self.deployments = resources.Deployments(self.client)
         self.vector_database = resources.VectorDatabase(self.client)
         self.agents = resources.Agents(self.client)
+        self.tools = resources.Tools(self.client)
         self.observability = resources.AgentObservability(self.client)
         self.explainability = resources.Explainability(self.client)
 
@@ -176,6 +177,7 @@ class AsyncSeekrFlow:
         self.deployments = resources.AsyncDeployments(self.client)
         self.vector_database = resources.AsyncVectorDatabase(self.client)
         self.agents = resources.AsyncAgents(self.client)
+        self.tools = resources.AsyncTools(self.client)
         self.observability = resources.AsyncAgentObservability(self.client)
         self.explainability = resources.AsyncExplainability(self.client)
 

{seekrai-0.5.24 → seekrai-0.5.26}/src/seekrai/resources/__init__.py

@@ -2,6 +2,7 @@ from seekrai.resources.agents import (
     AgentInference,
     AgentObservability,
     Agents,
+    AsyncAgentInference,
     AsyncAgentObservability,
     AsyncAgents,
 )
@@ -22,6 +23,7 @@ from seekrai.resources.images import AsyncImages, Images
 from seekrai.resources.ingestion import AsyncIngestion, Ingestion
 from seekrai.resources.models import AsyncModels, Models
 from seekrai.resources.projects import AsyncProjects, Projects
+from seekrai.resources.tools import AsyncTools, Tools
 from seekrai.resources.vectordb import AsyncVectorDatabase, VectorDatabase
 
 
@@ -42,8 +44,8 @@ __all__ = [
     "Files",
     "AsyncImages",
     "Images",
-    "Ingestion",
     "AsyncIngestion",
+    "Ingestion",
     "AsyncModels",
     "Models",
     "AsyncProjects",
@@ -52,11 +54,14 @@ __all__ = [
     "Deployments",
     "AsyncAgents",
     "Agents",
-    "AgentObservability",
     "AsyncAgentObservability",
-    "VectorDatabase",
+    "AgentObservability",
     "AsyncVectorDatabase",
+    "VectorDatabase",
+    "AsyncAgentInference",
     "AgentInference",
     "AsyncExplainability",
     "Explainability",
+    "AsyncTools",
+    "Tools",
 ]

{seekrai-0.5.24 → seekrai-0.5.26}/src/seekrai/resources/chat/completions.py

@@ -20,7 +20,7 @@ class ChatCompletions:
     def create(
         self,
         *,
-        messages: List[Dict[str, str]],
+        messages: List[Dict[str, Any]],
         model: str,
         max_completion_tokens: int | None = None,
         max_tokens: int | None = 512,
@@ -43,8 +43,11 @@ class ChatCompletions:
         Method to generate completions based on a given prompt using a specified model.
 
         Args:
-            messages (List[Dict[str, str]]): A list of messages in the format
-                `[{"role": seekrai.types.chat_completions.MessageRole, "content": TEXT}, ...]`
+            messages (List[Dict[str, Any]]): A list of messages in the format
+                `[{"role": seekrai.types.chat_completions.MessageRole, "content": TEXT}, ...]` or
+                `[{"role": seekrai.types.chat_completions.MessageRole, "content": PARTS}, ...]`
+                where PARTS is a list of content dicts, e.g. {"type": "text", "text": "..."} or
+                {"type": "image_url", "image_url": {"url": "data:image/jpeg;base64,..."}}
             model (str): The name of the model to query.
             max_completion_tokens (int, optional): The maximum number of tokens the output can contain.
             max_tokens (int, optional): The maximum number of tokens to generate.
@@ -141,7 +144,7 @@ class AsyncChatCompletions:
     async def create(
         self,
         *,
-        messages: List[Dict[str, str]],
+        messages: List[Dict[str, Any]],
         model: str,
         max_completion_tokens: int | None = None,
         max_tokens: int | None = 512,
@@ -164,8 +167,11 @@ class AsyncChatCompletions:
         Async method to generate completions based on a given prompt using a specified model.
 
         Args:
-            messages (List[Dict[str, str]]): A list of messages in the format
-                `[{"role": seekrai.types.chat_completions.MessageRole, "content": TEXT}, ...]`
+            messages (List[Dict[str, Any]]): A list of messages in the format
+                `[{"role": seekrai.types.chat_completions.MessageRole, "content": TEXT}, ...]` or
+                `[{"role": seekrai.types.chat_completions.MessageRole, "content": PARTS}, ...]`
+                where PARTS is a list of content dicts, e.g. {"type": "text", "text": "..."} or
+                {"type": "image_url", "image_url": {"url": "data:image/jpeg;base64,..."}}
             model (str): The name of the model to query.
             max_completion_tokens (int, optional): The maximum number of tokens the output can contain.
             max_tokens (int, optional): The maximum number of tokens to generate.

seekrai-0.5.26/src/seekrai/resources/tools.py

@@ -0,0 +1,384 @@
+"""
+Tool management resource for SeekrAI SDK.
+
+This module provides functionality to manage tools independently of agents,
+including creation, listing, retrieval, updating, deletion, and duplication.
+"""
+
+from typing import Any, List, Optional
+
+from pydantic import TypeAdapter
+
+from seekrai.abstract import api_requestor
+from seekrai.seekrflow_response import SeekrFlowResponse
+from seekrai.types import (
+    GetToolsResponse,
+    SeekrFlowClient,
+    SeekrFlowRequest,
+    ToolAgentSummaryResponse,
+    ToolDeleteResponse,
+    ToolResponse,
+)
+from seekrai.types.enums import ToolStatus, ToolType
+from seekrai.types.tools import CreateToolRequest, UpdateToolRequest
+
+
+TOOL_RESPONSE_ADAPTER: TypeAdapter[ToolResponse] = TypeAdapter(ToolResponse)
+
+
+class Tools:
+    """Tool management resource for managing tools independently of agents."""
+
+    def __init__(self, client: SeekrFlowClient) -> None:
+        self._client = client
+        self._requestor = api_requestor.APIRequestor(client=self._client)
+
+    def create(self, request: CreateToolRequest) -> ToolResponse:
+        """
+        Create a new tool.
+
+        Args:
+            request: Tool creation request containing name and typed tool object
+
+        Returns:
+            Created tool response
+        """
+        response, _, _ = self._requestor.request(
+            options=SeekrFlowRequest(
+                method="POST",
+                url="flow/tools",
+                params=request.model_dump(by_alias=True),
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return TOOL_RESPONSE_ADAPTER.validate_python(response.data)
+
+    def list(
+        self,
+        offset: int = 0,
+        limit: int = 100,
+        tool_type: Optional[ToolType] = None,
+        tool_status: Optional[ToolStatus] = None,
+    ) -> GetToolsResponse:
+        """
+        List tools with pagination and filtering.
+
+        Args:
+            offset: Number of records to skip
+            limit: Maximum number of records to return (1-1000)
+            tool_type: Filter by tool type
+            tool_status: Filter by tool status
+
+        Returns:
+            Paginated list of tools
+        """
+        params: dict[str, Any] = {
+            "offset": offset,
+            "limit": limit,
+        }
+
+        if tool_type is not None:
+            params["tool_type"] = tool_type.value
+        if tool_status is not None:
+            params["tool_status"] = tool_status.value
+
+        response, _, _ = self._requestor.request(
+            options=SeekrFlowRequest(
+                method="GET",
+                url="flow/tools",
+                params=params,
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return GetToolsResponse(**response.data)
+
+    def list_agents(self, tool_id: str) -> List[ToolAgentSummaryResponse]:
+        """
+        List agent summaries linked to this tool.
+
+        Args:
+            tool_id: Tool ID linked to agents
+
+        Returns:
+            List of agent summaries
+        """
+        response, _, _ = self._requestor.request(
+            options=SeekrFlowRequest(
+                method="GET",
+                url=f"flow/tools/{tool_id}/agents",
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        assert isinstance(response.data, list)
+        return [ToolAgentSummaryResponse(**agent_tool) for agent_tool in response.data]
+
+    def retrieve(self, tool_id: str) -> ToolResponse:
+        """
+        Retrieve a specific tool by ID.
+
+        Args:
+            tool_id: Tool ID to retrieve
+
+        Returns:
+            Tool response
+        """
+        response, _, _ = self._requestor.request(
+            options=SeekrFlowRequest(
+                method="GET",
+                url=f"flow/tools/{tool_id}",
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return TOOL_RESPONSE_ADAPTER.validate_python(response.data)
+
+    def update(self, tool_id: str, request: UpdateToolRequest) -> ToolResponse:
+        """
+        Update an existing tool.
+
+        Args:
+            tool_id: Tool ID to update
+            request: Tool update request containing updated fields
+
+        Returns:
+            Updated tool response
+        """
+        response, _, _ = self._requestor.request(
+            options=SeekrFlowRequest(
+                method="PUT",
+                url=f"flow/tools/{tool_id}",
+                params=request.model_dump(by_alias=True),
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return TOOL_RESPONSE_ADAPTER.validate_python(response.data)
+
+    def delete(self, tool_id: str) -> ToolDeleteResponse:
+        """
+        Delete a tool if it's not referenced by active agents.
+
+        Args:
+            tool_id: Tool ID to delete
+
+        Returns:
+            Deletion response indicating success/failure
+        """
+        response, _, _ = self._requestor.request(
+            options=SeekrFlowRequest(
+                method="DELETE",
+                url=f"flow/tools/{tool_id}",
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return ToolDeleteResponse(**response.data)
+
+    def duplicate(self, tool_id: str, name: Optional[str]) -> ToolResponse:
+        """
+        Duplicate an existing tool with a new name.
+
+        Args:
+            tool_id: Tool ID to duplicate
+            name (optional): Name for duplicated tool
+
+        Returns:
+            New tool response
+        """
+        params = {}
+        if name is not None:
+            params["name"] = name
+
+        response, _, _ = self._requestor.request(
+            options=SeekrFlowRequest(
+                method="POST",
+                url=f"flow/tools/{tool_id}/duplicate",
+                params=params,
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return TOOL_RESPONSE_ADAPTER.validate_python(response.data)
+
+
+class AsyncTools:
+    """Async tool management resource for managing tools independently of agents."""
+
+    def __init__(self, client: SeekrFlowClient) -> None:
+        self._client = client
+        self._requestor = api_requestor.APIRequestor(client=self._client)
+
+    async def create(self, request: CreateToolRequest) -> ToolResponse:
+        """
+        Create a new tool.
+
+        Args:
+            request: Tool creation request containing name and typed tool object
+
+        Returns:
+            Created tool response
+        """
+        response, _, _ = await self._requestor.arequest(
+            options=SeekrFlowRequest(
+                method="POST",
+                url="flow/tools",
+                params=request.model_dump(by_alias=True),
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return TOOL_RESPONSE_ADAPTER.validate_python(response.data)
+
+    async def list(
+        self,
+        offset: int = 0,
+        limit: int = 100,
+        tool_type: Optional[ToolType] = None,
+        tool_status: Optional[ToolStatus] = None,
+    ) -> GetToolsResponse:
+        """
+        List tools with pagination and filtering.
+
+        Args:
+            offset: Number of records to skip
+            limit: Maximum number of records to return (1-1000)
+            tool_type: Filter by tool type
+            tool_status: Filter by tool status
+
+        Returns:
+            Paginated list of tools
+        """
+        params: dict[str, Any] = {
+            "offset": offset,
+            "limit": limit,
+        }
+
+        if tool_type is not None:
+            params["tool_type"] = tool_type.value
+        if tool_status is not None:
+            params["tool_status"] = tool_status.value
+
+        response, _, _ = await self._requestor.arequest(
+            options=SeekrFlowRequest(
+                method="GET",
+                url="flow/tools",
+                params=params,
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return GetToolsResponse(**response.data)
+
+    async def list_agents(self, tool_id: str) -> List[ToolAgentSummaryResponse]:
+        """
+        List agent summaries linked to this tool.
+
+        Args:
+            tool_id: Tool ID linked to agents
+
+        Returns:
+            List of agent summaries
+        """
+        response, _, _ = await self._requestor.arequest(
+            options=SeekrFlowRequest(
+                method="GET",
+                url=f"flow/tools/{tool_id}/agents",
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        assert isinstance(response.data, list)
+        return [ToolAgentSummaryResponse(**agent_tool) for agent_tool in response.data]
+
+    async def retrieve(self, tool_id: str) -> ToolResponse:
+        """
+        Retrieve a specific tool by ID.
+
+        Args:
+            tool_id: Tool ID to retrieve
+
+        Returns:
+            Tool response
+        """
+        response, _, _ = await self._requestor.arequest(
+            options=SeekrFlowRequest(
+                method="GET",
+                url=f"flow/tools/{tool_id}",
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return TOOL_RESPONSE_ADAPTER.validate_python(response.data)
+
+    async def update(self, tool_id: str, request: UpdateToolRequest) -> ToolResponse:
+        """
+        Update an existing tool.
+
+        Args:
+            tool_id: Tool ID to update
+            request: Tool update request containing updated fields
+
+        Returns:
+            Updated tool response
+        """
+        response, _, _ = await self._requestor.arequest(
+            options=SeekrFlowRequest(
+                method="PUT",
+                url=f"flow/tools/{tool_id}",
+                params=request.model_dump(by_alias=True),
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return TOOL_RESPONSE_ADAPTER.validate_python(response.data)
+
+    async def delete(self, tool_id: str) -> ToolDeleteResponse:
+        """
+        Delete a tool if it's not referenced by active agents.
+
+        Args:
+            tool_id: Tool ID to delete
+
+        Returns:
+            Deletion response indicating success/failure
+        """
+        response, _, _ = await self._requestor.arequest(
+            options=SeekrFlowRequest(
+                method="DELETE",
+                url=f"flow/tools/{tool_id}",
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return ToolDeleteResponse(**response.data)
+
+    async def duplicate(
+        self,
+        tool_id: str,
+        name: Optional[str],
+    ) -> ToolResponse:
+        """
+        Duplicate an existing tool with a new name.
+
+        Args:
+            tool_id: Tool ID to duplicate
+            name (optional): Name for duplicated tool
+
+        Returns:
+            New tool response
+        """
+        params = {}
+        if name is not None:
+            params["name"] = name
+
+        response, _, _ = await self._requestor.arequest(
+            options=SeekrFlowRequest(
+                method="POST", url=f"flow/tools/{tool_id}/duplicate", params=params
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return TOOL_RESPONSE_ADAPTER.validate_python(response.data)

{seekrai-0.5.24 → seekrai-0.5.26}/src/seekrai/types/__init__.py

@@ -2,7 +2,6 @@ from seekrai.types.abstract import SeekrFlowClient
 from seekrai.types.agents import (
     Agent,
     AgentDeleteResponse,
-    AgentStatus,
     CreateAgentRequest,
     EnvConfig,
     FileSearch,
@@ -16,7 +15,6 @@ from seekrai.types.agents import (
     OutputGuardrail,
     OutputMessage,
     OutputText,
-    ReasoningEffort,
     Run,
     RunPython,
     RunPythonEnv,
@@ -44,7 +42,7 @@ from seekrai.types.agents import (
     ThreadStatus,
     Tool,
     ToolBase,
-    ToolType,
+    UpdateAgentRequest,
     WebSearch,
     WebSearchEnv,
 )
@@ -90,6 +88,7 @@ from seekrai.types.deployments import (
     NewDeploymentRequest,
 )
 from seekrai.types.embeddings import EmbeddingRequest, EmbeddingResponse
+from seekrai.types.enums import AgentStatus, ReasoningEffort, ToolStatus, ToolType
 from seekrai.types.files import (
     FileDeleteResponse,
     FileList,
@@ -120,6 +119,25 @@ from seekrai.types.projects import (
     Project,
     ProjectWithRuns,
 )
+from seekrai.types.tools import (
+    CreateFileSearch,
+    CreateRunPython,
+    CreateToolRequest,
+    CreateWebSearch,
+    FileSearchConfig,
+    FileSearchTool,
+    GetToolsResponse,
+    RunPythonConfig,
+    RunPythonTool,
+    ToolAgentSummaryResponse,
+    ToolDeleteResponse,
+    ToolResponse,
+    UpdateFileSearch,
+    UpdateRunPython,
+    UpdateToolRequest,
+    UpdateWebSearch,
+    WebSearchTool,
+)
 
 
 __all__ = [
@@ -209,12 +227,31 @@ __all__ = [
     "Agent",
     "AgentStatus",
     "CreateAgentRequest",
+    "UpdateAgentRequest",
     "ReasoningEffort",
     "AgentDeleteResponse",
     "ToolBase",
     "ToolType",
     "EnvConfig",
     "Tool",
+    "ToolStatus",
+    "FileSearchConfig",
+    "RunPythonConfig",
+    "CreateFileSearch",
+    "CreateWebSearch",
+    "CreateRunPython",
+    "CreateToolRequest",
+    "UpdateWebSearch",
+    "UpdateFileSearch",
+    "UpdateRunPython",
+    "UpdateToolRequest",
+    "ToolResponse",
+    "FileSearchTool",
+    "WebSearchTool",
+    "RunPythonTool",
+    "GetToolsResponse",
+    "ToolAgentSummaryResponse",
+    "ToolDeleteResponse",
     "FileSearch",
     "FileSearchEnv",
     "RunPython",

{seekrai-0.5.24 → seekrai-0.5.26}/src/seekrai/types/agents/__init__.py

@@ -1,9 +1,8 @@
 from seekrai.types.agents.agent import (
     Agent,
     AgentDeleteResponse,
-    AgentStatus,
     CreateAgentRequest,
-    ReasoningEffort,
+    UpdateAgentRequest,
 )
 from seekrai.types.agents.python_functions import (
     DeletePythonFunctionResponse,
@@ -46,7 +45,11 @@ from seekrai.types.agents.threads import (
     ThreadMessageContentType,
     ThreadStatus,
 )
-from seekrai.types.agents.tools import EnvConfig, Tool, ToolBase, ToolType
+from seekrai.types.agents.tools import (
+    EnvConfig,
+    Tool,
+    ToolBase,
+)
 from seekrai.types.agents.tools.schemas import (
     FileSearch,
     FileSearchEnv,
@@ -91,12 +94,10 @@ __all__ = [
     "ThreadMessage",
     "ThreadMessageContentType",
     "Agent",
-    "AgentStatus",
     "CreateAgentRequest",
-    "ReasoningEffort",
+    "UpdateAgentRequest",
     "AgentDeleteResponse",
     "ToolBase",
-    "ToolType",
     "EnvConfig",
     "Tool",
     "FileSearch",