Weapon set passives for command spells of the Djinns

Could we please get a function to make it so we can use the weapon set passives with the djinns on the new sorc ascendancy? As in being able to select which weapon we want to use for what djinn commands on the gem skills like we can do for any other either choose weapon set 1 or 2.