baml-py 0.205.0__cp38-abi3-macosx_10_12_x86_64.whl → 0.206.1__cp38-abi3-macosx_10_12_x86_64.whl

baml_py/__init__.py

@@ -3,6 +3,7 @@
 
 # Re-export the pyo3 API
 from .baml_py import (
+    AbortController,
     BamlRuntime,
     FunctionResult,
     FunctionResultStream,
@@ -24,6 +25,7 @@ from .stream import BamlStream, BamlSyncStream
 from .ctx_manager import CtxManager as BamlCtxManager
 
 __all__ = [
+    "AbortController",
     "BamlRuntime",
     "ClientRegistry",
     "BamlStream",

baml_py/baml_py.pyi

@@ -2,6 +2,8 @@ from __future__ import annotations
 from typing_extensions import Literal
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
+TickReason = Literal["Unknown"]
+
 def get_version() -> str:
     """Get the version of the BAML Python client."""
     ...
@@ -24,6 +26,35 @@ def set_log_max_chunk_length(length: int) -> None:
     """Set the maximum log chunk length for the BAML Python client."""
     ...
 
+class AbortController:
+    """Controller for cancelling BAML operations."""
+
+    def __init__(self, timeout_ms: Optional[int] = None) -> None: 
+        """
+        Creates a new abort controller with an optional timeout in milliseconds.
+        Once aborted, the AbortController will forever remain in an an aborted state.
+        The timeout will only start AFTER the object is passed to a BAML function.
+
+        Args:
+            timeout_ms: The timeout in milliseconds. If not provided, AbortController will not timeout.
+        """
+        ...
+    
+
+    def abort(self) -> None: 
+        """
+        Immediately abort all operations.
+        """
+        ...
+
+    @property
+    def aborted(self) -> bool: 
+        """
+        Check the state of this controller.
+        Once aborted, the AbortController will forever remain in an an aborted state.
+        """
+        ...
+
 class FunctionResult:
     """The result of a BAML function call.
 
@@ -109,7 +140,7 @@ class BamlPdfPy:
     @staticmethod
     def from_url(url: str) -> BamlPdfPy: ...
     @staticmethod
-    def from_base64(media_type: str, base64: str) -> BamlPdfPy: ...
+    def from_base64(base64: str) -> BamlPdfPy: ...
     def is_url(self) -> bool: ...
     def is_base64(self) -> bool: ...
     def as_url(self) -> str: ...
@@ -172,6 +203,7 @@ class BamlRuntime:
         cr: Optional[ClientRegistry],
         collectors: List[Collector],
         env_vars: Dict[str, str],
+        on_tick: Optional[Callable[[], None]],
     ) -> FunctionResultStream: ...
     def stream_function_sync(
         self,
@@ -183,6 +215,7 @@ class BamlRuntime:
         cr: Optional[ClientRegistry],
         collectors: List[Collector],
         env_vars: Dict[str, str],
+        on_tick: Optional[Callable[[], None]],
     ) -> SyncFunctionResultStream: ...
     def create_context_manager(self) -> RuntimeContextManager: ...
     def flush(self) -> None: ...

baml_py/errors.py

@@ -1,4 +1,5 @@
 from .baml_py import (
+    BamlAbortError,
     BamlError,
     BamlClientError,
     BamlInvalidArgumentError,
@@ -11,6 +12,7 @@ from .internal_monkeypatch import (
 
 
 __all__ = [
+    "BamlAbortError",
     "BamlError",
     "BamlClientError",
     "BamlClientHttpError",

{baml_py-0.205.0.dist-info → baml_py-0.206.1.dist-info}/METADATA

@@ -1,9 +1,10 @@
 Metadata-Version: 2.4
 Name: baml-py
-Version: 0.205.0
+Version: 0.206.1
 License-File: LICENSE
 Summary: BAML python bindings (pyproject.toml)
 Author-email: Boundary <contact@boundaryml.com>
+License-Expression: Apache-2.0
 Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
 
 BAML python bindings (readme.md)

baml_py-0.206.1.dist-info/RECORD

@@ -0,0 +1,16 @@
+baml_py-0.206.1.dist-info/METADATA,sha256=S0h1SPO7xQ2IYpOwDAY__wZI6nITNL8bvSnZVuzu9jw,304
+baml_py-0.206.1.dist-info/WHEEL,sha256=qq8T5RNN4nLzVDHWhr9aubnbXTyL36gsBivIyKvr8Qc,104
+baml_py-0.206.1.dist-info/entry_points.txt,sha256=9Uu_VcUjoI2qQMjVb0PRjEgI6pQ55WqBbzNparAPJyA,54
+baml_py-0.206.1.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+baml_py/__init__.py,sha256=_tXffmw8Be3rMDR8Zfivr1Gp-wL7baJ-guSoIRaOsBY,943
+baml_py/baml_py.abi3.so,sha256=-WeG3cDc8jdikC7fLLoIX9M6QZX2-Bg9QEgswJ6Aok8,47641576
+baml_py/baml_py.pyi,sha256=ha8JgEp2Sg45s_8Eu7OjWq29FMct4GqFQ7zOVs4Qt0Y,16794
+baml_py/ctx_manager.py,sha256=JwWLvxFT05BnYiguzB-k-g47r7mft74L3mYt8e3qtus,6081
+baml_py/errors.py,sha256=cpD5OlBuKIX8IbImwM_VgHzTz0gJqZJMhnEulXEajAE,431
+baml_py/internal_monkeypatch.py,sha256=JDwBPw4S8veD3nvJ13lFw8P5p29UOmDvvkgOy8eKA58,2106
+baml_py/logging.py,sha256=zM-yKPJ3LF4qpIptYQVr5zw_Gjimy3deWlTt4dOzVp0,190
+baml_py/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+baml_py/safe_import.py,sha256=ELnT9Jp1Z9ZqLZkbfmqYZ0vJpF-5dc2yshpryqn1UXE,2884
+baml_py/stream.py,sha256=pNfXDOa-6u4cjANaq071I6AevUx9N7DhDORc18Vh03k,6881
+baml_py/type_builder.py,sha256=wDrMuEr2-9r2rjFbNOXIGf2g-zdd6y3n7jlb8cFNHZc,5615
+baml_py-0.206.1.dist-info/RECORD,,

{baml_py-0.205.0.dist-info → baml_py-0.206.1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: maturin (1.9.3)
+Generator: maturin (1.9.4)
 Root-Is-Purelib: false
 Tag: cp38-abi3-macosx_10_12_x86_64

baml_py-0.205.0.dist-info/RECORD

@@ -1,16 +0,0 @@
-baml_py-0.205.0.dist-info/METADATA,sha256=8TuKollgGGupzfJaYYp4dGn2PGyFPoFWYCcJOsIgTWU,273
-baml_py-0.205.0.dist-info/WHEEL,sha256=hrXQVN4B0_e2LlpfUhUjieHy374NCAldv7m_G8ccdFw,104
-baml_py-0.205.0.dist-info/entry_points.txt,sha256=9Uu_VcUjoI2qQMjVb0PRjEgI6pQ55WqBbzNparAPJyA,54
-baml_py-0.205.0.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-baml_py/__init__.py,sha256=wBCtTkuBrwFdfo7AVI5Bxu81lh1wQgcfOU9bkkYAHAk,899
-baml_py/baml_py.abi3.so,sha256=tnVtKt8V8SSeOhyr1NqbkdsVy5jh3ZDSVBOa_ZjT1M8,47000424
-baml_py/baml_py.pyi,sha256=tGFuSIOaU6AhqK_g7eRJevuZ48DTlAp4bRbqaad067E,15807
-baml_py/ctx_manager.py,sha256=JwWLvxFT05BnYiguzB-k-g47r7mft74L3mYt8e3qtus,6081
-baml_py/errors.py,sha256=wqv7xT_-pVXQNxD5JbOrrr_CABCFuNrLrEhmEX8RVJ8,389
-baml_py/internal_monkeypatch.py,sha256=JDwBPw4S8veD3nvJ13lFw8P5p29UOmDvvkgOy8eKA58,2106
-baml_py/logging.py,sha256=zM-yKPJ3LF4qpIptYQVr5zw_Gjimy3deWlTt4dOzVp0,190
-baml_py/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-baml_py/safe_import.py,sha256=ELnT9Jp1Z9ZqLZkbfmqYZ0vJpF-5dc2yshpryqn1UXE,2884
-baml_py/stream.py,sha256=pNfXDOa-6u4cjANaq071I6AevUx9N7DhDORc18Vh03k,6881
-baml_py/type_builder.py,sha256=wDrMuEr2-9r2rjFbNOXIGf2g-zdd6y3n7jlb8cFNHZc,5615
-baml_py-0.205.0.dist-info/RECORD,,