Update app.py

app.py

@@ -333,15 +333,18 @@ def stream_agent(question: str,
                         yield round_state["final_text"] or " ", round_state["summary_text"] or " ", "\n".join(log_lines[-400:])
 
                     # Log tool call argument deltas (optional)
-                    elif etype in ("response.function_call_arguments.delta", "response.tool_call_arguments.delta"):
+                    elif etype in ("response.mcp_call_arguments.delta", "response.tool_call_arguments.delta"):
                         delta = getattr(event, "delta", None)
                         if delta:
                             log_lines.append(str(delta))
+                            round_state["summary_text"] += "\nQuery call: " + event.delta + "\n"
+                            yield round_state["final_text"] or " ", round_state["summary_text"] or " ", "\n".join(log_lines[-400:])
 
                     # Capture tool RESULT text and try to parse indices
-                    elif etype.startswith("response.tool_result"):
+                    elif etype.startswith("response.output_item.done"):
                         print("A tool output was detected")
-                        delta_text = getattr(event, "delta", "") or getattr(event, "output_text", "")
+                        print(event)
+                        delta_text = getattr(event.item, "output", "")
                         if delta_text:
                             tool_result_buffer += str(delta_text)
                             parsed_now = _maybe_parse_indices(tool_result_buffer)