seekrai 0.5.24__tar.gz → 0.5.25__tar.gz

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

@@ -1,9 +1,9 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: seekrai
-Version: 0.5.24
+Version: 0.5.25
 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.25}/pyproject.toml

@@ -13,7 +13,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "seekrai"
-version = "0.5.24"
+version = "0.5.25"
 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.25}/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.25}/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.25/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.25}/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.25}/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",

{seekrai-0.5.24 → seekrai-0.5.25}/src/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-0.5.24 → seekrai-0.5.25}/src/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-0.5.25/src/seekrai/types/agents/tools/schemas/file_search.py

@@ -0,0 +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
+from seekrai.types.enums import ToolType
+
+
+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-0.5.24 → seekrai-0.5.25}/src/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,
+        }
+    }

seekrai-0.5.25/src/seekrai/types/agents/tools/schemas/run_python.py

@@ -0,0 +1,16 @@
+from typing import Literal
+
+from seekrai.types.agents.tools.schemas.run_python_env import RunPythonEnv
+from seekrai.types.agents.tools.tool import ToolBase
+from seekrai.types.enums import ToolType
+
+
+class RunPython(ToolBase[Literal["run_python"], RunPythonEnv]):
+    name: Literal["run_python"] = ToolType.RUN_PYTHON.value
+    tool_env: RunPythonEnv
+
+    model_config = {
+        "json_schema_extra": {
+            "deprecated": True,
+        }
+    }

{seekrai-0.5.24 → seekrai-0.5.25}/src/seekrai/types/agents/tools/schemas/run_python_env.py

@@ -6,3 +6,9 @@ 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
+
+    model_config = {
+        "json_schema_extra": {
+            "deprecated": True,
+        }
+    }

seekrai-0.5.25/src/seekrai/types/agents/tools/schemas/web_search.py

@@ -0,0 +1,16 @@
+from typing import Literal
+
+from seekrai.types.agents.tools.schemas.web_search_env import WebSearchEnv
+from seekrai.types.agents.tools.tool import ToolBase
+from seekrai.types.enums import ToolType
+
+
+class WebSearch(ToolBase[Literal["web_search"], WebSearchEnv]):
+    name: Literal["web_search"] = ToolType.WEB_SEARCH.value
+    tool_env: WebSearchEnv
+
+    model_config = {
+        "json_schema_extra": {
+            "deprecated": True,
+        }
+    }

{seekrai-0.5.24 → seekrai-0.5.25}/src/seekrai/types/agents/tools/schemas/web_search_env.py

@@ -5,3 +5,9 @@ from seekrai.types.agents.tools.env_model_config import EnvConfig
 
 class WebSearchEnv(EnvConfig):
     web_search_tool_description: Optional[str] = None
+
+    model_config = {
+        "json_schema_extra": {
+            "deprecated": True,
+        }
+    }

seekrai-0.5.25/src/seekrai/types/agents/tools/tool.py

@@ -0,0 +1,23 @@
+from typing import Generic, TypeVar
+
+from pydantic import BaseModel
+
+from seekrai.types.agents.tools.env_model_config import EnvConfig
+from seekrai.types.enums import ToolType as SeekrToolType
+
+
+TName = TypeVar("TName", bound=str)
+TEnv = TypeVar("TEnv", bound=EnvConfig)
+
+ToolType = SeekrToolType
+
+
+class ToolBase(BaseModel, Generic[TName, TEnv]):
+    name: TName
+    tool_env: TEnv
+
+
+__all__ = [
+    "ToolBase",
+    "ToolType",
+]  # explicitly re-export ToolType from old namespace to keep backward compatibility

seekrai-0.5.25/src/seekrai/types/enums.py

@@ -0,0 +1,30 @@
+from enum import Enum
+
+
+class AgentStatus(str, Enum):
+    INACTIVE = "Inactive"
+    PENDING = "Pending"
+    ACTIVE = "Active"
+    UPDATING = "Updating"
+    FAILED = "Failed"
+
+
+class ReasoningEffort(str, Enum):
+    PERFORMANCE_OPTIMIZED = "performance_optimized"
+    SPEED_OPTIMIZED = "speed_optimized"
+
+
+class ToolStatus(str, Enum):
+    """Tool status enumeration."""
+
+    ACTIVE = "Active"
+    INACTIVE = "Inactive"
+    DEPRECATED = "Deprecated"
+
+
+class ToolType(str, Enum):
+    """Tool type enumeration used to discriminate the union."""
+
+    FILE_SEARCH = "file_search"
+    WEB_SEARCH = "web_search"
+    RUN_PYTHON = "run_python"

{seekrai-0.5.24 → seekrai-0.5.25}/src/seekrai/types/files.py

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

{seekrai-0.5.24 → seekrai-0.5.25}/src/seekrai/types/finetune.py

@@ -77,7 +77,7 @@ class FinetuneEventType(str, Enum):
 
 class FineTuneType(str, Enum):
     STANDARD = "STANDARD"
-    DPO = "DPO"
+    PREFERENCE = "PREFERENCE"
     GRPO = "GRPO"
 
 

seekrai-0.5.25/src/seekrai/types/tools.py

@@ -0,0 +1,141 @@
+from datetime import datetime
+from typing import Annotated, Literal, Optional, Union
+
+from pydantic import AfterValidator, ConfigDict, Field
+from pydantic_core.core_schema import ValidationInfo
+
+from seekrai.types.abstract import BaseModel
+from seekrai.types.enums import AgentStatus, ToolType
+
+
+def validate_length(value: str, info: ValidationInfo) -> str:
+    """Validate that string fields have meaningful content."""
+    if len(value) <= 0:
+        raise ValueError(f"{info.field_name} needs to have a length of at least 1")
+    if len(value.strip()) <= 0:
+        raise ValueError(f"{info.field_name} cannot be only whitespace")
+    return value
+
+
+class ToolConfig(BaseModel):
+    model_config = ConfigDict(
+        alias_generator=lambda k: k.upper(), populate_by_name=True
+    )
+
+
+class CreateTool(BaseModel):
+    model_config = ConfigDict(from_attributes=True, extra="forbid")
+
+    type: ToolType
+    name: Annotated[str, AfterValidator(validate_length)]
+    description: Annotated[str, AfterValidator(validate_length)]
+    config: ToolConfig
+
+
+class FileSearchConfig(ToolConfig):
+    file_search_index: str = Field(min_length=1)
+    embedding_model: Union[str, None] = None
+    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"
+    )
+
+
+class CreateFileSearch(CreateTool):
+    type: Literal[ToolType.FILE_SEARCH] = ToolType.FILE_SEARCH
+    config: FileSearchConfig
+
+
+class CreateWebSearch(CreateTool):
+    type: Literal[ToolType.WEB_SEARCH] = ToolType.WEB_SEARCH
+    config: ToolConfig = ToolConfig()
+
+
+class RunPythonConfig(ToolConfig):
+    function_ids: Union[list[str], None] = None
+
+
+class CreateRunPython(CreateTool):
+    type: Literal[ToolType.RUN_PYTHON] = ToolType.RUN_PYTHON
+    config: RunPythonConfig
+
+
+class Tool(BaseModel):
+    type: ToolType
+    name: Annotated[str, AfterValidator(validate_length)]
+    description: Union[str, None]
+    config: ToolConfig
+    id: str
+    version: int
+    created_at: datetime
+    updated_at: datetime
+
+
+class FileSearchTool(Tool):
+    type: Literal[ToolType.FILE_SEARCH] = ToolType.FILE_SEARCH
+    config: FileSearchConfig
+
+
+class WebSearchTool(Tool):
+    type: Literal[ToolType.WEB_SEARCH] = ToolType.WEB_SEARCH
+    config: ToolConfig = ToolConfig()
+
+
+class RunPythonTool(Tool):
+    type: Literal[ToolType.RUN_PYTHON] = ToolType.RUN_PYTHON
+    config: RunPythonConfig
+
+
+CreateToolRequest = Annotated[
+    Union[CreateFileSearch, CreateRunPython, CreateWebSearch],
+    Field(discriminator="type"),
+]
+
+ToolResponse = Annotated[
+    Union[FileSearchTool, WebSearchTool, RunPythonTool], Field(discriminator="type")
+]
+
+
+class UpdateFileSearch(CreateFileSearch):
+    pass
+
+
+class UpdateWebSearch(CreateWebSearch):
+    pass
+
+
+class UpdateRunPython(CreateRunPython):
+    pass
+
+
+UpdateToolRequest = Annotated[
+    Union[UpdateFileSearch, UpdateWebSearch, UpdateRunPython],
+    Field(discriminator="type"),
+]
+
+
+class GetToolsResponse(BaseModel):
+    """Response schema for paginated tool list."""
+
+    data: list[ToolResponse]
+    total: Optional[int] = None
+
+
+class ToolAgentSummaryResponse(BaseModel):
+    """Schema for the summary for agents linked to a given tool."""
+
+    model_config = ConfigDict(from_attributes=True)
+
+    id: str
+    name: str
+    status: AgentStatus
+
+
+class ToolDeleteResponse(BaseModel):
+    """Response schema for tool deletion."""
+
+    id: str
+    deleted: bool
+    message: str

seekrai-0.5.24/src/seekrai/types/agents/tools/schemas/file_search.py

@@ -1,9 +0,0 @@
-from typing import Literal
-
-from seekrai.types.agents.tools.schemas.file_search_env import FileSearchEnv
-from seekrai.types.agents.tools.tool import ToolBase, ToolType
-
-
-class FileSearch(ToolBase[FileSearchEnv]):
-    name: Literal[ToolType.FILE_SEARCH] = ToolType.FILE_SEARCH
-    tool_env: FileSearchEnv

seekrai-0.5.24/src/seekrai/types/agents/tools/schemas/run_python.py

@@ -1,9 +0,0 @@
-from typing import Literal
-
-from seekrai.types.agents.tools.schemas.run_python_env import RunPythonEnv
-from seekrai.types.agents.tools.tool import ToolBase, ToolType
-
-
-class RunPython(ToolBase[RunPythonEnv]):
-    name: Literal[ToolType.RUN_PYTHON] = ToolType.RUN_PYTHON
-    tool_env: RunPythonEnv

seekrai-0.5.24/src/seekrai/types/agents/tools/schemas/web_search.py

@@ -1,9 +0,0 @@
-from typing import Literal
-
-from seekrai.types.agents.tools.schemas.web_search_env import WebSearchEnv
-from seekrai.types.agents.tools.tool import ToolBase, ToolType
-
-
-class WebSearch(ToolBase[WebSearchEnv]):
-    name: Literal[ToolType.WEB_SEARCH] = ToolType.WEB_SEARCH
-    tool_env: WebSearchEnv

seekrai-0.5.24/src/seekrai/types/agents/tools/tool.py

@@ -1,20 +0,0 @@
-import enum
-from typing import Generic, TypeVar
-
-from pydantic import BaseModel
-
-from seekrai.types.agents.tools.env_model_config import EnvConfig
-
-
-class ToolType(str, enum.Enum):
-    FILE_SEARCH = "file_search"
-    WEB_SEARCH = "web_search"
-    RUN_PYTHON = "run_python"
-
-
-TEnv = TypeVar("TEnv", bound=EnvConfig)
-
-
-class ToolBase(BaseModel, Generic[TEnv]):
-    name: ToolType
-    tool_env: TEnv