baml-py 0.207.0__cp38-abi3-macosx_10_12_x86_64.whl → 0.208.0__cp38-abi3-macosx_10_12_x86_64.whl

baml_py/baml_py.pyi

@@ -335,6 +335,7 @@ class Collector:
     @property
     def usage(self) -> Usage: ...
     def id(self, function_log_id: str) -> Optional[FunctionLog]: ...
+    def clear(self) -> None: ...
     # For debugging
     @staticmethod
     def __function_call_count() -> int: ...

baml_py/internal_monkeypatch.py

@@ -8,42 +8,57 @@ from typing import Optional
 
 class BamlClientHttpError(BamlClientError):
     """Raised for HTTP-related client errors."""
-    def __init__(self, client_name: str, message: str, status_code: int):
+    def __init__(self, client_name: str, message: str, status_code: int, detailed_message: str):
         super().__init__(message)
         self.client_name = client_name
         self.status_code = status_code
         self.message = message
+        self.detailed_message = detailed_message
 
     def __str__(self):
+        if self.detailed_message:
+            return f"BamlClientHttpError(client_name={self.client_name}, message={self.message}, status_code={self.status_code}, detailed_message={self.detailed_message})"
         return f"BamlClientHttpError(client_name={self.client_name}, message={self.message}, status_code={self.status_code})"
 
     def __repr__(self):
+        if self.detailed_message:
+            return f"BamlClientHttpError(client_name={self.client_name}, message={self.message}, status_code={self.status_code}, detailed_message={self.detailed_message})"
         return f"BamlClientHttpError(client_name={self.client_name}, message={self.message}, status_code={self.status_code})"
 
 
 class BamlValidationError(BamlError):
-    def __init__(self, prompt: str, message: str, raw_output: str):
+    def __init__(self, prompt: str, message: str, raw_output: str, detailed_message: str):
         super().__init__(message)
         self.prompt = prompt
         self.message = message
         self.raw_output = raw_output
+        self.detailed_message = detailed_message
 
     def __str__(self):
+        if self.detailed_message:
+            return f"BamlValidationError(message={self.message}, raw_output={self.raw_output}, prompt={self.prompt}, detailed_message={self.detailed_message})"
         return f"BamlValidationError(message={self.message}, raw_output={self.raw_output}, prompt={self.prompt})"
 
     def __repr__(self):
+        if self.detailed_message:
+            return f"BamlValidationError(message={self.message}, raw_output={self.raw_output}, prompt={self.prompt}, detailed_message={self.detailed_message})"
         return f"BamlValidationError(message={self.message}, raw_output={self.raw_output}, prompt={self.prompt})"
 
 class BamlClientFinishReasonError(BamlError):
-    def __init__(self, prompt: str, message: str, raw_output: str, finish_reason: Optional[str]):
+    def __init__(self, prompt: str, message: str, raw_output: str, finish_reason: Optional[str], detailed_message: str):
         super().__init__(message)
         self.prompt = prompt
         self.message = message
         self.raw_output = raw_output
         self.finish_reason = finish_reason
+        self.detailed_message = detailed_message
 
     def __str__(self):
+        if self.detailed_message:
+            return f"BamlClientFinishReasonError(message={self.message}, raw_output={self.raw_output}, prompt={self.prompt}, finish_reason={self.finish_reason}, detailed_message={self.detailed_message})"
         return f"BamlClientFinishReasonError(message={self.message}, raw_output={self.raw_output}, prompt={self.prompt}, finish_reason={self.finish_reason})"
 
     def __repr__(self):
+        if self.detailed_message:
+            return f"BamlClientFinishReasonError(message={self.message}, raw_output={self.raw_output}, prompt={self.prompt}, finish_reason={self.finish_reason}, detailed_message={self.detailed_message})"
         return f"BamlClientFinishReasonError(message={self.message}, raw_output={self.raw_output}, prompt={self.prompt}, finish_reason={self.finish_reason})"

baml_py/stream.py

@@ -81,12 +81,32 @@ class BamlStream(Generic[PartialOutputType, FinalOutputType]):
             while True:
                 try:
                     event = self.__event_queue.get_nowait()
-                    if event is None:
-                        break
-                    if event.is_ok():
-                        yield self.__partial_coerce(event)
                 except queue.Empty:
-                    await asyncio.sleep(0.050)
+                    await asyncio.sleep(0.010)
+                    continue
+
+                if event is None:
+                    break
+
+                # Drain the queue to coalesce and keep only the most recent successful event.
+                latest_ok: Optional[FunctionResult] = event if event.is_ok() else None
+                done_seen = False
+                while True:
+                    try:
+                        nxt = self.__event_queue.get_nowait()
+                        if nxt is None:
+                            done_seen = True
+                            break
+                        if nxt.is_ok():
+                            latest_ok = nxt
+                    except queue.Empty:
+                        break
+
+                if latest_ok is not None:
+                    yield self.__partial_coerce(latest_ok)
+
+                if done_seen:
+                    break
         except Exception as e:
             raise e
         finally:
@@ -163,8 +183,26 @@ class BamlSyncStream(Generic[PartialOutputType, FinalOutputType]):
                 event = self.__event_queue.get()
                 if event is None:
                     break
-                if event.is_ok():
-                    yield self.__partial_coerce(event)
+
+                # Drain the queue to coalesce and keep only the most recent successful event.
+                latest_ok: Optional[FunctionResult] = event if event.is_ok() else None
+                done_seen = False
+                while True:
+                    try:
+                        nxt = self.__event_queue.get_nowait()
+                        if nxt is None:
+                            done_seen = True
+                            break
+                        if nxt.is_ok():
+                            latest_ok = nxt
+                    except queue.Empty:
+                        break
+
+                if latest_ok is not None:
+                    yield self.__partial_coerce(latest_ok)
+
+                if done_seen:
+                    break
         except Exception as e:
             raise e
         finally:

{baml_py-0.207.0.dist-info → baml_py-0.208.0.dist-info}/METADATA

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

baml_py-0.208.0.dist-info/RECORD

@@ -0,0 +1,16 @@
+baml_py-0.208.0.dist-info/METADATA,sha256=KucNdXX5xgHgSKWD_bil02ywbPiZpx04Ap2xDQozZlk,304
+baml_py-0.208.0.dist-info/WHEEL,sha256=qq8T5RNN4nLzVDHWhr9aubnbXTyL36gsBivIyKvr8Qc,104
+baml_py-0.208.0.dist-info/entry_points.txt,sha256=C5Qb3JdnQK_6tOFKhrAJnEJsMSQzSoluojZp7aDNZGk,86
+baml_py-0.208.0.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+baml_py/__init__.py,sha256=_tXffmw8Be3rMDR8Zfivr1Gp-wL7baJ-guSoIRaOsBY,943
+baml_py/baml_py.abi3.so,sha256=zGJGqnUxaynHxHR5-vDkQ0SbhK8SOxXPhNtChdpIYt8,49977808
+baml_py/baml_py.pyi,sha256=FdvYxwEooT96gj8PoWbCRwW2quzyiXqL4SJXZpG2SFg,16919
+baml_py/ctx_manager.py,sha256=JwWLvxFT05BnYiguzB-k-g47r7mft74L3mYt8e3qtus,6081
+baml_py/errors.py,sha256=cpD5OlBuKIX8IbImwM_VgHzTz0gJqZJMhnEulXEajAE,431
+baml_py/internal_monkeypatch.py,sha256=nuYufMyJQ3Nh3pWd1E7js_in2EZisFYzAD3N4KbCuVg,3598
+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=jcVYi3d6OXqrm39ne7ZMmLlTVIkDAuaFoerPhGs_ceg,8243
+baml_py/type_builder.py,sha256=wDrMuEr2-9r2rjFbNOXIGf2g-zdd6y3n7jlb8cFNHZc,5615
+baml_py-0.208.0.dist-info/RECORD,,

baml_py-0.207.0.dist-info/RECORD

@@ -1,16 +0,0 @@
-baml_py-0.207.0.dist-info/METADATA,sha256=MTrGM0Kr6NNjLawoGnPFvAtQGZl8sj75dusIdi6H4CI,304
-baml_py-0.207.0.dist-info/WHEEL,sha256=qq8T5RNN4nLzVDHWhr9aubnbXTyL36gsBivIyKvr8Qc,104
-baml_py-0.207.0.dist-info/entry_points.txt,sha256=C5Qb3JdnQK_6tOFKhrAJnEJsMSQzSoluojZp7aDNZGk,86
-baml_py-0.207.0.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-baml_py/__init__.py,sha256=_tXffmw8Be3rMDR8Zfivr1Gp-wL7baJ-guSoIRaOsBY,943
-baml_py/baml_py.abi3.so,sha256=YdkZa30Vlor4TRPYXQ-uDeFo43zcGl6MMlmgvrw8BWs,47835800
-baml_py/baml_py.pyi,sha256=zxoG-X_D_H4txwzDlgMltTIDWpyNRxn0Os1ll1BjEWk,16886
-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.207.0.dist-info/RECORD,,