minor update

Author: nanjiangwill

agent/display.py

@@ -100,12 +100,17 @@ def __init__(self, total_repos: int):
         )
         self.layout["progress"].split_row(
             Layout(name="pbar", ratio=4),
+            Layout(name="time", ratio=1),
             Layout(name="money", ratio=1),
         )
 
         self.layout["progress"]["pbar"].update(
             Panel(self.overall_progress, title="Overall Progress", border_style="blue")
         )
+        self.time_display = Text("Time Spent So Far: 0s", justify="center")
+        self.layout["progress"]["time"].update(
+            Panel(self.time_display, title="$$$$", border_style="blue")
+        )
         self.money_display = Text("Money Spent So Far: $0.00", justify="center")
         self.layout["progress"]["money"].update(
             Panel(self.money_display, title="$$$$", border_style="blue")
@@ -190,6 +195,24 @@ def update_agent_display(
                 Panel(text, title=title, border_style="blue")
             )
 
+    def update_time_display(self, time_in_seconds: int) -> None:
+        """Update the time display with the given time."""
+        days, remainder = divmod(time_in_seconds, 86400)
+        hours, remainder = divmod(remainder, 3600)
+        minutes, seconds = divmod(remainder, 60)
+        if days > 0:
+            time_str = f"{days}d {hours:02d}h {minutes:02d}m {seconds:02d}s"
+        elif hours > 0:
+            time_str = f"{hours:02d}h {minutes:02d}m {seconds:02d}s"
+        elif minutes > 0:
+            time_str = f"{minutes:02d}m {seconds:02d}s"
+        else:
+            time_str = f"{seconds:02d}s"
+        self.time_display = Text(f"Time Spent So Far: {time_str}", justify="center")
+        self.layout["progress"]["time"].update(
+            Panel(self.time_display, title="Time", border_style="blue")
+        )
+
     def update_backend_display(self, backend: str) -> None:
         """Update the backend display with the given backend."""
         self.backend_display = Text(f"Backend Using: {backend}", justify="center")

agent/run_agent.py

@@ -66,6 +66,10 @@ def run_agent_for_repo(
     repo_path = os.path.join(repo_base_dir, repo_name)
     repo_path = os.path.abspath(repo_path)
 
+    # TODO: remove this to make web3.py work
+    if repo_name == "web3-py":
+        repo_path = repo_path.replace("web3-py", "web3.py")
+
     src_dir = os.path.join(repo_path, example["src_dir"])
 
     try:
@@ -214,7 +218,11 @@ def run_agent(
         ]
         display.set_not_started_repos(not_started_repos)
 
-        display.update_repo_progress_num(display_repo_progress_num)
+        start_time = time.time()
+
+        display.update_repo_progress_num(
+            min(display_repo_progress_num, max_parallel_repos)
+        )
         display.update_backend_display(backend)
         display.update_log_dir_display(log_dir)
         display.update_agent_display(
@@ -248,6 +256,7 @@ def run_agent(
                     )
                     results.append(result)
 
+                last_time_update = 0
                 while any(not r.ready() for r in results):
                     try:
                         while not update_queue.empty():
@@ -268,6 +277,14 @@ def run_agent(
                                 )
                     except queue.Empty:
                         pass
+
+                    # Update time display every second
+                    current_time = time.time()
+                    if current_time - last_time_update >= 1:
+                        elapsed_time = int(current_time - start_time)
+                        display.update_time_display(elapsed_time)
+                        last_time_update = current_time
+
                     time.sleep(0.1)  # Small delay to prevent busy-waiting
 
                 # Final update after all repos are processed
@@ -286,5 +303,9 @@ def run_agent(
                         repo_name, file_name, money_spent = data
                         display.update_money_display(repo_name, file_name, money_spent)
 
+                # Final time update
+                elapsed_time = int(time.time() - start_time)
+                display.update_time_display(elapsed_time)
+
                 for result in results:
                     result.get()