Dreame Vacuum: Automation to clean only 1 room?

Hello,

I have configured the thir party integration “Dream Vacuum” to control my Dreame X40 Master and it is working. Is it possible to start the robot on some days via automation only to wipe 1 room?
Normally when it is started, it should clean the whole apartment, but on several days I would like to clean only one specific room or area. I know it’s possible in the official Dreame app, but I need to do this in HA to be able to consider other things.

action: dreame_vacuum.vacuum_clean_segment
target:
  entity_id: vacuum.x50_ultra_complete
data:
  segments: 1 or [1,2,4] #Room number(s) of the room(s) you want to clean.