From d9729985b11bc945a23e91b709eca2cf8fe127b8 Mon Sep 17 00:00:00 2001 From: Kerem Yilmaz Date: Mon, 4 Mar 2024 11:35:57 -0800 Subject: [PATCH] Null check for streamlit visualizer (#24) --- streamlit_app/visualizer/api.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/streamlit_app/visualizer/api.py b/streamlit_app/visualizer/api.py index c080fb83..a2104e9a 100644 --- a/streamlit_app/visualizer/api.py +++ b/streamlit_app/visualizer/api.py @@ -63,7 +63,8 @@ class SkyvernClient: response = requests.get(url, params=params, headers=headers) steps = response.json() for step in steps: - step["output"]["actions_and_results"] = json.dumps(step["output"]["actions_and_results"]) + if step.output and "actions_and_results" in step.output: + step["output"]["actions_and_results"] = json.dumps(step["output"]["actions_and_results"]) return steps def get_agent_task_video_artifact(self, task_id: str) -> dict[str, Any] | None: