Skip to content

Commit 90dd085

Browse files
committed
modified ScenarioStartRunParams to follow our previous pattern for use in the SDK
1 parent 19eb010 commit 90dd085

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
@@ -14,6 +14,7 @@
1414
from ..types.object_download_params import ObjectDownloadParams
1515
from ..types.blueprint_create_params import BlueprintCreateParams
1616
from ..types.devbox_upload_file_params import DevboxUploadFileParams
17+
from ..types.scenario_start_run_params import ScenarioStartRunBaseParams
1718
from ..types.devbox_create_tunnel_params import DevboxCreateTunnelParams
1819
from ..types.devbox_download_file_params import DevboxDownloadFileParams
1920
from ..types.devbox_execute_async_params import DevboxNiceExecuteAsyncParams
@@ -164,17 +165,5 @@ class SDKScenarioListParams(ScenarioListParams, BaseRequestOptions):
164165
pass
165166

166167

167-
class SDKScenarioRunParams(TypedDict, total=False):
168-
"""Parameters for starting a scenario run (excludes scenario_id which is set automatically)."""
169-
170-
benchmark_run_id: Optional[str]
171-
"""Benchmark to associate the run."""
172-
173-
metadata: Optional[dict[str, str]]
174-
"""User defined metadata to attach to the run for organization."""
175-
176-
run_name: Optional[str]
177-
"""Display name of the run."""
178-
179-
polling_config: Optional[PollingConfig]
180-
"""Configuration for polling behavior (used by run_and_await_env_ready)."""
168+
class SDKScenarioRunParams(ScenarioStartRunBaseParams, LongRequestOptions):
169+
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)