Merge pull request #562 from runloopai/gautam-score-metadata

feat: Add metadata to start_run_and_await_ready

Author: gautam-rl

src/runloop_api_client/resources/blueprints.py

@@ -20,11 +20,11 @@
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
+from ..pagination import SyncBlueprintsCursorIDPage, AsyncBlueprintsCursorIDPage
 from .._exceptions import RunloopError
 from ..lib.polling import PollingConfig, poll_until
-from ..lib.polling_async import async_poll_until
-from ..pagination import SyncBlueprintsCursorIDPage, AsyncBlueprintsCursorIDPage
 from .._base_client import AsyncPaginator, make_request_options
+from ..lib.polling_async import async_poll_until
 from ..types.blueprint_view import BlueprintView
 from ..types.blueprint_preview_view import BlueprintPreviewView
 from ..types.blueprint_build_logs_list_view import BlueprintBuildLogsListView

src/runloop_api_client/resources/devboxes/devboxes.py

@@ -93,7 +93,6 @@
 )
 from ..._exceptions import RunloopError
 from ...lib.polling import PollingConfig, poll_until
-from ...lib.polling_async import async_poll_until
 from ..._base_client import AsyncPaginator, make_request_options
 from .disk_snapshots import (
     DiskSnapshotsResource,
@@ -103,6 +102,7 @@
     DiskSnapshotsResourceWithStreamingResponse,
     AsyncDiskSnapshotsResourceWithStreamingResponse,
 )
+from ...lib.polling_async import async_poll_until
 from ...types.devbox_view import DevboxView
 from ...types.devbox_tunnel_view import DevboxTunnelView
 from ...types.devbox_snapshot_view import DevboxSnapshotView

src/runloop_api_client/resources/scenarios/scenarios.py

@@ -34,7 +34,6 @@
     maybe_transform,
     async_maybe_transform,
 )
-from ...lib.polling import PollingConfig
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
 from ..._response import (
@@ -44,6 +43,7 @@
     async_to_streamed_response_wrapper,
 )
 from ...pagination import SyncScenariosCursorIDPage, AsyncScenariosCursorIDPage
+from ...lib.polling import PollingConfig
 from ..._base_client import AsyncPaginator, make_request_options
 from ...types.scenario_view import ScenarioView
 from ...types.scenario_run_view import ScenarioRunView
@@ -422,6 +422,7 @@ def start_run_and_await_env_ready(
         *,
         scenario_id: str,
         benchmark_run_id: Optional[str] | NotGiven = NOT_GIVEN,
+        metadata: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
         run_name: Optional[str] | NotGiven = NOT_GIVEN,
         polling_config: PollingConfig | None = None,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -455,6 +456,7 @@ def start_run_and_await_env_ready(
         run = self.start_run(
             scenario_id=scenario_id,
             benchmark_run_id=benchmark_run_id,
+            metadata=metadata,
             run_name=run_name,
             extra_headers=extra_headers,
             extra_query=extra_query,
@@ -842,6 +844,7 @@ async def start_run_and_await_env_ready(
         self,
         scenario_id: str,
         benchmark_run_id: Optional[str] | NotGiven = NOT_GIVEN,
+        metadata: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
         run_name: Optional[str] | NotGiven = NOT_GIVEN,
         polling_config: PollingConfig | None = None,
     ) -> ScenarioRunView:
@@ -863,6 +866,7 @@ async def start_run_and_await_env_ready(
         run = await self.start_run(
             scenario_id=scenario_id,
             benchmark_run_id=benchmark_run_id,
+            metadata=metadata,
             run_name=run_name,
         )