diff --git a/bitia/__main__.py b/bitia/__main__.py index 8855fb7bb04ce3492bc4e4be91149e03e2be2b39..fa018422a1a49269d0d3ead2c807a378ea4c16d3 100644 --- a/bitia/__main__.py +++ b/bitia/__main__.py @@ -96,8 +96,8 @@ def submit_pipeline(user_input: str, server: str = DEFAULT_SERVER): def run_pipeline(user_input: str, server: str = DEFAULT_SERVER): """Run a pipeline""" pipeline_zip = user_input_to_pipeline(user_input) - containers = submit_job(pipeline_zip, server=server) - return _log_container(containers, server=server) + res = submit_job(pipeline_zip, server=server) + return _log_container(res["container"], server=server) @app.command() @@ -106,10 +106,10 @@ def version(): print(bitia.version()) -def _log_container(containers: T.List[T.Any], server: str): - for container in containers: - for line in _fetch_logs(container["Name"], server=server): - print(line.decode().rstrip()) +def _log_container(container: str, server: str): + logger.info(f"Fetching logs from {container}") + for line in _fetch_logs(container, server=server): + print(line.decode().rstrip()) def _check_server_status(server: str) -> int: @@ -240,7 +240,7 @@ def _fetch_logs(container: str, *, server): logger.info(f"Fetching logs for container `{container}`") return g_session.post( f"{server}/logs", - json=dict(container=container, follow=True, tail=-1), + json=dict(container=container, follow=True), stream=True, )