modal 0.72.4__py3-none-any.whl → 0.72.48__py3-none-any.whl

modal_proto/api_pb2.pyi

@@ -8,6 +8,7 @@ import google.protobuf.descriptor
 import google.protobuf.internal.containers
 import google.protobuf.internal.enum_type_wrapper
 import google.protobuf.message
+import google.protobuf.struct_pb2
 import google.protobuf.wrappers_pb2
 import sys
 import typing
@@ -179,6 +180,7 @@ class _ClientTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._Enu
     CLIENT_TYPE_WORKER: _ClientType.ValueType  # 2
     CLIENT_TYPE_CONTAINER: _ClientType.ValueType  # 3
     CLIENT_TYPE_WEB_SERVER: _ClientType.ValueType  # 5
+    CLIENT_TYPE_NOTEBOOK_KERNEL: _ClientType.ValueType  # 6
 
 class ClientType(_ClientType, metaclass=_ClientTypeEnumTypeWrapper): ...
 
@@ -187,6 +189,7 @@ CLIENT_TYPE_CLIENT: ClientType.ValueType  # 1
 CLIENT_TYPE_WORKER: ClientType.ValueType  # 2
 CLIENT_TYPE_CONTAINER: ClientType.ValueType  # 3
 CLIENT_TYPE_WEB_SERVER: ClientType.ValueType  # 5
+CLIENT_TYPE_NOTEBOOK_KERNEL: ClientType.ValueType  # 6
 global___ClientType = ClientType
 
 class _CloudProvider:
@@ -361,6 +364,7 @@ class _GPUTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTy
     GPU_TYPE_L4: _GPUType.ValueType  # 9
     GPU_TYPE_H100: _GPUType.ValueType  # 10
     GPU_TYPE_L40S: _GPUType.ValueType  # 11
+    GPU_TYPE_H200: _GPUType.ValueType  # 12
 
 class GPUType(_GPUType, metaclass=_GPUTypeEnumTypeWrapper): ...
 
@@ -373,6 +377,7 @@ GPU_TYPE_A100_80GB: GPUType.ValueType  # 8
 GPU_TYPE_L4: GPUType.ValueType  # 9
 GPU_TYPE_H100: GPUType.ValueType  # 10
 GPU_TYPE_L40S: GPUType.ValueType  # 11
+GPU_TYPE_H200: GPUType.ValueType  # 12
 global___GPUType = GPUType
 
 class _ObjectCreationType:
@@ -943,6 +948,7 @@ class AppGetLogsRequest(google.protobuf.message.Message):
     FUNCTION_ID_FIELD_NUMBER: builtins.int
     INPUT_ID_FIELD_NUMBER: builtins.int
     TASK_ID_FIELD_NUMBER: builtins.int
+    FUNCTION_CALL_ID_FIELD_NUMBER: builtins.int
     FILE_DESCRIPTOR_FIELD_NUMBER: builtins.int
     app_id: builtins.str
     timeout: builtins.float
@@ -950,6 +956,7 @@ class AppGetLogsRequest(google.protobuf.message.Message):
     function_id: builtins.str
     input_id: builtins.str
     task_id: builtins.str
+    function_call_id: builtins.str
     file_descriptor: global___FileDescriptor.ValueType
     def __init__(
         self,
@@ -960,9 +967,10 @@ class AppGetLogsRequest(google.protobuf.message.Message):
         function_id: builtins.str = ...,
         input_id: builtins.str = ...,
         task_id: builtins.str = ...,
+        function_call_id: builtins.str = ...,
         file_descriptor: global___FileDescriptor.ValueType = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["app_id", b"app_id", "file_descriptor", b"file_descriptor", "function_id", b"function_id", "input_id", b"input_id", "last_entry_id", b"last_entry_id", "task_id", b"task_id", "timeout", b"timeout"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["app_id", b"app_id", "file_descriptor", b"file_descriptor", "function_call_id", b"function_call_id", "function_id", b"function_id", "input_id", b"input_id", "last_entry_id", b"last_entry_id", "task_id", b"task_id", "timeout", b"timeout"]) -> None: ...
 
 global___AppGetLogsRequest = AppGetLogsRequest
 
@@ -1895,6 +1903,7 @@ class CheckpointInfo(google.protobuf.message.Message):
     SIZE_FIELD_NUMBER: builtins.int
     CHECKSUM_IS_FILE_INDEX_FIELD_NUMBER: builtins.int
     ORIGINAL_TASK_ID_FIELD_NUMBER: builtins.int
+    SNAPSHOT_VERSION_FIELD_NUMBER: builtins.int
     checksum: builtins.str
     status: global___CheckpointStatus.ValueType
     checkpoint_id: builtins.str
@@ -1902,6 +1911,8 @@ class CheckpointInfo(google.protobuf.message.Message):
     size: builtins.int
     checksum_is_file_index: builtins.bool
     original_task_id: builtins.str
+    snapshot_version: builtins.int
+    """snapshot version used to take the checkpoint"""
     def __init__(
         self,
         *,
@@ -1912,8 +1923,9 @@ class CheckpointInfo(google.protobuf.message.Message):
         size: builtins.int = ...,
         checksum_is_file_index: builtins.bool = ...,
         original_task_id: builtins.str = ...,
+        snapshot_version: builtins.int = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["checkpoint_id", b"checkpoint_id", "checksum", b"checksum", "checksum_is_file_index", b"checksum_is_file_index", "original_task_id", b"original_task_id", "runtime_fingerprint", b"runtime_fingerprint", "size", b"size", "status", b"status"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["checkpoint_id", b"checkpoint_id", "checksum", b"checksum", "checksum_is_file_index", b"checksum_is_file_index", "original_task_id", b"original_task_id", "runtime_fingerprint", b"runtime_fingerprint", "size", b"size", "snapshot_version", b"snapshot_version", "status", b"status"]) -> None: ...
 
 global___CheckpointInfo = CheckpointInfo
 
@@ -2228,6 +2240,7 @@ class CloudBucketMount(google.protobuf.message.Message):
     REQUESTER_PAYS_FIELD_NUMBER: builtins.int
     BUCKET_ENDPOINT_URL_FIELD_NUMBER: builtins.int
     KEY_PREFIX_FIELD_NUMBER: builtins.int
+    OIDC_AUTH_ROLE_ARN_FIELD_NUMBER: builtins.int
     bucket_name: builtins.str
     mount_path: builtins.str
     credentials_secret_id: builtins.str
@@ -2236,6 +2249,7 @@ class CloudBucketMount(google.protobuf.message.Message):
     requester_pays: builtins.bool
     bucket_endpoint_url: builtins.str
     key_prefix: builtins.str
+    oidc_auth_role_arn: builtins.str
     def __init__(
         self,
         *,
@@ -2247,13 +2261,16 @@ class CloudBucketMount(google.protobuf.message.Message):
         requester_pays: builtins.bool = ...,
         bucket_endpoint_url: builtins.str | None = ...,
         key_prefix: builtins.str | None = ...,
+        oidc_auth_role_arn: builtins.str | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["_bucket_endpoint_url", b"_bucket_endpoint_url", "_key_prefix", b"_key_prefix", "bucket_endpoint_url", b"bucket_endpoint_url", "key_prefix", b"key_prefix"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["_bucket_endpoint_url", b"_bucket_endpoint_url", "_key_prefix", b"_key_prefix", "bucket_endpoint_url", b"bucket_endpoint_url", "bucket_name", b"bucket_name", "bucket_type", b"bucket_type", "credentials_secret_id", b"credentials_secret_id", "key_prefix", b"key_prefix", "mount_path", b"mount_path", "read_only", b"read_only", "requester_pays", b"requester_pays"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_bucket_endpoint_url", b"_bucket_endpoint_url", "_key_prefix", b"_key_prefix", "_oidc_auth_role_arn", b"_oidc_auth_role_arn", "bucket_endpoint_url", b"bucket_endpoint_url", "key_prefix", b"key_prefix", "oidc_auth_role_arn", b"oidc_auth_role_arn"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_bucket_endpoint_url", b"_bucket_endpoint_url", "_key_prefix", b"_key_prefix", "_oidc_auth_role_arn", b"_oidc_auth_role_arn", "bucket_endpoint_url", b"bucket_endpoint_url", "bucket_name", b"bucket_name", "bucket_type", b"bucket_type", "credentials_secret_id", b"credentials_secret_id", "key_prefix", b"key_prefix", "mount_path", b"mount_path", "oidc_auth_role_arn", b"oidc_auth_role_arn", "read_only", b"read_only", "requester_pays", b"requester_pays"]) -> None: ...
     @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_bucket_endpoint_url", b"_bucket_endpoint_url"]) -> typing_extensions.Literal["bucket_endpoint_url"] | None: ...
     @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_key_prefix", b"_key_prefix"]) -> typing_extensions.Literal["key_prefix"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_oidc_auth_role_arn", b"_oidc_auth_role_arn"]) -> typing_extensions.Literal["oidc_auth_role_arn"] | None: ...
 
 global___CloudBucketMount = CloudBucketMount
 
@@ -3766,6 +3783,7 @@ class Function(google.protobuf.message.Message):
     METHOD_DEFINITIONS_FIELD_NUMBER: builtins.int
     METHOD_DEFINITIONS_SET_FIELD_NUMBER: builtins.int
     _EXPERIMENTAL_CUSTOM_SCALING_FIELD_NUMBER: builtins.int
+    CLOUD_PROVIDER_STR_FIELD_NUMBER: builtins.int
     module_name: builtins.str
     function_name: builtins.str
     @property
@@ -3794,6 +3812,7 @@ class Function(google.protobuf.message.Message):
     class_serialized: builtins.bytes
     task_idle_timeout_secs: builtins.int
     cloud_provider: global___CloudProvider.ValueType
+    """Deprecated at some point"""
     warm_pool_size: builtins.int
     web_url: builtins.str
     @property
@@ -3874,6 +3893,8 @@ class Function(google.protobuf.message.Message):
         """Mapping of method names to method definitions, only non-empty for class service functions"""
     method_definitions_set: builtins.bool
     _experimental_custom_scaling: builtins.bool
+    cloud_provider_str: builtins.str
+    """Supersedes cloud_provider"""
     def __init__(
         self,
         *,
@@ -3940,9 +3961,10 @@ class Function(google.protobuf.message.Message):
         method_definitions: collections.abc.Mapping[builtins.str, global___MethodDefinition] | None = ...,
         method_definitions_set: builtins.bool = ...,
         _experimental_custom_scaling: builtins.bool = ...,
+        cloud_provider_str: builtins.str = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["X_experimental_proxy_ip", b"X_experimental_proxy_ip", "_cloud_provider", b"_cloud_provider", "_experimental_proxy_ip", b"_experimental_proxy_ip", "_proxy_id", b"_proxy_id", "_scheduler_placement", b"_scheduler_placement", "checkpoint", b"checkpoint", "class_parameter_info", b"class_parameter_info", "cloud_provider", b"cloud_provider", "proxy_id", b"proxy_id", "pty_info", b"pty_info", "rate_limit", b"rate_limit", "resources", b"resources", "retry_policy", b"retry_policy", "schedule", b"schedule", "scheduler_placement", b"scheduler_placement", "web_url_info", b"web_url_info", "webhook_config", b"webhook_config"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["X_experimental_proxy_ip", b"X_experimental_proxy_ip", "_cloud_provider", b"_cloud_provider", "_experimental_buffer_containers", b"_experimental_buffer_containers", "_experimental_concurrent_cancellations", b"_experimental_concurrent_cancellations", "_experimental_custom_scaling", b"_experimental_custom_scaling", "_experimental_group_size", b"_experimental_group_size", "_experimental_proxy_ip", b"_experimental_proxy_ip", "_experimental_task_templates", b"_experimental_task_templates", "_experimental_task_templates_enabled", b"_experimental_task_templates_enabled", "_proxy_id", b"_proxy_id", "_scheduler_placement", b"_scheduler_placement", "app_name", b"app_name", "batch_linger_ms", b"batch_linger_ms", "batch_max_size", b"batch_max_size", "block_network", b"block_network", "checkpoint", b"checkpoint", "checkpointing_enabled", b"checkpointing_enabled", "class_parameter_info", b"class_parameter_info", "class_serialized", b"class_serialized", "cloud_bucket_mounts", b"cloud_bucket_mounts", "cloud_provider", b"cloud_provider", "concurrency_limit", b"concurrency_limit", "custom_domain_info", b"custom_domain_info", "definition_type", b"definition_type", "function_name", b"function_name", "function_serialized", b"function_serialized", "function_type", b"function_type", "i6pn_enabled", b"i6pn_enabled", "image_id", b"image_id", "is_auto_snapshot", b"is_auto_snapshot", "is_builder_function", b"is_builder_function", "is_checkpointing_function", b"is_checkpointing_function", "is_class", b"is_class", "is_method", b"is_method", "max_concurrent_inputs", b"max_concurrent_inputs", "max_inputs", b"max_inputs", "method_definitions", b"method_definitions", "method_definitions_set", b"method_definitions_set", "module_name", b"module_name", "mount_ids", b"mount_ids", "object_dependencies", b"object_dependencies", "proxy_id", b"proxy_id", "pty_info", b"pty_info", "rate_limit", b"rate_limit", "resources", b"resources", "retry_policy", b"retry_policy", "runtime", b"runtime", "runtime_debug", b"runtime_debug", "runtime_perf_record", b"runtime_perf_record", "s3_mounts", b"s3_mounts", "schedule", b"schedule", "scheduler_placement", b"scheduler_placement", "secret_ids", b"secret_ids", "shared_volume_mounts", b"shared_volume_mounts", "snapshot_debug", b"snapshot_debug", "target_concurrent_inputs", b"target_concurrent_inputs", "task_idle_timeout_secs", b"task_idle_timeout_secs", "timeout_secs", b"timeout_secs", "untrusted", b"untrusted", "use_function_id", b"use_function_id", "use_method_name", b"use_method_name", "volume_mounts", b"volume_mounts", "warm_pool_size", b"warm_pool_size", "web_url", b"web_url", "web_url_info", b"web_url_info", "webhook_config", b"webhook_config", "worker_id", b"worker_id"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["X_experimental_proxy_ip", b"X_experimental_proxy_ip", "_cloud_provider", b"_cloud_provider", "_experimental_buffer_containers", b"_experimental_buffer_containers", "_experimental_concurrent_cancellations", b"_experimental_concurrent_cancellations", "_experimental_custom_scaling", b"_experimental_custom_scaling", "_experimental_group_size", b"_experimental_group_size", "_experimental_proxy_ip", b"_experimental_proxy_ip", "_experimental_task_templates", b"_experimental_task_templates", "_experimental_task_templates_enabled", b"_experimental_task_templates_enabled", "_proxy_id", b"_proxy_id", "_scheduler_placement", b"_scheduler_placement", "app_name", b"app_name", "batch_linger_ms", b"batch_linger_ms", "batch_max_size", b"batch_max_size", "block_network", b"block_network", "checkpoint", b"checkpoint", "checkpointing_enabled", b"checkpointing_enabled", "class_parameter_info", b"class_parameter_info", "class_serialized", b"class_serialized", "cloud_bucket_mounts", b"cloud_bucket_mounts", "cloud_provider", b"cloud_provider", "cloud_provider_str", b"cloud_provider_str", "concurrency_limit", b"concurrency_limit", "custom_domain_info", b"custom_domain_info", "definition_type", b"definition_type", "function_name", b"function_name", "function_serialized", b"function_serialized", "function_type", b"function_type", "i6pn_enabled", b"i6pn_enabled", "image_id", b"image_id", "is_auto_snapshot", b"is_auto_snapshot", "is_builder_function", b"is_builder_function", "is_checkpointing_function", b"is_checkpointing_function", "is_class", b"is_class", "is_method", b"is_method", "max_concurrent_inputs", b"max_concurrent_inputs", "max_inputs", b"max_inputs", "method_definitions", b"method_definitions", "method_definitions_set", b"method_definitions_set", "module_name", b"module_name", "mount_ids", b"mount_ids", "object_dependencies", b"object_dependencies", "proxy_id", b"proxy_id", "pty_info", b"pty_info", "rate_limit", b"rate_limit", "resources", b"resources", "retry_policy", b"retry_policy", "runtime", b"runtime", "runtime_debug", b"runtime_debug", "runtime_perf_record", b"runtime_perf_record", "s3_mounts", b"s3_mounts", "schedule", b"schedule", "scheduler_placement", b"scheduler_placement", "secret_ids", b"secret_ids", "shared_volume_mounts", b"shared_volume_mounts", "snapshot_debug", b"snapshot_debug", "target_concurrent_inputs", b"target_concurrent_inputs", "task_idle_timeout_secs", b"task_idle_timeout_secs", "timeout_secs", b"timeout_secs", "untrusted", b"untrusted", "use_function_id", b"use_function_id", "use_method_name", b"use_method_name", "volume_mounts", b"volume_mounts", "warm_pool_size", b"warm_pool_size", "web_url", b"web_url", "web_url_info", b"web_url_info", "webhook_config", b"webhook_config", "worker_id", b"worker_id"]) -> None: ...
     @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["X_experimental_proxy_ip", b"X_experimental_proxy_ip"]) -> typing_extensions.Literal["_experimental_proxy_ip"] | None: ...
     @typing.overload
@@ -5327,17 +5349,21 @@ class GPUConfig(google.protobuf.message.Message):
     TYPE_FIELD_NUMBER: builtins.int
     COUNT_FIELD_NUMBER: builtins.int
     MEMORY_FIELD_NUMBER: builtins.int
+    GPU_TYPE_FIELD_NUMBER: builtins.int
     type: global___GPUType.ValueType
+    """Deprecated, at some point"""
     count: builtins.int
     memory: builtins.int
+    gpu_type: builtins.str
     def __init__(
         self,
         *,
         type: global___GPUType.ValueType = ...,
         count: builtins.int = ...,
         memory: builtins.int = ...,
+        gpu_type: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["count", b"count", "memory", b"memory", "type", b"type"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["count", b"count", "gpu_type", b"gpu_type", "memory", b"memory", "type", b"type"]) -> None: ...
 
 global___GPUConfig = GPUConfig
 
@@ -6035,6 +6061,198 @@ class NetworkAccess(google.protobuf.message.Message):
 
 global___NetworkAccess = NetworkAccess
 
+class NotebookKernelPublishResultsRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class ExecuteReply(google.protobuf.message.Message):
+        """See kernelshim.py for the differences between this and `ExecuteResult`.
+        https://jupyter-client.readthedocs.io/en/stable/messaging.html#execution-results
+        """
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        STATUS_FIELD_NUMBER: builtins.int
+        EXECUTION_COUNT_FIELD_NUMBER: builtins.int
+        status: builtins.str
+        execution_count: builtins.int
+        def __init__(
+            self,
+            *,
+            status: builtins.str = ...,
+            execution_count: builtins.int = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["execution_count", b"execution_count", "status", b"status"]) -> None: ...
+
+    class CellResult(google.protobuf.message.Message):
+        """IOPub message or reply received from the kernel for a cell."""
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        CELL_ID_FIELD_NUMBER: builtins.int
+        OUTPUT_FIELD_NUMBER: builtins.int
+        CLEAR_OUTPUT_FIELD_NUMBER: builtins.int
+        EXECUTE_REPLY_FIELD_NUMBER: builtins.int
+        cell_id: builtins.str
+        @property
+        def output(self) -> global___NotebookOutput:
+            """Persistent output that is saved in the notebook."""
+        clear_output: builtins.bool
+        """Clear all previous outputs of the cell."""
+        @property
+        def execute_reply(self) -> global___NotebookKernelPublishResultsRequest.ExecuteReply:
+            """Cell has finished executing, return the kernel's execute_reply."""
+        def __init__(
+            self,
+            *,
+            cell_id: builtins.str = ...,
+            output: global___NotebookOutput | None = ...,
+            clear_output: builtins.bool = ...,
+            execute_reply: global___NotebookKernelPublishResultsRequest.ExecuteReply | None = ...,
+        ) -> None: ...
+        def HasField(self, field_name: typing_extensions.Literal["clear_output", b"clear_output", "execute_reply", b"execute_reply", "output", b"output", "result_type", b"result_type"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing_extensions.Literal["cell_id", b"cell_id", "clear_output", b"clear_output", "execute_reply", b"execute_reply", "output", b"output", "result_type", b"result_type"]) -> None: ...
+        def WhichOneof(self, oneof_group: typing_extensions.Literal["result_type", b"result_type"]) -> typing_extensions.Literal["output", "clear_output", "execute_reply"] | None: ...
+
+    NOTEBOOK_ID_FIELD_NUMBER: builtins.int
+    RESULTS_FIELD_NUMBER: builtins.int
+    notebook_id: builtins.str
+    @property
+    def results(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___NotebookKernelPublishResultsRequest.CellResult]: ...
+    def __init__(
+        self,
+        *,
+        notebook_id: builtins.str = ...,
+        results: collections.abc.Iterable[global___NotebookKernelPublishResultsRequest.CellResult] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["notebook_id", b"notebook_id", "results", b"results"]) -> None: ...
+
+global___NotebookKernelPublishResultsRequest = NotebookKernelPublishResultsRequest
+
+class NotebookOutput(google.protobuf.message.Message):
+    """A single output from a notebook. When you execute a cell, it produces an
+    array of these outputs as the code runs.
+
+    https://github.com/jupyter/nbformat/blob/v5.10.4/nbformat/v4/nbformat.v4.schema.json#L301-L309
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class ExecuteResult(google.protobuf.message.Message):
+        """Result of executing a code cell."""
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        EXECUTION_COUNT_FIELD_NUMBER: builtins.int
+        DATA_FIELD_NUMBER: builtins.int
+        METADATA_FIELD_NUMBER: builtins.int
+        execution_count: builtins.int
+        @property
+        def data(self) -> google.protobuf.struct_pb2.Struct:
+            """mimebundle"""
+        @property
+        def metadata(self) -> google.protobuf.struct_pb2.Struct: ...
+        def __init__(
+            self,
+            *,
+            execution_count: builtins.int = ...,
+            data: google.protobuf.struct_pb2.Struct | None = ...,
+            metadata: google.protobuf.struct_pb2.Struct | None = ...,
+        ) -> None: ...
+        def HasField(self, field_name: typing_extensions.Literal["data", b"data", "metadata", b"metadata"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing_extensions.Literal["data", b"data", "execution_count", b"execution_count", "metadata", b"metadata"]) -> None: ...
+
+    class DisplayData(google.protobuf.message.Message):
+        """Data displayed as a result of code cell execution."""
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        DATA_FIELD_NUMBER: builtins.int
+        METADATA_FIELD_NUMBER: builtins.int
+        TRANSIENT_DISPLAY_ID_FIELD_NUMBER: builtins.int
+        @property
+        def data(self) -> google.protobuf.struct_pb2.Struct:
+            """mimebundle"""
+        @property
+        def metadata(self) -> google.protobuf.struct_pb2.Struct: ...
+        transient_display_id: builtins.str
+        """This should not be included in saved notebook."""
+        def __init__(
+            self,
+            *,
+            data: google.protobuf.struct_pb2.Struct | None = ...,
+            metadata: google.protobuf.struct_pb2.Struct | None = ...,
+            transient_display_id: builtins.str | None = ...,
+        ) -> None: ...
+        def HasField(self, field_name: typing_extensions.Literal["_transient_display_id", b"_transient_display_id", "data", b"data", "metadata", b"metadata", "transient_display_id", b"transient_display_id"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing_extensions.Literal["_transient_display_id", b"_transient_display_id", "data", b"data", "metadata", b"metadata", "transient_display_id", b"transient_display_id"]) -> None: ...
+        def WhichOneof(self, oneof_group: typing_extensions.Literal["_transient_display_id", b"_transient_display_id"]) -> typing_extensions.Literal["transient_display_id"] | None: ...
+
+    class Stream(google.protobuf.message.Message):
+        """Stream output from a code cell (stdout / stderr)."""
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        NAME_FIELD_NUMBER: builtins.int
+        TEXT_FIELD_NUMBER: builtins.int
+        name: builtins.str
+        """stdout | stderr"""
+        text: builtins.str
+        """multiline_string"""
+        def __init__(
+            self,
+            *,
+            name: builtins.str = ...,
+            text: builtins.str = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["name", b"name", "text", b"text"]) -> None: ...
+
+    class Error(google.protobuf.message.Message):
+        """Output of an error that occurred during code cell execution."""
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        ENAME_FIELD_NUMBER: builtins.int
+        EVALUE_FIELD_NUMBER: builtins.int
+        TRACEBACK_FIELD_NUMBER: builtins.int
+        ename: builtins.str
+        evalue: builtins.str
+        @property
+        def traceback(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
+        def __init__(
+            self,
+            *,
+            ename: builtins.str = ...,
+            evalue: builtins.str = ...,
+            traceback: collections.abc.Iterable[builtins.str] | None = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["ename", b"ename", "evalue", b"evalue", "traceback", b"traceback"]) -> None: ...
+
+    EXECUTE_RESULT_FIELD_NUMBER: builtins.int
+    DISPLAY_DATA_FIELD_NUMBER: builtins.int
+    STREAM_FIELD_NUMBER: builtins.int
+    ERROR_FIELD_NUMBER: builtins.int
+    @property
+    def execute_result(self) -> global___NotebookOutput.ExecuteResult: ...
+    @property
+    def display_data(self) -> global___NotebookOutput.DisplayData: ...
+    @property
+    def stream(self) -> global___NotebookOutput.Stream: ...
+    @property
+    def error(self) -> global___NotebookOutput.Error: ...
+    def __init__(
+        self,
+        *,
+        execute_result: global___NotebookOutput.ExecuteResult | None = ...,
+        display_data: global___NotebookOutput.DisplayData | None = ...,
+        stream: global___NotebookOutput.Stream | None = ...,
+        error: global___NotebookOutput.Error | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["display_data", b"display_data", "error", b"error", "execute_result", b"execute_result", "output_type", b"output_type", "stream", b"stream"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["display_data", b"display_data", "error", b"error", "execute_result", b"execute_result", "output_type", b"output_type", "stream", b"stream"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["output_type", b"output_type"]) -> typing_extensions.Literal["execute_result", "display_data", "stream", "error"] | None: ...
+
+global___NotebookOutput = NotebookOutput
+
 class Object(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -6194,6 +6412,36 @@ class Proxy(google.protobuf.message.Message):
 
 global___Proxy = Proxy
 
+class ProxyAddIpRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PROXY_ID_FIELD_NUMBER: builtins.int
+    proxy_id: builtins.str
+    def __init__(
+        self,
+        *,
+        proxy_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["proxy_id", b"proxy_id"]) -> None: ...
+
+global___ProxyAddIpRequest = ProxyAddIpRequest
+
+class ProxyAddIpResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PROXY_IP_FIELD_NUMBER: builtins.int
+    @property
+    def proxy_ip(self) -> global___ProxyIp: ...
+    def __init__(
+        self,
+        *,
+        proxy_ip: global___ProxyIp | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["proxy_ip", b"proxy_ip"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["proxy_ip", b"proxy_ip"]) -> None: ...
+
+global___ProxyAddIpResponse = ProxyAddIpResponse
+
 class ProxyCreateRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -6376,6 +6624,20 @@ class ProxyListResponse(google.protobuf.message.Message):
 
 global___ProxyListResponse = ProxyListResponse
 
+class ProxyRemoveIpRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PROXY_IP_FIELD_NUMBER: builtins.int
+    proxy_ip: builtins.str
+    def __init__(
+        self,
+        *,
+        proxy_ip: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["proxy_ip", b"proxy_ip"]) -> None: ...
+
+global___ProxyRemoveIpRequest = ProxyRemoveIpRequest
+
 class QueueClearRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -6846,6 +7108,9 @@ class Sandbox(google.protobuf.message.Message):
     I6PN_ENABLED_FIELD_NUMBER: builtins.int
     NETWORK_ACCESS_FIELD_NUMBER: builtins.int
     PROXY_ID_FIELD_NUMBER: builtins.int
+    ENABLE_SNAPSHOT_FIELD_NUMBER: builtins.int
+    SNAPSHOT_VERSION_FIELD_NUMBER: builtins.int
+    CLOUD_PROVIDER_STR_FIELD_NUMBER: builtins.int
     @property
     def entrypoint_args(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
     @property
@@ -6856,6 +7121,7 @@ class Sandbox(google.protobuf.message.Message):
     @property
     def resources(self) -> global___Resources: ...
     cloud_provider: global___CloudProvider.ValueType
+    """Deprecated at some point"""
     timeout_secs: builtins.int
     workdir: builtins.str
     @property
@@ -6882,6 +7148,16 @@ class Sandbox(google.protobuf.message.Message):
     def network_access(self) -> global___NetworkAccess:
         """Network access configuration beyond simple allow/block."""
     proxy_id: builtins.str
+    enable_snapshot: builtins.bool
+    """Enable snapshotting the sandbox (both memory and filesystem).
+    This doesn't need to be enabled to save the filesystem as an image (i.e. a filesystem-only snapshot).
+    """
+    snapshot_version: builtins.int
+    """Used to pin gVisor version for memory-snapshottable sandboxes.
+    This field is set by the server, not the client.
+    """
+    cloud_provider_str: builtins.str
+    """Supersedes cloud_provider"""
     def __init__(
         self,
         *,
@@ -6906,14 +7182,19 @@ class Sandbox(google.protobuf.message.Message):
         i6pn_enabled: builtins.bool = ...,
         network_access: global___NetworkAccess | None = ...,
         proxy_id: builtins.str | None = ...,
+        enable_snapshot: builtins.bool = ...,
+        snapshot_version: builtins.int | None = ...,
+        cloud_provider_str: builtins.str = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["_proxy_id", b"_proxy_id", "_scheduler_placement", b"_scheduler_placement", "_workdir", b"_workdir", "network_access", b"network_access", "open_ports", b"open_ports", "open_ports_oneof", b"open_ports_oneof", "proxy_id", b"proxy_id", "pty_info", b"pty_info", "resources", b"resources", "scheduler_placement", b"scheduler_placement", "workdir", b"workdir"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["_proxy_id", b"_proxy_id", "_scheduler_placement", b"_scheduler_placement", "_workdir", b"_workdir", "block_network", b"block_network", "cloud_bucket_mounts", b"cloud_bucket_mounts", "cloud_provider", b"cloud_provider", "entrypoint_args", b"entrypoint_args", "i6pn_enabled", b"i6pn_enabled", "image_id", b"image_id", "mount_ids", b"mount_ids", "network_access", b"network_access", "nfs_mounts", b"nfs_mounts", "open_ports", b"open_ports", "open_ports_oneof", b"open_ports_oneof", "proxy_id", b"proxy_id", "pty_info", b"pty_info", "resources", b"resources", "runtime_debug", b"runtime_debug", "s3_mounts", b"s3_mounts", "scheduler_placement", b"scheduler_placement", "secret_ids", b"secret_ids", "timeout_secs", b"timeout_secs", "volume_mounts", b"volume_mounts", "workdir", b"workdir", "worker_id", b"worker_id"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_proxy_id", b"_proxy_id", "_scheduler_placement", b"_scheduler_placement", "_snapshot_version", b"_snapshot_version", "_workdir", b"_workdir", "network_access", b"network_access", "open_ports", b"open_ports", "open_ports_oneof", b"open_ports_oneof", "proxy_id", b"proxy_id", "pty_info", b"pty_info", "resources", b"resources", "scheduler_placement", b"scheduler_placement", "snapshot_version", b"snapshot_version", "workdir", b"workdir"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_proxy_id", b"_proxy_id", "_scheduler_placement", b"_scheduler_placement", "_snapshot_version", b"_snapshot_version", "_workdir", b"_workdir", "block_network", b"block_network", "cloud_bucket_mounts", b"cloud_bucket_mounts", "cloud_provider", b"cloud_provider", "cloud_provider_str", b"cloud_provider_str", "enable_snapshot", b"enable_snapshot", "entrypoint_args", b"entrypoint_args", "i6pn_enabled", b"i6pn_enabled", "image_id", b"image_id", "mount_ids", b"mount_ids", "network_access", b"network_access", "nfs_mounts", b"nfs_mounts", "open_ports", b"open_ports", "open_ports_oneof", b"open_ports_oneof", "proxy_id", b"proxy_id", "pty_info", b"pty_info", "resources", b"resources", "runtime_debug", b"runtime_debug", "s3_mounts", b"s3_mounts", "scheduler_placement", b"scheduler_placement", "secret_ids", b"secret_ids", "snapshot_version", b"snapshot_version", "timeout_secs", b"timeout_secs", "volume_mounts", b"volume_mounts", "workdir", b"workdir", "worker_id", b"worker_id"]) -> None: ...
     @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_proxy_id", b"_proxy_id"]) -> typing_extensions.Literal["proxy_id"] | None: ...
     @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_scheduler_placement", b"_scheduler_placement"]) -> typing_extensions.Literal["scheduler_placement"] | None: ...
     @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_snapshot_version", b"_snapshot_version"]) -> typing_extensions.Literal["snapshot_version"] | None: ...
+    @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_workdir", b"_workdir"]) -> typing_extensions.Literal["workdir"] | None: ...
     @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["open_ports_oneof", b"open_ports_oneof"]) -> typing_extensions.Literal["open_ports"] | None: ...
@@ -6983,13 +7264,23 @@ class SandboxGetTaskIdRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     SANDBOX_ID_FIELD_NUMBER: builtins.int
+    TIMEOUT_FIELD_NUMBER: builtins.int
+    WAIT_UNTIL_READY_FIELD_NUMBER: builtins.int
     sandbox_id: builtins.str
+    timeout: builtins.float
+    """Legacy clients do not provide a timeout. New clients must always provide a timeout."""
+    wait_until_ready: builtins.bool
+    """If true, waits until the container's postStart hook has been run before returning. Useful for detecting init failures."""
     def __init__(
         self,
         *,
         sandbox_id: builtins.str = ...,
+        timeout: builtins.float | None = ...,
+        wait_until_ready: builtins.bool = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["sandbox_id", b"sandbox_id"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_timeout", b"_timeout", "timeout", b"timeout"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_timeout", b"_timeout", "sandbox_id", b"sandbox_id", "timeout", b"timeout", "wait_until_ready", b"wait_until_ready"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_timeout", b"_timeout"]) -> typing_extensions.Literal["timeout"] | None: ...
 
 global___SandboxGetTaskIdRequest = SandboxGetTaskIdRequest
 
@@ -6997,13 +7288,24 @@ class SandboxGetTaskIdResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     TASK_ID_FIELD_NUMBER: builtins.int
+    TASK_RESULT_FIELD_NUMBER: builtins.int
     task_id: builtins.str
+    """This is None if the sandbox was terminated before a task could be scheduled."""
+    @property
+    def task_result(self) -> global___GenericResult:
+        """If the task has already exited, this is the result."""
     def __init__(
         self,
         *,
-        task_id: builtins.str = ...,
+        task_id: builtins.str | None = ...,
+        task_result: global___GenericResult | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["task_id", b"task_id"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_task_id", b"_task_id", "_task_result", b"_task_result", "task_id", b"task_id", "task_result", b"task_result"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_task_id", b"_task_id", "_task_result", b"_task_result", "task_id", b"task_id", "task_result", b"task_result"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_task_id", b"_task_id"]) -> typing_extensions.Literal["task_id"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_task_result", b"_task_result"]) -> typing_extensions.Literal["task_result"] | None: ...
 
 global___SandboxGetTaskIdResponse = SandboxGetTaskIdResponse
 
@@ -7124,6 +7426,62 @@ class SandboxListResponse(google.protobuf.message.Message):
 
 global___SandboxListResponse = SandboxListResponse
 
+class SandboxRestoreRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SNAPSHOT_ID_FIELD_NUMBER: builtins.int
+    snapshot_id: builtins.str
+    def __init__(
+        self,
+        *,
+        snapshot_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["snapshot_id", b"snapshot_id"]) -> None: ...
+
+global___SandboxRestoreRequest = SandboxRestoreRequest
+
+class SandboxRestoreResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SANDBOX_ID_FIELD_NUMBER: builtins.int
+    sandbox_id: builtins.str
+    def __init__(
+        self,
+        *,
+        sandbox_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["sandbox_id", b"sandbox_id"]) -> None: ...
+
+global___SandboxRestoreResponse = SandboxRestoreResponse
+
+class SandboxSnapshotFromIdRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SNAPSHOT_ID_FIELD_NUMBER: builtins.int
+    snapshot_id: builtins.str
+    def __init__(
+        self,
+        *,
+        snapshot_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["snapshot_id", b"snapshot_id"]) -> None: ...
+
+global___SandboxSnapshotFromIdRequest = SandboxSnapshotFromIdRequest
+
+class SandboxSnapshotFromIdResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SNAPSHOT_ID_FIELD_NUMBER: builtins.int
+    snapshot_id: builtins.str
+    def __init__(
+        self,
+        *,
+        snapshot_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["snapshot_id", b"snapshot_id"]) -> None: ...
+
+global___SandboxSnapshotFromIdResponse = SandboxSnapshotFromIdResponse
+
 class SandboxSnapshotFsRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -7165,6 +7523,67 @@ class SandboxSnapshotFsResponse(google.protobuf.message.Message):
 
 global___SandboxSnapshotFsResponse = SandboxSnapshotFsResponse
 
+class SandboxSnapshotRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SANDBOX_ID_FIELD_NUMBER: builtins.int
+    sandbox_id: builtins.str
+    def __init__(
+        self,
+        *,
+        sandbox_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["sandbox_id", b"sandbox_id"]) -> None: ...
+
+global___SandboxSnapshotRequest = SandboxSnapshotRequest
+
+class SandboxSnapshotResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SNAPSHOT_ID_FIELD_NUMBER: builtins.int
+    snapshot_id: builtins.str
+    def __init__(
+        self,
+        *,
+        snapshot_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["snapshot_id", b"snapshot_id"]) -> None: ...
+
+global___SandboxSnapshotResponse = SandboxSnapshotResponse
+
+class SandboxSnapshotWaitRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SNAPSHOT_ID_FIELD_NUMBER: builtins.int
+    TIMEOUT_FIELD_NUMBER: builtins.int
+    snapshot_id: builtins.str
+    timeout: builtins.float
+    def __init__(
+        self,
+        *,
+        snapshot_id: builtins.str = ...,
+        timeout: builtins.float = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["snapshot_id", b"snapshot_id", "timeout", b"timeout"]) -> None: ...
+
+global___SandboxSnapshotWaitRequest = SandboxSnapshotWaitRequest
+
+class SandboxSnapshotWaitResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    RESULT_FIELD_NUMBER: builtins.int
+    @property
+    def result(self) -> global___GenericResult: ...
+    def __init__(
+        self,
+        *,
+        result: global___GenericResult | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["result", b"result"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["result", b"result"]) -> None: ...
+
+global___SandboxSnapshotWaitResponse = SandboxSnapshotWaitResponse
+
 class SandboxStdinWriteRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor