Skip to content

Commit 79ea205

Browse files
committed
exposed all scenario update parameters appropriately using typeddict
1 parent 23e625b commit 79ea205

3 files changed

Lines changed: 14 additions & 29 deletions

File tree

src/runloop_api_client/sdk/_types.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
from ..types.scenario_list_params import ScenarioListParams
1515
from ..types.blueprint_list_params import BlueprintListParams
1616
from ..types.object_download_params import ObjectDownloadParams
17+
from ..types.scenario_update_params import ScenarioUpdateParams
1718
from ..types.blueprint_create_params import BlueprintCreateParams
1819
from ..types.devbox_upload_file_params import DevboxUploadFileParams
1920
from ..types.scenario_start_run_params import ScenarioStartRunBaseParams
@@ -175,9 +176,13 @@ class SDKScenarioListParams(ScenarioListParams, BaseRequestOptions):
175176
pass
176177

177178

179+
class SDKScenarioUpdateParams(ScenarioUpdateParams, LongRequestOptions):
180+
pass
181+
182+
178183
class SDKScenarioRunAsyncParams(ScenarioStartRunBaseParams, LongRequestOptions):
179184
pass
180185

181186

182187
class SDKScenarioRunParams(ScenarioStartRunBaseParams, LongPollingRequestOptions):
183-
pass
188+
pass

src/runloop_api_client/sdk/async_scenario.py

Lines changed: 4 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,10 @@
22

33
from __future__ import annotations
44

5-
from typing import Dict, Optional
65
from typing_extensions import Unpack, override
76

87
from ..types import ScenarioView
9-
from ._types import BaseRequestOptions, LongRequestOptions, SDKScenarioRunAsyncParams, SDKScenarioRunParams
8+
from ._types import BaseRequestOptions, SDKScenarioRunParams, SDKScenarioUpdateParams, SDKScenarioRunAsyncParams
109
from .._client import AsyncRunloop
1110
from .async_scenario_run import AsyncScenarioRun
1211

@@ -65,28 +64,19 @@ async def get_info(
6564

6665
async def update(
6766
self,
68-
*,
69-
name: Optional[str] = None,
70-
metadata: Optional[Dict[str, str]] = None,
71-
**options: Unpack[LongRequestOptions],
67+
**params: Unpack[SDKScenarioUpdateParams],
7268
) -> ScenarioView:
7369
"""Update the scenario.
7470
7571
Only provided fields will be updated.
7672
77-
:param name: New name for the scenario
78-
:type name: Optional[str]
79-
:param metadata: New metadata for the scenario
80-
:type metadata: Optional[Dict[str, str]]
81-
:param options: Optional long-running request configuration
73+
:param params: See SDKScenarioUpdateParams for available parameters
8274
:return: Updated scenario info
8375
:rtype: ScenarioView
8476
"""
8577
return await self._client.scenarios.update(
8678
self._id,
87-
name=name,
88-
metadata=metadata,
89-
**options,
79+
**params,
9080
)
9181

9282
async def run_async(

src/runloop_api_client/sdk/scenario.py

Lines changed: 4 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,10 @@
22

33
from __future__ import annotations
44

5-
from typing import Dict, Optional
65
from typing_extensions import Unpack, override
76

87
from ..types import ScenarioView
9-
from ._types import BaseRequestOptions, LongRequestOptions, SDKScenarioRunAsyncParams, SDKScenarioRunParams
8+
from ._types import BaseRequestOptions, SDKScenarioRunParams, SDKScenarioUpdateParams, SDKScenarioRunAsyncParams
109
from .._client import Runloop
1110
from .scenario_run import ScenarioRun
1211

@@ -65,28 +64,19 @@ def get_info(
6564

6665
def update(
6766
self,
68-
*,
69-
name: Optional[str] = None,
70-
metadata: Optional[Dict[str, str]] = None,
71-
**options: Unpack[LongRequestOptions],
67+
**params: Unpack[SDKScenarioUpdateParams],
7268
) -> ScenarioView:
7369
"""Update the scenario.
7470
7571
Only provided fields will be updated.
7672
77-
:param name: New name for the scenario
78-
:type name: Optional[str]
79-
:param metadata: New metadata for the scenario
80-
:type metadata: Optional[Dict[str, str]]
81-
:param options: Optional long-running request configuration
73+
:param params: See SDKScenarioUpdateParams for available parameters
8274
:return: Updated scenario info
8375
:rtype: ScenarioView
8476
"""
8577
return self._client.scenarios.update(
8678
self._id,
87-
name=name,
88-
metadata=metadata,
89-
**options,
79+
**params,
9080
)
9181

9282
def run_async(

0 commit comments

Comments
 (0)