Upload app.py

app.py

@@ -65,6 +65,13 @@ class HealthResponse(BaseModel):
     message: str
     components: Dict[str, str]
 
+class SessionListResponse(BaseModel):
+    sessions: List[Dict[str, Any]]
+    total_count: int
+    limit: int
+    offset: int
+    has_more: bool
+
 # Global instances
 rag_system: RAGSystem = None
 session_manager: SessionManager = None
@@ -189,6 +196,33 @@ async def get_chat_history(session_id: str, limit: int = 10):
         logger.error(f"Error getting chat history: {e}")
         raise HTTPException(status_code=500, detail=f"Failed to get chat history: {str(e)}")
 
+@app.get("/sessions", response_model=SessionListResponse)
+async def list_all_sessions(limit: int = 100, offset: int = 0):
+    """List all sessions with pagination"""
+    if not session_manager:
+        raise HTTPException(status_code=503, detail="Session manager not initialized")
+    
+    try:
+        # Validate parameters
+        if limit < 1 or limit > 1000:
+            raise HTTPException(status_code=400, detail="Limit must be between 1 and 1000")
+        
+        if offset < 0:
+            raise HTTPException(status_code=400, detail="Offset must be non-negative")
+        
+        result = session_manager.list_all_sessions(limit=limit, offset=offset)
+        
+        if "error" in result:
+            raise HTTPException(status_code=500, detail=result["error"])
+        
+        return SessionListResponse(**result)
+        
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.error(f"Error listing sessions: {e}")
+        raise HTTPException(status_code=500, detail=f"Failed to list sessions: {str(e)}")
+
 @app.post("/chat", response_model=ChatResponse)
 async def chat(request: ChatRequest):
     """Main chat endpoint for legal questions with session support"""