Update api_routes_v2.py

api_routes_v2.py

@@ -86,11 +86,13 @@ def _save_conversation_to_s3(session_id: str, messages: List[Dict[str, Any]]):
             Body=json.dumps(messages, ensure_ascii=False),
             ContentType="application/json"
         )
-        # Update MongoDB with reference (idempotent)
+        # Update MongoDB with reference and stats (idempotent)
         session_manager.update_session(session_id, {
             "conversation_s3_key": key,
             "last_message_count": len(messages),
-            "last_updated": datetime.utcnow().isoformat() + "Z"
+            "last_activity": datetime.utcnow().isoformat() + "Z",
+            "last_updated": datetime.utcnow().isoformat() + "Z",
+            "stats.message_count": len(messages)
         })
     except Exception as e:
         print(f"CRITICAL ERROR saving conversation to S3: {e}")
@@ -819,7 +821,7 @@ async def get_all_sessions(
                     # REMOVED: current_file (not needed for session list)
                     "chat_name": session.get("chat_name"),  # CHANGE: added field
                     "stats": session.get("stats", {}),
-                    "total_messages": len(session.get("conversation_history", [])),
+                    "total_messages": session.get("last_message_count", 0),
                     "pipeline_executions_count": len(session.get("pipeline_executions", []))
                 })