seekrai 0.5.28__tar.gz → 0.5.29__tar.gz

{seekrai-0.5.28 → seekrai-0.5.29}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: seekrai
-Version: 0.5.28
+Version: 0.5.29
 Summary: Python client for SeekrAI
 License: Apache-2.0
 License-File: LICENSE

{seekrai-0.5.28 → seekrai-0.5.29}/pyproject.toml

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

{seekrai-0.5.28 → seekrai-0.5.29}/src/seekrai/types/__init__.py

@@ -1,6 +1,7 @@
 from seekrai.types.abstract import SeekrFlowClient
 from seekrai.types.agents import (
     Agent,
+    AgentAsToolLegacy,
     AgentDeleteResponse,
     CreateAgentRequest,
     EnvConfig,
@@ -47,6 +48,7 @@ from seekrai.types.agents import (
     WebSearchEnv,
 )
 from seekrai.types.agents.tools.schemas import (
+    AgentAsToolLegacy,
     FileSearch,
     FileSearchEnv,
     RunPython,
@@ -120,6 +122,9 @@ from seekrai.types.projects import (
     ProjectWithRuns,
 )
 from seekrai.types.tools import (
+    AgentAsTool,
+    AgentAsToolConfig,
+    CreateAgentAsTool,
     CreateFileSearch,
     CreateRunPython,
     CreateToolRequest,
@@ -132,6 +137,7 @@ from seekrai.types.tools import (
     ToolAgentSummaryResponse,
     ToolDeleteResponse,
     ToolResponse,
+    UpdateAgentAsTool,
     UpdateFileSearch,
     UpdateRunPython,
     UpdateToolRequest,
@@ -258,4 +264,9 @@ __all__ = [
     "RunPythonEnv",
     "WebSearch",
     "WebSearchEnv",
+    "AgentAsToolLegacy",
+    "AgentAsTool",
+    "CreateAgentAsTool",
+    "AgentAsToolConfig",
+    "UpdateAgentAsTool",
 ]

{seekrai-0.5.28 → seekrai-0.5.29}/src/seekrai/types/agents/__init__.py

@@ -51,6 +51,7 @@ from seekrai.types.agents.tools import (
     ToolBase,
 )
 from seekrai.types.agents.tools.schemas import (
+    AgentAsToolLegacy,
     FileSearch,
     FileSearchEnv,
     RunPython,
@@ -106,6 +107,7 @@ __all__ = [
     "RunPythonEnv",
     "WebSearch",
     "WebSearchEnv",
+    "AgentAsToolLegacy",
     "PythonFunctionBase",
     "PythonFunctionResponse",
     "DeletePythonFunctionResponse",

{seekrai-0.5.28 → seekrai-0.5.29}/src/seekrai/types/agents/tools/__init__.py

@@ -1,5 +1,6 @@
 from seekrai.types.agents.tools.env_model_config import EnvConfig
 from seekrai.types.agents.tools.schemas import (
+    AgentAsToolLegacy,
     FileSearch,
     FileSearchEnv,
     RunPython,
@@ -23,4 +24,5 @@ __all__ = [
     "RunPythonEnv",
     "WebSearch",
     "WebSearchEnv",
+    "AgentAsToolLegacy",
 ]

{seekrai-0.5.28 → seekrai-0.5.29}/src/seekrai/types/agents/tools/schemas/__init__.py

@@ -1,3 +1,4 @@
+from seekrai.types.agents.tools.schemas.agent_as_tool import AgentAsToolLegacy
 from seekrai.types.agents.tools.schemas.file_search import FileSearch
 from seekrai.types.agents.tools.schemas.file_search_env import FileSearchEnv
 from seekrai.types.agents.tools.schemas.run_python import RunPython
@@ -13,4 +14,5 @@ __all__ = [
     "RunPythonEnv",
     "WebSearch",
     "WebSearchEnv",
+    "AgentAsToolLegacy",
 ]

seekrai-0.5.29/src/seekrai/types/agents/tools/schemas/agent_as_tool.py

@@ -0,0 +1,16 @@
+from typing import Literal
+
+from seekrai.types.agents.tools import EnvConfig
+from seekrai.types.agents.tools.tool import ToolBase, ToolType
+
+
+class AgentAsToolLegacy(ToolBase[Literal["agent_as_tool"], EnvConfig]):
+    name: Literal["agent_as_tool"] = ToolType.AGENT_AS_TOOL.value
+    description: str
+    agent_id: str
+
+    model_config = {
+        "json_schema_extra": {
+            "deprecated": True,
+        }
+    }

{seekrai-0.5.28 → seekrai-0.5.29}/src/seekrai/types/agents/tools/tool_types.py

@@ -2,9 +2,13 @@ from typing import Annotated, Union
 
 from pydantic import Field
 
+from seekrai.types.agents.tools.schemas.agent_as_tool import AgentAsToolLegacy
 from seekrai.types.agents.tools.schemas.file_search import FileSearch
 from seekrai.types.agents.tools.schemas.run_python import RunPython
 from seekrai.types.agents.tools.schemas.web_search import WebSearch
 
 
-Tool = Annotated[Union[FileSearch, WebSearch, RunPython], Field(discriminator="name")]
+Tool = Annotated[
+    Union[FileSearch, WebSearch, RunPython, AgentAsToolLegacy],
+    Field(discriminator="name"),
+]

{seekrai-0.5.28 → seekrai-0.5.29}/src/seekrai/types/enums.py

@@ -28,3 +28,4 @@ class ToolType(str, Enum):
     FILE_SEARCH = "file_search"
     WEB_SEARCH = "web_search"
     RUN_PYTHON = "run_python"
+    AGENT_AS_TOOL = "agent_as_tool"

{seekrai-0.5.28 → seekrai-0.5.29}/src/seekrai/types/tools.py

@@ -62,6 +62,15 @@ class CreateRunPython(CreateTool):
     config: RunPythonConfig
 
 
+class AgentAsToolConfig(ToolConfig):
+    agent_id: str
+
+
+class CreateAgentAsTool(CreateTool):
+    type: Literal[ToolType.AGENT_AS_TOOL] = ToolType.AGENT_AS_TOOL
+    config: AgentAsToolConfig
+
+
 class Tool(BaseModel):
     type: ToolType
     name: Annotated[str, AfterValidator(validate_length)]
@@ -88,13 +97,19 @@ class RunPythonTool(Tool):
     config: RunPythonConfig
 
 
+class AgentAsTool(Tool):
+    type: Literal[ToolType.AGENT_AS_TOOL] = ToolType.AGENT_AS_TOOL
+    config: AgentAsToolConfig
+
+
 CreateToolRequest = Annotated[
-    Union[CreateFileSearch, CreateRunPython, CreateWebSearch],
+    Union[CreateFileSearch, CreateRunPython, CreateWebSearch, CreateAgentAsTool],
     Field(discriminator="type"),
 ]
 
 ToolResponse = Annotated[
-    Union[FileSearchTool, WebSearchTool, RunPythonTool], Field(discriminator="type")
+    Union[FileSearchTool, WebSearchTool, RunPythonTool, AgentAsTool],
+    Field(discriminator="type"),
 ]
 
 
@@ -110,6 +125,10 @@ class UpdateRunPython(CreateRunPython):
     pass
 
 
+class UpdateAgentAsTool(CreateAgentAsTool):
+    pass
+
+
 UpdateToolRequest = Annotated[
     Union[UpdateFileSearch, UpdateWebSearch, UpdateRunPython],
     Field(discriminator="type"),
@@ -117,7 +136,7 @@ UpdateToolRequest = Annotated[
 
 
 class GetToolsResponse(BaseModel):
-    """Response schema for paginated tool list."""
+    """Response schema for a paginated tool list."""
 
     data: list[ToolResponse]
     total: Optional[int] = None