Skip to content

Commit c2662ab

Browse files
committed
fixed pagination for scorer, scenario and network policy list methods
1 parent c3bebb5 commit c2662ab

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

src/runloop_api_client/sdk/async_.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -557,7 +557,7 @@ async def list(self, **params: Unpack[SDKScorerListParams]) -> list[AsyncScorer]
557557
:rtype: list[AsyncScorer]
558558
"""
559559
page = await self._client.scenarios.scorers.list(**params)
560-
return [AsyncScorer(self._client, item.id) async for item in page]
560+
return [AsyncScorer(self._client, item.id) for item in page.scorers]
561561

562562

563563
class AsyncAgentOps:
@@ -818,7 +818,7 @@ async def list(self, **params: Unpack[SDKScenarioListParams]) -> list[AsyncScena
818818
:rtype: list[AsyncScenario]
819819
"""
820820
page = await self._client.scenarios.list(**params)
821-
return [AsyncScenario(self._client, item.id) async for item in page]
821+
return [AsyncScenario(self._client, item.id) for item in page.scenarios]
822822

823823

824824
class AsyncBenchmarkOps:
@@ -920,8 +920,8 @@ async def list(self, **params: Unpack[SDKNetworkPolicyListParams]) -> list[Async
920920
:return: List of network policies
921921
:rtype: list[AsyncNetworkPolicy]
922922
"""
923-
page = self._client.network_policies.list(**params)
924-
return [AsyncNetworkPolicy(self._client, item.id) async for item in page]
923+
page = await self._client.network_policies.list(**params)
924+
return [AsyncNetworkPolicy(self._client, item.id) for item in page.network_policies]
925925

926926

927927
class AsyncRunloopSDK:

src/runloop_api_client/sdk/sync.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -552,7 +552,7 @@ def list(self, **params: Unpack[SDKScorerListParams]) -> list[Scorer]:
552552
:rtype: list[Scorer]
553553
"""
554554
page = self._client.scenarios.scorers.list(**params)
555-
return [Scorer(self._client, item.id) for item in page]
555+
return [Scorer(self._client, item.id) for item in page.scorers]
556556

557557

558558
class AgentOps:
@@ -843,7 +843,7 @@ def list(self, **params: Unpack[SDKScenarioListParams]) -> list[Scenario]:
843843
:rtype: list[Scenario]
844844
"""
845845
page = self._client.scenarios.list(**params)
846-
return [Scenario(self._client, item.id) for item in page]
846+
return [Scenario(self._client, item.id) for item in page.scenarios]
847847

848848

849849
class BenchmarkOps:
@@ -946,7 +946,7 @@ def list(self, **params: Unpack[SDKNetworkPolicyListParams]) -> list[NetworkPoli
946946
:rtype: list[NetworkPolicy]
947947
"""
948948
page = self._client.network_policies.list(**params)
949-
return [NetworkPolicy(self._client, item.id) for item in page]
949+
return [NetworkPolicy(self._client, item.id) for item in page.network_policies]
950950

951951

952952
class RunloopSDK:

0 commit comments

Comments
 (0)