seekrai 0.5.28__py3-none-any.whl → 0.5.29__py3-none-any.whl

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/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/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/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/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/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/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/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

{seekrai-0.5.28.dist-info → seekrai-0.5.29.dist-info}/METADATA

@@ -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.dist-info → seekrai-0.5.29.dist-info}/RECORD

@@ -30,17 +30,18 @@ seekrai/resources/resource_base.py,sha256=rFIHFeqKPiAEbMYcMiIGHIym7qxwmh-EGsWiZc
 seekrai/resources/tools.py,sha256=NH47V0Ks6yXLr0yzSVegWGKJNF1PblUhkYThz0BBKO0,11259
 seekrai/resources/vectordb.py,sha256=1uUsyCUJdVAVUnua9RJWqf-j0HDD74EPeERUBjqVvmg,15797
 seekrai/seekrflow_response.py,sha256=5RFEQzamDy7sTSDkxSsZQThZ3biNmeCPeHWdrFId5Go,1320
-seekrai/types/__init__.py,sha256=-9Z2sTV_qsdQ0VOoh3_R8amGFJpQiESFxcTOrTXgCC4,5816
+seekrai/types/__init__.py,sha256=c27Om5Yv8f-V-d9reA96ecPjc5ldeVnNLDjxrdi-VTY,6067
 seekrai/types/abstract.py,sha256=TqWFQV_6bPblywfCH-r8FCkXWvPkc9KlJ4QVgyrnaMc,642
-seekrai/types/agents/__init__.py,sha256=yEyuhWaiBP0e5l669O0FqSEnRMPhJC473JgYG7YFPUs,2356
+seekrai/types/agents/__init__.py,sha256=Ukt-s8yz1sLNQuQUYxfi5_77Ni_oGqetjqBajW5sWx4,2404
 seekrai/types/agents/agent.py,sha256=VIMF6gXhlLViVsCBGgPo54UdJqpMjb_ystNNx9n94FY,1543
 seekrai/types/agents/observability.py,sha256=kOpBrNsrOzSbf-AsRBWAxRB9FqO0063CshUr-r2hHBE,963
 seekrai/types/agents/python_functions.py,sha256=31Jm46gryHsgNsC7nlivuyY0TSko58y2YVxsu_7bEAg,653
 seekrai/types/agents/runs.py,sha256=JgvasNM_eesGbwnqeAbRzPNvbzLAG1_N2IU7M7OHD1w,4933
 seekrai/types/agents/threads.py,sha256=TinCMKv1bi5LzboDyCx1XI4Zzd8UzUZos4VOrTNhmEc,6835
-seekrai/types/agents/tools/__init__.py,sha256=Uh9kzxdUQw8RPpeazUqP3FYMmbfLzeSldGgL8dD4K5U,509
+seekrai/types/agents/tools/__init__.py,sha256=91FbBuKKMxloI-iebI7FYLcWIJT8dwz5tyAnteaXkpU,557
 seekrai/types/agents/tools/env_model_config.py,sha256=9POx2DPwfSXgoaziJv7QvKeMrhMsYD1exnanSRK48vw,177
-seekrai/types/agents/tools/schemas/__init__.py,sha256=_HQoqgBk4T1IvlD0Yt3REz0_x_oVFTgITTd7H2ZJXws,562
+seekrai/types/agents/tools/schemas/__init__.py,sha256=D4m_2mTjQM_zWPcalKgoGWq6aKdrCqZRvc2d6FOtJQo,666
+seekrai/types/agents/tools/schemas/agent_as_tool.py,sha256=ypNRKgwZDKzNyHnbLNbr7dcUIOwFYgHBujuOgPVob5Q,420
 seekrai/types/agents/tools/schemas/file_search.py,sha256=0e9jL4B7-GSYGrkGBA4I6PHd2vP6vz8KfLSp2DIeg2A,459
 seekrai/types/agents/tools/schemas/file_search_env.py,sha256=VGM3fkBjbp0wJVwgtgxKw_kEKP0qwvkGpJQZsvlOtcY,683
 seekrai/types/agents/tools/schemas/run_python.py,sha256=Gq-LAXyXswW_IhHBfdEvw3WZuuI62R9HRfj5YoLp06c,451
@@ -48,14 +49,14 @@ seekrai/types/agents/tools/schemas/run_python_env.py,sha256=0GcyrKILbRRz-6CufG9i
 seekrai/types/agents/tools/schemas/web_search.py,sha256=jAF9OMQbUODd5egzbct5CgFtMm-0Y62lvXFRbEONAlk,451
 seekrai/types/agents/tools/schemas/web_search_env.py,sha256=R3fGXV43ZacbgoSy-M43rsTe5FFPU-t2VPMBnTVj7M8,283
 seekrai/types/agents/tools/tool.py,sha256=XdbYzSn0Rr-m9s34Aa1NDQ-35Kpzk4rcaXY3dKrg-Ys,514
-seekrai/types/agents/tools/tool_types.py,sha256=1tF_kE6Z_zzuZpOAK1HrHsHkXFPEoK0PdYv-pbTLfkY,360
+seekrai/types/agents/tools/tool_types.py,sha256=5MtxL-DY2VUA0RHPO7T3BluRsoYCWZLXeHkEww5SVcA,469
 seekrai/types/alignment.py,sha256=nWcc4kQLs40-T0_HC3MnGkLd-StwBvwCXQrjUVJ5dEI,2973
 seekrai/types/chat_completions.py,sha256=-ganbaf5wDlNIj6zWGjcmkCwTIoCUthyVO9vuBcQiwk,3759
 seekrai/types/common.py,sha256=YI1pE-i_lDLU2o6FjoINdIhPXsV9lUl2MeAg2aRtT-M,2062
 seekrai/types/completions.py,sha256=lm9AFdZR3Xg5AHPkV-qETHikkwMJmkHrLGr5GG-YR-M,2171
 seekrai/types/deployments.py,sha256=a0zew1DuB9vPQXcBT2R4Tdn_8z5qleh6V6i4T4xyYZo,1798
 seekrai/types/embeddings.py,sha256=OANoLNOs0aceS8NppVvvcNYQbF7-pAOAmcr30pw64OU,749
-seekrai/types/enums.py,sha256=sQ1CW-ctbhpV2jM1cEAEy7ZUdzZa0IC85YvycjvudHE,633
+seekrai/types/enums.py,sha256=v6HDfpgKJ__fCouWYPBbfD6ZjsXCMrnNt9zTuALBsp0,669
 seekrai/types/error.py,sha256=uTKISs9aRC4_6zwirtNkanxepN8KY-SqCq0kNbfZylQ,370
 seekrai/types/explainability.py,sha256=Ih-8hCm5r22EMMtr83cDy8vePo7_Ik7UdUcXhsj5Zm0,835
 seekrai/types/files.py,sha256=kOy4s8D4tlsenyWmiiEyAS0jDAdxMScBu5j1GwQCf3E,2808
@@ -64,7 +65,7 @@ seekrai/types/images.py,sha256=Fusj8OhVYFsT8kz636lRGGivLbPXo_ZNgakKwmzJi3U,914
 seekrai/types/ingestion.py,sha256=uUdKOR4xqSfAXWQOR1UOltSlOnuyAwKVA1Q2a6Yslk8,919
 seekrai/types/models.py,sha256=9Z0nvLdlAfpF8mNRW5-IqBdDHoE-3qQ5przmIDJgwLo,1345
 seekrai/types/projects.py,sha256=JFgpZdovia8Orcnhp6QkIEAXzyPCfKT_bUiwjxUaHHQ,670
-seekrai/types/tools.py,sha256=k6CcZwFhP_is-9KwZhRMWhhp6-z6HKwbbnrg0LOsS2I,3525
+seekrai/types/tools.py,sha256=VkpEFVyTEGk3lzBky6lh7sAQF9iWGZfFPRI417tavLI,3936
 seekrai/types/vectordb.py,sha256=Fw3ZrEvWU5OHzXNIUXtD24JVG7pCULfudX_WDOazJnk,2454
 seekrai/utils/__init__.py,sha256=dfbiYEc47EBVRkq6C4O9y6tTGuPuV3LbV3__v01Mbds,658
 seekrai/utils/_log.py,sha256=Cayw5B394H2WGVTXPXS2AN8znQdxsgrLqADXgqmokvU,1649
@@ -72,8 +73,8 @@ seekrai/utils/api_helpers.py,sha256=0Y8BblNIr9h_R12zdmhkxgTlxgoRkbq84QNi4nNWGu8,
 seekrai/utils/files.py,sha256=7ixn_hgV-6pEhYqLyOp-EN0o8c1CzUwJzX9n3PQ5oqo,7164
 seekrai/utils/tools.py,sha256=jgJTL-dOIouDbEJLdQpQfpXhqaz_poQYS52adyUtBjo,1781
 seekrai/version.py,sha256=q6iGQVFor8zXiPP5F-3vy9TndOxKv5JXbaNJ2kdOQws,125
-seekrai-0.5.28.dist-info/METADATA,sha256=LQLnopWdiNd6l3OJP92tv-DzgJWOa_jw_0lbXooI2JE,4788
-seekrai-0.5.28.dist-info/WHEEL,sha256=kJCRJT_g0adfAJzTx2GUMmS80rTJIVHRCfG0DQgLq3o,88
-seekrai-0.5.28.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
-seekrai-0.5.28.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-seekrai-0.5.28.dist-info/RECORD,,
+seekrai-0.5.29.dist-info/METADATA,sha256=bHPgRE1YAJ86dCrS3VCEoE2rMt7lh63iy2tnZNiCAh0,4788
+seekrai-0.5.29.dist-info/WHEEL,sha256=kJCRJT_g0adfAJzTx2GUMmS80rTJIVHRCfG0DQgLq3o,88
+seekrai-0.5.29.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
+seekrai-0.5.29.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+seekrai-0.5.29.dist-info/RECORD,,