Null check for streamlit visualizer (#24)

2024-03-04 11:35:57 -08:00
parent 0917fa0702
commit d9729985b1
1 changed files with 2 additions and 1 deletions
--- 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: