seekrai 0.5.5__tar.gz → 0.5.7__tar.gz

{seekrai-0.5.5 → seekrai-0.5.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: seekrai
-Version: 0.5.5
+Version: 0.5.7
 Summary: Python client for SeekrAI
 License: Apache-2.0
 Author: SeekrFlow
@@ -20,7 +20,7 @@ Requires-Dist: filelock (>=3.13.1,<4.0.0)
 Requires-Dist: httpx[http2] (>=0.28.0,<0.29.0)
 Requires-Dist: numpy (>=1.23.5) ; python_version < "3.12"
 Requires-Dist: numpy (>=1.26.0) ; python_version >= "3.12"
-Requires-Dist: pillow (>=10.3.0,<11.0.0)
+Requires-Dist: pillow (>=10.2.0,<11.0.0)
 Requires-Dist: pyarrow (>=10.0.1)
 Requires-Dist: pydantic (>=2.7,<3.0)
 Requires-Dist: requests (>=2.31.0,<3.0.0)

{seekrai-0.5.5 → seekrai-0.5.7}/pyproject.toml

@@ -14,7 +14,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "seekrai"
-version = "0.5.5"
+version = "0.5.7"
 authors = [
     "SeekrFlow <support@seekr.com>"
 ]
@@ -40,7 +40,7 @@ pydantic = "^2.7"
 filelock = "^3.13.1"
 eval-type-backport = ">=0.1.3,<0.3.0"
 click = "^8.1.7"
-pillow = "^10.3.0"
+pillow = "^10.2.0"
 pyarrow = ">=10.0.1"
 numpy = [
     { version = ">=1.23.5", python = "<3.12" },

{seekrai-0.5.5 → seekrai-0.5.7}/src/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-0.5.5 → seekrai-0.5.7}/src/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-0.5.5 → seekrai-0.5.7}/src/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-0.5.5 → seekrai-0.5.7}/src/seekrai/types/agents/agent.py

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

{seekrai-0.5.5 → seekrai-0.5.7}/src/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-0.5.7/src/seekrai/types/agents/tools/schemas/__init__.py

@@ -0,0 +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-0.5.5 → seekrai-0.5.7}/src/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-0.5.7/src/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-0.5.7/src/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-0.5.7/src/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-0.5.7/src/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-0.5.7/src/seekrai/types/agents/tools/tool.py

@@ -0,0 +1,20 @@
+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

seekrai-0.5.7/src/seekrai/types/agents/tools/tool_types.py

@@ -0,0 +1,10 @@
+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[Union[FileSearch, WebSearch, RunPython], Field(discriminator="name")]

seekrai-0.5.5/src/seekrai/types/agents/tools/schemas/__init__.py

@@ -1,8 +0,0 @@
-from seekrai.types.agents.tools.schemas.file_search import FileSearch
-from seekrai.types.agents.tools.schemas.file_search_env import FileSearchEnv
-
-
-__all__ = [
-    "FileSearch",
-    "FileSearchEnv",
-]

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

@@ -1,14 +0,0 @@
-import enum
-
-from pydantic import BaseModel
-
-from seekrai.types.agents.tools.tool_env_types import Env
-
-
-class ToolType(str, enum.Enum):
-    FILE_SEARCH = "file_search"
-
-
-class ToolBase(BaseModel):
-    name: ToolType
-    tool_env: Env

seekrai-0.5.5/src/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

seekrai-0.5.5/src/seekrai/types/agents/tools/tool_types.py

@@ -1,10 +0,0 @@
-from typing import Annotated
-
-from pydantic import Field
-
-from seekrai.types.agents.tools.schemas.file_search import FileSearch
-
-
-Tool = Annotated[
-    FileSearch, Field(discriminator="name")
-]  # will be a Union of tools when more are added