From aefba2046a6e056ccddb7e5e48a4551dc3a6cf82 Mon Sep 17 00:00:00 2001
From: XianBW <36835909+XianBW@users.noreply.github.com>
Date: Wed, 25 Sep 2024 14:25:25 +0800
Subject: [PATCH] change docker logs print (#326)

---
 rdagent/utils/env.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/rdagent/utils/env.py b/rdagent/utils/env.py
index a6405617..91bb621c 100644
--- a/rdagent/utils/env.py
+++ b/rdagent/utils/env.py
@@ -24,8 +24,9 @@ import docker.models.containers
 from pydantic import BaseModel
 from pydantic_settings import BaseSettings
 from rich import print
-from rich.progress import Progress, TextColumn
+from rich.progress import Progress, SpinnerColumn, TextColumn
 from rich.rule import Rule
+from rich.table import Table
 
 from rdagent.log import rdagent_logger as logger
 
@@ -203,7 +204,7 @@ class DockerEnv(Env[DockerConf]):
             )
             if isinstance(resp_stream, str):
                 logger.info(resp_stream)
-            with Progress(TextColumn("{task.description}")) as p:
+            with Progress(SpinnerColumn(), TextColumn("{task.description}")) as p:
                 task = p.add_task("[cyan]Building image...")
                 for part in resp_stream:
                     status_dict = json.loads(part)
@@ -308,6 +309,13 @@ class DockerEnv(Env[DockerConf]):
             )
             logs = container.logs(stream=True)
             print(Rule("[bold green]Docker Logs Begin[/bold green]", style="dark_orange"))
+            table = Table(title="Run Info", show_header=False)
+            table.add_column("Key", style="bold cyan")
+            table.add_column("Value", style="bold magenta")
+            table.add_row("Entry", entry)
+            table.add_row("Env", "\n".join(f"{k}:{v}" for k, v in env.items()))
+            table.add_row("Volumns", "\n".join(f"{k}:{v}" for k, v in volumns.items()))
+            print(table)
             for log in logs:
                 decoded_log = log.strip().decode()
                 print(decoded_log)