Update research_agent/agent.py

research_agent/agent.py

@@ -149,4 +149,4 @@ def write_report_stream(config: AgentConfig, writer_model, tavily_client, embedd
     final_bibliography = "\n".join(f"- {url}" for url in sorted(list(report_state["all_source_urls"])))
     report_state["full_report_text"] += f"## Master Bibliography\n\n{final_bibliography}"
 
-    return report_state["full_report_text"]
+    yield report_state["full_report_text"]