switch to aide offical method of timeout

Author: raycai420

aide/interpreter.py

@@ -305,17 +305,16 @@ def run(self, code: str, reset_session=True) -> ExecutionResult:
         # read all stdout/stderr from child up to the EOF marker
         # waiting until the queue is empty is not enough since
         # the feeder thread in child might still be adding to the queue
-        drain_start = time.time()
+        start_collect = time.time()
         while not self.result_outq.empty() or not output or output[-1] != "<|EOF|>":
-            logger.debug("[Parent] starting drain; qsize=%d", self.result_outq.qsize())
             try:
-                output.append(self.result_outq.get(timeout=5))
+                # Add 5-second timeout for output collection
+                if time.time() - start_collect > 5:
+                    logger.warning("Output collection timed out")
+                    break
+                output.append(self.result_outq.get(timeout=1))
             except queue.Empty:
-                logger.error(
-                    "[Parent] drain timeout - q empty, no EOF after %ds",
-                    time.time() - drain_start,
-                )
-                break
+                continue
         output.pop()  # remove the EOF marker
 
         e_cls_name, exc_info, exc_stack = state[1:]