chore(stainless): generate watch_evictions as an SSE stream (#10404)

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 120
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/runloop-ai/runloop-f18b1bba718700c841b3904207cd8ba2c066421a691845d6283bbbb361cf161d.yml
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/runloop-ai/runloop-6fb96a2de54d260cad2b76366cada4eda15335275dbab7f6c108c1bb29acf309.yml
 openapi_spec_hash: 6044f4a4da13881c5f82fcc394c1f43b
-config_hash: d9a5b7a4227236a55c89de56b56f4aec
+config_hash: 92c04f0b12249cb2d2bcf9e503a9cbf9

api.md

@@ -201,6 +201,7 @@ Types:
 ```python
 from runloop_api_client.types import (
     DevboxAsyncExecutionDetailView,
+    DevboxEvictionEventView,
     DevboxExecutionDetailView,
     DevboxKillExecutionRequest,
     DevboxListView,
@@ -214,7 +215,6 @@ from runloop_api_client.types import (
     TunnelView,
     DevboxCreateSSHKeyResponse,
     DevboxReadFileContentsResponse,
-    DevboxWatchEvictionsResponse,
 )
 ```
 
@@ -245,7 +245,7 @@ Methods:
 - <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/{devbox_id}/executions/{execution_id}/wait_for_status">client.devboxes.<a href="./src/runloop_api_client/resources/devboxes/devboxes.py">wait_for_command</a>(execution_id, \*, devbox_id, \*\*<a href="src/runloop_api_client/types/devbox_wait_for_command_params.py">params</a>) -> <a href="./src/runloop_api_client/types/devbox_async_execution_detail_view.py">DevboxAsyncExecutionDetailView</a></code>
-- <code title="get /v1/devboxes/watch_evictions">client.devboxes.<a href="./src/runloop_api_client/resources/devboxes/devboxes.py">watch_evictions</a>() -> <a href="./src/runloop_api_client/types/devbox_watch_evictions_response.py">DevboxWatchEvictionsResponse</a></code>
+- <code title="get /v1/devboxes/watch_evictions">client.devboxes.<a href="./src/runloop_api_client/resources/devboxes/devboxes.py">watch_evictions</a>() -> <a href="./src/runloop_api_client/types/devbox_eviction_event_view.py">DevboxEvictionEventView</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

@@ -69,6 +69,7 @@
     LONG_POLL_CLIENT_BUFFER_SECONDS,
     EXEC_LONG_POLL_SERVER_MAX_SECONDS,
 )
+from ..._streaming import Stream, AsyncStream
 from ...pagination import (
     SyncDevboxesCursorIDPage,
     AsyncDevboxesCursorIDPage,
@@ -94,10 +95,10 @@
 from ...types.shared_params.mount import Mount
 from ...types.devbox_snapshot_view import DevboxSnapshotView
 from ...types.shared.launch_parameters import LaunchParameters as SharedLaunchParameters
+from ...types.devbox_eviction_event_view import DevboxEvictionEventView
 from ...types.devbox_resource_usage_view import DevboxResourceUsageView
 from ...types.devbox_execution_detail_view import DevboxExecutionDetailView
 from ...types.devbox_create_ssh_key_response import DevboxCreateSSHKeyResponse
-from ...types.devbox_watch_evictions_response import DevboxWatchEvictionsResponse
 from ...types.shared_params.launch_parameters import LaunchParameters
 from ...types.devbox_async_execution_detail_view import DevboxAsyncExecutionDetailView
 from ...types.shared_params.code_mount_parameters import CodeMountParameters
@@ -1752,7 +1753,7 @@ def watch_evictions(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> DevboxWatchEvictionsResponse:
+    ) -> Stream[DevboxEvictionEventView]:
         """
         Subscribe, via server-sent events, to pending infrastructure evictions for every
         Devbox in the account. On connect the stream emits one event per Devbox that
@@ -1765,7 +1766,9 @@ def watch_evictions(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=DevboxWatchEvictionsResponse,
+            cast_to=DevboxEvictionEventView,
+            stream=True,
+            stream_cls=Stream[DevboxEvictionEventView],
         )
 
     def write_file_contents(
@@ -3452,7 +3455,7 @@ async def watch_evictions(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> DevboxWatchEvictionsResponse:
+    ) -> AsyncStream[DevboxEvictionEventView]:
         """
         Subscribe, via server-sent events, to pending infrastructure evictions for every
         Devbox in the account. On connect the stream emits one event per Devbox that
@@ -3465,7 +3468,9 @@ async def watch_evictions(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=DevboxWatchEvictionsResponse,
+            cast_to=DevboxEvictionEventView,
+            stream=True,
+            stream_cls=AsyncStream[DevboxEvictionEventView],
         )
 
     async def write_file_contents(

src/runloop_api_client/types/__init__.py

@@ -110,6 +110,7 @@
 from .scenario_start_run_params import ScenarioStartRunParams as ScenarioStartRunParams
 from .benchmark_start_run_params import BenchmarkStartRunParams as BenchmarkStartRunParams
 from .blueprint_build_parameters import BlueprintBuildParameters as BlueprintBuildParameters
+from .devbox_eviction_event_view import DevboxEvictionEventView as DevboxEvictionEventView
 from .devbox_execute_sync_params import DevboxExecuteSyncParams as DevboxExecuteSyncParams
 from .devbox_resource_usage_view import DevboxResourceUsageView as DevboxResourceUsageView
 from .gateway_config_list_params import GatewayConfigListParams as GatewayConfigListParams
@@ -139,7 +140,6 @@
 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 .devbox_watch_evictions_response import DevboxWatchEvictionsResponse as DevboxWatchEvictionsResponse
 from .devbox_read_file_contents_params import DevboxReadFileContentsParams as DevboxReadFileContentsParams
 from .benchmark_update_scenarios_params import BenchmarkUpdateScenariosParams as BenchmarkUpdateScenariosParams
 from .devbox_list_disk_snapshots_params import DevboxListDiskSnapshotsParams as DevboxListDiskSnapshotsParams

…types/devbox_watch_evictions_response.py …ient/types/devbox_eviction_event_view.pysrc/runloop_api_client/types/devbox_watch_evictions_response.py renamed to src/runloop_api_client/types/devbox_eviction_event_view.py src/runloop_api_client/types/devbox_watch_evictions_response.py renamed to src/runloop_api_client/types/devbox_eviction_event_view.py

@@ -2,10 +2,10 @@
 
 from .._models import BaseModel
 
-__all__ = ["DevboxWatchEvictionsResponse"]
+__all__ = ["DevboxEvictionEventView"]
 
 
-class DevboxWatchEvictionsResponse(BaseModel):
+class DevboxEvictionEventView(BaseModel):
     devbox_id: str
     """The ID of the Devbox with a pending eviction."""
 

tests/api_resources/test_devboxes.py

@@ -20,7 +20,6 @@
     DevboxResourceUsageView,
     DevboxExecutionDetailView,
     DevboxCreateSSHKeyResponse,
-    DevboxWatchEvictionsResponse,
     DevboxAsyncExecutionDetailView,
 )
 from runloop_api_client._response import (
@@ -1193,26 +1192,25 @@ def test_path_params_wait_for_command(self, client: Runloop) -> None:
 
     @parametrize
     def test_method_watch_evictions(self, client: Runloop) -> None:
-        devbox = client.devboxes.watch_evictions()
-        assert_matches_type(DevboxWatchEvictionsResponse, devbox, path=["response"])
+        devbox_stream = client.devboxes.watch_evictions()
+        devbox_stream.response.close()
 
     @parametrize
     def test_raw_response_watch_evictions(self, client: Runloop) -> None:
         response = client.devboxes.with_raw_response.watch_evictions()
 
-        assert response.is_closed is True
         assert response.http_request.headers.get("X-Stainless-Lang") == "python"
-        devbox = response.parse()
-        assert_matches_type(DevboxWatchEvictionsResponse, devbox, path=["response"])
+        stream = response.parse()
+        stream.close()
 
     @parametrize
     def test_streaming_response_watch_evictions(self, client: Runloop) -> None:
         with client.devboxes.with_streaming_response.watch_evictions() as response:
             assert not response.is_closed
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
 
-            devbox = response.parse()
-            assert_matches_type(DevboxWatchEvictionsResponse, devbox, path=["response"])
+            stream = response.parse()
+            stream.close()
 
         assert cast(Any, response.is_closed) is True
 
@@ -2887,26 +2885,25 @@ async def test_path_params_wait_for_command(self, async_client: AsyncRunloop) ->
 
     @parametrize
     async def test_method_watch_evictions(self, async_client: AsyncRunloop) -> None:
-        devbox = await async_client.devboxes.watch_evictions()
-        assert_matches_type(DevboxWatchEvictionsResponse, devbox, path=["response"])
+        devbox_stream = await async_client.devboxes.watch_evictions()
+        await devbox_stream.response.aclose()
 
     @parametrize
     async def test_raw_response_watch_evictions(self, async_client: AsyncRunloop) -> None:
         response = await async_client.devboxes.with_raw_response.watch_evictions()
 
-        assert response.is_closed is True
         assert response.http_request.headers.get("X-Stainless-Lang") == "python"
-        devbox = await response.parse()
-        assert_matches_type(DevboxWatchEvictionsResponse, devbox, path=["response"])
+        stream = await response.parse()
+        await stream.close()
 
     @parametrize
     async def test_streaming_response_watch_evictions(self, async_client: AsyncRunloop) -> None:
         async with async_client.devboxes.with_streaming_response.watch_evictions() as response:
             assert not response.is_closed
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
 
-            devbox = await response.parse()
-            assert_matches_type(DevboxWatchEvictionsResponse, devbox, path=["response"])
+            stream = await response.parse()
+            await stream.close()
 
         assert cast(Any, response.is_closed) is True