baml-py 0.207.1__cp38-abi3-win_amd64.whl → 0.208.0__cp38-abi3-win_amd64.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.1.dist-info → baml_py-0.208.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: baml-py
-Version: 0.207.1
+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=X4Kj5f65Bbxj2Cm0uavOCDWwrn7nWjbUaLTrduDkIdg,305
+baml_py-0.208.0.dist-info/WHEEL,sha256=7bfl5v0wbVhXZba613g0x-n2obNNfpQuN8I1cQ4oaU8,94
+baml_py-0.208.0.dist-info/entry_points.txt,sha256=C5Qb3JdnQK_6tOFKhrAJnEJsMSQzSoluojZp7aDNZGk,86
+baml_py-0.208.0.dist-info/licenses/LICENSE,sha256=WtjCEwlcVzkh1ziO35P2qfVEkLjr87Flro7xlHz3CEY,11556
+baml_py/__init__.py,sha256=iya4VPU6rBDBAjRnRVTjr0lsagfUV289picJ8HGZA34,991
+baml_py/baml_py.pyd,sha256=yrIFfOv7DhmSX1cT1RChfIQditt6NnyXfSlREZqpuF4,56374272
+baml_py/baml_py.pyi,sha256=P3NiOckhJwhdYN98yTiEyUh5kZ_nwNPHVQrRvnsMV2I,17475
+baml_py/ctx_manager.py,sha256=KLsEi-clqnabNKR-R2hCoBd3SxY5uxZqBl1AFgfejng,6268
+baml_py/errors.py,sha256=IVgxDwBuB_4S2WxYq3DMko_OVY2clNxNallEMDSwFAQ,453
+baml_py/internal_monkeypatch.py,sha256=aiEc9WaQ0UEi84evpuY3E_xkrmsysFsMLH504urL_u4,3662
+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=RiKSTf8tKMVVm2b1y7ZPkoFhGJ1LsxEYHAQ1Xy6w8Vs,8468
+baml_py/type_builder.py,sha256=qTEQSU3aZn-D9r_KS0z2sI6yo0XXbXA0h6gxN4nqcrU,5804
+baml_py-0.208.0.dist-info/RECORD,,

baml_py-0.207.1.dist-info/RECORD

@@ -1,16 +0,0 @@
-baml_py-0.207.1.dist-info/METADATA,sha256=C2Ep_vE1jRdunXGoa7Vde1QI5CI1vCkdsX314gGLO_M,305
-baml_py-0.207.1.dist-info/WHEEL,sha256=7bfl5v0wbVhXZba613g0x-n2obNNfpQuN8I1cQ4oaU8,94
-baml_py-0.207.1.dist-info/entry_points.txt,sha256=C5Qb3JdnQK_6tOFKhrAJnEJsMSQzSoluojZp7aDNZGk,86
-baml_py-0.207.1.dist-info/licenses/LICENSE,sha256=WtjCEwlcVzkh1ziO35P2qfVEkLjr87Flro7xlHz3CEY,11556
-baml_py/__init__.py,sha256=iya4VPU6rBDBAjRnRVTjr0lsagfUV289picJ8HGZA34,991
-baml_py/baml_py.pyd,sha256=lYDbvnZzk4ZaQRIu2GZBdZaiUO6DB472zHYCeurDELE,54031360
-baml_py/baml_py.pyi,sha256=gg13h4UKKXzSMg9OrgHL0Gm06qG36LBCxLdIrco-_SA,17441
-baml_py/ctx_manager.py,sha256=KLsEi-clqnabNKR-R2hCoBd3SxY5uxZqBl1AFgfejng,6268
-baml_py/errors.py,sha256=IVgxDwBuB_4S2WxYq3DMko_OVY2clNxNallEMDSwFAQ,453
-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=qTEQSU3aZn-D9r_KS0z2sI6yo0XXbXA0h6gxN4nqcrU,5804
-baml_py-0.207.1.dist-info/RECORD,,