Spaces:
Sleeping
Sleeping
Commit
·
2fc4e76
1
Parent(s):
ea1e68e
Update services/session_manager.py
Browse files- 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,
|