baml-py 0.201.0__cp38-abi3-musllinux_1_1_aarch64.whl → 0.202.0__cp38-abi3-musllinux_1_1_aarch64.whl

baml_py/__init__.py

@@ -9,6 +9,8 @@ from .baml_py import (
     BamlImagePy as Image,
     BamlAudioPy as Audio,
     invoke_runtime_cli,
+    BamlPdfPy as Pdf,
+    BamlVideoPy as Video,
     ClientRegistry,
     # Collector utilities
     Collector,
@@ -31,6 +33,8 @@ __all__ = [
     "FunctionResultStream",
     "Image",
     "Audio",
+    "Pdf",
+    "Video",
     "invoke_runtime_cli",
     # Collector types
     "Collector",

baml_py/baml_py.pyi

@@ -6,7 +6,9 @@ def get_version() -> str:
     """Get the version of the BAML Python client."""
     ...
 
-def set_log_level(level: Literal["DEBUG", "INFO", "WARN", "ERROR", "OFF"] | str) -> None:
+def set_log_level(
+    level: Literal["DEBUG", "INFO", "WARN", "ERROR", "OFF"] | str,
+) -> None:
     """Set the log level for the BAML Python client."""
     ...
 
@@ -39,7 +41,13 @@ class FunctionResult:
     def __str__(self) -> str: ...
     # Returns True if the function call was successful, False otherwise
     def is_ok(self) -> bool: ...
-    def cast_to(self, enum_module: Any, class_module: Any, partial_class_module: Any, allow_partials: bool) -> Any: ...
+    def cast_to(
+        self,
+        enum_module: Any,
+        class_module: Any,
+        partial_class_module: Any,
+        allow_partials: bool,
+    ) -> Any: ...
 
     # This is a debug function that returns the internal representation of the response
     # This is not to be relied upon and is subject to change
@@ -97,6 +105,26 @@ class BamlAudioPy:
     def as_url(self) -> str: ...
     def as_base64(self) -> Tuple[str, str]: ...
 
+class BamlPdfPy:
+    @staticmethod
+    def from_url(url: str) -> BamlPdfPy: ...
+    @staticmethod
+    def from_base64(media_type: str, base64: str) -> BamlPdfPy: ...
+    def is_url(self) -> bool: ...
+    def is_base64(self) -> bool: ...
+    def as_url(self) -> str: ...
+    def as_base64(self) -> Tuple[str, str]: ...
+
+class BamlVideoPy:
+    @staticmethod
+    def from_url(url: str) -> BamlVideoPy: ...
+    @staticmethod
+    def from_base64(media_type: str, base64: str) -> BamlVideoPy: ...
+    def is_url(self) -> bool: ...
+    def is_base64(self) -> bool: ...
+    def as_url(self) -> str: ...
+    def as_base64(self) -> Tuple[str, str]: ...
+
 class RuntimeContextManager:
     def upsert_tags(self, tags: Dict[str, Any]) -> None: ...
     def deep_clone(self) -> RuntimeContextManager: ...
@@ -196,7 +224,6 @@ class BamlRuntime:
         env_vars: Dict[str, str],
     ) -> Any: ...
 
-
 class LogEventMetadata:
     event_id: str
     parent_id: Optional[str]
@@ -280,7 +307,9 @@ class Collector:
     def __print_storage() -> None: ...
 
 class FunctionLog:
-    def __init__(self, name: str, args: Dict[str, Any], result: FunctionResult) -> None: ...
+    def __init__(
+        self, name: str, args: Dict[str, Any], result: FunctionResult
+    ) -> None: ...
     def __str__(self) -> str: ...
     @property
     def id(self) -> str: ...
@@ -301,9 +330,16 @@ class FunctionLog:
     @property
     def selected_call(self) -> Optional[Union[LLMCall, LLMStreamCall]]: ...
 
-
 class LLMCall:
-    def __init__(self, client_name: str, provider: str, timing: Timing, request: Dict[str, Any], response: Dict[str, Any], usage: Usage) -> None: ...
+    def __init__(
+        self,
+        client_name: str,
+        provider: str,
+        timing: Timing,
+        request: Dict[str, Any],
+        response: Dict[str, Any],
+        usage: Usage,
+    ) -> None: ...
     @property
     def selected(self) -> bool: ...
     @property
@@ -320,38 +356,80 @@ class LLMCall:
     @property
     def client_name(self) -> str: ...
 
+class LLMStreamCall:
+    def __init__(
+        self,
+        client_name: str,
+        provider: str,
+        timing: StreamTiming,
+        request: Dict[str, Any],
+        response: Dict[str, Any],
+        usage: Usage,
+    ) -> None: ...
+    @property
+    def selected(self) -> bool: ...
+    @property
+    def http_request(self) -> Optional[HTTPRequest]: ...
+    @property
+    def http_response(self) -> Optional[HTTPResponse]: ...
+    @property
+    def usage(self) -> Usage: ...
+    @property
+    def timing(self) -> Timing: ...
+    @property
+    def provider(self) -> str: ...
 
-class LLMStreamCall(LLMCall):
-    def __init__(self, client_name: str, provider: str, timing: StreamTiming, request: Dict[str, Any], response: Dict[str, Any],
-    usage: Usage) -> None: ...
-    # TODO: add chunks
-    # def chunks(self) -> List[str]: ...
+    # The baml client name / clientregistry name
+    @property
+    def client_name(self) -> str: ...
     @property
     def timing(self) -> StreamTiming: ...
+    def sse_responses(self) -> Optional[List[SSEResponse]]: ...
+
+class SSEResponse:
+    @property
+    def text(self) -> str: ...
+    def json(self) -> Optional[Any]: ...
 
 class Usage:
-    def __init__(self, input_tokens: Optional[int], output_tokens: Optional[int]) -> None: ...
+    def __init__(
+        self, input_tokens: Optional[int], output_tokens: Optional[int]
+    ) -> None: ...
     @property
     def input_tokens(self) -> Optional[int]: ...
     @property
     def output_tokens(self) -> Optional[int]: ...
 
 class Timing:
-    def __init__(self, start_time: str, end_time: str, duration: Optional[int], time_to_first_parsed_ms: Optional[int]) -> None: ...
+    def __init__(
+        self,
+        start_time: str,
+        end_time: str,
+        duration: Optional[int],
+        time_to_first_parsed_ms: Optional[int],
+    ) -> None: ...
     @property
     def start_time_utc_ms(self) -> int: ...
     @property
     def duration_ms(self) -> Optional[int]: ...
-    @property
-    def time_to_first_parsed_ms(self) -> Optional[int]: ...
 
 class StreamTiming(Timing):
-    def __init__(self, start_time_utc_ms: int, duration_ms: int, time_to_first_parsed_ms: int, time_to_first_token_ms: int) -> None: ...
+    def __init__(
+        self,
+        start_time_utc_ms: int,
+        duration_ms: int,
+        time_to_first_parsed_ms: int,
+        time_to_first_token_ms: int,
+    ) -> None: ...
+    @property
+    def start_time_utc_ms(self) -> int: ...
     @property
-    def time_to_first_token_ms(self) -> Optional[int]: ...
+    def duration_ms(self) -> Optional[int]: ...
 
 class HTTPRequest:
-    def __init__(self, url: str, method: str, headers: Dict[str, Any], body: str) -> None: ...
+    def __init__(
+        self, url: str, method: str, headers: Dict[str, Any], body: str
+    ) -> None: ...
     @property
     def id(self) -> str: ...
     @property

baml_py/safe_import.py

@@ -66,7 +66,7 @@ https://boundaryml.com/discord
             raise ImportError(f"""
 {msg}
 
-Please upgrade baml-py to version "{target_version}".
+Please set baml-py to version "{target_version}".
 
 $ pip install {self._target_package_name()}
 $ uv add {self._target_package_name()}

{baml_py-0.201.0.dist-info → baml_py-0.202.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: baml-py
-Version: 0.201.0
+Version: 0.202.0
 License-File: LICENSE
 Summary: BAML python bindings (pyproject.toml)
 Author: Boundary <contact@boundaryml.com>

{baml_py-0.201.0.dist-info → baml_py-0.202.0.dist-info}/RECORD

@@ -1,17 +1,17 @@
-baml_py-0.201.0.dist-info/METADATA,sha256=JropL9HMS5Kattful84VALE7D6qCzPfttqMvWi8HaRU,335
-baml_py-0.201.0.dist-info/WHEEL,sha256=B43xczSjoNhPSK5OHg1ebG0-FCDNoF4LtGHeVtc4gB8,106
-baml_py-0.201.0.dist-info/entry_points.txt,sha256=9Uu_VcUjoI2qQMjVb0PRjEgI6pQ55WqBbzNparAPJyA,54
-baml_py-0.201.0.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+baml_py-0.202.0.dist-info/METADATA,sha256=bzzA3IH8sHtXSgD-iHMwzqwDZ6m5syywfhDTu083PkA,335
+baml_py-0.202.0.dist-info/WHEEL,sha256=VBqZ7yWNVuAHzgY5FGTKps0uNrM9AISwCCgWYYUr00M,106
+baml_py-0.202.0.dist-info/entry_points.txt,sha256=9Uu_VcUjoI2qQMjVb0PRjEgI6pQ55WqBbzNparAPJyA,54
+baml_py-0.202.0.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
 baml_py.libs/libgcc_s-e52197c3.so.1,sha256=vkPW1Auw6CH9Bjk7frmX3hry_1H9c0tRI0Ncyg71WUI,724137
-baml_py/__init__.py,sha256=QWyuR-eLmqkiRW5v9RKF83knRj_HG8fyTlp6Sse9D3o,827
-baml_py/baml_py.abi3.so,sha256=lIxJA9HQge3aSe0Klg0Xx4rtcRYXtmFilN3SMK1U46U,49222873
-baml_py/baml_py.pyi,sha256=Ev8oqkdiQhp0tb9-62xEy3ZtMN43uyJIxP68ZNlGD98,13995
+baml_py/__init__.py,sha256=wBCtTkuBrwFdfo7AVI5Bxu81lh1wQgcfOU9bkkYAHAk,899
+baml_py/baml_py.abi3.so,sha256=ub6Y-WE3dQMh78BMJlpKrkpbAs6u99buOhTPKObzn80,51057929
+baml_py/baml_py.pyi,sha256=WWXAiJE737ENYUxjeJWXYs67ktUVhZP-bDwdcNEHTuo,15461
 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=turgUpn9B4G273ZuDVjfZ_CkA2qmFQyiP-ZCPhtJO4M,2888
+baml_py/safe_import.py,sha256=ELnT9Jp1Z9ZqLZkbfmqYZ0vJpF-5dc2yshpryqn1UXE,2884
 baml_py/stream.py,sha256=pNfXDOa-6u4cjANaq071I6AevUx9N7DhDORc18Vh03k,6881
 baml_py/type_builder.py,sha256=HIAlses70C5DWNWgx3ZwsLeGt5-tviWXCXZiyyWedSg,6374
-baml_py-0.201.0.dist-info/RECORD,,
+baml_py-0.202.0.dist-info/RECORD,,

{baml_py-0.201.0.dist-info → baml_py-0.202.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: maturin (1.9.0)
+Generator: maturin (1.9.1)
 Root-Is-Purelib: false
 Tag: cp38-abi3-musllinux_1_1_aarch64