Google Assistant Component 'type' option

I just tried to set up all my devices to work with Google Assistant using it’s component everything works fine, thanks!
I have some smart wall plugs which control lights and sadly in the configuration for an entity in the google assistant component there is no option to change the device type under the entity_config so that a switch could be changed into a light for example, to allow commands like “Hey Google, turn off all lights”.

Would it be possible to implement something like this?

I’m also on the lookout for this feature to represent a “light” as a “fireplace” on google home.