modal 0.62.115__py3-none-any.whl → 0.72.13__py3-none-any.whl

modal/_container_io_manager.pyi

@@ -1,412 +0,0 @@
-import _asyncio
-import asyncio.locks
-import asyncio.queues
-import modal.client
-import modal.running_app
-import modal_proto.api_pb2
-import synchronicity.combined_types
-import typing
-import typing_extensions
-
-class UserException(Exception):
-    ...
-
-class Sentinel:
-    ...
-
-class _ContainerIOManager:
-    cancelled_input_ids: typing.Set[str]
-    task_id: str
-    function_id: str
-    app_id: str
-    function_def: modal_proto.api_pb2.Function
-    checkpoint_id: typing.Union[str, None]
-    calls_completed: int
-    total_user_time: float
-    current_input_id: typing.Union[str, None]
-    current_input_started_at: typing.Union[float, None]
-    _input_concurrency: typing.Union[int, None]
-    _semaphore: typing.Union[asyncio.locks.Semaphore, None]
-    _environment_name: str
-    _waiting_for_checkpoint: bool
-    _heartbeat_loop: typing.Union[_asyncio.Task, None]
-    _is_interactivity_enabled: bool
-    _fetching_inputs: bool
-    _client: modal.client._Client
-    _GENERATOR_STOP_SENTINEL: typing.ClassVar[Sentinel]
-    _singleton: typing.ClassVar[typing.Union[_ContainerIOManager, None]]
-
-    def _init(self, container_args: modal_proto.api_pb2.ContainerArguments, client: modal.client._Client):
-        ...
-
-    @staticmethod
-    def __new__(cls, container_args: modal_proto.api_pb2.ContainerArguments, client: modal.client._Client) -> _ContainerIOManager:
-        ...
-
-    @classmethod
-    def _reset_singleton(cls):
-        ...
-
-    async def _run_heartbeat_loop(self):
-        ...
-
-    async def _heartbeat_handle_cancellations(self) -> bool:
-        ...
-
-    def heartbeats(self) -> typing.AsyncContextManager[None]:
-        ...
-
-    def stop_heartbeat(self):
-        ...
-
-    async def get_app_objects(self) -> modal.running_app.RunningApp:
-        ...
-
-    async def get_serialized_function(self) -> typing.Tuple[typing.Union[typing.Any, None], typing.Callable]:
-        ...
-
-    def serialize(self, obj: typing.Any) -> bytes:
-        ...
-
-    def deserialize(self, data: bytes) -> typing.Any:
-        ...
-
-    def serialize_data_format(self, obj: typing.Any, data_format: int) -> bytes:
-        ...
-
-    def deserialize_data_format(self, data: bytes, data_format: int) -> typing.Any:
-        ...
-
-    def get_data_in(self, function_call_id: str) -> typing.AsyncIterator[typing.Any]:
-        ...
-
-    async def put_data_out(self, function_call_id: str, start_index: int, data_format: int, messages_bytes: typing.List[typing.Any]) -> None:
-        ...
-
-    async def generator_output_task(self, function_call_id: str, data_format: int, message_rx: asyncio.queues.Queue) -> None:
-        ...
-
-    async def _queue_create(self, size: int) -> asyncio.queues.Queue:
-        ...
-
-    async def _queue_put(self, queue: asyncio.queues.Queue, value: typing.Any) -> None:
-        ...
-
-    async def populate_input_blobs(self, item: modal_proto.api_pb2.FunctionInput):
-        ...
-
-    def get_average_call_time(self) -> float:
-        ...
-
-    def get_max_inputs_to_fetch(self):
-        ...
-
-    def _generate_inputs(self) -> typing.AsyncIterator[typing.Tuple[str, str, modal_proto.api_pb2.FunctionInput]]:
-        ...
-
-    def run_inputs_outputs(self, input_concurrency: int = 1) -> typing.AsyncIterator[typing.Tuple[str, str, typing.Any, typing.Any]]:
-        ...
-
-    async def _push_output(self, input_id, started_at: float, data_format=0, **kwargs):
-        ...
-
-    def serialize_exception(self, exc: BaseException) -> typing.Union[bytes, None]:
-        ...
-
-    def serialize_traceback(self, exc: BaseException) -> typing.Tuple[typing.Union[bytes, None], typing.Union[bytes, None]]:
-        ...
-
-    def handle_user_exception(self) -> typing.AsyncContextManager[None]:
-        ...
-
-    def handle_input_exception(self, input_id, started_at: float) -> typing.AsyncContextManager[None]:
-        ...
-
-    async def complete_call(self, started_at):
-        ...
-
-    async def push_output(self, input_id, started_at: float, data: typing.Any, data_format: int) -> None:
-        ...
-
-    async def restore(self) -> None:
-        ...
-
-    async def checkpoint(self) -> None:
-        ...
-
-    async def volume_commit(self, volume_ids: typing.List[str]) -> None:
-        ...
-
-    async def interact(self):
-        ...
-
-    @classmethod
-    def stop_fetching_inputs(cls):
-        ...
-
-
-class ContainerIOManager:
-    cancelled_input_ids: typing.Set[str]
-    task_id: str
-    function_id: str
-    app_id: str
-    function_def: modal_proto.api_pb2.Function
-    checkpoint_id: typing.Union[str, None]
-    calls_completed: int
-    total_user_time: float
-    current_input_id: typing.Union[str, None]
-    current_input_started_at: typing.Union[float, None]
-    _input_concurrency: typing.Union[int, None]
-    _semaphore: typing.Union[asyncio.locks.Semaphore, None]
-    _environment_name: str
-    _waiting_for_checkpoint: bool
-    _heartbeat_loop: typing.Union[_asyncio.Task, None]
-    _is_interactivity_enabled: bool
-    _fetching_inputs: bool
-    _client: modal.client.Client
-    _GENERATOR_STOP_SENTINEL: typing.ClassVar[Sentinel]
-    _singleton: typing.ClassVar[typing.Union[ContainerIOManager, None]]
-
-    def __init__(self, /, *args, **kwargs):
-        ...
-
-    def _init(self, container_args: modal_proto.api_pb2.ContainerArguments, client: modal.client.Client):
-        ...
-
-    @classmethod
-    def _reset_singleton(cls):
-        ...
-
-    class ___run_heartbeat_loop_spec(typing_extensions.Protocol):
-        def __call__(self):
-            ...
-
-        async def aio(self, *args, **kwargs):
-            ...
-
-    _run_heartbeat_loop: ___run_heartbeat_loop_spec
-
-    class ___heartbeat_handle_cancellations_spec(typing_extensions.Protocol):
-        def __call__(self) -> bool:
-            ...
-
-        async def aio(self, *args, **kwargs) -> bool:
-            ...
-
-    _heartbeat_handle_cancellations: ___heartbeat_handle_cancellations_spec
-
-    class __heartbeats_spec(typing_extensions.Protocol):
-        def __call__(self) -> synchronicity.combined_types.AsyncAndBlockingContextManager[None]:
-            ...
-
-        def aio(self) -> typing.AsyncContextManager[None]:
-            ...
-
-    heartbeats: __heartbeats_spec
-
-    def stop_heartbeat(self):
-        ...
-
-    class __get_app_objects_spec(typing_extensions.Protocol):
-        def __call__(self) -> modal.running_app.RunningApp:
-            ...
-
-        async def aio(self, *args, **kwargs) -> modal.running_app.RunningApp:
-            ...
-
-    get_app_objects: __get_app_objects_spec
-
-    class __get_serialized_function_spec(typing_extensions.Protocol):
-        def __call__(self) -> typing.Tuple[typing.Union[typing.Any, None], typing.Callable]:
-            ...
-
-        async def aio(self, *args, **kwargs) -> typing.Tuple[typing.Union[typing.Any, None], typing.Callable]:
-            ...
-
-    get_serialized_function: __get_serialized_function_spec
-
-    def serialize(self, obj: typing.Any) -> bytes:
-        ...
-
-    def deserialize(self, data: bytes) -> typing.Any:
-        ...
-
-    def serialize_data_format(self, obj: typing.Any, data_format: int) -> bytes:
-        ...
-
-    def deserialize_data_format(self, data: bytes, data_format: int) -> typing.Any:
-        ...
-
-    class __get_data_in_spec(typing_extensions.Protocol):
-        def __call__(self, function_call_id: str) -> typing.Iterator[typing.Any]:
-            ...
-
-        def aio(self, function_call_id: str) -> typing.AsyncIterator[typing.Any]:
-            ...
-
-    get_data_in: __get_data_in_spec
-
-    class __put_data_out_spec(typing_extensions.Protocol):
-        def __call__(self, function_call_id: str, start_index: int, data_format: int, messages_bytes: typing.List[typing.Any]) -> None:
-            ...
-
-        async def aio(self, *args, **kwargs) -> None:
-            ...
-
-    put_data_out: __put_data_out_spec
-
-    class __generator_output_task_spec(typing_extensions.Protocol):
-        def __call__(self, function_call_id: str, data_format: int, message_rx: asyncio.queues.Queue) -> None:
-            ...
-
-        async def aio(self, *args, **kwargs) -> None:
-            ...
-
-    generator_output_task: __generator_output_task_spec
-
-    class ___queue_create_spec(typing_extensions.Protocol):
-        def __call__(self, size: int) -> asyncio.queues.Queue:
-            ...
-
-        async def aio(self, *args, **kwargs) -> asyncio.queues.Queue:
-            ...
-
-    _queue_create: ___queue_create_spec
-
-    class ___queue_put_spec(typing_extensions.Protocol):
-        def __call__(self, queue: asyncio.queues.Queue, value: typing.Any) -> None:
-            ...
-
-        async def aio(self, *args, **kwargs) -> None:
-            ...
-
-    _queue_put: ___queue_put_spec
-
-    class __populate_input_blobs_spec(typing_extensions.Protocol):
-        def __call__(self, item: modal_proto.api_pb2.FunctionInput):
-            ...
-
-        async def aio(self, *args, **kwargs):
-            ...
-
-    populate_input_blobs: __populate_input_blobs_spec
-
-    def get_average_call_time(self) -> float:
-        ...
-
-    def get_max_inputs_to_fetch(self):
-        ...
-
-    class ___generate_inputs_spec(typing_extensions.Protocol):
-        def __call__(self) -> typing.Iterator[typing.Tuple[str, str, modal_proto.api_pb2.FunctionInput]]:
-            ...
-
-        def aio(self) -> typing.AsyncIterator[typing.Tuple[str, str, modal_proto.api_pb2.FunctionInput]]:
-            ...
-
-    _generate_inputs: ___generate_inputs_spec
-
-    class __run_inputs_outputs_spec(typing_extensions.Protocol):
-        def __call__(self, input_concurrency: int = 1) -> typing.Iterator[typing.Tuple[str, str, typing.Any, typing.Any]]:
-            ...
-
-        def aio(self, input_concurrency: int = 1) -> typing.AsyncIterator[typing.Tuple[str, str, typing.Any, typing.Any]]:
-            ...
-
-    run_inputs_outputs: __run_inputs_outputs_spec
-
-    class ___push_output_spec(typing_extensions.Protocol):
-        def __call__(self, input_id, started_at: float, data_format=0, **kwargs):
-            ...
-
-        async def aio(self, *args, **kwargs):
-            ...
-
-    _push_output: ___push_output_spec
-
-    def serialize_exception(self, exc: BaseException) -> typing.Union[bytes, None]:
-        ...
-
-    def serialize_traceback(self, exc: BaseException) -> typing.Tuple[typing.Union[bytes, None], typing.Union[bytes, None]]:
-        ...
-
-    class __handle_user_exception_spec(typing_extensions.Protocol):
-        def __call__(self) -> synchronicity.combined_types.AsyncAndBlockingContextManager[None]:
-            ...
-
-        def aio(self) -> typing.AsyncContextManager[None]:
-            ...
-
-    handle_user_exception: __handle_user_exception_spec
-
-    class __handle_input_exception_spec(typing_extensions.Protocol):
-        def __call__(self, input_id, started_at: float) -> synchronicity.combined_types.AsyncAndBlockingContextManager[None]:
-            ...
-
-        def aio(self, input_id, started_at: float) -> typing.AsyncContextManager[None]:
-            ...
-
-    handle_input_exception: __handle_input_exception_spec
-
-    class __complete_call_spec(typing_extensions.Protocol):
-        def __call__(self, started_at):
-            ...
-
-        async def aio(self, *args, **kwargs):
-            ...
-
-    complete_call: __complete_call_spec
-
-    class __push_output_spec(typing_extensions.Protocol):
-        def __call__(self, input_id, started_at: float, data: typing.Any, data_format: int) -> None:
-            ...
-
-        async def aio(self, *args, **kwargs) -> None:
-            ...
-
-    push_output: __push_output_spec
-
-    class __restore_spec(typing_extensions.Protocol):
-        def __call__(self) -> None:
-            ...
-
-        async def aio(self, *args, **kwargs) -> None:
-            ...
-
-    restore: __restore_spec
-
-    class __checkpoint_spec(typing_extensions.Protocol):
-        def __call__(self) -> None:
-            ...
-
-        async def aio(self, *args, **kwargs) -> None:
-            ...
-
-    checkpoint: __checkpoint_spec
-
-    class __volume_commit_spec(typing_extensions.Protocol):
-        def __call__(self, volume_ids: typing.List[str]) -> None:
-            ...
-
-        async def aio(self, *args, **kwargs) -> None:
-            ...
-
-    volume_commit: __volume_commit_spec
-
-    class __interact_spec(typing_extensions.Protocol):
-        def __call__(self):
-            ...
-
-        async def aio(self, *args, **kwargs):
-            ...
-
-    interact: __interact_spec
-
-    @classmethod
-    def stop_fetching_inputs(cls):
-        ...
-
-
-MAX_OUTPUT_BATCH_SIZE: int
-
-RTT_S: float

modal/_sandbox_shell.py

@@ -1,49 +0,0 @@
-# Copyright Modal Labs 2024
-import asyncio
-
-from ._utils.shell_utils import connect_to_terminal, write_to_fd
-from .sandbox import _Sandbox
-
-
-async def connect_to_sandbox(sandbox: _Sandbox):
-    """
-    Connects the current terminal to the Sandbox process.
-    """
-
-    async def _handle_input(data: bytes, _):
-        sandbox.stdin.write(data)
-        await sandbox.stdin.drain.aio()  # type: ignore
-
-    async def _stream_to_stdout(on_connect: asyncio.Event) -> int:
-        return await _stream_logs_to_stdout(sandbox, on_connect)
-
-    await connect_to_terminal(_handle_input, _stream_to_stdout, pty=True)
-
-
-async def _stream_logs_to_stdout(sandbox: _Sandbox, on_connect: asyncio.Event) -> int:
-    """
-    Streams sandbox output logs to the current terminal's stdout.
-
-    The on_connect event will be set when the client connects to the running process,
-    and the event loop will be released.
-    """
-
-    # we are connected if we received at least one message from the server
-    # (the server will send an empty message when the process spawns)
-    connected = False
-
-    # Since the sandbox process will run in a PTY, stderr will go to the PTY
-    # slave. The PTY shell will then relay data from PTY master to stdout.
-    # Therefore, we only need to stream from/to stdout here.
-    async for message in sandbox.stdout:
-        await write_to_fd(1, message.encode("utf-8"))
-
-        if not connected:
-            connected = True
-            on_connect.set()
-            # give up the event loop
-            await asyncio.sleep(0)
-
-    # Right now we don't propagate the exit_status to the TaskLogs, so setting
-    # exit status to 0.
-    return 0

modal/app_utils.py

@@ -1,20 +0,0 @@
-# Copyright Modal Labs 2024
-# Note: this is a temporary module until we've (1) deleted the current app.py (3) renamed stub.py to app.py
-from typing import List, Optional
-
-from modal_proto import api_pb2
-
-from ._utils.async_utils import synchronize_api
-from .client import _Client
-from .object import _get_environment_name
-
-
-async def _list_apps(env: Optional[str] = None, client: Optional[_Client] = None) -> List[api_pb2.AppStats]:
-    """List apps in a given Modal environment."""
-    if client is None:
-        client = await _Client.from_env()
-    resp: api_pb2.AppListResponse = await client.stub.AppList(api_pb2.AppListRequest(environment_name=_get_environment_name(env)))
-    return list(resp.apps)
-
-
-list_apps = synchronize_api(_list_apps)

modal/app_utils.pyi

@@ -1,17 +0,0 @@
-import modal.client
-import modal_proto.api_pb2
-import typing
-import typing_extensions
-
-async def _list_apps(env: typing.Union[str, None] = None, client: typing.Union[modal.client._Client, None] = None) -> typing.List[modal_proto.api_pb2.AppStats]:
-    ...
-
-
-class __list_apps_spec(typing_extensions.Protocol):
-    def __call__(self, env: typing.Union[str, None] = None, client: typing.Union[modal.client.Client, None] = None) -> typing.List[modal_proto.api_pb2.AppStats]:
-        ...
-
-    async def aio(self, *args, **kwargs) -> typing.List[modal_proto.api_pb2.AppStats]:
-        ...
-
-list_apps: __list_apps_spec

modal/execution_context.pyi

@@ -1,37 +0,0 @@
-import _contextvars
-import typing
-import typing_extensions
-
-def is_local() -> bool:
-    ...
-
-
-async def _interact() -> None:
-    ...
-
-
-class __interact_spec(typing_extensions.Protocol):
-    def __call__(self) -> None:
-        ...
-
-    async def aio(self, *args, **kwargs) -> None:
-        ...
-
-interact: __interact_spec
-
-
-def current_input_id() -> typing.Union[str, None]:
-    ...
-
-
-def current_function_call_id() -> typing.Union[str, None]:
-    ...
-
-
-def _set_current_context_ids(input_id: str, function_call_id: str) -> typing.Callable[[], None]:
-    ...
-
-
-_current_input_id: _contextvars.ContextVar
-
-_current_function_call_id: _contextvars.ContextVar

modal/shared_volume.py

@@ -1,23 +0,0 @@
-# Copyright Modal Labs 2023
-
-from ._utils.async_utils import synchronize_api
-from .exception import deprecation_error
-
-
-class _SharedVolume:
-    def __init__(self, *args, **kwargs):
-        """`SharedVolume` is deprecated. We recommend `Volume` (https://modal.com/docs/guide/volumes) instead."""
-        deprecation_error((2023, 7, 5), _SharedVolume.__init__.__doc__)
-
-    @staticmethod
-    def new(*args, **kwargs):
-        """`SharedVolume` is deprecated. We recommend `Volume` (https://modal.com/docs/guide/volumes) instead."""
-        deprecation_error((2023, 7, 5), _SharedVolume.new.__doc__)
-
-    @staticmethod
-    def persisted(*args, **kwargs):
-        """`SharedVolume` is deprecated. We recommend `Volume` (https://modal.com/docs/guide/volumes) instead."""
-        deprecation_error((2023, 7, 5), _SharedVolume.persisted.__doc__)
-
-
-SharedVolume = synchronize_api(_SharedVolume)

modal/shared_volume.pyi

@@ -1,24 +0,0 @@
-class _SharedVolume:
-    def __init__(self, *args, **kwargs):
-        ...
-
-    @staticmethod
-    def new(*args, **kwargs):
-        ...
-
-    @staticmethod
-    def persisted(*args, **kwargs):
-        ...
-
-
-class SharedVolume:
-    def __init__(self, *args, **kwargs):
-        ...
-
-    @staticmethod
-    def new(*args, **kwargs):
-        ...
-
-    @staticmethod
-    def persisted(*args, **kwargs):
-        ...