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.
         """