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

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

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

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

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

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

@@ -891,6 +891,13 @@ class CeleryExecutor(object):
             self.get_thread_pool, _poll_task_state, self.app, task_id
         )
 
+    async def get_task_info(self, task_id):
+        def _get_info(celery_app, task_id):
+            return celery_app.AsyncResult(task_id).info
+        return await asyncio.get_running_loop().run_in_executor(
+            self.get_thread_pool, _get_info, self.app, task_id,
+        )
+
     async def get_task_result(self, task_id, remove=False):
         def _try_get_task_data(celery_app, task_id):
             async_result = celery_app.AsyncResult(task_id)
@@ -1018,10 +1025,12 @@ class FastAPICeleryTaskRoute(Route):
                 data = {"task_state": "UNKNOWN", "task_id": task_id}
             elif state in ["AWAITING", "STARTED"]:
                 data = {"task_state": state, "task_id": task_id}
+                if state == "STARTED":
+                    info = await self.celery_executor.get_task_info(task_id)
+                    if info and isinstance(info, dict):
+                        data["progress"] = info
             elif state == "SUCCESS":
                 data = {"task_state": state, "task_id": task_id, "data": result}
-            elif state == "STARTED":
-                data = {"task_status": state, "task_id": task_id}
             else:
                 raise ValueError(f"Unknown task state {state}!")
             return CommonModels.DataResponseModel(data=data)

{streamlit_octostar_utils-0.5.0.dev16 → streamlit_octostar_utils-0.5.0.dev17}/streamlit_octostar_utils/api_crafter/nifi.py

@@ -1446,6 +1446,7 @@ class NifiEntity(object):
         source_entity_relationship=SOURCE_FRAGMENT_ENTITY_RELATIONSHIP,
     ):
         fields = {
+            **{k: v for k, v in self.record if k.startswith("fragment") and v is not None},
             **fields,
             "os_parent_uid": self.record["os_entity_uid"],
             "source_entity_uid": source_entity_uid,