cp

Author: wall-rl

src/runloop_api_client/sdk/__init__.py

@@ -5,20 +5,23 @@
 
 from __future__ import annotations
 
-from .sync import DevboxOps, ScorerOps, RunloopSDK, SnapshotOps, BlueprintOps, StorageObjectOps
+from .sync import AgentOps, DevboxOps, ScorerOps, RunloopSDK, SnapshotOps, BlueprintOps, StorageObjectOps
 from .async_ import (
     AsyncDevboxOps,
     AsyncScorerOps,
     AsyncRunloopSDK,
     AsyncSnapshotOps,
     AsyncBlueprintOps,
     AsyncStorageObjectOps,
+    AsyncAgentOps,
 )
+from .agent import Agent
 from .devbox import Devbox, NamedShell
 from .scorer import Scorer
 from .snapshot import Snapshot
 from .blueprint import Blueprint
 from .execution import Execution
+from .async_agent import AsyncAgent
 from .async_devbox import AsyncDevbox, AsyncNamedShell
 from .async_scorer import AsyncScorer
 from .async_snapshot import AsyncSnapshot
@@ -34,6 +37,8 @@
     "RunloopSDK",
     "AsyncRunloopSDK",
     # Management interfaces
+    "AgentOps",
+    "AsyncAgentOps",
     "DevboxOps",
     "AsyncDevboxOps",
     "BlueprintOps",
@@ -45,6 +50,8 @@
     "StorageObjectOps",
     "AsyncStorageObjectOps",
     # Resource classes
+    "Agent",
+    "AsyncAgent",
     "Devbox",
     "AsyncDevbox",
     "Execution",

src/runloop_api_client/sdk/_types.py

@@ -5,6 +5,8 @@
 from ..lib.polling import PollingConfig
 from ..types.devboxes import DiskSnapshotListParams, DiskSnapshotUpdateParams
 from ..types.scenarios import ScorerListParams, ScorerCreateParams, ScorerUpdateParams, ScorerValidateParams
+from ..types.agent_list_params import AgentListParams
+from ..types.agent_create_params import AgentCreateParams
 from ..types.devbox_list_params import DevboxListParams
 from ..types.object_list_params import ObjectListParams
 from ..types.devbox_create_params import DevboxCreateParams, DevboxBaseCreateParams
@@ -157,3 +159,11 @@ class SDKScorerUpdateParams(ScorerUpdateParams, LongRequestOptions):
 
 class SDKScorerValidateParams(ScorerValidateParams, LongRequestOptions):
     pass
+
+
+class SDKAgentCreateParams(AgentCreateParams, LongRequestOptions):
+    pass
+
+
+class SDKAgentListParams(AgentListParams, RequestOptions):
+    pass

src/runloop_api_client/sdk/agent.py

@@ -0,0 +1,58 @@
+"""Agent resource class for synchronous operations."""
+
+from __future__ import annotations
+
+from typing_extensions import Unpack, override
+
+from ._types import (
+    RequestOptions,
+)
+from .._client import Runloop
+from ..types.agent_view import AgentView
+
+
+class Agent:
+    """Wrapper around synchronous agent operations."""
+
+    def __init__(
+        self,
+        client: Runloop,
+        agent_id: str,
+    ) -> None:
+        """Initialize the wrapper.
+
+        :param client: Generated Runloop client
+        :type client: Runloop
+        :param agent_id: Agent identifier returned by the API
+        :type agent_id: str
+        """
+        self._client = client
+        self._id = agent_id
+
+    @override
+    def __repr__(self) -> str:
+        return f"<Agent id={self._id!r}>"
+
+    @property
+    def id(self) -> str:
+        """Return the agent identifier.
+
+        :return: Unique agent ID
+        :rtype: str
+        """
+        return self._id
+
+    def get_info(
+        self,
+        **options: Unpack[RequestOptions],
+    ) -> AgentView:
+        """Retrieve the latest agent information.
+
+        :param options: Optional request configuration
+        :return: Agent details
+        :rtype: AgentView
+        """
+        return self._client.agents.retrieve(
+            self._id,
+            **options,
+        )

src/runloop_api_client/sdk/async_.py

@@ -14,9 +14,11 @@
 
 from ._types import (
     LongRequestOptions,
+    SDKAgentListParams,
     SDKDevboxListParams,
     SDKObjectListParams,
     SDKScorerListParams,
+    SDKAgentCreateParams,
     SDKDevboxCreateParams,
     SDKObjectCreateParams,
     SDKScorerCreateParams,
@@ -28,6 +30,7 @@
 from .._types import Timeout, NotGiven, not_given
 from .._client import DEFAULT_MAX_RETRIES, AsyncRunloop
 from ._helpers import detect_content_type
+from .async_agent import AsyncAgent
 from .async_devbox import AsyncDevbox
 from .async_scorer import AsyncScorer
 from .async_snapshot import AsyncSnapshot
@@ -541,6 +544,66 @@ async def list(self, **params: Unpack[SDKScorerListParams]) -> list[AsyncScorer]
         """
         page = await self._client.scenarios.scorers.list(**params)
         return [AsyncScorer(self._client, item.id) async for item in page]
+        
+class AsyncAgentOps:
+    """High-level async manager for creating and managing agents.
+
+    Accessed via ``runloop.agent`` from :class:`AsyncRunloopSDK`, provides
+    coroutines to create, retrieve, and list agents.
+
+    Example:
+        >>> runloop = AsyncRunloopSDK()
+        >>> agent = await runloop.agent.create(name="my-agent")
+        >>> agents = await runloop.agent.list(limit=10)
+    """
+
+    def __init__(self, client: AsyncRunloop) -> None:
+        """Initialize the manager.
+
+        :param client: Generated AsyncRunloop client to wrap
+        :type client: AsyncRunloop
+        """
+        self._client = client
+
+    async def create(
+        self,
+        **params: Unpack[SDKAgentCreateParams],
+    ) -> AsyncAgent:
+        """Create a new agent.
+
+        :param params: See :typeddict:`~runloop_api_client.sdk._types.SDKAgentCreateParams` for available parameters
+        :return: Wrapper bound to the newly created agent
+        :rtype: AsyncAgent
+        """
+        agent_view = await self._client.agents.create(
+            **params,
+        )
+        return AsyncAgent(self._client, agent_view.id)
+
+    def from_id(self, agent_id: str) -> AsyncAgent:
+        """Attach to an existing agent by ID.
+
+        :param agent_id: Existing agent ID
+        :type agent_id: str
+        :return: Wrapper bound to the requested agent
+        :rtype: AsyncAgent
+        """
+        return AsyncAgent(self._client, agent_id)
+
+    async def list(
+        self,
+        **params: Unpack[SDKAgentListParams],
+    ) -> list[AsyncAgent]:
+        """List agents accessible to the caller.
+
+        :param params: See :typeddict:`~runloop_api_client.sdk._types.SDKAgentListParams` for available parameters
+        :return: Collection of agent wrappers
+        :rtype: list[AsyncAgent]
+        """
+        page = await self._client.agents.list(
+            **params,
+        )
+        return [AsyncAgent(self._client, item.id) for item in page.agents]
 
 
 class AsyncRunloopSDK:
@@ -552,6 +615,8 @@ class AsyncRunloopSDK:
 
     :ivar api: Direct access to the generated async REST API client
     :vartype api: AsyncRunloop
+    :ivar agent: High-level async interface for agent management.
+    :vartype agent: AsyncAgentOps
     :ivar devbox: High-level async interface for devbox management
     :vartype devbox: AsyncDevboxOps
     :ivar blueprint: High-level async interface for blueprint management
@@ -572,6 +637,7 @@ class AsyncRunloopSDK:
     """
 
     api: AsyncRunloop
+    agent: AsyncAgentOps
     devbox: AsyncDevboxOps
     blueprint: AsyncBlueprintOps
     scorer: AsyncScorerOps
@@ -616,6 +682,7 @@ def __init__(
             http_client=http_client,
         )
 
+        self.agent = AsyncAgentOps(self.api)
         self.devbox = AsyncDevboxOps(self.api)
         self.blueprint = AsyncBlueprintOps(self.api)
         self.scorer = AsyncScorerOps(self.api)

src/runloop_api_client/sdk/async_agent.py

@@ -0,0 +1,58 @@
+"""Agent resource class for asynchronous operations."""
+
+from __future__ import annotations
+
+from typing_extensions import Unpack, override
+
+from ._types import (
+    RequestOptions,
+)
+from .._client import AsyncRunloop
+from ..types.agent_view import AgentView
+
+
+class AsyncAgent:
+    """Async wrapper around agent operations."""
+
+    def __init__(
+        self,
+        client: AsyncRunloop,
+        agent_id: str,
+    ) -> None:
+        """Initialize the wrapper.
+
+        :param client: Generated AsyncRunloop client
+        :type client: AsyncRunloop
+        :param agent_id: Agent identifier returned by the API
+        :type agent_id: str
+        """
+        self._client = client
+        self._id = agent_id
+
+    @override
+    def __repr__(self) -> str:
+        return f"<AsyncAgent id={self._id!r}>"
+
+    @property
+    def id(self) -> str:
+        """Return the agent identifier.
+
+        :return: Unique agent ID
+        :rtype: str
+        """
+        return self._id
+
+    async def get_info(
+        self,
+        **options: Unpack[RequestOptions],
+    ) -> AgentView:
+        """Retrieve the latest agent information.
+
+        :param options: Optional request configuration
+        :return: Agent details
+        :rtype: AgentView
+        """
+        return await self._client.agents.retrieve(
+            self._id,
+            **options,
+        )