feat(api): api update

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 103
+configured_endpoints: 104
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/runloop-ai%2Frunloop-a8e2289b32b616e0bd7e15995a6553d0335bd21e30617a013d07450f8f5a7a20.yml
 openapi_spec_hash: cd13a7af1ffe64c7ce8753f87786e01f
-config_hash: b9de5d96e095a7e749a8c1fc3536c308
+config_hash: 98a0f112ecc9445456962dab3136d7b6

api.md

@@ -108,6 +108,7 @@ Methods:
 - <code title="post /v1/devboxes/{id}/snapshot_disk_async">client.devboxes.<a href="./src/runloop_api_client/resources/devboxes/devboxes.py">snapshot_disk_async</a>(id, \*\*<a href="src/runloop_api_client/types/devbox_snapshot_disk_async_params.py">params</a>) -> <a href="./src/runloop_api_client/types/devbox_snapshot_view.py">DevboxSnapshotView</a></code>
 - <code title="post /v1/devboxes/{id}/suspend">client.devboxes.<a href="./src/runloop_api_client/resources/devboxes/devboxes.py">suspend</a>(id) -> <a href="./src/runloop_api_client/types/devbox_view.py">DevboxView</a></code>
 - <code title="post /v1/devboxes/{id}/upload_file">client.devboxes.<a href="./src/runloop_api_client/resources/devboxes/devboxes.py">upload_file</a>(id, \*\*<a href="src/runloop_api_client/types/devbox_upload_file_params.py">params</a>) -> object</code>
+- <code title="post /v1/devboxes/{id}/wait_for_status">client.devboxes.<a href="./src/runloop_api_client/resources/devboxes/devboxes.py">wait_for_command</a>(id, \*\*<a href="src/runloop_api_client/types/devbox_wait_for_command_params.py">params</a>) -> <a href="./src/runloop_api_client/types/devbox_view.py">DevboxView</a></code>
 - <code title="post /v1/devboxes/{id}/write_file_contents">client.devboxes.<a href="./src/runloop_api_client/resources/devboxes/devboxes.py">write_file_contents</a>(id, \*\*<a href="src/runloop_api_client/types/devbox_write_file_contents_params.py">params</a>) -> <a href="./src/runloop_api_client/types/devbox_execution_detail_view.py">DevboxExecutionDetailView</a></code>
 
 

src/runloop_api_client/resources/devboxes/devboxes.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Dict, Mapping, Iterable, Optional, TypedDict, cast
+from typing import Dict, List, Mapping, Iterable, Optional, TypedDict, cast
 from typing_extensions import Literal
 
 import httpx
@@ -35,6 +35,7 @@
     devbox_execute_async_params,
     devbox_remove_tunnel_params,
     devbox_snapshot_disk_params,
+    devbox_wait_for_command_params,
     devbox_read_file_contents_params,
     devbox_list_disk_snapshots_params,
     devbox_snapshot_disk_async_params,
@@ -1407,6 +1408,66 @@ def upload_file(
             cast_to=object,
         )
 
+    def wait_for_command(
+        self,
+        id: str,
+        *,
+        statuses: List[
+            Literal[
+                "provisioning", "initializing", "running", "suspending", "suspended", "resuming", "failure", "shutdown"
+            ]
+        ],
+        timeout_seconds: Optional[int] | NotGiven = NOT_GIVEN,
+        # 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.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        idempotency_key: str | None = None,
+    ) -> DevboxView:
+        """
+        Polls the Devbox's status until it reaches one of the desired statuses or times
+        out.
+
+        Args:
+          statuses: The Devbox statuses to wait for. At least one status must be provided. The
+              devbox will be returned as soon as it reaches any of the provided statuses.
+
+          timeout_seconds: (Optional) Timeout in seconds to wait for the status, up to 30 seconds. Defaults
+              to 10 seconds.
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+
+          idempotency_key: Specify a custom idempotency key for this request
+        """
+        if not id:
+            raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
+        return self._post(
+            f"/v1/devboxes/{id}/wait_for_status",
+            body=maybe_transform(
+                {
+                    "statuses": statuses,
+                    "timeout_seconds": timeout_seconds,
+                },
+                devbox_wait_for_command_params.DevboxWaitForCommandParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                idempotency_key=idempotency_key,
+            ),
+            cast_to=DevboxView,
+        )
+
     def write_file_contents(
         self,
         id: str,
@@ -2735,6 +2796,66 @@ async def upload_file(
             cast_to=object,
         )
 
+    async def wait_for_command(
+        self,
+        id: str,
+        *,
+        statuses: List[
+            Literal[
+                "provisioning", "initializing", "running", "suspending", "suspended", "resuming", "failure", "shutdown"
+            ]
+        ],
+        timeout_seconds: Optional[int] | NotGiven = NOT_GIVEN,
+        # 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.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        idempotency_key: str | None = None,
+    ) -> DevboxView:
+        """
+        Polls the Devbox's status until it reaches one of the desired statuses or times
+        out.
+
+        Args:
+          statuses: The Devbox statuses to wait for. At least one status must be provided. The
+              devbox will be returned as soon as it reaches any of the provided statuses.
+
+          timeout_seconds: (Optional) Timeout in seconds to wait for the status, up to 30 seconds. Defaults
+              to 10 seconds.
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+
+          idempotency_key: Specify a custom idempotency key for this request
+        """
+        if not id:
+            raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
+        return await self._post(
+            f"/v1/devboxes/{id}/wait_for_status",
+            body=await async_maybe_transform(
+                {
+                    "statuses": statuses,
+                    "timeout_seconds": timeout_seconds,
+                },
+                devbox_wait_for_command_params.DevboxWaitForCommandParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                idempotency_key=idempotency_key,
+            ),
+            cast_to=DevboxView,
+        )
+
     async def write_file_contents(
         self,
         id: str,
@@ -2862,6 +2983,9 @@ def __init__(self, devboxes: DevboxesResource) -> None:
         self.upload_file = to_raw_response_wrapper(
             devboxes.upload_file,
         )
+        self.wait_for_command = to_raw_response_wrapper(
+            devboxes.wait_for_command,
+        )
         self.write_file_contents = to_raw_response_wrapper(
             devboxes.write_file_contents,
         )
@@ -2959,6 +3083,9 @@ def __init__(self, devboxes: AsyncDevboxesResource) -> None:
         self.upload_file = async_to_raw_response_wrapper(
             devboxes.upload_file,
         )
+        self.wait_for_command = async_to_raw_response_wrapper(
+            devboxes.wait_for_command,
+        )
         self.write_file_contents = async_to_raw_response_wrapper(
             devboxes.write_file_contents,
         )
@@ -3056,6 +3183,9 @@ def __init__(self, devboxes: DevboxesResource) -> None:
         self.upload_file = to_streamed_response_wrapper(
             devboxes.upload_file,
         )
+        self.wait_for_command = to_streamed_response_wrapper(
+            devboxes.wait_for_command,
+        )
         self.write_file_contents = to_streamed_response_wrapper(
             devboxes.write_file_contents,
         )
@@ -3153,6 +3283,9 @@ def __init__(self, devboxes: AsyncDevboxesResource) -> None:
         self.upload_file = async_to_streamed_response_wrapper(
             devboxes.upload_file,
         )
+        self.wait_for_command = async_to_streamed_response_wrapper(
+            devboxes.wait_for_command,
+        )
         self.write_file_contents = async_to_streamed_response_wrapper(
             devboxes.write_file_contents,
         )

src/runloop_api_client/types/__init__.py

@@ -86,6 +86,7 @@
 from .scoring_contract_update_param import ScoringContractUpdateParam as ScoringContractUpdateParam
 from .blueprint_build_logs_list_view import BlueprintBuildLogsListView as BlueprintBuildLogsListView
 from .devbox_create_ssh_key_response import DevboxCreateSSHKeyResponse as DevboxCreateSSHKeyResponse
+from .devbox_wait_for_command_params import DevboxWaitForCommandParams as DevboxWaitForCommandParams
 from .repository_connection_list_view import RepositoryConnectionListView as RepositoryConnectionListView
 from .repository_inspection_list_view import RepositoryInspectionListView as RepositoryInspectionListView
 from .devbox_read_file_contents_params import DevboxReadFileContentsParams as DevboxReadFileContentsParams

src/runloop_api_client/types/devbox_wait_for_command_params.py

@@ -0,0 +1,29 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import List, Optional
+from typing_extensions import Literal, Required, TypedDict
+
+__all__ = ["DevboxWaitForCommandParams"]
+
+
+class DevboxWaitForCommandParams(TypedDict, total=False):
+    statuses: Required[
+        List[
+            Literal[
+                "provisioning", "initializing", "running", "suspending", "suspended", "resuming", "failure", "shutdown"
+            ]
+        ]
+    ]
+    """The Devbox statuses to wait for.
+
+    At least one status must be provided. The devbox will be returned as soon as it
+    reaches any of the provided statuses.
+    """
+
+    timeout_seconds: Optional[int]
+    """(Optional) Timeout in seconds to wait for the status, up to 30 seconds.
+
+    Defaults to 10 seconds.
+    """

tests/api_resources/test_devboxes.py

@@ -980,6 +980,57 @@ def test_path_params_upload_file(self, client: Runloop) -> None:
                 path="path",
             )
 
+    @parametrize
+    def test_method_wait_for_command(self, client: Runloop) -> None:
+        devbox = client.devboxes.wait_for_command(
+            id="id",
+            statuses=["provisioning"],
+        )
+        assert_matches_type(DevboxView, devbox, path=["response"])
+
+    @parametrize
+    def test_method_wait_for_command_with_all_params(self, client: Runloop) -> None:
+        devbox = client.devboxes.wait_for_command(
+            id="id",
+            statuses=["provisioning"],
+            timeout_seconds=0,
+        )
+        assert_matches_type(DevboxView, devbox, path=["response"])
+
+    @parametrize
+    def test_raw_response_wait_for_command(self, client: Runloop) -> None:
+        response = client.devboxes.with_raw_response.wait_for_command(
+            id="id",
+            statuses=["provisioning"],
+        )
+
+        assert response.is_closed is True
+        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+        devbox = response.parse()
+        assert_matches_type(DevboxView, devbox, path=["response"])
+
+    @parametrize
+    def test_streaming_response_wait_for_command(self, client: Runloop) -> None:
+        with client.devboxes.with_streaming_response.wait_for_command(
+            id="id",
+            statuses=["provisioning"],
+        ) as response:
+            assert not response.is_closed
+            assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+
+            devbox = response.parse()
+            assert_matches_type(DevboxView, devbox, path=["response"])
+
+        assert cast(Any, response.is_closed) is True
+
+    @parametrize
+    def test_path_params_wait_for_command(self, client: Runloop) -> None:
+        with pytest.raises(ValueError, match=r"Expected a non-empty value for `id` but received ''"):
+            client.devboxes.with_raw_response.wait_for_command(
+                id="",
+                statuses=["provisioning"],
+            )
+
     @parametrize
     def test_method_write_file_contents(self, client: Runloop) -> None:
         devbox = client.devboxes.write_file_contents(
@@ -2256,6 +2307,57 @@ async def test_path_params_upload_file(self, async_client: AsyncRunloop) -> None
                 path="path",
             )
 
+    @parametrize
+    async def test_method_wait_for_command(self, async_client: AsyncRunloop) -> None:
+        devbox = await async_client.devboxes.wait_for_command(
+            id="id",
+            statuses=["provisioning"],
+        )
+        assert_matches_type(DevboxView, devbox, path=["response"])
+
+    @parametrize
+    async def test_method_wait_for_command_with_all_params(self, async_client: AsyncRunloop) -> None:
+        devbox = await async_client.devboxes.wait_for_command(
+            id="id",
+            statuses=["provisioning"],
+            timeout_seconds=0,
+        )
+        assert_matches_type(DevboxView, devbox, path=["response"])
+
+    @parametrize
+    async def test_raw_response_wait_for_command(self, async_client: AsyncRunloop) -> None:
+        response = await async_client.devboxes.with_raw_response.wait_for_command(
+            id="id",
+            statuses=["provisioning"],
+        )
+
+        assert response.is_closed is True
+        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+        devbox = await response.parse()
+        assert_matches_type(DevboxView, devbox, path=["response"])
+
+    @parametrize
+    async def test_streaming_response_wait_for_command(self, async_client: AsyncRunloop) -> None:
+        async with async_client.devboxes.with_streaming_response.wait_for_command(
+            id="id",
+            statuses=["provisioning"],
+        ) as response:
+            assert not response.is_closed
+            assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+
+            devbox = await response.parse()
+            assert_matches_type(DevboxView, devbox, path=["response"])
+
+        assert cast(Any, response.is_closed) is True
+
+    @parametrize
+    async def test_path_params_wait_for_command(self, async_client: AsyncRunloop) -> None:
+        with pytest.raises(ValueError, match=r"Expected a non-empty value for `id` but received ''"):
+            await async_client.devboxes.with_raw_response.wait_for_command(
+                id="",
+                statuses=["provisioning"],
+            )
+
     @parametrize
     async def test_method_write_file_contents(self, async_client: AsyncRunloop) -> None:
         devbox = await async_client.devboxes.write_file_contents(