Update api_routes_v2.py

api_routes_v2.py

@@ -634,6 +634,149 @@ def upload_stream_to_s3(chat_id: str, file: UploadFile) -> str:
     return s3_uri
 
 
+# ========================
+# GET ALL SESSIONS ENDPOINT
+# ========================
+
+@router.get("/sessions", response_model=Dict[str, Any])
+async def get_all_sessions(
+    limit: int = 100,
+    skip: int = 0,
+    include_stats: bool = False
+):
+    """
+    Get all session IDs from the database with optional pagination
+    
+    Args:
+        limit: Maximum number of sessions to return (default: 100)
+        skip: Number of sessions to skip (for pagination)
+        include_stats: If True, include session statistics
+    
+    Returns:
+        List of sessions with basic info or full stats
+    """
+    try:
+        all_session_ids = session_manager.get_all_session_ids()
+        
+        if not all_session_ids:
+            return {"sessions": [], "pagination": {"total": 0, "returned": 0}}
+        
+        # Apply pagination
+        total_sessions = len(all_session_ids)
+        paginated_ids = all_session_ids[skip:skip + limit]
+        
+        if not include_stats:
+            # Return just session IDs with pagination info
+            sessions_basic = [
+                {
+                    "session_id": sid,
+                    "created_at": None,
+                    "last_activity": None
+                }
+                for sid in paginated_ids
+            ]
+            
+            return {
+                "sessions": sessions_basic,
+                "pagination": {
+                    "total": total_sessions,
+                    "returned": len(sessions_basic),
+                    "limit": limit,
+                    "skip": skip,
+                    "has_more": total_sessions > (skip + limit)
+                }
+            }
+        
+        # Include detailed statistics for each session
+        sessions_with_stats = []
+        for session_id in paginated_ids:
+            session = session_manager.get_session(session_id)
+            if session:
+                # Format datetime objects for JSON serialization
+                created_at = session.get("created_at")
+                last_activity = session.get("last_activity")
+                
+                if isinstance(created_at, datetime):
+                    created_at = created_at.isoformat()
+                if isinstance(last_activity, datetime):
+                    last_activity = last_activity.isoformat()
+                
+                sessions_with_stats.append({
+                    "session_id": session_id,
+                    "user_id": session.get("user_id"),
+                    "created_at": created_at,
+                    "last_activity": last_activity,
+                    "state": session.get("state", "unknown"),
+                    "current_file": session.get("current_file"),
+                    "stats": session.get("stats", {}),
+                    "total_messages": len(session.get("conversation_history", [])),
+                    "pipeline_executions_count": len(session.get("pipeline_executions", []))
+                })
+        
+        return {
+            "sessions": sessions_with_stats,
+            "pagination": {
+                "total": total_sessions,
+                "returned": len(sessions_with_stats),
+                "limit": limit,
+                "skip": skip,
+                "has_more": total_sessions > (skip + limit)
+            }
+        }
+    
+    except Exception as e:
+        raise HTTPException(
+            status_code=500,
+            detail=f"Error retrieving sessions: {str(e)}"
+        )
+
+
+# ========================
+# GET SESSION HISTORY ENDPOINT (with session_id in response)
+# ========================
+
+@router.get("/sessions/{session_id}/history", response_model=Dict[str, Any])
+async def get_session_history(
+    session_id: str,
+    limit: int = 50
+):
+    """
+    Get conversation history for a session
+    
+    Args:
+        session_id: Session identifier
+        limit: Maximum number of messages to return (default: 50)
+    
+    Returns:
+        Session history including session ID
+    """
+    try:
+        # Get the history using session manager
+        history = session_manager.get_session_history(session_id, limit)
+        
+        # Format datetime objects to ISO strings for JSON serialization
+        formatted_history = []
+        for msg in history:
+            msg_copy = msg.copy()
+            if "timestamp" in msg_copy and isinstance(msg_copy["timestamp"], datetime):
+                msg_copy["timestamp"] = msg_copy["timestamp"].isoformat()
+            formatted_history.append(msg_copy)
+        
+        # Return with session_id included
+        return {
+            "session_id": session_id,
+            "history": formatted_history,
+            "count": len(formatted_history),
+            "limit": limit
+        }
+    
+    except Exception as e:
+        raise HTTPException(
+            status_code=500,
+            detail=f"Error retrieving session history: {str(e)}"
+        )
+
+
 # ========================
 # UNIFIED CHAT (non-streaming)
 # ========================