redhairedshanks1 commited on
Commit
2fc4e76
·
1 Parent(s): ea1e68e

Update services/session_manager.py

Browse files
Files changed (1) hide show
  1. services/session_manager.py +18 -0
services/session_manager.py CHANGED
@@ -285,6 +285,24 @@ class SessionManager:
285
  history = session.get("conversation_history", [])
286
  return history[-limit:] if len(history) > limit else history
287
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
288
  def get_pipeline_executions(
289
  self,
290
  session_id: Optional[str] = None,
 
285
  history = session.get("conversation_history", [])
286
  return history[-limit:] if len(history) > limit else history
287
 
288
+ def get_all_session_ids(self) -> List[str]:
289
+ """
290
+ Get all session IDs from the database
291
+
292
+ Returns:
293
+ List of session IDs
294
+ """
295
+ if self.sessions_col is None:
296
+ return []
297
+
298
+ try:
299
+ sessions = self.sessions_col.find({}, {"session_id": 1, "_id": 0})
300
+ session_ids = [session["session_id"] for session in sessions]
301
+ return session_ids
302
+ except Exception as e:
303
+ print(f"⚠️ Error retrieving session IDs: {e}")
304
+ return []
305
+
306
  def get_pipeline_executions(
307
  self,
308
  session_id: Optional[str] = None,