Agent Get Pending Message (flyteorg#1834)

Signed-off-by: Future Outlier <[email protected]> Signed-off-by: Kevin Su <[email protected]> Co-authored-by: Future Outlier <[email protected]> Co-authored-by: Kevin Su <[email protected]> Co-authored-by: Kevin Su <[email protected]> Signed-off-by: Rafael Raposo <[email protected]>
RRap0so · Dec 15, 2023 · 774f857 · 774f857
1 parent fc86ee9
commit 774f857
Show file tree

Hide file tree

Showing 6 changed files with 224 additions and 96 deletions.
diff --git a/flytekit/extend/backend/base_agent.py b/flytekit/extend/backend/base_agent.py
@@ -215,7 +215,7 @@ async def _get(self, resource_meta: bytes) -> GetTaskResponse:
             else:
                 res = self._agent.get(grpc_ctx, resource_meta)
             state = res.resource.state
-            logger.info(f"Task state: {state}")
+            logger.info(f"Task state: {state}, State message: {res.resource.message}")
         return res
 
     def signal_handler(self, resource_meta: bytes, signum: int, frame: FrameType) -> typing.Any:

diff --git a/plugins/flytekit-papermill/dev-requirements.in b/plugins/flytekit-papermill/dev-requirements.in
@@ -1,4 +1,4 @@
-flyteidl>=1.3.0
+flyteidl>=1.10.0
 -e file:../../.#egg=flytekitplugins-pod&subdirectory=plugins/flytekit-k8s-pod
 -e file:../../.#egg=flytekitplugins-spark&subdirectory=plugins/flytekit-spark
 -e file:../../.#egg=flytekitplugins-awsbatch&subdirectory=plugins/flytekit-aws-batch