shift args and format

tode-rl · tode-rl · commit 6a47c1b5b7f0 · 2025-08-19T13:42:00.000-07:00
diff --git a/src/runloop_api_client/lib/polling_async.py b/src/runloop_api_client/lib/polling_async.py
@@ -4,7 +4,8 @@
 
 from .polling import PollingConfig, PollingTimeout
 
-T = TypeVar('T')
+T = TypeVar("T")
+
 
 async def async_poll_until(
     retriever: Callable[[], Awaitable[T]],
@@ -14,27 +15,27 @@ async def async_poll_until(
 ) -> T:
     """
     Poll until a condition is met or timeout/max attempts are reached.
-    
+
     Args:
         retriever: Async or sync callable that returns the object to check
         is_terminal: Callable that returns True when polling should stop
         config: Optional polling configuration
         on_error: Optional error handler that can return a value to continue polling
                  or re-raise the exception to stop polling
-        
+
     Returns:
         The final state of the polled object
-        
+
     Raises:
         PollingTimeout: When max attempts or timeout is reached
     """
     if config is None:
         config = PollingConfig()
-    
+
     attempts = 0
     start_time = time.time()
     last_result: Union[T, None] = None
-    
+
     while True:
         try:
             last_result = await retriever()
@@ -43,23 +44,17 @@ async def async_poll_until(
                 last_result = on_error(e)
             else:
                 raise
-        
+
         if is_terminal(last_result):
             return last_result
-            
+
         attempts += 1
         if attempts >= config.max_attempts:
-            raise PollingTimeout(
-                f"Exceeded maximum attempts ({config.max_attempts})",
-                last_result
-            )
-            
+            raise PollingTimeout(f"Exceeded maximum attempts ({config.max_attempts})", last_result)
+
         if config.timeout_seconds is not None:
             elapsed = time.time() - start_time
             if elapsed >= config.timeout_seconds:
-                raise PollingTimeout(
-                    f"Exceeded timeout of {config.timeout_seconds} seconds",
-                    last_result
-                )
-                
+                raise PollingTimeout(f"Exceeded timeout of {config.timeout_seconds} seconds", last_result)
+
         await asyncio.sleep(config.interval_seconds)
diff --git a/src/runloop_api_client/resources/devboxes/devboxes.py b/src/runloop_api_client/resources/devboxes/devboxes.py
@@ -1576,13 +1576,20 @@ async def create_and_await_running(
             PollingTimeout: If polling times out before devbox is running
             RunloopError: If devbox enters a non-running terminal state
         """
-        # Pass all create_args to the underlying create method
-        devbox = await self.create(**(create_args or {}))
 
         # Extract polling config and other request args
         if request_args is None:
             request_args = {}
 
+        # Pass all create_args, relevant request args to the underlying create method
+        devbox = await self.create(
+            **(create_args or {}),
+            extra_headers=request_args.get("extra_headers", None),
+            extra_query=request_args.get("extra_query", None),
+            extra_body=request_args.get("extra_body", None),
+            timeout=request_args.get("timeout", None),
+        )
+
         return await self.await_running(
             devbox.id,
             polling_config=request_args.get("polling_config", None),
diff --git a/src/runloop_api_client/resources/scenarios/runs.py b/src/runloop_api_client/resources/scenarios/runs.py
@@ -316,7 +316,7 @@ def await_scored(
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> ScenarioRunView:
         """Wait for a scenario run to be scored.
-        
+
         Args:
             id: The ID of the scenario run to wait for
             polling_config: Optional polling configuration
@@ -332,13 +332,10 @@ def await_scored(
             PollingTimeout: If polling times out before scenario run is scored
             RunloopError: If scenario run enters a non-scored terminal state
         """
+
         def retrieve_run() -> ScenarioRunView:
             return self.retrieve(
-                id,
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout
+                id, extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             )
 
         def is_done_scoring(run: ScenarioRunView) -> bool:
@@ -347,9 +344,7 @@ def is_done_scoring(run: ScenarioRunView) -> bool:
         run = poll_until(retrieve_run, is_done_scoring, polling_config)
 
         if run.state != "scored":
-            raise RunloopError(
-                f"Scenario run entered non-scored state unexpectedly: {run.state}"
-            )
+            raise RunloopError(f"Scenario run entered non-scored state unexpectedly: {run.state}")
 
         return run
 
@@ -366,7 +361,7 @@ def score_and_await(
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> ScenarioRunView:
         """Score a scenario run and wait for it to be scored.
-        
+
         Args:
             id: The ID of the scenario run to score and wait for
             polling_config: Optional polling configuration
@@ -412,7 +407,7 @@ def score_and_complete(
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> ScenarioRunView:
         """Score a scenario run, wait for it to be scored, then complete it.
-        
+
         Args:
             id: The ID of the scenario run to score, wait for, and complete
             polling_config: Optional polling configuration
@@ -445,6 +440,7 @@ def score_and_complete(
             timeout=timeout,
         )
 
+
 class AsyncRunsResource(AsyncAPIResource):
     @cached_property
     def with_raw_response(self) -> AsyncRunsResourceWithRawResponse:
@@ -728,7 +724,7 @@ async def await_scored(
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> ScenarioRunView:
         """Wait for a scenario run to be scored.
-        
+
         Args:
             id: The ID of the scenario run to wait for
             polling_config: Optional polling configuration
@@ -744,13 +740,10 @@ async def await_scored(
             PollingTimeout: If polling times out before scenario run is scored
             RunloopError: If scenario run enters a non-scored terminal state
         """
+
         async def retrieve_run() -> ScenarioRunView:
             return await self.retrieve(
-                id,
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout
+                id, extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             )
 
         def is_done_scoring(run: ScenarioRunView) -> bool:
@@ -759,9 +752,7 @@ def is_done_scoring(run: ScenarioRunView) -> bool:
         run = await async_poll_until(retrieve_run, is_done_scoring, polling_config)
 
         if run.state != "scored":
-            raise RunloopError(
-                f"Scenario run entered non-scored state unexpectedly: {run.state}"
-            )
+            raise RunloopError(f"Scenario run entered non-scored state unexpectedly: {run.state}")
 
         return run
 
@@ -778,7 +769,7 @@ async def score_and_await(
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> ScenarioRunView:
         """Score a scenario run and wait for it to be scored.
-        
+
         Args:
             id: The ID of the scenario run to score and wait for
             polling_config: Optional polling configuration
@@ -824,7 +815,7 @@ async def score_and_complete(
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> ScenarioRunView:
         """Score a scenario run, wait for it to be scored, then complete it.
-        
+
         Args:
             id: The ID of the scenario run to score, wait for, and complete
             polling_config: Optional polling configuration
diff --git a/src/runloop_api_client/resources/scenarios/scenarios.py b/src/runloop_api_client/resources/scenarios/scenarios.py
@@ -926,6 +926,7 @@ async def start_run_and_await_env_ready(
 
         return run
 
+
 class ScenariosResourceWithRawResponse:
     def __init__(self, scenarios: ScenariosResource) -> None:
         self._scenarios = scenarios
diff --git a/uv.lock b/uv.lock
