Add methods to empty dustbin and set clean mode

Author: lboue

roborock/devices/traits/b01/q10/vacuum.py

@@ -1,6 +1,6 @@
 """Traits for Q10 B01 devices."""
 
-from roborock.data.b01_q10.b01_q10_code_mappings import B01_Q10_DP
+from roborock.data.b01_q10.b01_q10_code_mappings import B01_Q10_DP, YXCleanType
 
 from .command import CommandTrait
 
@@ -54,3 +54,17 @@ async def return_to_dock(self) -> None:
             command=B01_Q10_DP.START_DOCK_TASK,
             params={},
         )
+
+    async def empty_dustbin(self) -> None:
+        """Empty the dustbin at the dock."""
+        await self._command.send(
+            command=B01_Q10_DP.START_DOCK_TASK,
+            params=2,
+        )
+
+    async def set_clean_mode(self, mode: YXCleanType) -> None:
+        """Set the cleaning mode (vacuum, mop, or both)."""
+        await self._command.send(
+            command=B01_Q10_DP.CLEAN_MODE,
+            params=mode.code,
+        )