modal 0.72.37__py3-none-any.whl → 0.72.39__py3-none-any.whl

modal/functions.pyi

@@ -134,6 +134,8 @@ ReturnType = typing.TypeVar("ReturnType", covariant=True)
 
 OriginalReturnType = typing.TypeVar("OriginalReturnType", covariant=True)
 
+SUPERSELF = typing.TypeVar("SUPERSELF", covariant=True)
+
 class _Function(typing.Generic[P, ReturnType, OriginalReturnType], modal._object._Object):
     _info: typing.Optional[modal._utils.function_utils.FunctionInfo]
     _serve_mounts: frozenset[modal.mount._Mount]
@@ -144,7 +146,7 @@ class _Function(typing.Generic[P, ReturnType, OriginalReturnType], modal._object
     _is_method: bool
     _spec: typing.Optional[_FunctionSpec]
     _tag: str
-    _raw_f: typing.Optional[typing.Callable[..., typing.Any]]
+    _raw_f: typing.Optional[collections.abc.Callable[..., typing.Any]]
     _build_args: dict
     _is_generator: typing.Optional[bool]
     _cluster_size: typing.Optional[int]
@@ -255,10 +257,10 @@ class _Function(typing.Generic[P, ReturnType, OriginalReturnType], modal._object
     def local(self, *args: P.args, **kwargs: P.kwargs) -> OriginalReturnType: ...
     async def _experimental_spawn(self, *args: P.args, **kwargs: P.kwargs) -> _FunctionCall[ReturnType]: ...
     async def spawn(self, *args: P.args, **kwargs: P.kwargs) -> _FunctionCall[ReturnType]: ...
-    def get_raw_f(self) -> typing.Callable[..., typing.Any]: ...
+    def get_raw_f(self) -> collections.abc.Callable[..., typing.Any]: ...
     async def get_current_stats(self) -> FunctionStats: ...
 
-    class __map_spec(typing_extensions.Protocol):
+    class __map_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(
             self, *input_iterators, kwargs={}, order_outputs: bool = True, return_exceptions: bool = False
         ) -> modal._utils.async_utils.AsyncOrSyncIterable: ...
@@ -270,9 +272,9 @@ class _Function(typing.Generic[P, ReturnType, OriginalReturnType], modal._object
             return_exceptions: bool = False,
         ) -> typing.AsyncGenerator[typing.Any, None]: ...
 
-    map: __map_spec
+    map: __map_spec[typing_extensions.Self]
 
-    class __starmap_spec(typing_extensions.Protocol):
+    class __starmap_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(
             self,
             input_iterator: typing.Iterable[typing.Sequence[typing.Any]],
@@ -290,13 +292,13 @@ class _Function(typing.Generic[P, ReturnType, OriginalReturnType], modal._object
             return_exceptions: bool = False,
         ) -> typing.AsyncIterable[typing.Any]: ...
 
-    starmap: __starmap_spec
+    starmap: __starmap_spec[typing_extensions.Self]
 
-    class __for_each_spec(typing_extensions.Protocol):
+    class __for_each_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, *input_iterators, kwargs={}, ignore_exceptions: bool = False): ...
         async def aio(self, *input_iterators, kwargs={}, ignore_exceptions: bool = False): ...
 
-    for_each: __for_each_spec
+    for_each: __for_each_spec[typing_extensions.Self]
 
 ReturnType_INNER = typing.TypeVar("ReturnType_INNER", covariant=True)
 
@@ -312,7 +314,7 @@ class Function(typing.Generic[P, ReturnType, OriginalReturnType], modal.object.O
     _is_method: bool
     _spec: typing.Optional[_FunctionSpec]
     _tag: str
-    _raw_f: typing.Optional[typing.Callable[..., typing.Any]]
+    _raw_f: typing.Optional[collections.abc.Callable[..., typing.Any]]
     _build_args: dict
     _is_generator: typing.Optional[bool]
     _cluster_size: typing.Optional[int]
@@ -376,11 +378,11 @@ class Function(typing.Generic[P, ReturnType, OriginalReturnType], modal.object.O
         kwargs: dict[str, typing.Any],
     ) -> Function: ...
 
-    class __keep_warm_spec(typing_extensions.Protocol):
+    class __keep_warm_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, warm_pool_size: int) -> None: ...
         async def aio(self, warm_pool_size: int) -> None: ...
 
-    keep_warm: __keep_warm_spec
+    keep_warm: __keep_warm_spec[typing_extensions.Self]
 
     @classmethod
     def from_name(
@@ -429,7 +431,7 @@ class Function(typing.Generic[P, ReturnType, OriginalReturnType], modal.object.O
     @property
     def cluster_size(self) -> int: ...
 
-    class ___map_spec(typing_extensions.Protocol):
+    class ___map_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(
             self, input_queue: modal.parallel_map.SynchronizedQueue, order_outputs: bool, return_exceptions: bool
         ) -> typing.Generator[typing.Any, None, None]: ...
@@ -437,70 +439,70 @@ class Function(typing.Generic[P, ReturnType, OriginalReturnType], modal.object.O
             self, input_queue: modal.parallel_map.SynchronizedQueue, order_outputs: bool, return_exceptions: bool
         ) -> collections.abc.AsyncGenerator[typing.Any, None]: ...
 
-    _map: ___map_spec
+    _map: ___map_spec[typing_extensions.Self]
 
-    class ___call_function_spec(typing_extensions.Protocol[ReturnType_INNER]):
+    class ___call_function_spec(typing_extensions.Protocol[ReturnType_INNER, SUPERSELF]):
         def __call__(self, args, kwargs) -> ReturnType_INNER: ...
         async def aio(self, args, kwargs) -> ReturnType_INNER: ...
 
-    _call_function: ___call_function_spec[ReturnType]
+    _call_function: ___call_function_spec[ReturnType, typing_extensions.Self]
 
-    class ___call_function_nowait_spec(typing_extensions.Protocol):
+    class ___call_function_nowait_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, args, kwargs, function_call_invocation_type: int) -> _Invocation: ...
         async def aio(self, args, kwargs, function_call_invocation_type: int) -> _Invocation: ...
 
-    _call_function_nowait: ___call_function_nowait_spec
+    _call_function_nowait: ___call_function_nowait_spec[typing_extensions.Self]
 
-    class ___call_generator_spec(typing_extensions.Protocol):
+    class ___call_generator_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, args, kwargs): ...
         def aio(self, args, kwargs): ...
 
-    _call_generator: ___call_generator_spec
+    _call_generator: ___call_generator_spec[typing_extensions.Self]
 
-    class ___call_generator_nowait_spec(typing_extensions.Protocol):
+    class ___call_generator_nowait_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, args, kwargs): ...
         async def aio(self, args, kwargs): ...
 
-    _call_generator_nowait: ___call_generator_nowait_spec
+    _call_generator_nowait: ___call_generator_nowait_spec[typing_extensions.Self]
 
-    class __remote_spec(typing_extensions.Protocol[P_INNER, ReturnType_INNER]):
+    class __remote_spec(typing_extensions.Protocol[ReturnType_INNER, P_INNER, SUPERSELF]):
         def __call__(self, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> ReturnType_INNER: ...
         async def aio(self, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> ReturnType_INNER: ...
 
-    remote: __remote_spec[P, ReturnType]
+    remote: __remote_spec[ReturnType, P, typing_extensions.Self]
 
-    class __remote_gen_spec(typing_extensions.Protocol):
+    class __remote_gen_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, *args, **kwargs) -> typing.Generator[typing.Any, None, None]: ...
         def aio(self, *args, **kwargs) -> collections.abc.AsyncGenerator[typing.Any, None]: ...
 
-    remote_gen: __remote_gen_spec
+    remote_gen: __remote_gen_spec[typing_extensions.Self]
 
     def _is_local(self): ...
     def _get_info(self) -> modal._utils.function_utils.FunctionInfo: ...
     def _get_obj(self) -> typing.Optional[modal.cls.Obj]: ...
     def local(self, *args: P.args, **kwargs: P.kwargs) -> OriginalReturnType: ...
 
-    class ___experimental_spawn_spec(typing_extensions.Protocol[P_INNER, ReturnType_INNER]):
+    class ___experimental_spawn_spec(typing_extensions.Protocol[ReturnType_INNER, P_INNER, SUPERSELF]):
         def __call__(self, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> FunctionCall[ReturnType_INNER]: ...
         async def aio(self, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> FunctionCall[ReturnType_INNER]: ...
 
-    _experimental_spawn: ___experimental_spawn_spec[P, ReturnType]
+    _experimental_spawn: ___experimental_spawn_spec[ReturnType, P, typing_extensions.Self]
 
-    class __spawn_spec(typing_extensions.Protocol[P_INNER, ReturnType_INNER]):
+    class __spawn_spec(typing_extensions.Protocol[ReturnType_INNER, P_INNER, SUPERSELF]):
         def __call__(self, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> FunctionCall[ReturnType_INNER]: ...
         async def aio(self, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> FunctionCall[ReturnType_INNER]: ...
 
-    spawn: __spawn_spec[P, ReturnType]
+    spawn: __spawn_spec[ReturnType, P, typing_extensions.Self]
 
-    def get_raw_f(self) -> typing.Callable[..., typing.Any]: ...
+    def get_raw_f(self) -> collections.abc.Callable[..., typing.Any]: ...
 
-    class __get_current_stats_spec(typing_extensions.Protocol):
+    class __get_current_stats_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self) -> FunctionStats: ...
         async def aio(self) -> FunctionStats: ...
 
-    get_current_stats: __get_current_stats_spec
+    get_current_stats: __get_current_stats_spec[typing_extensions.Self]
 
-    class __map_spec(typing_extensions.Protocol):
+    class __map_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(
             self, *input_iterators, kwargs={}, order_outputs: bool = True, return_exceptions: bool = False
         ) -> modal._utils.async_utils.AsyncOrSyncIterable: ...
@@ -512,9 +514,9 @@ class Function(typing.Generic[P, ReturnType, OriginalReturnType], modal.object.O
             return_exceptions: bool = False,
         ) -> typing.AsyncGenerator[typing.Any, None]: ...
 
-    map: __map_spec
+    map: __map_spec[typing_extensions.Self]
 
-    class __starmap_spec(typing_extensions.Protocol):
+    class __starmap_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(
             self,
             input_iterator: typing.Iterable[typing.Sequence[typing.Any]],
@@ -532,13 +534,13 @@ class Function(typing.Generic[P, ReturnType, OriginalReturnType], modal.object.O
             return_exceptions: bool = False,
         ) -> typing.AsyncIterable[typing.Any]: ...
 
-    starmap: __starmap_spec
+    starmap: __starmap_spec[typing_extensions.Self]
 
-    class __for_each_spec(typing_extensions.Protocol):
+    class __for_each_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, *input_iterators, kwargs={}, ignore_exceptions: bool = False): ...
         async def aio(self, *input_iterators, kwargs={}, ignore_exceptions: bool = False): ...
 
-    for_each: __for_each_spec
+    for_each: __for_each_spec[typing_extensions.Self]
 
 class _FunctionCall(typing.Generic[ReturnType], modal._object._Object):
     _is_generator: bool
@@ -559,29 +561,29 @@ class FunctionCall(typing.Generic[ReturnType], modal.object.Object):
     def __init__(self, *args, **kwargs): ...
     def _invocation(self): ...
 
-    class __get_spec(typing_extensions.Protocol[ReturnType_INNER]):
+    class __get_spec(typing_extensions.Protocol[ReturnType_INNER, SUPERSELF]):
         def __call__(self, timeout: typing.Optional[float] = None) -> ReturnType_INNER: ...
         async def aio(self, timeout: typing.Optional[float] = None) -> ReturnType_INNER: ...
 
-    get: __get_spec[ReturnType]
+    get: __get_spec[ReturnType, typing_extensions.Self]
 
-    class __get_gen_spec(typing_extensions.Protocol):
+    class __get_gen_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self) -> typing.Generator[typing.Any, None, None]: ...
         def aio(self) -> collections.abc.AsyncGenerator[typing.Any, None]: ...
 
-    get_gen: __get_gen_spec
+    get_gen: __get_gen_spec[typing_extensions.Self]
 
-    class __get_call_graph_spec(typing_extensions.Protocol):
+    class __get_call_graph_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self) -> list[modal.call_graph.InputInfo]: ...
         async def aio(self) -> list[modal.call_graph.InputInfo]: ...
 
-    get_call_graph: __get_call_graph_spec
+    get_call_graph: __get_call_graph_spec[typing_extensions.Self]
 
-    class __cancel_spec(typing_extensions.Protocol):
+    class __cancel_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, terminate_containers: bool = False): ...
         async def aio(self, terminate_containers: bool = False): ...
 
-    cancel: __cancel_spec
+    cancel: __cancel_spec[typing_extensions.Self]
 
     class __from_id_spec(typing_extensions.Protocol):
         def __call__(

modal/image.pyi

@@ -53,11 +53,11 @@ def _get_image_builder_version(
 ) -> typing.Literal["2023.12", "2024.04", "2024.10"]: ...
 def _create_context_mount(
     docker_commands: collections.abc.Sequence[str],
-    ignore_fn: typing.Callable[[pathlib.Path], bool],
+    ignore_fn: collections.abc.Callable[[pathlib.Path], bool],
     context_dir: pathlib.Path,
 ) -> typing.Optional[modal.mount._Mount]: ...
 def _create_context_mount_function(
-    ignore: typing.Union[collections.abc.Sequence[str], typing.Callable[[pathlib.Path], bool]],
+    ignore: typing.Union[collections.abc.Sequence[str], collections.abc.Callable[[pathlib.Path], bool]],
     dockerfile_cmds: list[str] = [],
     dockerfile_path: typing.Optional[pathlib.Path] = None,
     context_mount: typing.Optional[modal.mount._Mount] = None,
@@ -98,14 +98,16 @@ class _Image(modal._object._Object):
         *,
         base_images: typing.Optional[dict[str, _Image]] = None,
         dockerfile_function: typing.Optional[
-            typing.Callable[[typing.Literal["2023.12", "2024.04", "2024.10"]], DockerfileSpec]
+            collections.abc.Callable[[typing.Literal["2023.12", "2024.04", "2024.10"]], DockerfileSpec]
         ] = None,
         secrets: typing.Optional[collections.abc.Sequence[modal.secret._Secret]] = None,
         gpu_config: typing.Optional[modal_proto.api_pb2.GPUConfig] = None,
         build_function: typing.Optional[modal.functions._Function] = None,
         build_function_input: typing.Optional[modal_proto.api_pb2.FunctionInput] = None,
         image_registry_config: typing.Optional[_ImageRegistryConfig] = None,
-        context_mount_function: typing.Optional[typing.Callable[[], typing.Optional[modal.mount._Mount]]] = None,
+        context_mount_function: typing.Optional[
+            collections.abc.Callable[[], typing.Optional[modal.mount._Mount]]
+        ] = None,
         force_build: bool = False,
         _namespace: int = 1,
         _do_assert_no_mount_layers: bool = True,
@@ -120,7 +122,7 @@ class _Image(modal._object._Object):
         remote_path: str,
         *,
         copy: bool = False,
-        ignore: typing.Union[collections.abc.Sequence[str], typing.Callable[[pathlib.Path], bool]] = [],
+        ignore: typing.Union[collections.abc.Sequence[str], collections.abc.Callable[[pathlib.Path], bool]] = [],
     ) -> _Image: ...
     def copy_local_file(
         self, local_path: typing.Union[str, pathlib.Path], remote_path: typing.Union[str, pathlib.Path] = "./"
@@ -130,14 +132,14 @@ class _Image(modal._object._Object):
         *module_names: str,
         copy: bool = False,
         ignore: typing.Union[
-            collections.abc.Sequence[str], typing.Callable[[pathlib.Path], bool]
+            collections.abc.Sequence[str], collections.abc.Callable[[pathlib.Path], bool]
         ] = modal.file_pattern_matcher.NON_PYTHON_FILES,
     ) -> _Image: ...
     def copy_local_dir(
         self,
         local_path: typing.Union[str, pathlib.Path],
         remote_path: typing.Union[str, pathlib.Path] = ".",
-        ignore: typing.Union[collections.abc.Sequence[str], typing.Callable[[pathlib.Path], bool]] = [],
+        ignore: typing.Union[collections.abc.Sequence[str], collections.abc.Callable[[pathlib.Path], bool]] = [],
     ) -> _Image: ...
     @staticmethod
     async def from_id(image_id: str, client: typing.Optional[modal.client._Client] = None) -> _Image: ...
@@ -216,7 +218,7 @@ class _Image(modal._object._Object):
         context_mount: typing.Optional[modal.mount._Mount] = None,
         force_build: bool = False,
         ignore: typing.Union[
-            collections.abc.Sequence[str], typing.Callable[[pathlib.Path], bool]
+            collections.abc.Sequence[str], collections.abc.Callable[[pathlib.Path], bool]
         ] = modal.image.AUTO_DOCKERIGNORE,
     ) -> _Image: ...
     def entrypoint(self, entrypoint_commands: list[str]) -> _Image: ...
@@ -304,7 +306,7 @@ class _Image(modal._object._Object):
         gpu: typing.Union[None, bool, str, modal.gpu._GPUConfig] = None,
         add_python: typing.Optional[str] = None,
         ignore: typing.Union[
-            collections.abc.Sequence[str], typing.Callable[[pathlib.Path], bool]
+            collections.abc.Sequence[str], collections.abc.Callable[[pathlib.Path], bool]
         ] = modal.image.AUTO_DOCKERIGNORE,
     ) -> _Image: ...
     @staticmethod
@@ -318,7 +320,7 @@ class _Image(modal._object._Object):
     ) -> _Image: ...
     def run_function(
         self,
-        raw_f: typing.Callable[..., typing.Any],
+        raw_f: collections.abc.Callable[..., typing.Any],
         secrets: collections.abc.Sequence[modal.secret._Secret] = (),
         gpu: typing.Union[
             None, bool, str, modal.gpu._GPUConfig, list[typing.Union[None, bool, str, modal.gpu._GPUConfig]]
@@ -345,6 +347,8 @@ class _Image(modal._object._Object):
     def imports(self): ...
     def _logs(self) -> typing.AsyncGenerator[str, None]: ...
 
+SUPERSELF = typing.TypeVar("SUPERSELF", covariant=True)
+
 class Image(modal.object.Object):
     force_build: bool
     inside_exceptions: list[Exception]
@@ -365,14 +369,16 @@ class Image(modal.object.Object):
         *,
         base_images: typing.Optional[dict[str, Image]] = None,
         dockerfile_function: typing.Optional[
-            typing.Callable[[typing.Literal["2023.12", "2024.04", "2024.10"]], DockerfileSpec]
+            collections.abc.Callable[[typing.Literal["2023.12", "2024.04", "2024.10"]], DockerfileSpec]
         ] = None,
         secrets: typing.Optional[collections.abc.Sequence[modal.secret.Secret]] = None,
         gpu_config: typing.Optional[modal_proto.api_pb2.GPUConfig] = None,
         build_function: typing.Optional[modal.functions.Function] = None,
         build_function_input: typing.Optional[modal_proto.api_pb2.FunctionInput] = None,
         image_registry_config: typing.Optional[_ImageRegistryConfig] = None,
-        context_mount_function: typing.Optional[typing.Callable[[], typing.Optional[modal.mount.Mount]]] = None,
+        context_mount_function: typing.Optional[
+            collections.abc.Callable[[], typing.Optional[modal.mount.Mount]]
+        ] = None,
         force_build: bool = False,
         _namespace: int = 1,
         _do_assert_no_mount_layers: bool = True,
@@ -387,7 +393,7 @@ class Image(modal.object.Object):
         remote_path: str,
         *,
         copy: bool = False,
-        ignore: typing.Union[collections.abc.Sequence[str], typing.Callable[[pathlib.Path], bool]] = [],
+        ignore: typing.Union[collections.abc.Sequence[str], collections.abc.Callable[[pathlib.Path], bool]] = [],
     ) -> Image: ...
     def copy_local_file(
         self, local_path: typing.Union[str, pathlib.Path], remote_path: typing.Union[str, pathlib.Path] = "./"
@@ -397,14 +403,14 @@ class Image(modal.object.Object):
         *module_names: str,
         copy: bool = False,
         ignore: typing.Union[
-            collections.abc.Sequence[str], typing.Callable[[pathlib.Path], bool]
+            collections.abc.Sequence[str], collections.abc.Callable[[pathlib.Path], bool]
         ] = modal.file_pattern_matcher.NON_PYTHON_FILES,
     ) -> Image: ...
     def copy_local_dir(
         self,
         local_path: typing.Union[str, pathlib.Path],
         remote_path: typing.Union[str, pathlib.Path] = ".",
-        ignore: typing.Union[collections.abc.Sequence[str], typing.Callable[[pathlib.Path], bool]] = [],
+        ignore: typing.Union[collections.abc.Sequence[str], collections.abc.Callable[[pathlib.Path], bool]] = [],
     ) -> Image: ...
 
     class __from_id_spec(typing_extensions.Protocol):
@@ -488,7 +494,7 @@ class Image(modal.object.Object):
         context_mount: typing.Optional[modal.mount.Mount] = None,
         force_build: bool = False,
         ignore: typing.Union[
-            collections.abc.Sequence[str], typing.Callable[[pathlib.Path], bool]
+            collections.abc.Sequence[str], collections.abc.Callable[[pathlib.Path], bool]
         ] = modal.image.AUTO_DOCKERIGNORE,
     ) -> Image: ...
     def entrypoint(self, entrypoint_commands: list[str]) -> Image: ...
@@ -576,7 +582,7 @@ class Image(modal.object.Object):
         gpu: typing.Union[None, bool, str, modal.gpu._GPUConfig] = None,
         add_python: typing.Optional[str] = None,
         ignore: typing.Union[
-            collections.abc.Sequence[str], typing.Callable[[pathlib.Path], bool]
+            collections.abc.Sequence[str], collections.abc.Callable[[pathlib.Path], bool]
         ] = modal.image.AUTO_DOCKERIGNORE,
     ) -> Image: ...
     @staticmethod
@@ -590,7 +596,7 @@ class Image(modal.object.Object):
     ) -> Image: ...
     def run_function(
         self,
-        raw_f: typing.Callable[..., typing.Any],
+        raw_f: collections.abc.Callable[..., typing.Any],
         secrets: collections.abc.Sequence[modal.secret.Secret] = (),
         gpu: typing.Union[
             None, bool, str, modal.gpu._GPUConfig, list[typing.Union[None, bool, str, modal.gpu._GPUConfig]]
@@ -616,10 +622,10 @@ class Image(modal.object.Object):
     def workdir(self, path: typing.Union[str, pathlib.PurePosixPath]) -> Image: ...
     def imports(self): ...
 
-    class ___logs_spec(typing_extensions.Protocol):
+    class ___logs_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self) -> typing.Generator[str, None, None]: ...
         def aio(self) -> typing.AsyncGenerator[str, None]: ...
 
-    _logs: ___logs_spec
+    _logs: ___logs_spec[typing_extensions.Self]
 
 SUPPORTED_PYTHON_SERIES: dict[typing.Literal["2023.12", "2024.04", "2024.10"], list[str]]

modal/io_streams.pyi

@@ -50,6 +50,8 @@ class _StreamWriter:
 
 T_INNER = typing.TypeVar("T_INNER", covariant=True)
 
+SUPERSELF = typing.TypeVar("SUPERSELF", covariant=True)
+
 class StreamReader(typing.Generic[T]):
     _stream: typing.Optional[collections.abc.AsyncGenerator[typing.Optional[bytes], None]]
 
@@ -66,25 +68,25 @@ class StreamReader(typing.Generic[T]):
     @property
     def file_descriptor(self) -> int: ...
 
-    class __read_spec(typing_extensions.Protocol[T_INNER]):
+    class __read_spec(typing_extensions.Protocol[T_INNER, SUPERSELF]):
         def __call__(self) -> T_INNER: ...
         async def aio(self) -> T_INNER: ...
 
-    read: __read_spec[T]
+    read: __read_spec[T, typing_extensions.Self]
 
-    class ___consume_container_process_stream_spec(typing_extensions.Protocol):
+    class ___consume_container_process_stream_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self): ...
         async def aio(self): ...
 
-    _consume_container_process_stream: ___consume_container_process_stream_spec
+    _consume_container_process_stream: ___consume_container_process_stream_spec[typing_extensions.Self]
 
-    class ___stream_container_process_spec(typing_extensions.Protocol):
+    class ___stream_container_process_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self) -> typing.Generator[tuple[typing.Optional[bytes], str], None, None]: ...
         def aio(self) -> collections.abc.AsyncGenerator[tuple[typing.Optional[bytes], str], None]: ...
 
-    _stream_container_process: ___stream_container_process_spec
+    _stream_container_process: ___stream_container_process_spec[typing_extensions.Self]
 
-    class ___get_logs_spec(typing_extensions.Protocol):
+    class ___get_logs_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(
             self, skip_empty_messages: bool = True
         ) -> typing.Generator[typing.Optional[bytes], None, None]: ...
@@ -92,13 +94,13 @@ class StreamReader(typing.Generic[T]):
             self, skip_empty_messages: bool = True
         ) -> collections.abc.AsyncGenerator[typing.Optional[bytes], None]: ...
 
-    _get_logs: ___get_logs_spec
+    _get_logs: ___get_logs_spec[typing_extensions.Self]
 
-    class ___get_logs_by_line_spec(typing_extensions.Protocol):
+    class ___get_logs_by_line_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self) -> typing.Generator[typing.Optional[bytes], None, None]: ...
         def aio(self) -> collections.abc.AsyncGenerator[typing.Optional[bytes], None]: ...
 
-    _get_logs_by_line: ___get_logs_by_line_spec
+    _get_logs_by_line: ___get_logs_by_line_spec[typing_extensions.Self]
 
     def __iter__(self) -> typing.Iterator[T]: ...
     def __aiter__(self) -> collections.abc.AsyncIterator[T]: ...
@@ -115,8 +117,8 @@ class StreamWriter:
     def write(self, data: typing.Union[bytes, bytearray, memoryview, str]) -> None: ...
     def write_eof(self) -> None: ...
 
-    class __drain_spec(typing_extensions.Protocol):
+    class __drain_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self) -> None: ...
         async def aio(self) -> None: ...
 
-    drain: __drain_spec
+    drain: __drain_spec[typing_extensions.Self]