Update research_manager.py
Browse files- research_manager.py +3 -10
research_manager.py
CHANGED
@@ -13,8 +13,6 @@ class ResearchManagerAgent:
|
|
13 |
query: str,
|
14 |
clarifying_questions: list[str],
|
15 |
clarifying_answers: list[str],
|
16 |
-
send_email_flag: bool = False,
|
17 |
-
recipient_email: Optional[str] = None,
|
18 |
):
|
19 |
""" Run the deep research process using user-provided clarification answers. """
|
20 |
trace_id = gen_trace_id()
|
@@ -35,14 +33,9 @@ class ResearchManagerAgent:
|
|
35 |
yield "Searches complete, writing report..."
|
36 |
report = await self.write_report(query, search_results)
|
37 |
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
yield "Email sent"
|
42 |
-
else:
|
43 |
-
yield "Skipping email step"
|
44 |
-
|
45 |
-
yield "Email sent"
|
46 |
yield report.markdown_report
|
47 |
|
48 |
async def plan_searches(self, query: str, questions: list[str], answers: list[str]) -> WebSearchPlan:
|
|
|
13 |
query: str,
|
14 |
clarifying_questions: list[str],
|
15 |
clarifying_answers: list[str],
|
|
|
|
|
16 |
):
|
17 |
""" Run the deep research process using user-provided clarification answers. """
|
18 |
trace_id = gen_trace_id()
|
|
|
33 |
yield "Searches complete, writing report..."
|
34 |
report = await self.write_report(query, search_results)
|
35 |
|
36 |
+
yield "Report written, sending email..."
|
37 |
+
await self.send_email(report)
|
38 |
+
yield "Email sent, research complete"
|
|
|
|
|
|
|
|
|
|
|
39 |
yield report.markdown_report
|
40 |
|
41 |
async def plan_searches(self, query: str, questions: list[str], answers: list[str]) -> WebSearchPlan:
|