Skip to content

Commit 41efaa8

Browse files
committed
fix: print error
# What does this PR do? ## Test Plan
1 parent b664564 commit 41efaa8

File tree

1 file changed

+13
-1
lines changed
  • src/llama_stack_client/lib/agents

1 file changed

+13
-1
lines changed

src/llama_stack_client/lib/agents/agent.py

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -257,7 +257,19 @@ def create_turn(
257257
chunks = [x for x in self._create_turn_streaming(messages, session_id, toolgroups, documents)]
258258
if not chunks:
259259
raise Exception("Turn did not complete")
260-
return chunks[-1].event.payload.turn
260+
261+
last_chunk = chunks[-1]
262+
try:
263+
return last_chunk.event.payload.turn
264+
except AttributeError:
265+
if hasattr(last_chunk, "error"):
266+
if "message" in last_chunk.error:
267+
error_msg = last_chunk.error["message"]
268+
else:
269+
error_msg = str(last_chunk.error)
270+
else:
271+
error_msg = str(last_chunk)
272+
raise RuntimeError(f"Turn did not complete. Error: {error_msg}") from None
261273

262274
def _create_turn_streaming(
263275
self,

0 commit comments

Comments
 (0)