streamlit-octostar-utils 0.4.2.dev16__tar.gz → 0.4.2.dev17__tar.gz

{streamlit_octostar_utils-0.4.2.dev16 → streamlit_octostar_utils-0.4.2.dev17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: streamlit-octostar-utils
-Version: 0.4.2.dev16
+Version: 0.4.2.dev17
 Summary: 
 License: MIT
 License-File: LICENSE

{streamlit_octostar_utils-0.4.2.dev16 → streamlit_octostar_utils-0.4.2.dev17}/pyproject.toml

@@ -5,7 +5,7 @@ include = '\.pyi?$'
 
 [tool.poetry]
 name = "streamlit-octostar-utils"
-version = "0.4.2-dev.16"
+version = "0.4.2-dev.17"
 description = ""
 license = "MIT"
 authors = ["Octostar"]

{streamlit_octostar_utils-0.4.2.dev16 → streamlit_octostar_utils-0.4.2.dev17}/streamlit_octostar_utils/api_crafter/celery.py

@@ -28,7 +28,7 @@ logging.getLogger("pottery").setLevel(logging.WARNING)
 from celery.app.defaults import DEFAULTS as CELERY_DEFAULTS
 import urllib
 
-from .fastapi import Route, CommonModels, DefaultErrorRoute
+from .fastapi import Route, CommonModels, DefaultErrorRoute, RemoteTraceback
 
 
 class RedisFileLock:
@@ -700,7 +700,12 @@ class CeleryExecutor(object):
 
     async def get_task_result(self, task_id, remove=False):
         def _try_get_task_data(celery_app, task_id):
-            celery_app.AsyncResult(task_id).get()  # will raise if the task raised an exception
+            async_result = celery_app.AsyncResult(task_id)
+            if async_result.state in ("FAILURE", "RETRY", "REVOKED"):
+                exc = async_result.result
+                if not isinstance(exc, BaseException):
+                    exc = RuntimeError(str(exc))
+                raise exc from RemoteTraceback(async_result.traceback)
 
         def _read_task_data(out_folder, task_id):
             serialized_data = CelerySerialized(folder=out_folder, redis_client=self.redis_client)

{streamlit_octostar_utils-0.4.2.dev16 → streamlit_octostar_utils-0.4.2.dev17}/streamlit_octostar_utils/api_crafter/fastapi.py

@@ -267,6 +267,11 @@ class ErrorLogFilter(logging.Filter):
         return True
 
 
+class RemoteTraceback(Exception):
+    """Holds a formatted traceback string from a remote worker process."""
+    pass
+
+
 class DefaultErrorRoute:
     DEFAULT_STATUS_CODE_MAPPINGS = {
         StarletteHTTPException: lambda exc: exc.status_code,
@@ -329,7 +334,11 @@ class DefaultErrorRoute:
         if len(message) > MAX_ERROR_MESSAGE_BYTES:
             message = message[-MAX_ERROR_MESSAGE_BYTES:]
         try:
-            tcbk = "\n".join(traceback.format_exception(exc))
+            remote_cause = getattr(exc, "__cause__", None)
+            if isinstance(remote_cause, RemoteTraceback):
+                tcbk = str(remote_cause)
+            else:
+                tcbk = "\n".join(traceback.format_exception(exc))
             if len(tcbk) > MAX_ERROR_TRACEBACK_BYTES:
                 tcbk = tcbk[-MAX_ERROR_TRACEBACK_BYTES:]
         except: