From ade4511a8718ee090981f36c28d9e3d4da2b900c Mon Sep 17 00:00:00 2001 From: ThomasTaroni Date: Fri, 25 Apr 2025 18:39:33 +0200 Subject: [PATCH] Add report generation step to the main async process The `generate_report` method is now explicitly called within the main process, ensuring the researcher's report generation is initiated properly. Updated its return type annotation for clarity in `deepresearch.py`. --- src/main.py | 1 + src/phoenix_technologies/gptresearch/deepresearch.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/main.py b/src/main.py index 3bbc6cf..b223552 100644 --- a/src/main.py +++ b/src/main.py @@ -37,6 +37,7 @@ async def get_report_endpoint(request: ReportRequest): try: # Call the asynchronous get_report function generator = ReportGenerator(request.query, request.report_type) + await generator.generate_report() async for log in generator: yield log except Exception as e: diff --git a/src/phoenix_technologies/gptresearch/deepresearch.py b/src/phoenix_technologies/gptresearch/deepresearch.py index add36e3..d8e0adc 100644 --- a/src/phoenix_technologies/gptresearch/deepresearch.py +++ b/src/phoenix_technologies/gptresearch/deepresearch.py @@ -27,7 +27,7 @@ class ReportGenerator: self.researcher = GPTResearcher(query, report_type, websocket=self.custom_logs_handler) - async def generate_report(self): + async def generate_report(self) -> CustomLogsHandler: """ Conducts research and generates the report along with additional information. """