expose request options in push()

Author: sid-rl

src/runloop_api_client/sdk/async_scenario_builder.py

@@ -3,9 +3,10 @@
 from __future__ import annotations
 
 from typing import Dict, List, Iterable, Optional
-from typing_extensions import Self, Literal, override
+from typing_extensions import Self, Unpack, Literal, override
 
 from ..types import ScenarioCreateParams, ScenarioEnvironmentParam
+from ._types import LongRequestOptions
 from .._client import AsyncRunloop
 from .async_scenario import AsyncScenario
 from .async_snapshot import AsyncSnapshot
@@ -443,13 +444,14 @@ def _build_params(self) -> ScenarioCreateParams:
 
         return params
 
-    async def push(self) -> AsyncScenario:
+    async def push(self, **options: Unpack[LongRequestOptions]) -> AsyncScenario:
         """Create the scenario on the platform.
 
+        :param options: Optional long-running request configuration
         :raises ValueError: If required fields are missing
         :return: Created scenario wrapper
         :rtype: AsyncScenario
         """
         params = self._build_params()
-        scenario_view = await self._client.scenarios.create(**params)
+        scenario_view = await self._client.scenarios.create(**params, **options)
         return AsyncScenario(self._client, scenario_view.id)

src/runloop_api_client/sdk/scenario_builder.py

@@ -3,9 +3,10 @@
 from __future__ import annotations
 
 from typing import Dict, List, Iterable, Optional
-from typing_extensions import Self, Literal, override
+from typing_extensions import Self, Unpack, Literal, override
 
 from ..types import ScenarioCreateParams, ScenarioEnvironmentParam
+from ._types import LongRequestOptions
 from .._client import Runloop
 from .scenario import Scenario
 from .snapshot import Snapshot
@@ -443,13 +444,14 @@ def _build_params(self) -> ScenarioCreateParams:
 
         return params
 
-    def push(self) -> Scenario:
+    def push(self, **options: Unpack[LongRequestOptions]) -> Scenario:
         """Create the scenario on the platform.
 
+        :param options: Optional long-running request configuration
         :raises ValueError: If required fields are missing
         :return: Created scenario wrapper
         :rtype: Scenario
         """
         params = self._build_params()
-        scenario_view = self._client.scenarios.create(**params)
+        scenario_view = self._client.scenarios.create(**params, **options)
         return Scenario(self._client, scenario_view.id)