Copper Golem Flower Filtering

Hello! I have a suggestion regarding copper golems ability to only search a few chest before resetting. I was thinking if the copper golems flower mechanic could act as a filter for which chest to search! For instance, let’s say you add a red flower to its head it would only look in chests that have a red tint like Mangrove chest. This would mean that if you have a chest room full of 64 chests you could mix up the colors to allow copper golems with different flowers to sort larger amounts of materials without just making golems search more chests. It could also prevent confusion between golems, though you would probably need to make it so that when a golem gets confused from their chest limit they put the item back in the chest.