baml-py 0.90.1__cp38-abi3-macosx_10_12_x86_64.whl → 0.200.0__cp38-abi3-macosx_10_12_x86_64.whl

baml_py/baml_py.pyi

@@ -103,6 +103,8 @@ class RuntimeContextManager:
     def context_depth(self) -> int: ...
 
 class BamlRuntime:
+    def __getstate__(self) -> Tuple[str, Dict[str, str]]: ...
+    def __setstate__(self, state: Tuple[str, Dict[str, str]]) -> None: ...
     @staticmethod
     def from_directory(directory: str, env_vars: Dict[str, str]) -> BamlRuntime: ...
     async def call_function(

baml_py/ctx_manager.py

@@ -31,6 +31,16 @@ prev_ctx_manager: typing.Optional["CtxManager"] = None
 
 
 class CtxManager:
+    def __setstate__(self, state: typing.Dict[str, typing.Any]) -> None:
+        self.rt = state["rt"]
+        self.ctx = contextvars.ContextVar[typing.Dict[int, RuntimeContextManager]](
+            "baml_ctx", default={current_thread_id(): self.rt.create_context_manager()}
+        )
+        atexit.register(self.rt.flush)
+    
+    def __getstate__(self) -> typing.Dict[str, typing.Any]:
+        return {"rt": self.rt}
+
     def __new__(cls, *args, **kwargs):
         if prev_ctx_manager is not None:
             return prev_ctx_manager

baml_py/stream.py

@@ -50,7 +50,7 @@ class BamlStream(Generic[PartialOutputType, FinalOutputType]):
             retval = await self.__ffi_stream.done(self.__ctx_manager)
 
             self.__future.set_result(retval)
-           
+
             return retval
         except Exception as e:
             self.__future.set_exception(e)

{baml_py-0.90.1.dist-info → baml_py-0.200.0.dist-info}/METADATA

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

baml_py-0.200.0.dist-info/RECORD

@@ -0,0 +1,16 @@
+baml_py-0.200.0.dist-info/METADATA,sha256=EbvmXr3GAXbMB2z23KV0HZx_OOd1_As-TL_tFOVpC9g,335
+baml_py-0.200.0.dist-info/WHEEL,sha256=EGuPi0n_1hkOWYoFFR6b_9qqSL6dMSNZbfmSoiDcLVo,104
+baml_py-0.200.0.dist-info/entry_points.txt,sha256=9Uu_VcUjoI2qQMjVb0PRjEgI6pQ55WqBbzNparAPJyA,54
+baml_py-0.200.0.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+baml_py/__init__.py,sha256=QWyuR-eLmqkiRW5v9RKF83knRj_HG8fyTlp6Sse9D3o,827
+baml_py/baml_py.abi3.so,sha256=HTW028Lh3_9QreEn1isirzRTgwKDthXEPF6iSGCMuwc,43121240
+baml_py/baml_py.pyi,sha256=Ev8oqkdiQhp0tb9-62xEy3ZtMN43uyJIxP68ZNlGD98,13995
+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/stream.py,sha256=pNfXDOa-6u4cjANaq071I6AevUx9N7DhDORc18Vh03k,6881
+baml_py/type_builder.py,sha256=HIAlses70C5DWNWgx3ZwsLeGt5-tviWXCXZiyyWedSg,6374
+baml_py-0.200.0.dist-info/RECORD,,

{baml_py-0.90.1.dist-info → baml_py-0.200.0.dist-info}/WHEEL

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

baml_py-0.90.1.dist-info/RECORD

@@ -1,16 +0,0 @@
-baml_py-0.90.1.dist-info/METADATA,sha256=dfqOnYNMNtHAl8j8oo_CqkZI1KaPu5fS0WkoERvr3YI,334
-baml_py-0.90.1.dist-info/WHEEL,sha256=-VeZVZVER2HyvPbjF_EpJfcVOYx0xLg34sdR3QVUxMw,104
-baml_py-0.90.1.dist-info/entry_points.txt,sha256=9Uu_VcUjoI2qQMjVb0PRjEgI6pQ55WqBbzNparAPJyA,54
-baml_py-0.90.1.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-baml_py/__init__.py,sha256=QWyuR-eLmqkiRW5v9RKF83knRj_HG8fyTlp6Sse9D3o,827
-baml_py/baml_py.abi3.so,sha256=ivftwi4VS71K2_zTLF0egOoDWY8e5B6J5jkVFWdi6Vg,40902064
-baml_py/baml_py.pyi,sha256=YYyXU1tLGLhz2OMRm9lroIAaKmQ0RZfATQ5Dea8q73s,13858
-baml_py/ctx_manager.py,sha256=p6Y4X1qfUExLeBeJKcVuf4WS9NrSabjRcRFrPIasTxY,5660
-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/stream.py,sha256=RoHvdlYi1lap7DN0sCUA-H5HtAfxxePnm1nIe6BRTTs,6892
-baml_py/type_builder.py,sha256=HIAlses70C5DWNWgx3ZwsLeGt5-tviWXCXZiyyWedSg,6374
-baml_py-0.90.1.dist-info/RECORD,,