jcleee commited on
Commit
45ae27a
·
verified ·
1 Parent(s): 90e5bc9

Update Gradio_UI.py

Browse files
Files changed (1) hide show
  1. Gradio_UI.py +7 -3
Gradio_UI.py CHANGED
@@ -155,11 +155,13 @@ def stream_to_gradio(
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(
161
  role="assistant",
162
- content=final_answer.to_string().strip(),
163
  )
164
  elif isinstance(final_answer, AgentImage):
165
  yield gr.ChatMessage(
@@ -172,7 +174,9 @@ def stream_to_gradio(
172
  content={"path": final_answer.to_string(), "mime_type": "audio/wav"},
173
  )
174
  else:
175
- yield gr.ChatMessage(role="assistant", content=f"**Final answer:** {str(final_answer)}")
 
 
176
 
177
 
178
  class GradioUI:
 
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
+ # Strip any wrapper or formatting for exact match evaluation
160
  if isinstance(final_answer, AgentText):
161
+ content = final_answer.to_string().strip()
162
  yield gr.ChatMessage(
163
  role="assistant",
164
+ content=content,
165
  )
166
  elif isinstance(final_answer, AgentImage):
167
  yield gr.ChatMessage(
 
174
  content={"path": final_answer.to_string(), "mime_type": "audio/wav"},
175
  )
176
  else:
177
+ # Fall back to raw string with no prefix
178
+ yield gr.ChatMessage(role="assistant", content=str(final_answer).strip())
179
+
180
 
181
 
182
  class GradioUI: