Bug fix: Unwrap final_answer in Gradio_UI
#413
by
m-bano
- opened
- Gradio_UI.py +1 -1
Gradio_UI.py
CHANGED
|
@@ -154,7 +154,7 @@ def stream_to_gradio(
|
|
| 154 |
yield message
|
| 155 |
|
| 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(
|
|
|
|
| 154 |
yield message
|
| 155 |
|
| 156 |
final_answer = step_log # Last log is the run's final_answer
|
| 157 |
+
final_answer = handle_agent_output_types(final_answer.final_answer)
|
| 158 |
|
| 159 |
if isinstance(final_answer, AgentText):
|
| 160 |
yield gr.ChatMessage(
|