baml-py 0.201.0__cp38-abi3-win_arm64.whl → 0.202.1__cp38-abi3-win_arm64.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.1.dist-info}/METADATA

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

baml_py-0.202.1.dist-info/RECORD

@@ -0,0 +1,16 @@
+baml_py-0.202.1.dist-info/METADATA,sha256=MyU0gcPBwALw-N2plZC0M8Eg6N_WvZLFyRzkXxo-BDg,336
+baml_py-0.202.1.dist-info/WHEEL,sha256=brtZzFSYSpN-7dgVTVyZRllMDwYqodWyQDQRZMiNrIo,94
+baml_py-0.202.1.dist-info/entry_points.txt,sha256=9Uu_VcUjoI2qQMjVb0PRjEgI6pQ55WqBbzNparAPJyA,54
+baml_py-0.202.1.dist-info/licenses/LICENSE,sha256=WtjCEwlcVzkh1ziO35P2qfVEkLjr87Flro7xlHz3CEY,11556
+baml_py/__init__.py,sha256=0Nnk5nLOhfu7whIfBsWUkXWyBnRl2QuyGQVfoi3ia0c,945
+baml_py/baml_py.pyd,sha256=oOsJFZk5djCddcG2U1KiZdbQcdmsIeKZF2VCjg1o4a0,41475584
+baml_py/baml_py.pyi,sha256=oXK82Wzd5lJEzGhWDmOPoXV3QoKuvBnmEPUQCp0hIwU,15974
+baml_py/ctx_manager.py,sha256=KLsEi-clqnabNKR-R2hCoBd3SxY5uxZqBl1AFgfejng,6268
+baml_py/errors.py,sha256=Qesr_IinnvsmRAw-hkExlLcFKvY4Zyi66JGoujxJ-sM,409
+baml_py/internal_monkeypatch.py,sha256=eX9LK0xtZqVpWYhCEqLPqYmP2NKzharE5j64h7CFFxI,2155
+baml_py/logging.py,sha256=x5sJibE7KfT4YsnNgmyXbhzHu4xOwLtc4trbYHsYmzc,193
+baml_py/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+baml_py/safe_import.py,sha256=gV6nMqBx-O6INutLxGSdfKzgtICx4WjXV40uz_NKwhQ,2971
+baml_py/stream.py,sha256=cY9biE_V3SqOEQFqWTttW3wPOh7RhOHOD1fDifEylR8,7068
+baml_py/type_builder.py,sha256=0VhIU1sA5IgFksKN9rMWWXRtsKE8MX4i8qosvGdMtG0,6578
+baml_py-0.202.1.dist-info/RECORD,,

{baml_py-0.201.0.dist-info → baml_py-0.202.1.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-win_arm64

baml_py-0.201.0.dist-info/RECORD

@@ -1,16 +0,0 @@
-baml_py-0.201.0.dist-info/METADATA,sha256=dlREla8THs9NMhe51vOyLCLQzxOE7nrkWICjV6nNZeE,336
-baml_py-0.201.0.dist-info/WHEEL,sha256=bJVluvRCzevgnqQwri7z33LPshy2FtyoozeY5cUvjgQ,94
-baml_py-0.201.0.dist-info/entry_points.txt,sha256=9Uu_VcUjoI2qQMjVb0PRjEgI6pQ55WqBbzNparAPJyA,54
-baml_py-0.201.0.dist-info/licenses/LICENSE,sha256=WtjCEwlcVzkh1ziO35P2qfVEkLjr87Flro7xlHz3CEY,11556
-baml_py/__init__.py,sha256=goPPklYWODB3Bi7g3AxHInixcRAvLSUS_aC2nMd8NLg,869
-baml_py/baml_py.pyd,sha256=eH_c55LKPg-5JTIk29Uk0u9VMctkcTTTFJK1PONwInE,39216128
-baml_py/baml_py.pyi,sha256=Ge26n3ilsCvFeshf5KB4_nZMt9VOtGCJNcd1IGbz7E8,14430
-baml_py/ctx_manager.py,sha256=KLsEi-clqnabNKR-R2hCoBd3SxY5uxZqBl1AFgfejng,6268
-baml_py/errors.py,sha256=Qesr_IinnvsmRAw-hkExlLcFKvY4Zyi66JGoujxJ-sM,409
-baml_py/internal_monkeypatch.py,sha256=eX9LK0xtZqVpWYhCEqLPqYmP2NKzharE5j64h7CFFxI,2155
-baml_py/logging.py,sha256=x5sJibE7KfT4YsnNgmyXbhzHu4xOwLtc4trbYHsYmzc,193
-baml_py/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-baml_py/safe_import.py,sha256=rZe6oMYrESyNTkkkpuPsJKBloqzF5EAio3T2Zl9WqXg,2975
-baml_py/stream.py,sha256=cY9biE_V3SqOEQFqWTttW3wPOh7RhOHOD1fDifEylR8,7068
-baml_py/type_builder.py,sha256=0VhIU1sA5IgFksKN9rMWWXRtsKE8MX4i8qosvGdMtG0,6578
-baml_py-0.201.0.dist-info/RECORD,,