According to my research, the Lenovo Legion Go uses a 2242 NVMe SSD with M key, unfortunately we could only find a suitable 1 TB variant in our range:
OWC 1.0TB Aura P13 Pro M.2 2242 NVMe SSD High Performance (1000 GB, M.2)