Skip to content

Commit ed187e7

Browse files
committed
modified ScenarioStartRunParams to follow our previous pattern for use in the SDK
1 parent 5e07fee commit ed187e7

2 files changed

Lines changed: 9 additions & 18 deletions

File tree

src/runloop_api_client/sdk/_types.py

Lines changed: 3 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
from ..types.object_download_params import ObjectDownloadParams
1717
from ..types.blueprint_create_params import BlueprintCreateParams
1818
from ..types.devbox_upload_file_params import DevboxUploadFileParams
19+
from ..types.scenario_start_run_params import ScenarioStartRunBaseParams
1920
from ..types.devbox_create_tunnel_params import DevboxCreateTunnelParams
2021
from ..types.devbox_download_file_params import DevboxDownloadFileParams
2122
from ..types.devbox_execute_async_params import DevboxNiceExecuteAsyncParams
@@ -174,17 +175,5 @@ class SDKScenarioListParams(ScenarioListParams, BaseRequestOptions):
174175
pass
175176

176177

177-
class SDKScenarioRunParams(TypedDict, total=False):
178-
"""Parameters for starting a scenario run (excludes scenario_id which is set automatically)."""
179-
180-
benchmark_run_id: Optional[str]
181-
"""Benchmark to associate the run."""
182-
183-
metadata: Optional[dict[str, str]]
184-
"""User defined metadata to attach to the run for organization."""
185-
186-
run_name: Optional[str]
187-
"""Display name of the run."""
188-
189-
polling_config: Optional[PollingConfig]
190-
"""Configuration for polling behavior (used by run_and_await_env_ready)."""
178+
class SDKScenarioRunParams(ScenarioStartRunBaseParams, LongRequestOptions):
179+
pass

src/runloop_api_client/types/scenario_start_run_params.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,7 @@
1111
__all__ = ["ScenarioStartRunParams"]
1212

1313

14-
class ScenarioStartRunParams(TypedDict, total=False):
15-
scenario_id: Required[str]
16-
"""ID of the Scenario to run."""
17-
14+
class ScenarioStartRunBaseParams(TypedDict, total=False):
1815
benchmark_run_id: Optional[str]
1916
"""Benchmark to associate the run."""
2017

@@ -26,3 +23,8 @@ class ScenarioStartRunParams(TypedDict, total=False):
2623

2724
run_profile: Annotated[Optional[RunProfile], PropertyInfo(alias="runProfile")]
2825
"""Runtime configuration to use for this benchmark run"""
26+
27+
28+
class ScenarioStartRunParams(ScenarioStartRunBaseParams):
29+
scenario_id: Required[str]
30+
"""ID of the Scenario to run."""

0 commit comments

Comments
 (0)