fix(internal_user_endpoints.py): add sort functionality to user list endpoint

litellm/proxy/management_endpoints/internal_user_endpoints.py

@@ -925,9 +925,16 @@ async def get_users(
     page_size: int = fastapi.Query(
         default=25, ge=1, le=100, description="Number of items per page"
     ),
+    sort_by: Optional[str] = fastapi.Query(
+        default=None,
+        description="Column to sort by (e.g. 'user_id', 'user_email', 'created_at', 'spend')",
+    ),
+    sort_order: str = fastapi.Query(
+        default="asc", description="Sort order ('asc' or 'desc')"
+    ),
 ):
     """
-    Get a paginated list of users with filtering options.
+    Get a paginated list of users with filtering and sorting options.
 
     Parameters:
         role: Optional[str]
@@ -946,9 +953,10 @@ async def get_users(
             The page number to return
         page_size: int
             The number of items per page
-
-    Returns:
-        UserListResponse with filtered and paginated users
+        sort_by: Optional[str]
+            Column to sort by (e.g. 'user_id', 'user_email', 'created_at', 'spend')
+        sort_order: Optional[str]
+            Sort order ('asc' or 'desc')
     """
     from litellm.proxy.proxy_server import prisma_client
 
@@ -986,11 +994,43 @@ async def get_users(
 
     ## Filter any none fastapi.Query params - e.g. where_conditions: {'user_email': {'contains': Query(None), 'mode': 'insensitive'}, 'teams': {'has': Query(None)}}
     where_conditions = {k: v for k, v in where_conditions.items() if v is not None}
+
+    # Build order_by conditions
+    order_by: Dict[str, str] = {}
+    if sort_by:
+        # Validate sort_by is a valid column
+        valid_columns = [
+            "user_id",
+            "user_email",
+            "created_at",
+            "spend",
+            "user_alias",
+            "user_role",
+        ]
+        if sort_by not in valid_columns:
+            raise HTTPException(
+                status_code=400,
+                detail={
+                    "error": f"Invalid sort column. Must be one of: {', '.join(valid_columns)}"
+                },
+            )
+
+        # Validate sort_order
+        if sort_order.lower() not in ["asc", "desc"]:
+            raise HTTPException(
+                status_code=400,
+                detail={"error": "Invalid sort order. Must be 'asc' or 'desc'"},
+            )
+
+        order_by[sort_by] = sort_order.lower()
+
     users = await prisma_client.db.litellm_usertable.find_many(
         where=where_conditions,
         skip=skip,
         take=page_size,
-        order={"created_at": "desc"},
+        order=order_by
+        if order_by
+        else {"created_at": "desc"},  # Default to created_at desc if no sort specified
     )
 
     # Get total count of user rows