fix: Fix typing issues in new device traits (#577)

Author: allenporter

roborock/devices/traits/v1/command.py

@@ -1,6 +1,7 @@
 from typing import Any
 
 from roborock import RoborockCommand
+from roborock.protocols.v1_protocol import ParamsType
 
 
 class CommandTrait:
@@ -14,7 +15,7 @@ def __post_init__(self) -> None:
         """
         self._rpc_channel = None
 
-    async def send(self, command: RoborockCommand | str, params: dict[str, Any] | None = None) -> Any:
+    async def send(self, command: RoborockCommand | str, params: ParamsType = None) -> Any:
         """Send a command to the device."""
         if not self._rpc_channel:
             raise ValueError("Device trait in invalid state")

roborock/devices/traits/v1/valley_electricity_timer.py

@@ -5,7 +5,7 @@
 _ENABLED_PARAM = "enabled"
 
 
-class ValleyElectricityTimerTrait(ValleyElectricityTimer, common.V1TraitMixin):
+class ValleyElectricityTimerTrait(ValleyElectricityTimer, common.V1TraitMixin, common.RoborockSwitchBase):
     """Trait for managing Valley Electricity Timer settings on Roborock devices."""
 
     command = RoborockCommand.GET_VALLEY_ELECTRICITY_TIMER