seekrai 0.5.5__py3-none-any.whl → 0.5.6__py3-none-any.whl

seekrai/resources/agents/agents.py

@@ -7,6 +7,7 @@ from seekrai.types.agents.agent import (
     Agent,
     AgentDeleteResponse,
     CreateAgentRequest,
+    UpdateAgentRequest,
 )
 
 
@@ -140,6 +141,28 @@ class Agents:
         assert isinstance(response, SeekrFlowResponse)
         return AgentDeleteResponse(**response.data)
 
+    def update(self, agent_id: str, request: UpdateAgentRequest) -> Agent:
+        """
+        Update an existing agent's configuration.
+
+        Args:
+            agent_id: The ID of the agent to update.
+            request: The request object containing updated agent config.
+
+        Returns:
+            The updated agent.
+        """
+        response, _, _ = self._requestor.request(
+            options=SeekrFlowRequest(
+                method="PUT",
+                url=f"flow/agents/{agent_id}/update",
+                params=request.model_dump(),
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return Agent(**response.data)
+
 
 class AsyncAgents:
     def __init__(self, client: SeekrFlowClient) -> None:
@@ -270,3 +293,25 @@ class AsyncAgents:
 
         assert isinstance(response, SeekrFlowResponse)
         return AgentDeleteResponse(**response.data)
+
+    async def update(self, agent_id: str, request: UpdateAgentRequest) -> Agent:
+        """
+        Update an existing agent's configuration.
+
+        Args:
+            agent_id: The ID of the agent to update.
+            request: The request object containing updated agent config.
+
+        Returns:
+            The updated agent.
+        """
+        response, _, _ = await self._requestor.arequest(
+            options=SeekrFlowRequest(
+                method="PUT",
+                url=f"flow/agents/{agent_id}/update",
+                params=request.model_dump(),
+            ),
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return Agent(**response.data)

seekrai/types/__init__.py

@@ -4,7 +4,6 @@ from seekrai.types.agents import (
     AgentDeleteResponse,
     AgentStatus,
     CreateAgentRequest,
-    Env,
     EnvConfig,
     FileSearch,
     FileSearchEnv,
@@ -19,6 +18,8 @@ from seekrai.types.agents import (
     OutputText,
     ReasoningEffort,
     Run,
+    RunPython,
+    RunPythonEnv,
     RunRequest,
     RunResponse,
     RunStatus,
@@ -44,8 +45,17 @@ from seekrai.types.agents import (
     Tool,
     ToolBase,
     ToolType,
+    WebSearch,
+    WebSearchEnv,
+)
+from seekrai.types.agents.tools.schemas import (
+    FileSearch,
+    FileSearchEnv,
+    RunPython,
+    RunPythonEnv,
+    WebSearch,
+    WebSearchEnv,
 )
-from seekrai.types.agents.tools.schemas import FileSearch, FileSearchEnv
 from seekrai.types.alignment import (
     AlignmentEstimationRequest,
     AlignmentEstimationResponse,
@@ -194,8 +204,11 @@ __all__ = [
     "ToolBase",
     "ToolType",
     "EnvConfig",
-    "Env",
     "Tool",
     "FileSearch",
     "FileSearchEnv",
+    "RunPython",
+    "RunPythonEnv",
+    "WebSearch",
+    "WebSearchEnv",
 ]

seekrai/types/agents/__init__.py

@@ -41,8 +41,15 @@ from seekrai.types.agents.threads import (
     ThreadMessageContentType,
     ThreadStatus,
 )
-from seekrai.types.agents.tools import Env, EnvConfig, Tool, ToolBase, ToolType
-from seekrai.types.agents.tools.schemas import FileSearch, FileSearchEnv
+from seekrai.types.agents.tools import EnvConfig, Tool, ToolBase, ToolType
+from seekrai.types.agents.tools.schemas import (
+    FileSearch,
+    FileSearchEnv,
+    RunPython,
+    RunPythonEnv,
+    WebSearch,
+    WebSearchEnv,
+)
 
 
 __all__ = [
@@ -86,8 +93,11 @@ __all__ = [
     "ToolBase",
     "ToolType",
     "EnvConfig",
-    "Env",
     "Tool",
     "FileSearch",
     "FileSearchEnv",
+    "RunPython",
+    "RunPythonEnv",
+    "WebSearch",
+    "WebSearchEnv",
 ]

seekrai/types/agents/agent.py

@@ -47,3 +47,7 @@ class Agent(BaseModel):
 class AgentDeleteResponse(BaseModel):
     id: str
     deleted: bool
+
+
+class UpdateAgentRequest(CreateAgentRequest):
+    pass

seekrai/types/agents/tools/__init__.py

@@ -1,7 +1,13 @@
 from seekrai.types.agents.tools.env_model_config import EnvConfig
-from seekrai.types.agents.tools.schemas import FileSearch, FileSearchEnv
+from seekrai.types.agents.tools.schemas import (
+    FileSearch,
+    FileSearchEnv,
+    RunPython,
+    RunPythonEnv,
+    WebSearch,
+    WebSearchEnv,
+)
 from seekrai.types.agents.tools.tool import ToolBase, ToolType
-from seekrai.types.agents.tools.tool_env_types import Env
 from seekrai.types.agents.tools.tool_types import Tool
 
 
@@ -13,4 +19,8 @@ __all__ = [
     "Tool",
     "FileSearch",
     "FileSearchEnv",
+    "RunPython",
+    "RunPythonEnv",
+    "WebSearch",
+    "WebSearchEnv",
 ]

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

@@ -1,8 +1,16 @@
 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
+from seekrai.types.agents.tools.schemas.run_python_env import RunPythonEnv
+from seekrai.types.agents.tools.schemas.web_search import WebSearch
+from seekrai.types.agents.tools.schemas.web_search_env import WebSearchEnv
 
 
 __all__ = [
     "FileSearch",
     "FileSearchEnv",
+    "RunPython",
+    "RunPythonEnv",
+    "WebSearch",
+    "WebSearchEnv",
 ]

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

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

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

@@ -0,0 +1,9 @@
+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/types/agents/tools/schemas/run_python_env.py

@@ -0,0 +1,7 @@
+from typing import Optional
+
+from seekrai.types.agents.tools.env_model_config import EnvConfig
+
+
+class RunPythonEnv(EnvConfig):
+    run_python_tool_desc: Optional[str] = None

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

@@ -0,0 +1,9 @@
+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/types/agents/tools/schemas/web_search_env.py

@@ -0,0 +1,7 @@
+from typing import Optional
+
+from seekrai.types.agents.tools.env_model_config import EnvConfig
+
+
+class WebSearchEnv(EnvConfig):
+    web_search_tool_description: Optional[str] = None

seekrai/types/agents/tools/tool.py

@@ -1,14 +1,20 @@
 import enum
+from typing import Generic, TypeVar
 
 from pydantic import BaseModel
 
-from seekrai.types.agents.tools.tool_env_types import Env
+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"
 
 
-class ToolBase(BaseModel):
+TEnv = TypeVar("TEnv", bound=EnvConfig)
+
+
+class ToolBase(BaseModel, Generic[TEnv]):
     name: ToolType
-    tool_env: Env
+    tool_env: TEnv

seekrai/types/agents/tools/tool_types.py

@@ -1,10 +1,10 @@
-from typing import Annotated
+from typing import Annotated, Union
 
 from pydantic import Field
 
 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[
-    FileSearch, Field(discriminator="name")
-]  # will be a Union of tools when more are added
+Tool = Annotated[Union[FileSearch, WebSearch, RunPython], Field(discriminator="name")]

{seekrai-0.5.5.dist-info → seekrai-0.5.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: seekrai
-Version: 0.5.5
+Version: 0.5.6
 Summary: Python client for SeekrAI
 License: Apache-2.0
 Author: SeekrFlow

{seekrai-0.5.5.dist-info → seekrai-0.5.6.dist-info}/RECORD

@@ -9,7 +9,7 @@ seekrai/filemanager.py,sha256=bO2OvjZ9Cx5r2vr3Ocpd_0qVc3owRDT2LCU4Zmp2uDY,15489
 seekrai/resources/__init__.py,sha256=EPMOqI3mvpzMjNJIr2Da2tkDqSq_lntMph2O-j4pyIA,1501
 seekrai/resources/agents/__init__.py,sha256=qPdo3vMZUaGZPdZCNYL0hjtX-T6yAlnpE_zc5otkjak,373
 seekrai/resources/agents/agent_inference.py,sha256=oNk8UK39XY9mRaBdE5icpi-nSWiRKHgQBcq-DIhSht4,9219
-seekrai/resources/agents/agents.py,sha256=Cs184v2jYQRlk5MmsAH_hRI93D7qS6mJpuCLGIxH3as,7996
+seekrai/resources/agents/agents.py,sha256=d3C4VaXq4q5WlUr0f8txdRD3hvEKKvz4UQA-a_sLgso,9385
 seekrai/resources/agents/threads.py,sha256=BwZ2_6wlezsb12PQjEw1fgdJh5S83SPgD6qZQoGvyIM,14544
 seekrai/resources/alignment.py,sha256=IOKlKK2I9_NhS9pwcrsd9-5OO7lVT8Uw0y_wuGHOnyA,5839
 seekrai/resources/chat/__init__.py,sha256=KmtPupgECtEN80NyvcnSmieTAFXhwmVxhMHP0qhspA4,618
@@ -27,20 +27,23 @@ seekrai/resources/projects.py,sha256=Nmuh0_BwWoAO89r-p0ZEM8p4NHIH1EUeP83ivRoW5hw
 seekrai/resources/resource_base.py,sha256=rFIHFeqKPiAEbMYcMiIGHIym7qxwmh-EGsWiZcMDHdk,224
 seekrai/resources/vectordb.py,sha256=RRULyuldM4A0RlveBNZWFrav7l405wm89ua3k3Bqkgc,14527
 seekrai/seekrflow_response.py,sha256=5RFEQzamDy7sTSDkxSsZQThZ3biNmeCPeHWdrFId5Go,1320
-seekrai/types/__init__.py,sha256=rgFu__IgCM706ziux2wDfT5GNVyo2SWDWvNS0XF7nrw,4466
+seekrai/types/__init__.py,sha256=N72VsnmOdvLxnhBFLgfugIkT0tqr-M7WVFpAFtXEjLI,4665
 seekrai/types/abstract.py,sha256=TqWFQV_6bPblywfCH-r8FCkXWvPkc9KlJ4QVgyrnaMc,642
-seekrai/types/agents/__init__.py,sha256=gQBTRS5YjUvbz_uexwJ55QGMDeXytnd_-cFPxz926p0,2026
-seekrai/types/agents/agent.py,sha256=QZKp0xPo70regWKohWrkuTlmkvnN_2St7uFkX8M14yY,1075
+seekrai/types/agents/__init__.py,sha256=akuEDZDs4-OzvPIt0nupF63vBmI_CPfSLKsQUHTl1s8,2163
+seekrai/types/agents/agent.py,sha256=85D4GeHF-bYYnPirJSi1MbFg_2uFE2fSEmAHV9LxZfQ,1132
 seekrai/types/agents/runs.py,sha256=BZRwrdRWizOcoMJ1r8FKq06znIfd25MNVTP7DxZBQWA,3780
 seekrai/types/agents/threads.py,sha256=TinCMKv1bi5LzboDyCx1XI4Zzd8UzUZos4VOrTNhmEc,6835
-seekrai/types/agents/tools/__init__.py,sha256=-rdY30tqMSSapMKPFOYa-6crd6tVOUvHCzOW1UXxO0U,442
+seekrai/types/agents/tools/__init__.py,sha256=4MmlL13JLhWgMUlL3TKfegiA-IXGG06YellZTSTVFC8,537
 seekrai/types/agents/tools/env_model_config.py,sha256=9POx2DPwfSXgoaziJv7QvKeMrhMsYD1exnanSRK48vw,177
-seekrai/types/agents/tools/schemas/__init__.py,sha256=sWNKRzmnWoTv3cJ5k0q0tVuWeOlRhDMEvVeClF1r4p8,202
-seekrai/types/agents/tools/schemas/file_search.py,sha256=sBH3nHgp2aIlbDsv6B3Er8LFzq6VZSXXzgN_toGVY28,289
+seekrai/types/agents/tools/schemas/__init__.py,sha256=_HQoqgBk4T1IvlD0Yt3REz0_x_oVFTgITTd7H2ZJXws,562
+seekrai/types/agents/tools/schemas/file_search.py,sha256=7reLq-S3jMnHlGgP0ihZrU2S4Mcaizoz4IXm3SqCj4A,304
 seekrai/types/agents/tools/schemas/file_search_env.py,sha256=3iS69R8DpLPTJY2WhiCbOMV47-vXvMjBA7csgfIxKeU,582
-seekrai/types/agents/tools/tool.py,sha256=VVreXuKz1LX7L1En2wTYnrCt6Za2KK2pco0RH9PxgX8,235
-seekrai/types/agents/tools/tool_env_types.py,sha256=qQDRbRBU4VGLb6uLIq_G4o-cdgZ5DBwBCa4m1Gw7OYg,151
-seekrai/types/agents/tools/tool_types.py,sha256=Auciez5JSSGjczfG38UAuuCeyRKs-ikP3bh2VAEUPFo,242
+seekrai/types/agents/tools/schemas/run_python.py,sha256=CMw9C0T9UBWfHSW6mRhNDIIe_uGfxjgRkPIbqGlQxjo,297
+seekrai/types/agents/tools/schemas/run_python_env.py,sha256=lvmGSg-u6CoGSEGhAHHkmDffFj6kpFDERIwAhggzgwI,175
+seekrai/types/agents/tools/schemas/web_search.py,sha256=HrcVIC-SQGlBIXvJ0MxXYVQe6JnlgacleViXIo5OrGw,297
+seekrai/types/agents/tools/schemas/web_search_env.py,sha256=aHaricIcXaWz9N4G_gEipstANPbJPf41F9VZC5LwPog,182
+seekrai/types/agents/tools/tool.py,sha256=rX-SoN-XgweiHQIKRhNxBWK4ClNjEiXv-YaLdejpgmc,397
+seekrai/types/agents/tools/tool_types.py,sha256=1tF_kE6Z_zzuZpOAK1HrHsHkXFPEoK0PdYv-pbTLfkY,360
 seekrai/types/alignment.py,sha256=t1jS_lylUF7zbnf91ceXc3v6HzPmegUCkc3_eypTsSg,2109
 seekrai/types/chat_completions.py,sha256=Z7H1MkMgb4O0O5LDMKotQqhjGVCYk5eBeZ8n--RJpf8,3736
 seekrai/types/common.py,sha256=YI1pE-i_lDLU2o6FjoINdIhPXsV9lUl2MeAg2aRtT-M,2062
@@ -62,8 +65,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.5.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-seekrai-0.5.5.dist-info/METADATA,sha256=3hWqTgeliuL8gotG8dPW8oz5QIJUcgmYPHkJ6GD_1tU,4780
-seekrai-0.5.5.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-seekrai-0.5.5.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
-seekrai-0.5.5.dist-info/RECORD,,
+seekrai-0.5.6.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+seekrai-0.5.6.dist-info/METADATA,sha256=QptNWacnlAuFAUABIfoCqR-XyBvNS0L-zqjKTqWwObA,4780
+seekrai-0.5.6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+seekrai-0.5.6.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
+seekrai-0.5.6.dist-info/RECORD,,

seekrai/types/agents/tools/tool_env_types.py

@@ -1,4 +0,0 @@
-from seekrai.types.agents.tools.schemas.file_search_env import FileSearchEnv
-
-
-Env = FileSearchEnv  # will be a Union of tool envs when more are added