modal 1.0.4.dev12__py3-none-any.whl → 1.0.5__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (67) hide show
  1. modal/_clustered_functions.pyi +13 -3
  2. modal/_functions.py +84 -46
  3. modal/_partial_function.py +1 -1
  4. modal/_runtime/container_io_manager.pyi +222 -40
  5. modal/_runtime/execution_context.pyi +60 -6
  6. modal/_serialization.py +25 -2
  7. modal/_tunnel.pyi +380 -12
  8. modal/_utils/async_utils.py +1 -1
  9. modal/_utils/blob_utils.py +56 -19
  10. modal/_utils/function_utils.py +33 -7
  11. modal/_utils/grpc_utils.py +11 -4
  12. modal/app.py +5 -5
  13. modal/app.pyi +658 -48
  14. modal/cli/run.py +2 -1
  15. modal/client.pyi +224 -36
  16. modal/cloud_bucket_mount.pyi +192 -4
  17. modal/cls.py +7 -7
  18. modal/cls.pyi +442 -35
  19. modal/container_process.pyi +103 -14
  20. modal/dict.py +4 -4
  21. modal/dict.pyi +453 -51
  22. modal/environments.pyi +41 -9
  23. modal/exception.py +6 -2
  24. modal/experimental/__init__.py +90 -0
  25. modal/experimental/ipython.py +11 -7
  26. modal/file_io.pyi +236 -45
  27. modal/functions.pyi +573 -65
  28. modal/gpu.py +1 -1
  29. modal/image.py +1 -1
  30. modal/image.pyi +1256 -74
  31. modal/io_streams.py +8 -4
  32. modal/io_streams.pyi +348 -38
  33. modal/mount.pyi +261 -31
  34. modal/network_file_system.py +3 -3
  35. modal/network_file_system.pyi +307 -26
  36. modal/object.pyi +48 -9
  37. modal/parallel_map.py +93 -19
  38. modal/parallel_map.pyi +160 -15
  39. modal/partial_function.pyi +255 -14
  40. modal/proxy.py +1 -1
  41. modal/proxy.pyi +28 -3
  42. modal/queue.py +4 -4
  43. modal/queue.pyi +447 -30
  44. modal/runner.pyi +160 -22
  45. modal/sandbox.py +8 -7
  46. modal/sandbox.pyi +310 -50
  47. modal/schedule.py +1 -1
  48. modal/secret.py +2 -2
  49. modal/secret.pyi +164 -15
  50. modal/snapshot.pyi +25 -4
  51. modal/token_flow.pyi +28 -8
  52. modal/volume.py +41 -4
  53. modal/volume.pyi +693 -59
  54. {modal-1.0.4.dev12.dist-info → modal-1.0.5.dist-info}/METADATA +3 -3
  55. {modal-1.0.4.dev12.dist-info → modal-1.0.5.dist-info}/RECORD +67 -67
  56. modal_proto/api.proto +56 -0
  57. modal_proto/api_grpc.py +48 -0
  58. modal_proto/api_pb2.py +874 -780
  59. modal_proto/api_pb2.pyi +194 -8
  60. modal_proto/api_pb2_grpc.py +100 -0
  61. modal_proto/api_pb2_grpc.pyi +32 -0
  62. modal_proto/modal_api_grpc.py +3 -0
  63. modal_version/__init__.py +1 -1
  64. {modal-1.0.4.dev12.dist-info → modal-1.0.5.dist-info}/WHEEL +0 -0
  65. {modal-1.0.4.dev12.dist-info → modal-1.0.5.dist-info}/entry_points.txt +0 -0
  66. {modal-1.0.4.dev12.dist-info → modal-1.0.5.dist-info}/licenses/LICENSE +0 -0
  67. {modal-1.0.4.dev12.dist-info → modal-1.0.5.dist-info}/top_level.txt +0 -0
@@ -9,6 +9,11 @@ class PartialFunction(
9
9
  modal._partial_function.P, modal._partial_function.ReturnType, modal._partial_function.OriginalReturnType
10
10
  ]
11
11
  ):
12
+ """Object produced by a decorator in the `modal` namespace
13
+
14
+ The object will eventually by consumed by an App decorator.
15
+ """
16
+
12
17
  raw_f: typing.Optional[collections.abc.Callable[modal._partial_function.P, modal._partial_function.ReturnType]]
13
18
  user_cls: typing.Optional[type]
14
19
  flags: modal._partial_function._PartialFunctionFlags
@@ -27,11 +32,20 @@ class PartialFunction(
27
32
  self,
28
33
  flags: modal._partial_function._PartialFunctionFlags,
29
34
  params: modal._partial_function._PartialFunctionParams,
30
- ) -> typing_extensions.Self: ...
31
- def validate_flag_composition(self) -> None: ...
35
+ ) -> typing_extensions.Self:
36
+ """Implement decorator composition by combining the flags and params."""
37
+ ...
38
+
39
+ def validate_flag_composition(self) -> None:
40
+ """Validate decorator composition based on PartialFunctionFlags."""
41
+ ...
42
+
32
43
  def validate_obj_compatibility(
33
44
  self, decorator_name: str, require_sync: bool = False, require_nullary: bool = False
34
- ) -> None: ...
45
+ ) -> None:
46
+ """Enforce compatibility with the wrapped object; called from individual decorator functions."""
47
+ ...
48
+
35
49
  def _get_raw_f(self) -> collections.abc.Callable[modal._partial_function.P, modal._partial_function.ReturnType]: ...
36
50
  def _is_web_endpoint(self) -> bool: ...
37
51
  def __get__(
@@ -43,7 +57,22 @@ class PartialFunction(
43
57
 
44
58
  def method(
45
59
  _warn_parentheses_missing=None, *, is_generator: typing.Optional[bool] = None
46
- ) -> modal._partial_function._MethodDecoratorType: ...
60
+ ) -> modal._partial_function._MethodDecoratorType:
61
+ """Decorator for methods that should be transformed into a Modal Function registered against this class's App.
62
+
63
+ **Usage:**
64
+
65
+ ```python
66
+ @app.cls(cpu=8)
67
+ class MyCls:
68
+
69
+ @modal.method()
70
+ def f(self):
71
+ ...
72
+ ```
73
+ """
74
+ ...
75
+
47
76
  def web_endpoint(
48
77
  _warn_parentheses_missing=None,
49
78
  *,
@@ -62,7 +91,25 @@ def web_endpoint(
62
91
  ]
63
92
  ],
64
93
  PartialFunction[modal._partial_function.P, modal._partial_function.ReturnType, modal._partial_function.ReturnType],
65
- ]: ...
94
+ ]:
95
+ """Register a basic web endpoint with this application.
96
+
97
+ DEPRECATED: This decorator has been renamed to `@modal.fastapi_endpoint`.
98
+
99
+ This is the simple way to create a web endpoint on Modal. The function
100
+ behaves as a [FastAPI](https://fastapi.tiangolo.com/) handler and should
101
+ return a response object to the caller.
102
+
103
+ Endpoints created with `@modal.web_endpoint` are meant to be simple, single
104
+ request handlers and automatically have
105
+ [CORS](https://developer.mozilla.org/en-US/docs/Web/HTTP/CORS) enabled.
106
+ For more flexibility, use `@modal.asgi_app`.
107
+
108
+ To learn how to use Modal with popular web frameworks, see the
109
+ [guide on web endpoints](https://modal.com/docs/guide/webhooks).
110
+ """
111
+ ...
112
+
66
113
  def fastapi_endpoint(
67
114
  _warn_parentheses_missing=None,
68
115
  *,
@@ -81,7 +128,24 @@ def fastapi_endpoint(
81
128
  ]
82
129
  ],
83
130
  PartialFunction[modal._partial_function.P, modal._partial_function.ReturnType, modal._partial_function.ReturnType],
84
- ]: ...
131
+ ]:
132
+ """Convert a function into a basic web endpoint by wrapping it with a FastAPI App.
133
+
134
+ Modal will internally use [FastAPI](https://fastapi.tiangolo.com/) to expose a
135
+ simple, single request handler. If you are defining your own `FastAPI` application
136
+ (e.g. if you want to define multiple routes), use `@modal.asgi_app` instead.
137
+
138
+ The endpoint created with this decorator will automatically have
139
+ [CORS](https://developer.mozilla.org/en-US/docs/Web/HTTP/CORS) enabled
140
+ and can leverage many of FastAPI's features.
141
+
142
+ For more information on using Modal with popular web frameworks, see our
143
+ [guide on web endpoints](https://modal.com/docs/guide/webhooks).
144
+
145
+ *Added in v0.73.82*: This function replaces the deprecated `@web_endpoint` decorator.
146
+ """
147
+ ...
148
+
85
149
  def asgi_app(
86
150
  _warn_parentheses_missing=None,
87
151
  *,
@@ -97,7 +161,30 @@ def asgi_app(
97
161
  ]
98
162
  ],
99
163
  PartialFunction,
100
- ]: ...
164
+ ]:
165
+ """Decorator for registering an ASGI app with a Modal function.
166
+
167
+ Asynchronous Server Gateway Interface (ASGI) is a standard for Python
168
+ synchronous and asynchronous apps, supported by all popular Python web
169
+ libraries. This is an advanced decorator that gives full flexibility in
170
+ defining one or more web endpoints on Modal.
171
+
172
+ **Usage:**
173
+
174
+ ```python
175
+ from typing import Callable
176
+
177
+ @app.function()
178
+ @modal.asgi_app()
179
+ def create_asgi() -> Callable:
180
+ ...
181
+ ```
182
+
183
+ To learn how to use Modal with popular web frameworks, see the
184
+ [guide on web endpoints](https://modal.com/docs/guide/webhooks).
185
+ """
186
+ ...
187
+
101
188
  def wsgi_app(
102
189
  _warn_parentheses_missing=None,
103
190
  *,
@@ -113,7 +200,30 @@ def wsgi_app(
113
200
  ]
114
201
  ],
115
202
  PartialFunction,
116
- ]: ...
203
+ ]:
204
+ """Decorator for registering a WSGI app with a Modal function.
205
+
206
+ Web Server Gateway Interface (WSGI) is a standard for synchronous Python web apps.
207
+ It has been [succeeded by the ASGI interface](https://asgi.readthedocs.io/en/latest/introduction.html#wsgi-compatibility)
208
+ which is compatible with ASGI and supports additional functionality such as web sockets.
209
+ Modal supports ASGI via [`asgi_app`](https://modal.com/docs/reference/modal.asgi_app).
210
+
211
+ **Usage:**
212
+
213
+ ```python
214
+ from typing import Callable
215
+
216
+ @app.function()
217
+ @modal.wsgi_app()
218
+ def create_wsgi() -> Callable:
219
+ ...
220
+ ```
221
+
222
+ To learn how to use this decorator with popular web frameworks, see the
223
+ [guide on web endpoints](https://modal.com/docs/guide/webhooks).
224
+ """
225
+ ...
226
+
117
227
  def web_server(
118
228
  port: int,
119
229
  *,
@@ -130,20 +240,84 @@ def web_server(
130
240
  ]
131
241
  ],
132
242
  PartialFunction,
133
- ]: ...
243
+ ]:
244
+ """Decorator that registers an HTTP web server inside the container.
245
+
246
+ This is similar to `@asgi_app` and `@wsgi_app`, but it allows you to expose a full HTTP server
247
+ listening on a container port. This is useful for servers written in other languages like Rust,
248
+ as well as integrating with non-ASGI frameworks like aiohttp and Tornado.
249
+
250
+ **Usage:**
251
+
252
+ ```python
253
+ import subprocess
254
+
255
+ @app.function()
256
+ @modal.web_server(8000)
257
+ def my_file_server():
258
+ subprocess.Popen("python -m http.server -d / 8000", shell=True)
259
+ ```
260
+
261
+ The above example starts a simple file server, displaying the contents of the root directory.
262
+ Here, requests to the web endpoint will go to external port 8000 on the container. The
263
+ `http.server` module is included with Python, but you could run anything here.
264
+
265
+ Internally, the web server is transparently converted into a web endpoint by Modal, so it has
266
+ the same serverless autoscaling behavior as other web endpoints.
267
+
268
+ For more info, see the [guide on web endpoints](https://modal.com/docs/guide/webhooks).
269
+ """
270
+ ...
271
+
134
272
  def build(
135
273
  _warn_parentheses_missing=None, *, force: bool = False, timeout: int = 86400
136
274
  ) -> collections.abc.Callable[
137
275
  [typing.Union[PartialFunction, collections.abc.Callable[[typing.Any], typing.Any]]], PartialFunction
138
- ]: ...
276
+ ]:
277
+ """mdmd:hidden
278
+ Decorator for methods that execute at _build time_ to create a new Image layer.
279
+
280
+ **Deprecated**: This function is deprecated. We recommend using `modal.Volume`
281
+ to store large assets (such as model weights) instead of writing them to the
282
+ Image during the build process. For other use cases, you can replace this
283
+ decorator with the `Image.run_function` method.
284
+
285
+ **Usage**
286
+
287
+ ```python notest
288
+ @app.cls(gpu="A10G")
289
+ class AlpacaLoRAModel:
290
+ @build()
291
+ def download_models(self):
292
+ model = LlamaForCausalLM.from_pretrained(
293
+ base_model,
294
+ )
295
+ PeftModel.from_pretrained(model, lora_weights)
296
+ LlamaTokenizer.from_pretrained(base_model)
297
+ ```
298
+ """
299
+ ...
300
+
139
301
  def enter(
140
302
  _warn_parentheses_missing=None, *, snap: bool = False
141
303
  ) -> collections.abc.Callable[
142
304
  [typing.Union[PartialFunction, collections.abc.Callable[[typing.Any], typing.Any]]], PartialFunction
143
- ]: ...
305
+ ]:
306
+ """Decorator for methods which should be executed when a new container is started.
307
+
308
+ See the [lifeycle function guide](https://modal.com/docs/guide/lifecycle-functions#enter) for more information.
309
+ """
310
+ ...
311
+
144
312
  def exit(
145
313
  _warn_parentheses_missing=None,
146
- ) -> collections.abc.Callable[[collections.abc.Callable[[typing.Any], typing.Any]], PartialFunction]: ...
314
+ ) -> collections.abc.Callable[[collections.abc.Callable[[typing.Any], typing.Any]], PartialFunction]:
315
+ """Decorator for methods which should be executed when a container is about to exit.
316
+
317
+ See the [lifeycle function guide](https://modal.com/docs/guide/lifecycle-functions#exit) for more information.
318
+ """
319
+ ...
320
+
147
321
  def batched(
148
322
  _warn_parentheses_missing=None, *, max_batch_size: int, wait_ms: int
149
323
  ) -> collections.abc.Callable[
@@ -156,7 +330,33 @@ def batched(
156
330
  ]
157
331
  ],
158
332
  PartialFunction[modal._partial_function.P, modal._partial_function.ReturnType, modal._partial_function.ReturnType],
159
- ]: ...
333
+ ]:
334
+ """Decorator for functions or class methods that should be batched.
335
+
336
+ **Usage**
337
+
338
+ ```python
339
+ # Stack the decorator under `@app.function()` to enable dynamic batching
340
+ @app.function()
341
+ @modal.batched(max_batch_size=4, wait_ms=1000)
342
+ async def batched_multiply(xs: list[int], ys: list[int]) -> list[int]:
343
+ return [x * y for x, y in zip(xs, ys)]
344
+
345
+ # call batched_multiply with individual inputs
346
+ # batched_multiply.remote.aio(2, 100)
347
+
348
+ # With `@app.cls()`, apply the decorator to a method (this may change in the future)
349
+ @app.cls()
350
+ class BatchedClass:
351
+ @modal.batched(max_batch_size=4, wait_ms=1000)
352
+ def batched_multiply(self, xs: list[int], ys: list[int]) -> list[int]:
353
+ return [x * y for x, y in zip(xs, ys)]
354
+ ```
355
+
356
+ See the [dynamic batching guide](https://modal.com/docs/guide/dynamic-batching) for more information.
357
+ """
358
+ ...
359
+
160
360
  def concurrent(
161
361
  _warn_parentheses_missing=None, *, max_inputs: int, target_inputs: typing.Optional[int] = None
162
362
  ) -> collections.abc.Callable[
@@ -169,4 +369,45 @@ def concurrent(
169
369
  ]
170
370
  ],
171
371
  PartialFunction[modal._partial_function.P, modal._partial_function.ReturnType, modal._partial_function.ReturnType],
172
- ]: ...
372
+ ]:
373
+ """Decorator that allows individual containers to handle multiple inputs concurrently.
374
+
375
+ The concurrency mechanism depends on whether the function is async or not:
376
+ - Async functions will run inputs on a single thread as asyncio tasks.
377
+ - Synchronous functions will use multi-threading. The code must be thread-safe.
378
+
379
+ Input concurrency will be most useful for workflows that are IO-bound
380
+ (e.g., making network requests) or when running an inference server that supports
381
+ dynamic batching.
382
+
383
+ When `target_inputs` is set, Modal's autoscaler will try to provision resources
384
+ such that each container is running that many inputs concurrently, rather than
385
+ autoscaling based on `max_inputs`. Containers may burst up to up to `max_inputs`
386
+ if resources are insufficient to remain at the target concurrency, e.g. when the
387
+ arrival rate of inputs increases. This can trade-off a small increase in average
388
+ latency to avoid larger tail latencies from input queuing.
389
+
390
+ **Examples:**
391
+ ```python
392
+ # Stack the decorator under `@app.function()` to enable input concurrency
393
+ @app.function()
394
+ @modal.concurrent(max_inputs=100)
395
+ async def f(data):
396
+ # Async function; will be scheduled as asyncio task
397
+ ...
398
+
399
+ # With `@app.cls()`, apply the decorator at the class level, not on individual methods
400
+ @app.cls()
401
+ @modal.concurrent(max_inputs=100, target_inputs=80)
402
+ class C:
403
+ @modal.method()
404
+ def f(self, data):
405
+ # Sync function; must be thread-safe
406
+ ...
407
+
408
+ ```
409
+
410
+ *Added in v0.73.148:* This decorator replaces the `allow_concurrent_inputs` parameter
411
+ in `@app.function()` and `@app.cls()`.
412
+ """
413
+ ...
modal/proxy.py CHANGED
@@ -12,7 +12,7 @@ class _Proxy(_Object, type_prefix="pr"):
12
12
  """Proxy objects give your Modal containers a static outbound IP address.
13
13
 
14
14
  This can be used for connecting to a remote address with network whitelist, for example
15
- a database. See [the guide](/docs/guide/proxy-ips) for more information.
15
+ a database. See [the guide](https://modal.com/docs/guide/proxy-ips) for more information.
16
16
  """
17
17
 
18
18
  @staticmethod
modal/proxy.pyi CHANGED
@@ -3,10 +3,35 @@ import modal.object
3
3
  import typing
4
4
 
5
5
  class _Proxy(modal._object._Object):
6
+ """Proxy objects give your Modal containers a static outbound IP address.
7
+
8
+ This can be used for connecting to a remote address with network whitelist, for example
9
+ a database. See [the guide](https://modal.com/docs/guide/proxy-ips) for more information.
10
+ """
6
11
  @staticmethod
7
- def from_name(name: str, *, environment_name: typing.Optional[str] = None) -> _Proxy: ...
12
+ def from_name(name: str, *, environment_name: typing.Optional[str] = None) -> _Proxy:
13
+ """Reference a Proxy by its name.
14
+
15
+ In contrast to most other Modal objects, new Proxy objects must be
16
+ provisioned via the Dashboard and cannot be created on the fly from code.
17
+ """
18
+ ...
8
19
 
9
20
  class Proxy(modal.object.Object):
10
- def __init__(self, *args, **kwargs): ...
21
+ """Proxy objects give your Modal containers a static outbound IP address.
22
+
23
+ This can be used for connecting to a remote address with network whitelist, for example
24
+ a database. See [the guide](https://modal.com/docs/guide/proxy-ips) for more information.
25
+ """
26
+ def __init__(self, *args, **kwargs):
27
+ """mdmd:hidden"""
28
+ ...
29
+
11
30
  @staticmethod
12
- def from_name(name: str, *, environment_name: typing.Optional[str] = None) -> Proxy: ...
31
+ def from_name(name: str, *, environment_name: typing.Optional[str] = None) -> Proxy:
32
+ """Reference a Proxy by its name.
33
+
34
+ In contrast to most other Modal objects, new Proxy objects must be
35
+ provisioned via the Dashboard and cannot be created on the fly from code.
36
+ """
37
+ ...
modal/queue.py CHANGED
@@ -64,7 +64,7 @@ class _Queue(_Object, type_prefix="qu"):
64
64
  assert queue.get() == 42
65
65
  ```
66
66
 
67
- For more examples, see the [guide](/docs/guide/dicts-and-queues#modal-queues).
67
+ For more examples, see the [guide](https://modal.com/docs/guide/dicts-and-queues#modal-queues).
68
68
 
69
69
  **Queue partitions (beta)**
70
70
 
@@ -154,9 +154,9 @@ class _Queue(_Object, type_prefix="qu"):
154
154
  ) -> "_Queue":
155
155
  """Reference a named Queue, creating if necessary.
156
156
 
157
- In contrast to `modal.Queue.lookup`, this is a lazy method
158
- the defers hydrating the local object with metadata from
159
- Modal servers until the first time it is actually used.
157
+ This is a lazy method the defers hydrating the local
158
+ object with metadata from Modal servers until the first
159
+ time it is actually used.
160
160
 
161
161
  ```python
162
162
  q = modal.Queue.from_name("my-queue", create_if_missing=True)