feat(AppClient): wait for logs after an error (#327)

fal-ai · Oct 10, 2024 · 4aba448 · 4aba448
1 parent 9aa6fc3
commit 4aba448
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/projects/fal/src/fal/app.py b/projects/fal/src/fal/app.py
@@ -93,7 +93,12 @@ def __call__(self, data):
                 json=data.dict() if hasattr(data, "dict") else dict(data),
                 timeout=self.timeout,
             )
-            resp.raise_for_status()
+            try:
+                resp.raise_for_status()
+            except httpx.HTTPStatusError:
+                # allow logs to be printed before raising the exception
+                time.sleep(1)
+                raise
             resp_dict = resp.json()
 
         if not self.return_type: