fix(binaries): don't synchronously block in astream_to_file (#28)

orbcorp · Nov 3, 2023 · a63dc23 · a63dc23
1 parent 8adf54e
commit a63dc23
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/orb/_base_client.py b/src/orb/_base_client.py
@@ -1758,9 +1758,10 @@ async def aiter_raw(self, chunk_size: Optional[int] = None) -> AsyncIterator[byt
 
     @override
     async def astream_to_file(self, file: str | os.PathLike[str]) -> None:
-        with open(file, mode="wb") as f:
+        path = anyio.Path(file)
+        async with await path.open(mode="wb") as f:
             async for data in self.response.aiter_bytes():
-                f.write(data)
+                await f.write(data)
 
     @override
     async def aclose(self) -> None: