Genuine Kohler Fill Valve Service Kit, 400A Series.
Cap color doesn't matter. Look at shank, if grey use this valve and if black use GP1083167.
Manufacturer:
Kohler
Part Number:
GP1138930