From f0573a09d9ef956370e28d01db4577fd8e0f9767 Mon Sep 17 00:00:00 2001
From: "jinli.yl" <jinli.yl@alibaba-inc.com>
Date: Wed, 14 Jan 2026 00:02:26 +0800
Subject: [PATCH] feat(bench): add progress bar for session processing in
 halumem evaluation

- Integrated tqdm library for progress tracking
- Added progress bar display for user sessions processing
- Implemented session-by-session progress updates with custom description
- Maintained existing session processing logic while adding visual feedback
- Preserved all original functionality including logging and data handling
---
 bench/halumem/eval_baseline_simple.py | 29 +++++++++++++++++----------
 1 file changed, 18 insertions(+), 11 deletions(-)

diff --git a/bench/halumem/eval_baseline_simple.py b/bench/halumem/eval_baseline_simple.py
index 441ffe28..f06d69f5 100644
--- a/bench/halumem/eval_baseline_simple.py
+++ b/bench/halumem/eval_baseline_simple.py
@@ -24,6 +24,7 @@ from pathlib import Path
 from typing import Any
 
 from loguru import logger
+from tqdm import tqdm
 
 from eval_tools import evaluation_for_question2
 from llms import llm_request_for_json
@@ -425,17 +426,23 @@ class HaluMemBaselineEvaluator:
         
         logger.info(f"Processing user: {user_name}")
         
-        for idx, session in enumerate(user_data["sessions"]):
-            logger.info(f"  Session {idx + 1}/{len(user_data['sessions'])}")
-            
-            session_data = await self.process_session(
-                session=session,
-                session_id=idx,
-                user_name=user_name,
-                uuid=uuid
-            )
-            
-            self.file_manager.save_session(user_name, idx, session_data)
+        # Create progress bar for sessions
+        with tqdm(
+            total=len(user_data["sessions"]),
+            desc=f"Sessions [{user_name}]",
+            leave=False,
+            ncols=100
+        ) as pbar:
+            for idx, session in enumerate(user_data["sessions"]):
+                session_data = await self.process_session(
+                    session=session,
+                    session_id=idx,
+                    user_name=user_name,
+                    uuid=uuid
+                )
+                
+                self.file_manager.save_session(user_name, idx, session_data)
+                pbar.update(1)
         
         return {"uuid": uuid, "user_name": user_name, "status": "ok"}