🚑 quickfix: fixed the progress bar display

MLSysOps · Nov 19, 2024 · 0e1ca7e · 0e1ca7e
1 parent 905b1b5
commit 0e1ca7e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/mle/cli.py b/mle/cli.py
@@ -217,7 +217,11 @@ def chat(model, build_mem):
 
             for file_path in source_files:
                 raw_code = read_file(file_path)
-                progress.update(process_task, advance=1, description=f"Adding memory...")
+                progress.update(
+                    process_task,
+                    advance=1,
+                    description=f"Adding {os.path.basename(file_path)} to memory..."
+                )
 
                 chunks = chunker.chunk(raw_code, token_limit=100)
                 memory.add(