Am I able to edit what drops when frogs eat magma cubes?

I want to add red, orange, and blue froglights, and I was thinking of having each variant have an equal chance of giving you one of two colors since i'm lazy and didnt wanna make more frog variants. (warm frogs either give you red or purple, temperate frogs either give you orange or yellow, and cold frogs give you either green or blue). is this possible without creating custom code? i know how to edit loot tables but im not sure if theres one specifically for the conversion of magma cubes to froglights. thanks! :3