diff --git a/src/runloop_api_client/sdk/_types.py b/src/runloop_api_client/sdk/_types.py index aa7e0208a..4d2accc46 100644 --- a/src/runloop_api_client/sdk/_types.py +++ b/src/runloop_api_client/sdk/_types.py @@ -4,6 +4,7 @@ from ..types import ( InputContext, ScenarioView, + AxonListParams, AgentListParams, AxonCreateParams, DevboxListParams, @@ -190,6 +191,10 @@ class SDKAgentListParams(AgentListParams, BaseRequestOptions): pass +class SDKAxonListParams(AxonListParams, BaseRequestOptions): + pass + + class SDKAxonCreateParams(AxonCreateParams, LongRequestOptions): pass diff --git a/src/runloop_api_client/sdk/async_.py b/src/runloop_api_client/sdk/async_.py index c2a237da8..d02a6a9f1 100644 --- a/src/runloop_api_client/sdk/async_.py +++ b/src/runloop_api_client/sdk/async_.py @@ -11,6 +11,7 @@ import httpx from ._types import ( + SDKAxonListParams, BaseRequestOptions, LongRequestOptions, SDKAgentListParams, @@ -544,9 +545,14 @@ def from_id(self, axon_id: str) -> AsyncAxon: """Get an AsyncAxon instance for an existing axon ID.""" return AsyncAxon(self._client, axon_id) - async def list(self, **options: Unpack[BaseRequestOptions]) -> list[AsyncAxon]: - """[Beta] List all active axons.""" - result = await self._client.axons.list(**options) + async def list(self, **params: Unpack[SDKAxonListParams]) -> list[AsyncAxon]: + """[Beta] List all active axons. + + :param params: See :typeddict:`~runloop_api_client.sdk._types.SDKAxonListParams` for available parameters + :return: Collection of axon wrappers + :rtype: list[AsyncAxon] + """ + result = await self._client.axons.list(**params) return [AsyncAxon(self._client, axon.id) for axon in result.axons] diff --git a/src/runloop_api_client/sdk/sync.py b/src/runloop_api_client/sdk/sync.py index 0fa9f8cea..e6b672159 100644 --- a/src/runloop_api_client/sdk/sync.py +++ b/src/runloop_api_client/sdk/sync.py @@ -12,6 +12,7 @@ from .axon import Axon from .agent import Agent from ._types import ( + SDKAxonListParams, BaseRequestOptions, LongRequestOptions, SDKAgentListParams, @@ -539,9 +540,14 @@ def from_id(self, axon_id: str) -> Axon: """Get an Axon instance for an existing axon ID.""" return Axon(self._client, axon_id) - def list(self, **options: Unpack[BaseRequestOptions]) -> list[Axon]: - """[Beta] List all active axons.""" - result = self._client.axons.list(**options) + def list(self, **params: Unpack[SDKAxonListParams]) -> list[Axon]: + """[Beta] List all active axons. + + :param params: See :typeddict:`~runloop_api_client.sdk._types.SDKAxonListParams` for available parameters + :return: Collection of axon wrappers + :rtype: list[Axon] + """ + result = self._client.axons.list(**params) return [Axon(self._client, axon.id) for axon in result.axons] diff --git a/uv.lock b/uv.lock index bac2c9a14..536ecad8e 100644 --- a/uv.lock +++ b/uv.lock @@ -2386,7 +2386,7 @@ wheels = [ [[package]] name = "runloop-api-client" -version = "1.13.2" +version = "1.13.3" source = { editable = "." } dependencies = [ { name = "anyio" },