diff --git a/src/main.py b/src/main.py index 02ca60f..0b432c8 100644 --- a/src/main.py +++ b/src/main.py @@ -42,7 +42,7 @@ async def get_report_endpoint(request: ReportRequest): generator.generate_report() yield "Report generation completed successfully!\n" index = 0 - while not generator.is_complete(): + while not generator.complete: # If there are more logs to send, yield them if index < len(custom_logs_handler.logs): log_entry = custom_logs_handler.logs[index] @@ -52,7 +52,7 @@ async def get_report_endpoint(request: ReportRequest): # Wait briefly to avoid aggressive looping await asyncio.sleep(0.1) # Stop if processing is complete and no more logs remain - if generator.researcher.is_complete(): + if generator.researcher.complete: break except Exception as e: diff --git a/src/phoenix_technologies/gptresearch/deepresearch.py b/src/phoenix_technologies/gptresearch/deepresearch.py index 77ad753..61f1aec 100644 --- a/src/phoenix_technologies/gptresearch/deepresearch.py +++ b/src/phoenix_technologies/gptresearch/deepresearch.py @@ -28,9 +28,6 @@ class ReportGenerator: def init(self) -> CustomLogsHandler: return self.custom_logs_handler - def is_complete(self): - return self.complete - async def generate_report(self) -> None: """ Conducts research and generates the report along with additional information.