You should be able to use any M.2-based SSD that comes in the 22mm x 42mm package. While you have this physical size limit for the M.2 SSD, you should not have any limit on the amount of storage provided by the M.2 SSD.
As an example, here is a link to a 512GB M.2 SSD that will fit within this drive bay: https://www.amazon.com/dp/B07CJ4BSK8.
Hope this helps,
S
