Add Qdrant stats endpoint for DataView

api.py

@@ -12,6 +12,7 @@ from src.graph import get_graph
 from src.graph import get_graph
 from src.utils.llm import set_large_model_override, get_available_large_models
 from src.utils.firebase import delete_user_permanently, verify_id_token
+from src.utils.ingestion import get_qdrant_client
 from fastapi import Header
 
 
@@ -201,6 +202,44 @@ async def delete_user(req: DeleteUserRequest, authorization: str = Header(None))
         raise HTTPException(500, str(e))
 
 
+@app.get("/api/qdrant/stats")
+async def get_qdrant_stats():
+    """Get Qdrant database statistics."""
+    try:
+        client = get_qdrant_client()
+        collections = client.get_collections().collections
+        
+        stats = {
+            "collections": [],
+            "total_vectors": 0,
+            "total_size_bytes": 0
+        }
+        
+        for collection in collections:
+            try:
+                collection_info = client.get_collection(collection.name)
+                vector_count = collection_info.points_count or 0
+                
+                stats["collections"].append({
+                    "name": collection.name,
+                    "vectors": vector_count,
+                    "status": "healthy"
+                })
+                stats["total_vectors"] += vector_count
+            except Exception as e:
+                print(f"Error getting collection {collection.name}: {e}")
+                stats["collections"].append({
+                    "name": collection.name,
+                    "vectors": 0,
+                    "status": "error"
+                })
+        
+        return stats
+    except Exception as e:
+        print(f"Qdrant stats error: {e}")
+        raise HTTPException(500, f"Failed to get Qdrant stats: {str(e)}")
+
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=8000)