From f6baa0cfa0467b0cf44977b36d1fa3fbd38c9646 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?V=C3=ADctor=20Mayoral=20Vilches?= Date: Wed, 28 May 2025 09:06:30 +0200 Subject: [PATCH] Add banner to replay MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Víctor Mayoral Vilches --- tools/replay.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tools/replay.py b/tools/replay.py index 5eb70b40..fd34a63d 100644 --- a/tools/replay.py +++ b/tools/replay.py @@ -48,6 +48,7 @@ from cai.util import ( color ) from cai.sdk.agents.run_to_jsonl import get_token_stats, load_history_from_jsonl +from cai.repl.ui.banner import display_banner # Initialize console object for rich printing console = Console() @@ -373,6 +374,10 @@ Examples: def main(): """Main function to process JSONL files and generate replay output.""" + # Display banner + display_banner(console) + print("\n") + # Parse command line arguments args = parse_arguments() @@ -472,4 +477,4 @@ def main(): if __name__ == "__main__": - main() \ No newline at end of file + main()