fix(sdk): expose axon list params (#773)

Author: sid-rl

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
 

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]
 
 

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]