Merge pull request #635 from runloopai/james/run_profile

Add run_profile when starting a scenario

Author: dines-rl

src/runloop_api_client/resources/scenarios/scenarios.py

@@ -460,6 +460,7 @@ def start_run_and_await_env_ready(
         benchmark_run_id: Optional[str] | NotGiven = NOT_GIVEN,
         metadata: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
         run_name: Optional[str] | NotGiven = NOT_GIVEN,
+        run_profile: Optional[scenario_start_run_params.RunProfile] | 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.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -475,6 +476,7 @@ def start_run_and_await_env_ready(
             scenario_id: ID of the Scenario to run
             benchmark_run_id: Benchmark to associate the run
             run_name: Display name of the run
+            run_profile: Runtime configuration to use for this benchmark run
             polling_config: Optional polling configuration
             extra_headers: Send extra headers
             extra_query: Add additional query parameters to the request
@@ -494,6 +496,7 @@ def start_run_and_await_env_ready(
             benchmark_run_id=benchmark_run_id,
             metadata=metadata,
             run_name=run_name,
+            run_profile=run_profile,
             extra_headers=extra_headers,
             extra_query=extra_query,
             extra_body=extra_body,
@@ -913,6 +916,7 @@ async def start_run_and_await_env_ready(
         benchmark_run_id: Optional[str] | NotGiven = NOT_GIVEN,
         metadata: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
         run_name: Optional[str] | NotGiven = NOT_GIVEN,
+        run_profile: Optional[scenario_start_run_params.RunProfile] | NotGiven = NOT_GIVEN,
         polling_config: PollingConfig | None = None,
     ) -> ScenarioRunView:
         """Start a new ScenarioRun and wait for its environment to be ready.
@@ -921,6 +925,7 @@ async def start_run_and_await_env_ready(
             scenario_id: ID of the Scenario to run
             benchmark_run_id: Benchmark to associate the run
             run_name: Display name of the run
+            run_profile: Runtime configuration to use for this benchmark run
             polling_config: Optional polling configuration
 
         Returns:
@@ -935,6 +940,7 @@ async def start_run_and_await_env_ready(
             benchmark_run_id=benchmark_run_id,
             metadata=metadata,
             run_name=run_name,
+            run_profile=run_profile,
         )
 
         await self._client.devboxes.await_running(