seekrai 0.5.17__py3-none-any.whl → 0.5.25__py3-none-any.whl

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/resources/vectordb.py

@@ -250,6 +250,21 @@ class VectorDatabase(ResourceBase):
         # The endpoint returns 204 No Content
         return None
 
+    def delete_file(self, database_id: str, file_id: str) -> None:
+        """Delete a file from a vector database."""
+        requestor = api_requestor.APIRequestor(client=self._client)
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="DELETE",
+                url=f"flow/vectordb/{database_id}/files/{file_id}",
+            ),
+            stream=False,
+        )
+
+        # The endpoint returns 204 No Content
+        return None
+
 
 class AsyncVectorDatabase(ResourceBase):
     async def create(
@@ -484,3 +499,18 @@ class AsyncVectorDatabase(ResourceBase):
 
         # The endpoint returns 204 No Content
         return None
+
+    async def delete_file(self, database_id: str, file_id: str) -> None:
+        """Delete a file from a vector database asynchronously."""
+        requestor = api_requestor.APIRequestor(client=self._client)
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="DELETE",
+                url=f"flow/vectordb/{database_id}/files/{file_id}",
+            ),
+            stream=False,
+        )
+
+        # The endpoint returns 204 No Content
+        return None

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,
 )
@@ -61,9 +59,13 @@ from seekrai.types.alignment import (
     AlignmentEstimationResponse,
     AlignmentJobStatus,
     AlignmentList,
+    AlignmentOutput,
     AlignmentRequest,
     AlignmentResponse,
     AlignmentType,
+    SystemPrompt,
+    SystemPromptCreateRequest,
+    SystemPromptUpdateRequest,
 )
 from seekrai.types.chat_completions import (
     ChatCompletionChunk,
@@ -86,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,
@@ -102,6 +105,7 @@ from seekrai.types.finetune import (
     FinetuneRequest,
     FinetuneResponse,
     InfrastructureConfig,
+    LoRAConfig,
     TrainingConfig,
 )
 from seekrai.types.images import (
@@ -115,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__ = [
@@ -135,6 +158,7 @@ __all__ = [
     "FinetuneDownloadResult",
     "InfrastructureConfig",
     "TrainingConfig",
+    "LoRAConfig",
     "FileRequest",
     "FileResponse",
     "FileList",
@@ -151,8 +175,12 @@ __all__ = [
     "AlignmentEstimationResponse",
     "AlignmentResponse",
     "AlignmentJobStatus",
+    "AlignmentOutput",
     "AlignmentList",
     "AlignmentType",
+    "SystemPrompt",
+    "SystemPromptCreateRequest",
+    "SystemPromptUpdateRequest",
     "Project",
     "ProjectWithRuns",
     "GetProjectsResponse",
@@ -199,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/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",

seekrai/types/agents/agent.py

@@ -1,31 +1,34 @@
-import enum
 from datetime import datetime
 from typing import Optional
 
-from pydantic import BaseModel, ConfigDict, Field
+from pydantic import BaseModel, ConfigDict, Field, model_validator
 
 from seekrai.types.agents.tools.tool_types import Tool
-
-
-class AgentStatus(str, enum.Enum):
-    INACTIVE = "Inactive"
-    PENDING = "Pending"
-    ACTIVE = "Active"
-    FAILED = "Failed"
-
-
-class ReasoningEffort(str, enum.Enum):
-    PERFORMANCE_OPTIMIZED = "performance_optimized"
-    SPEED_OPTIMIZED = "speed_optimized"
+from seekrai.types.enums import AgentStatus, ReasoningEffort
 
 
 class CreateAgentRequest(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
     name: str
     instructions: str
-    tools: list[Tool]
+    tools: Optional[list[Tool]] = Field(
+        deprecated="This field is deprecated. Use tool_ids instead.",
+        default=None,
+    )
+    tool_ids: Optional[list[str]] = None
     model_id: str
     reasoning_effort: Optional[ReasoningEffort] = None
 
+    @model_validator(mode="after")
+    def validate_tools_or_tool_ids(self) -> "CreateAgentRequest":
+        """Validate that either tools or tool_ids is provided, but not both."""
+        if self.tools is not None and self.tool_ids is not None:
+            raise ValueError(
+                "Only one of 'tools' or 'tool_ids' can be provided, but both were provided."
+            )
+        return self
+
 
 class Agent(BaseModel):
     model_config = ConfigDict(from_attributes=True)

seekrai/types/agents/tools/__init__.py

@@ -7,15 +7,15 @@ from seekrai.types.agents.tools.schemas import (
     WebSearch,
     WebSearchEnv,
 )
-from seekrai.types.agents.tools.tool import ToolBase, ToolType
-from seekrai.types.agents.tools.tool_types import Tool
+from seekrai.types.agents.tools.tool import ToolBase
+from seekrai.types.agents.tools.tool_types import (
+    Tool,
+)
 
 
 __all__ = [
     "ToolBase",
-    "ToolType",
     "EnvConfig",
-    "Env",
     "Tool",
     "FileSearch",
     "FileSearchEnv",

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

@@ -1,9 +1,16 @@
 from typing import Literal
 
 from seekrai.types.agents.tools.schemas.file_search_env import FileSearchEnv
-from seekrai.types.agents.tools.tool import ToolBase, ToolType
+from seekrai.types.agents.tools.tool import ToolBase
+from seekrai.types.enums import ToolType
 
 
-class FileSearch(ToolBase[FileSearchEnv]):
-    name: Literal[ToolType.FILE_SEARCH] = ToolType.FILE_SEARCH
+class FileSearch(ToolBase[Literal["file_search"], FileSearchEnv]):
+    name: Literal["file_search"] = ToolType.FILE_SEARCH.value
     tool_env: FileSearchEnv
+
+    model_config = {
+        "json_schema_extra": {
+            "deprecated": True,
+        }
+    }

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

@@ -13,3 +13,9 @@ class FileSearchEnv(EnvConfig):
     score_threshold: float = Field(
         default=0, ge=0, lt=1.0, description="Score must be ≥ 0.0 and < 1.0"
     )
+
+    model_config = {
+        "json_schema_extra": {
+            "deprecated": True,
+        }
+    }