nicolagheza commited on
Commit
486d40a
·
verified ·
1 Parent(s): 01c3cf1

Update Gradio_UI.py

Browse files
Files changed (1) hide show
  1. Gradio_UI.py +3 -0
Gradio_UI.py CHANGED
@@ -156,6 +156,9 @@ def stream_to_gradio(
156
  final_answer = step_log # Last log is the run's final_answer
157
  final_answer = handle_agent_output_types(final_answer)
158
 
 
 
 
159
  if isinstance(final_answer, AgentText):
160
  yield gr.ChatMessage(
161
  role="assistant",
 
156
  final_answer = step_log # Last log is the run's final_answer
157
  final_answer = handle_agent_output_types(final_answer)
158
 
159
+ if final_answer.final_answer:
160
+ final_answer = final_answer.final_answer
161
+
162
  if isinstance(final_answer, AgentText):
163
  yield gr.ChatMessage(
164
  role="assistant",