mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-29 00:17:44 +00:00
change docker logs print (#326)
This commit is contained in:
+10
-2
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user