feat: add WebSocket process attach and PTY support

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 89
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/kernel%2Fkernel-8e4a29d23d2882fcb0864606091790fd58bffa4f5d5c8d081052b72ad47b215b.yml
-openapi_spec_hash: fc82d930dad739ac01e3c2bddba7bf61
-config_hash: d0585c44724cd7deecea60a5ffae69df
+configured_endpoints: 90
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/kernel%2Fkernel-6a44c851ec955b997558a8524eb641355ff3097474f40772b8ea2fef5bee4134.yml
+openapi_spec_hash: 155ee005a1b43e1c11e843de91e9f509
+config_hash: 6cbbf855a29bc675f35ddb1106ea9083

api.md

@@ -154,6 +154,7 @@ Types:
 from kernel.types.browsers import (
     ProcessExecResponse,
     ProcessKillResponse,
+    ProcessResizeResponse,
     ProcessSpawnResponse,
     ProcessStatusResponse,
     ProcessStdinResponse,
@@ -165,6 +166,7 @@ Methods:
 
 - <code title="post /browsers/{id}/process/exec">client.browsers.process.<a href="./src/kernel/resources/browsers/process.py">exec</a>(id, \*\*<a href="src/kernel/types/browsers/process_exec_params.py">params</a>) -> <a href="./src/kernel/types/browsers/process_exec_response.py">ProcessExecResponse</a></code>
 - <code title="post /browsers/{id}/process/{process_id}/kill">client.browsers.process.<a href="./src/kernel/resources/browsers/process.py">kill</a>(process_id, \*, id, \*\*<a href="src/kernel/types/browsers/process_kill_params.py">params</a>) -> <a href="./src/kernel/types/browsers/process_kill_response.py">ProcessKillResponse</a></code>
+- <code title="post /browsers/{id}/process/{process_id}/resize">client.browsers.process.<a href="./src/kernel/resources/browsers/process.py">resize</a>(process_id, \*, id, \*\*<a href="src/kernel/types/browsers/process_resize_params.py">params</a>) -> <a href="./src/kernel/types/browsers/process_resize_response.py">ProcessResizeResponse</a></code>
 - <code title="post /browsers/{id}/process/spawn">client.browsers.process.<a href="./src/kernel/resources/browsers/process.py">spawn</a>(id, \*\*<a href="src/kernel/types/browsers/process_spawn_params.py">params</a>) -> <a href="./src/kernel/types/browsers/process_spawn_response.py">ProcessSpawnResponse</a></code>
 - <code title="get /browsers/{id}/process/{process_id}/status">client.browsers.process.<a href="./src/kernel/resources/browsers/process.py">status</a>(process_id, \*, id) -> <a href="./src/kernel/types/browsers/process_status_response.py">ProcessStatusResponse</a></code>
 - <code title="post /browsers/{id}/process/{process_id}/stdin">client.browsers.process.<a href="./src/kernel/resources/browsers/process.py">stdin</a>(process_id, \*, id, \*\*<a href="src/kernel/types/browsers/process_stdin_params.py">params</a>) -> <a href="./src/kernel/types/browsers/process_stdin_response.py">ProcessStdinResponse</a></code>

src/kernel/resources/browsers/process.py

@@ -19,11 +19,18 @@
 )
 from ..._streaming import Stream, AsyncStream
 from ..._base_client import make_request_options
-from ...types.browsers import process_exec_params, process_kill_params, process_spawn_params, process_stdin_params
+from ...types.browsers import (
+    process_exec_params,
+    process_kill_params,
+    process_spawn_params,
+    process_stdin_params,
+    process_resize_params,
+)
 from ...types.browsers.process_exec_response import ProcessExecResponse
 from ...types.browsers.process_kill_response import ProcessKillResponse
 from ...types.browsers.process_spawn_response import ProcessSpawnResponse
 from ...types.browsers.process_stdin_response import ProcessStdinResponse
+from ...types.browsers.process_resize_response import ProcessResizeResponse
 from ...types.browsers.process_status_response import ProcessStatusResponse
 from ...types.browsers.process_stdout_stream_response import ProcessStdoutStreamResponse
 
@@ -156,16 +163,68 @@ def kill(
             cast_to=ProcessKillResponse,
         )
 
+    def resize(
+        self,
+        process_id: str,
+        *,
+        id: str,
+        cols: int,
+        rows: int,
+        # 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,
+    ) -> ProcessResizeResponse:
+        """
+        Resize a PTY-backed process terminal
+
+        Args:
+          cols: New terminal columns.
+
+          rows: New terminal rows.
+
+          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
+        """
+        if not id:
+            raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
+        if not process_id:
+            raise ValueError(f"Expected a non-empty value for `process_id` but received {process_id!r}")
+        return self._post(
+            f"/browsers/{id}/process/{process_id}/resize",
+            body=maybe_transform(
+                {
+                    "cols": cols,
+                    "rows": rows,
+                },
+                process_resize_params.ProcessResizeParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=ProcessResizeResponse,
+        )
+
     def spawn(
         self,
         id: str,
         *,
         command: str,
+        allocate_tty: bool | Omit = omit,
         args: SequenceNotStr[str] | Omit = omit,
         as_root: bool | Omit = omit,
         as_user: Optional[str] | Omit = omit,
+        cols: int | Omit = omit,
         cwd: Optional[str] | Omit = omit,
         env: Dict[str, str] | Omit = omit,
+        rows: int | Omit = omit,
         timeout_sec: Optional[int] | Omit = omit,
         # 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.
@@ -180,16 +239,22 @@ def spawn(
         Args:
           command: Executable or shell command to run.
 
+          allocate_tty: Allocate a pseudo-terminal (PTY) for interactive shells.
+
           args: Command arguments.
 
           as_root: Run the process with root privileges.
 
           as_user: Run the process as this user.
 
+          cols: Initial terminal columns. Only used when allocate_tty is true.
+
           cwd: Working directory (absolute path) to run the command in.
 
           env: Environment variables to set for the process.
 
+          rows: Initial terminal rows. Only used when allocate_tty is true.
+
           timeout_sec: Maximum execution time in seconds.
 
           extra_headers: Send extra headers
@@ -207,11 +272,14 @@ def spawn(
             body=maybe_transform(
                 {
                     "command": command,
+                    "allocate_tty": allocate_tty,
                     "args": args,
                     "as_root": as_root,
                     "as_user": as_user,
+                    "cols": cols,
                     "cwd": cwd,
                     "env": env,
+                    "rows": rows,
                     "timeout_sec": timeout_sec,
                 },
                 process_spawn_params.ProcessSpawnParams,
@@ -464,16 +532,68 @@ async def kill(
             cast_to=ProcessKillResponse,
         )
 
+    async def resize(
+        self,
+        process_id: str,
+        *,
+        id: str,
+        cols: int,
+        rows: int,
+        # 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,
+    ) -> ProcessResizeResponse:
+        """
+        Resize a PTY-backed process terminal
+
+        Args:
+          cols: New terminal columns.
+
+          rows: New terminal rows.
+
+          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
+        """
+        if not id:
+            raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
+        if not process_id:
+            raise ValueError(f"Expected a non-empty value for `process_id` but received {process_id!r}")
+        return await self._post(
+            f"/browsers/{id}/process/{process_id}/resize",
+            body=await async_maybe_transform(
+                {
+                    "cols": cols,
+                    "rows": rows,
+                },
+                process_resize_params.ProcessResizeParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=ProcessResizeResponse,
+        )
+
     async def spawn(
         self,
         id: str,
         *,
         command: str,
+        allocate_tty: bool | Omit = omit,
         args: SequenceNotStr[str] | Omit = omit,
         as_root: bool | Omit = omit,
         as_user: Optional[str] | Omit = omit,
+        cols: int | Omit = omit,
         cwd: Optional[str] | Omit = omit,
         env: Dict[str, str] | Omit = omit,
+        rows: int | Omit = omit,
         timeout_sec: Optional[int] | Omit = omit,
         # 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.
@@ -488,16 +608,22 @@ async def spawn(
         Args:
           command: Executable or shell command to run.
 
+          allocate_tty: Allocate a pseudo-terminal (PTY) for interactive shells.
+
           args: Command arguments.
 
           as_root: Run the process with root privileges.
 
           as_user: Run the process as this user.
 
+          cols: Initial terminal columns. Only used when allocate_tty is true.
+
           cwd: Working directory (absolute path) to run the command in.
 
           env: Environment variables to set for the process.
 
+          rows: Initial terminal rows. Only used when allocate_tty is true.
+
           timeout_sec: Maximum execution time in seconds.
 
           extra_headers: Send extra headers
@@ -515,11 +641,14 @@ async def spawn(
             body=await async_maybe_transform(
                 {
                     "command": command,
+                    "allocate_tty": allocate_tty,
                     "args": args,
                     "as_root": as_root,
                     "as_user": as_user,
+                    "cols": cols,
                     "cwd": cwd,
                     "env": env,
+                    "rows": rows,
                     "timeout_sec": timeout_sec,
                 },
                 process_spawn_params.ProcessSpawnParams,
@@ -656,6 +785,9 @@ def __init__(self, process: ProcessResource) -> None:
         self.kill = to_raw_response_wrapper(
             process.kill,
         )
+        self.resize = to_raw_response_wrapper(
+            process.resize,
+        )
         self.spawn = to_raw_response_wrapper(
             process.spawn,
         )
@@ -680,6 +812,9 @@ def __init__(self, process: AsyncProcessResource) -> None:
         self.kill = async_to_raw_response_wrapper(
             process.kill,
         )
+        self.resize = async_to_raw_response_wrapper(
+            process.resize,
+        )
         self.spawn = async_to_raw_response_wrapper(
             process.spawn,
         )
@@ -704,6 +839,9 @@ def __init__(self, process: ProcessResource) -> None:
         self.kill = to_streamed_response_wrapper(
             process.kill,
         )
+        self.resize = to_streamed_response_wrapper(
+            process.resize,
+        )
         self.spawn = to_streamed_response_wrapper(
             process.spawn,
         )
@@ -728,6 +866,9 @@ def __init__(self, process: AsyncProcessResource) -> None:
         self.kill = async_to_streamed_response_wrapper(
             process.kill,
         )
+        self.resize = async_to_streamed_response_wrapper(
+            process.resize,
+        )
         self.spawn = async_to_streamed_response_wrapper(
             process.spawn,
         )

src/kernel/types/browsers/__init__.py

@@ -21,10 +21,12 @@
 from .f_list_files_response import FListFilesResponse as FListFilesResponse
 from .process_exec_response import ProcessExecResponse as ProcessExecResponse
 from .process_kill_response import ProcessKillResponse as ProcessKillResponse
+from .process_resize_params import ProcessResizeParams as ProcessResizeParams
 from .replay_start_response import ReplayStartResponse as ReplayStartResponse
 from .computer_scroll_params import ComputerScrollParams as ComputerScrollParams
 from .process_spawn_response import ProcessSpawnResponse as ProcessSpawnResponse
 from .process_stdin_response import ProcessStdinResponse as ProcessStdinResponse
+from .process_resize_response import ProcessResizeResponse as ProcessResizeResponse
 from .process_status_response import ProcessStatusResponse as ProcessStatusResponse
 from .computer_press_key_params import ComputerPressKeyParams as ComputerPressKeyParams
 from .computer_type_text_params import ComputerTypeTextParams as ComputerTypeTextParams

src/kernel/types/browsers/process_resize_params.py

@@ -0,0 +1,17 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, TypedDict
+
+__all__ = ["ProcessResizeParams"]
+
+
+class ProcessResizeParams(TypedDict, total=False):
+    id: Required[str]
+
+    cols: Required[int]
+    """New terminal columns."""
+
+    rows: Required[int]
+    """New terminal rows."""

src/kernel/types/browsers/process_resize_response.py

@@ -0,0 +1,12 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from ..._models import BaseModel
+
+__all__ = ["ProcessResizeResponse"]
+
+
+class ProcessResizeResponse(BaseModel):
+    """Generic OK response."""
+
+    ok: bool
+    """Indicates success."""

src/kernel/types/browsers/process_spawn_params.py

@@ -14,6 +14,9 @@ class ProcessSpawnParams(TypedDict, total=False):
     command: Required[str]
     """Executable or shell command to run."""
 
+    allocate_tty: bool
+    """Allocate a pseudo-terminal (PTY) for interactive shells."""
+
     args: SequenceNotStr[str]
     """Command arguments."""
 
@@ -23,11 +26,17 @@ class ProcessSpawnParams(TypedDict, total=False):
     as_user: Optional[str]
     """Run the process as this user."""
 
+    cols: int
+    """Initial terminal columns. Only used when allocate_tty is true."""
+
     cwd: Optional[str]
     """Working directory (absolute path) to run the command in."""
 
     env: Dict[str, str]
     """Environment variables to set for the process."""
 
+    rows: int
+    """Initial terminal rows. Only used when allocate_tty is true."""
+
     timeout_sec: Optional[int]
     """Maximum execution time in seconds."""