Skip to content

Commit 6baf371

Browse files
committed
feat: complete SDK wrapper cancellation_token propagation
- Update AsyncScenarioRun.await_env_ready to pass cancellation_token - Completes cancellation support across all SDK polling operations Part of porting TypeScript PR #765 features to Python SDK.
1 parent 0d0f863 commit 6baf371

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/runloop_api_client/sdk/async_scenario_run.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,11 @@ async def await_env_ready(
106106
:return: Scenario run state after environment is ready
107107
:rtype: ScenarioRunView
108108
"""
109-
await self._client.devboxes.await_running(self._devbox_id, polling_config=options.get("polling_config"))
109+
await self._client.devboxes.await_running(
110+
self._devbox_id,
111+
polling_config=options.get("polling_config"),
112+
cancellation_token=options.get("cancellation_token"),
113+
)
110114
return await self.get_info(**filter_params(options, BaseRequestOptions))
111115

112116
async def score(

0 commit comments

Comments
 (0)