baml-py 0.204.0__cp38-abi3-musllinux_1_1_x86_64.whl → 0.206.0__cp38-abi3-musllinux_1_1_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: ...
@@ -223,6 +256,7 @@ class BamlRuntime:
         cr: Optional[ClientRegistry],
         env_vars: Dict[str, str],
     ) -> Any: ...
+    def disassemble(self, function_name: str) -> None: ...
 
 class LogEventMetadata:
     event_id: str

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.204.0.dist-info → baml_py-0.206.0.dist-info}/METADATA

@@ -1,9 +1,10 @@
 Metadata-Version: 2.4
 Name: baml-py
-Version: 0.204.0
+Version: 0.206.0
 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.204.0.dist-info → baml_py-0.206.0.dist-info}/RECORD

@@ -1,17 +1,17 @@
-baml_py-0.204.0.dist-info/METADATA,sha256=zKv1tSF-R_g6Ki-sHKY5KILNj5qaV5mnv_HZDa6xnMs,273
-baml_py-0.204.0.dist-info/WHEEL,sha256=vSv-hP6muoZ7COiPtF7hZk_jEzLvv2VRUWDp6tnOxBU,105
-baml_py-0.204.0.dist-info/entry_points.txt,sha256=9Uu_VcUjoI2qQMjVb0PRjEgI6pQ55WqBbzNparAPJyA,54
-baml_py-0.204.0.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+baml_py-0.206.0.dist-info/METADATA,sha256=FcHOfHQpdx-ToGTRxWz4zHHiJeAz1S3YNOCwibVs_dc,304
+baml_py-0.206.0.dist-info/WHEEL,sha256=RyuTMqZ-GvVRmySrlkZCJvWJb4QsexmNN-AIFn-SgE4,105
+baml_py-0.206.0.dist-info/entry_points.txt,sha256=9Uu_VcUjoI2qQMjVb0PRjEgI6pQ55WqBbzNparAPJyA,54
+baml_py-0.206.0.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
 baml_py.libs/libgcc_s-1e52349c.so.1,sha256=Ani0u_W_tB8ESsFePO4D2mn2lrgWLhyFdw6RETxBTYM,437537
-baml_py/__init__.py,sha256=wBCtTkuBrwFdfo7AVI5Bxu81lh1wQgcfOU9bkkYAHAk,899
-baml_py/baml_py.abi3.so,sha256=CNvJHjCKkzPcfTYW4fSr0UiDrTm1UEIoHtIWYlZGWhI,55985361
-baml_py/baml_py.pyi,sha256=jqSIfnsL2pKGPHJz_U_VrviWP3FErnOT-6TOCCrCefY,15748
+baml_py/__init__.py,sha256=_tXffmw8Be3rMDR8Zfivr1Gp-wL7baJ-guSoIRaOsBY,943
+baml_py/baml_py.abi3.so,sha256=YA5k5FlyiLm6IHG_XAt768zgFCJbpxb2Xj8UA4sIE0I,57586913
+baml_py/baml_py.pyi,sha256=ha8JgEp2Sg45s_8Eu7OjWq29FMct4GqFQ7zOVs4Qt0Y,16794
 baml_py/ctx_manager.py,sha256=JwWLvxFT05BnYiguzB-k-g47r7mft74L3mYt8e3qtus,6081
-baml_py/errors.py,sha256=wqv7xT_-pVXQNxD5JbOrrr_CABCFuNrLrEhmEX8RVJ8,389
+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.204.0.dist-info/RECORD,,
+baml_py-0.206.0.dist-info/RECORD,,

{baml_py-0.204.0.dist-info → baml_py-0.206.0.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-musllinux_1_1_x86_64