Revert "Remove @SortDimensionId"

Revert submission 25032702-DocumentsUI-lib

Reason for revert: broke document browsing

Reverted changes: /q/submissionid:25032702-DocumentsUI-lib

Bug: 307873144
Bug: 307882692
Change-Id: Ie0eefdf383fa5cb5339bc5b67608cb3c42057788
6 files changed