fal 1.5.15__py3-none-any.whl → 1.5.16__py3-none-any.whl

fal/_fal_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '1.5.15'
-__version_tuple__ = version_tuple = (1, 5, 15)
+__version__ = version = '1.5.16'
+__version_tuple__ = version_tuple = (1, 5, 16)

fal/apps.py

@@ -261,7 +261,7 @@ class _WSConnection:
     _ws: Connection
     _buffer: str | bytes | None = None
 
-    def run(self, arguments: dict[str, Any]) -> dict[str, Any]:
+    def run(self, arguments: dict[str, Any]) -> bytes:
         """Run an inference task on the app and return the result."""
         self.send(arguments)
         return self.recv()
@@ -318,34 +318,28 @@ class _WSConnection:
 
             return json_payload
 
-    def _recv_response(self) -> Any:
-        import msgpack
-
-        body: bytes = b""
+    def _recv_response(self) -> Iterator[str | bytes]:
         while True:
             try:
                 with self._recv() as res:
                     if self._is_meta(res):
-                        # Keep the meta message for later
+                        # Raise so we dont consume the message
                         raise _MetaMessageFound()
 
-                    if isinstance(res, str):
-                        return res
-                    else:
-                        body += res
+                    yield res
             except _MetaMessageFound:
                 break
 
-        if not body:
-            raise ValueError("Empty response body")
-
-        return msgpack.unpackb(body)
-
-    def recv(self) -> Any:
+    def recv(self) -> bytes:
         start = self._recv_meta("start")
         request_id = start["request_id"]
 
-        response = self._recv_response()
+        response = b""
+        for part in self._recv_response():
+            if isinstance(part, str):
+                response += part.encode()
+            else:
+                response += part
 
         end = self._recv_meta("end")
         if end["request_id"] != request_id:
@@ -353,6 +347,17 @@ class _WSConnection:
 
         return response
 
+    def stream(self) -> Iterator[str | bytes]:
+        start = self._recv_meta("start")
+        request_id = start["request_id"]
+
+        yield from self._recv_response()
+
+        # Make sure we consume the end message
+        end = self._recv_meta("end")
+        if end["request_id"] != request_id:
+            raise ValueError("Mismatched request_id in end message")
+
 
 @contextmanager
 def ws(app_id: str, *, path: str = "") -> Iterator[_WSConnection]:

{fal-1.5.15.dist-info → fal-1.5.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fal
-Version: 1.5.15
+Version: 1.5.16
 Summary: fal is an easy-to-use Serverless Python Framework
 Author: Features & Labels <support@fal.ai>
 Requires-Python: >=3.8

{fal-1.5.15.dist-info → fal-1.5.16.dist-info}/RECORD

@@ -1,11 +1,11 @@
 fal/__init__.py,sha256=wXs1G0gSc7ZK60-bHe-B2m0l_sA6TrFk4BxY0tMoLe8,784
 fal/__main__.py,sha256=4JMK66Wj4uLZTKbF-sT3LAxOsr6buig77PmOkJCRRxw,83
-fal/_fal_version.py,sha256=299zYncL2JUBDyICATkgM5qBt_4iztg0ZseMesrRfvI,413
+fal/_fal_version.py,sha256=mHfZXquAABTFJcCTtEKL7lltRkyfT160CZojEvwd1Aw,413
 fal/_serialization.py,sha256=rD2YiSa8iuzCaZohZwN_MPEB-PpSKbWRDeaIDpTEjyY,7653
 fal/_version.py,sha256=EBGqrknaf1WygENX-H4fBefLvHryvJBBGtVJetaB0NY,266
 fal/api.py,sha256=xTtPvDqaEHsq2lFsMwRZiHb4hzjVY3y6lV-xbzkSetI,43375
 fal/app.py,sha256=nLku84uTyK2VJRH_dGe_Ym8fNRsTvC3_5yolgjh9wlY,22429
-fal/apps.py,sha256=-s3xVOclIQxAevLdbMcyhGkPGsx8x0VAZbRoYQG-F_I,10800
+fal/apps.py,sha256=RpmElElJnDYjsTRQOdNYiJwd74GEOGYA38L5O5GzNEg,11068
 fal/container.py,sha256=V7riyyq8AZGwEX9QaqRQDZyDN_bUKeRKV1OOZArXjL0,622
 fal/files.py,sha256=QgfYfMKmNobMPufrAP_ga1FKcIAlSbw18Iar1-0qepo,2650
 fal/flags.py,sha256=oWN_eidSUOcE9wdPK_77si3A1fpgOC0UEERPsvNLIMc,842
@@ -128,8 +128,8 @@ openapi_fal_rest/models/workflow_node_type.py,sha256=-FzyeY2bxcNmizKbJI8joG7byRi
 openapi_fal_rest/models/workflow_schema.py,sha256=4K5gsv9u9pxx2ItkffoyHeNjBBYf6ur5bN4m_zePZNY,2019
 openapi_fal_rest/models/workflow_schema_input.py,sha256=2OkOXWHTNsCXHWS6EGDFzcJKkW5FIap-2gfO233EvZQ,1191
 openapi_fal_rest/models/workflow_schema_output.py,sha256=EblwSPAGfWfYVWw_WSSaBzQVju296is9o28rMBAd0mc,1196
-fal-1.5.15.dist-info/METADATA,sha256=C4sigcaEzk4Yv8fuA7rJtKcr0AhaH3zyS7Qb1PQrFF4,3997
-fal-1.5.15.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-fal-1.5.15.dist-info/entry_points.txt,sha256=32zwTUC1U1E7nSTIGCoANQOQ3I7-qHG5wI6gsVz5pNU,37
-fal-1.5.15.dist-info/top_level.txt,sha256=r257X1L57oJL8_lM0tRrfGuXFwm66i1huwQygbpLmHw,21
-fal-1.5.15.dist-info/RECORD,,
+fal-1.5.16.dist-info/METADATA,sha256=rXM_O4xtD7xvwYD3gp1Bsh-Dtgc7M3IvsTv3EiVjVlg,3997
+fal-1.5.16.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+fal-1.5.16.dist-info/entry_points.txt,sha256=32zwTUC1U1E7nSTIGCoANQOQ3I7-qHG5wI6gsVz5pNU,37
+fal-1.5.16.dist-info/top_level.txt,sha256=r257X1L57oJL8_lM0tRrfGuXFwm66i1huwQygbpLmHw,21
+fal-1.5.16.dist-info/RECORD,,