Handle the case where we may end before we get any output

ajhai · ajhai · commit 137b785b0f3d · 2024-11-13T23:48:53.000-08:00
diff --git a/llmstack/apps/runner/agent_actor.py b/llmstack/apps/runner/agent_actor.py
@@ -130,7 +130,11 @@ async def _process_output(self):
                         {
                             "output": {
                                 **self._agent_outputs,
-                                "output": self._stitched_data["agent"][str(message_index)].data.content[0].data,
+                                "output": (
+                                    self._stitched_data["agent"][str(message_index)].data.content[0].data
+                                    if str(message_index) in self._stitched_data["agent"]
+                                    else ""
+                                ),
                             },
                             "chunks": self._stitched_data,
                         }

Original file line number	Diff line number	Diff line change
`@@ -130,7 +130,11 @@ async def _process_output(self):`
`130`	`130`	`{`
`131`	`131`	`"output": {`
`132`	`132`	`**self._agent_outputs,`
`133`		`- "output": self._stitched_data["agent"][str(message_index)].data.content[0].data,`
	`133`	`+ "output": (`
	`134`	`+ self._stitched_data["agent"][str(message_index)].data.content[0].data`
	`135`	`+ if str(message_index) in self._stitched_data["agent"]`
	`136`	`+ else ""`
	`137`	`+ ),`
`134`	`138`	`},`
`135`	`139`	`"chunks": self._stitched_data,`
`136`	`140`	`}`