feat: Add video_memory to request the 4GiB VRAM GPU browser tier

Stainless-Generated-From: f581fece2ccb216a10620562f96439429da85527

Author: kernel-internal[bot]

api.md

@@ -131,6 +131,7 @@ from kernel.types import (
     BrowserReplResult,
     BrowserReplTextContent,
     BrowserUsage,
+    BrowserVideoMemory,
     Profile,
     Tags,
     VaultReference,

src/kernel/resources/browsers/browsers.py

@@ -25,6 +25,7 @@
     AsyncFsResourceWithStreamingResponse,
 )
 from ...types import (
+    BrowserVideoMemory,
     BrowserMemoryRequest,
     browser_curl_params,
     browser_list_params,
@@ -97,6 +98,7 @@
     AsyncPlaywrightResourceWithStreamingResponse,
 )
 from ...types.browser_repl_result import BrowserReplResult
+from ...types.browser_video_memory import BrowserVideoMemory
 from ...types.browser_curl_response import BrowserCurlResponse
 from ...types.browser_list_response import BrowserListResponse
 from ...types.vault_reference_param import VaultReferenceParam
@@ -206,6 +208,7 @@ def create(
         telemetry: Optional[browser_create_params.Telemetry] | Omit = omit,
         timeout_seconds: int | Omit = omit,
         vaults: Iterable[VaultReferenceParam] | Omit = omit,
+        video_memory: BrowserVideoMemory | Omit = omit,
         viewport: BrowserViewport | 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.
@@ -293,6 +296,10 @@ def create(
           vaults: Project-scoped vaults to link to the browser session. Links are immutable after
               creation.
 
+          video_memory: Video memory (VRAM) for a GPU browser session. Requires gpu=true. Defaults to
+              2GiB, which comes with 4 vCPU and 6GiB memory. 4GiB comes with 8 vCPU and 12GiB
+              memory.
+
           viewport: Initial browser window size in pixels with optional refresh rate. If omitted,
               image defaults apply (1920x1080@25). For GPU images, the default is
               1920x1080@60. Arbitrary viewport dimensions and refresh rates are accepted.
@@ -337,6 +344,7 @@ def create(
                     "telemetry": telemetry,
                     "timeout_seconds": timeout_seconds,
                     "vaults": vaults,
+                    "video_memory": video_memory,
                     "viewport": viewport,
                 },
                 browser_create_params.BrowserCreateParams,
@@ -915,6 +923,7 @@ async def create(
         telemetry: Optional[browser_create_params.Telemetry] | Omit = omit,
         timeout_seconds: int | Omit = omit,
         vaults: Iterable[VaultReferenceParam] | Omit = omit,
+        video_memory: BrowserVideoMemory | Omit = omit,
         viewport: BrowserViewport | 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.
@@ -1002,6 +1011,10 @@ async def create(
           vaults: Project-scoped vaults to link to the browser session. Links are immutable after
               creation.
 
+          video_memory: Video memory (VRAM) for a GPU browser session. Requires gpu=true. Defaults to
+              2GiB, which comes with 4 vCPU and 6GiB memory. 4GiB comes with 8 vCPU and 12GiB
+              memory.
+
           viewport: Initial browser window size in pixels with optional refresh rate. If omitted,
               image defaults apply (1920x1080@25). For GPU images, the default is
               1920x1080@60. Arbitrary viewport dimensions and refresh rates are accepted.
@@ -1046,6 +1059,7 @@ async def create(
                     "telemetry": telemetry,
                     "timeout_seconds": timeout_seconds,
                     "vaults": vaults,
+                    "video_memory": video_memory,
                     "viewport": viewport,
                 },
                 browser_create_params.BrowserCreateParams,

src/kernel/types/__init__.py

@@ -65,6 +65,7 @@
 from .vault_upsert_params import VaultUpsertParams as VaultUpsertParams
 from .browser_proxy_config import BrowserProxyConfig as BrowserProxyConfig
 from .browser_repl_content import BrowserReplContent as BrowserReplContent
+from .browser_video_memory import BrowserVideoMemory as BrowserVideoMemory
 from .proxy_check_response import ProxyCheckResponse as ProxyCheckResponse
 from .search_create_params import SearchCreateParams as SearchCreateParams
 from .vault_encryption_key import VaultEncryptionKey as VaultEncryptionKey

src/kernel/types/browser_create_params.py

@@ -6,6 +6,7 @@
 from typing_extensions import Literal, TypedDict
 
 from .tags_param import TagsParam
+from .browser_video_memory import BrowserVideoMemory
 from .vault_reference_param import VaultReferenceParam
 from .browser_memory_request import BrowserMemoryRequest
 from .browser_proxy_config_param import BrowserProxyConfigParam
@@ -158,6 +159,13 @@ class BrowserCreateParams(TypedDict, total=False):
     Links are immutable after creation.
     """
 
+    video_memory: BrowserVideoMemory
+    """Video memory (VRAM) for a GPU browser session.
+
+    Requires gpu=true. Defaults to 2GiB, which comes with 4 vCPU and 6GiB memory.
+    4GiB comes with 8 vCPU and 12GiB memory.
+    """
+
     viewport: BrowserViewport
     """
     Initial browser window size in pixels with optional refresh rate. If omitted,

src/kernel/types/browser_create_response.py

@@ -12,6 +12,7 @@
 from .browser_memory import BrowserMemory
 from .vault_reference import VaultReference
 from .browser_pool_ref import BrowserPoolRef
+from .browser_video_memory import BrowserVideoMemory
 from .browser_network_config import BrowserNetworkConfig
 from .shared.browser_viewport import BrowserViewport
 from .browsers.browser_telemetry_config import BrowserTelemetryConfig
@@ -135,6 +136,9 @@ class BrowserCreateResponse(BaseModel):
     vaults: Optional[List[VaultReference]] = None
     """Vaults linked when the browser session was created."""
 
+    video_memory: Optional[BrowserVideoMemory] = None
+    """Video memory (VRAM) of the GPU browser session. Only present when gpu is true."""
+
     viewport: Optional[BrowserViewport] = None
     """
     Initial browser window size in pixels with optional refresh rate. If omitted,

src/kernel/types/browser_list_response.py

@@ -12,6 +12,7 @@
 from .browser_memory import BrowserMemory
 from .vault_reference import VaultReference
 from .browser_pool_ref import BrowserPoolRef
+from .browser_video_memory import BrowserVideoMemory
 from .browser_network_config import BrowserNetworkConfig
 from .shared.browser_viewport import BrowserViewport
 from .browsers.browser_telemetry_config import BrowserTelemetryConfig
@@ -135,6 +136,9 @@ class BrowserListResponse(BaseModel):
     vaults: Optional[List[VaultReference]] = None
     """Vaults linked when the browser session was created."""
 
+    video_memory: Optional[BrowserVideoMemory] = None
+    """Video memory (VRAM) of the GPU browser session. Only present when gpu is true."""
+
     viewport: Optional[BrowserViewport] = None
     """
     Initial browser window size in pixels with optional refresh rate. If omitted,

src/kernel/types/browser_memory.py

@@ -4,4 +4,4 @@
 
 __all__ = ["BrowserMemory"]
 
-BrowserMemory: TypeAlias = Literal["1GiB", "2GiB", "6GiB", "8GiB", "16GiB"]
+BrowserMemory: TypeAlias = Literal["1GiB", "2GiB", "6GiB", "8GiB", "12GiB", "16GiB"]

src/kernel/types/browser_pool_acquire_response.py

@@ -12,6 +12,7 @@
 from .browser_memory import BrowserMemory
 from .vault_reference import VaultReference
 from .browser_pool_ref import BrowserPoolRef
+from .browser_video_memory import BrowserVideoMemory
 from .browser_network_config import BrowserNetworkConfig
 from .shared.browser_viewport import BrowserViewport
 from .browsers.browser_telemetry_config import BrowserTelemetryConfig
@@ -135,6 +136,9 @@ class BrowserPoolAcquireResponse(BaseModel):
     vaults: Optional[List[VaultReference]] = None
     """Vaults linked when the browser session was created."""
 
+    video_memory: Optional[BrowserVideoMemory] = None
+    """Video memory (VRAM) of the GPU browser session. Only present when gpu is true."""
+
     viewport: Optional[BrowserViewport] = None
     """
     Initial browser window size in pixels with optional refresh rate. If omitted,

src/kernel/types/browser_retrieve_response.py

@@ -12,6 +12,7 @@
 from .browser_memory import BrowserMemory
 from .vault_reference import VaultReference
 from .browser_pool_ref import BrowserPoolRef
+from .browser_video_memory import BrowserVideoMemory
 from .browser_network_config import BrowserNetworkConfig
 from .shared.browser_viewport import BrowserViewport
 from .browsers.browser_telemetry_config import BrowserTelemetryConfig
@@ -135,6 +136,9 @@ class BrowserRetrieveResponse(BaseModel):
     vaults: Optional[List[VaultReference]] = None
     """Vaults linked when the browser session was created."""
 
+    video_memory: Optional[BrowserVideoMemory] = None
+    """Video memory (VRAM) of the GPU browser session. Only present when gpu is true."""
+
     viewport: Optional[BrowserViewport] = None
     """
     Initial browser window size in pixels with optional refresh rate. If omitted,

src/kernel/types/browser_update_response.py

@@ -12,6 +12,7 @@
 from .browser_memory import BrowserMemory
 from .vault_reference import VaultReference
 from .browser_pool_ref import BrowserPoolRef
+from .browser_video_memory import BrowserVideoMemory
 from .browser_network_config import BrowserNetworkConfig
 from .shared.browser_viewport import BrowserViewport
 from .browsers.browser_telemetry_config import BrowserTelemetryConfig
@@ -135,6 +136,9 @@ class BrowserUpdateResponse(BaseModel):
     vaults: Optional[List[VaultReference]] = None
     """Vaults linked when the browser session was created."""
 
+    video_memory: Optional[BrowserVideoMemory] = None
+    """Video memory (VRAM) of the GPU browser session. Only present when gpu is true."""
+
     viewport: Optional[BrowserViewport] = None
     """
     Initial browser window size in pixels with optional refresh rate. If omitted,