Great, I have the round one.
Searching in the forum I found this Finally - a cheap WIRELESS switch that dims! Xiaomi Switch Gen1 but in the automation is used long_click_press as click type, I think the that the switch is used with the xiaomi gateway.
Can you post your script @aherbjornsen?