File tree Expand file tree Collapse file tree 1 file changed +13
-1
lines changed
src/llama_stack_client/lib/agents Expand file tree Collapse file tree 1 file changed +13
-1
lines changed Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments