fix: Use correct API capability types for heater temperature and fan speed (#13)

Temperature control was sending devices.capabilities.range/temperature
(plain integer) but heaters require devices.capabilities.temperature_setting/
targetTemperature with a STRUCT value {"temperature": N, "unit": "Celsius"}.
Fan speed looked for a non-existent devices.capabilities.mode/fanSpeed
capability instead of reading workMode options from devices.capabilities.work_mode.

Author: Tom Lasswell

custom_components/govee/coordinator.py

@@ -40,6 +40,7 @@
     MusicModeCommand,
     PowerCommand,
     SceneCommand,
+    TemperatureSettingCommand,
     ToggleCommand,
     create_dreamview_command,
 )
@@ -883,6 +884,8 @@ def _apply_optimistic_update(
         elif isinstance(command, ModeCommand):
             if command.mode_instance == INSTANCE_HDMI_SOURCE:
                 state.apply_optimistic_hdmi_source(command.value)
+        elif isinstance(command, TemperatureSettingCommand):
+            state.heater_temperature = command.temperature
         elif isinstance(command, MusicModeCommand):
             # Look up mode name from device capabilities for display
             device = self._devices.get(device_id)

custom_components/govee/manifest.json

@@ -16,6 +16,6 @@
     "cryptography>=41.0.0"
   ],
   "ssdp": [],
-  "version": "2026.2.7",
+  "version": "2026.2.8",
   "zeroconf": []
 }

custom_components/govee/models/__init__.py

@@ -16,6 +16,7 @@
     RangeCommand,
     SceneCommand,
     SegmentColorCommand,
+    TemperatureSettingCommand,
     ToggleCommand,
     WorkModeCommand,
     create_dreamview_command,
@@ -54,6 +55,7 @@
     "WorkModeCommand",
     "ModeCommand",
     "MusicModeCommand",
+    "TemperatureSettingCommand",
     "create_dreamview_command",
     "create_night_light_command",
 ]

custom_components/govee/models/commands.py

@@ -18,6 +18,7 @@
     CAPABILITY_ON_OFF,
     CAPABILITY_RANGE,
     CAPABILITY_SEGMENT_COLOR,
+    CAPABILITY_TEMPERATURE_SETTING,
     CAPABILITY_TOGGLE,
     CAPABILITY_WORK_MODE,
     INSTANCE_BRIGHTNESS,
@@ -31,6 +32,7 @@
     INSTANCE_POWER,
     INSTANCE_SCENE,
     INSTANCE_SEGMENT_COLOR,
+    INSTANCE_TARGET_TEMPERATURE,
     INSTANCE_WORK_MODE,
 )
 from .state import RGBColor
@@ -381,3 +383,26 @@ def get_value(self) -> dict[str, Any]:
         if self.rgb is not None and self.auto_color == 0:
             value["rgb"] = self.rgb
         return value
+
+
+@dataclass(frozen=True)
+class TemperatureSettingCommand(DeviceCommand):
+    """Command to set heater target temperature via STRUCT payload.
+
+    Heaters use the temperature_setting capability with a STRUCT value
+    containing temperature and unit fields.
+    """
+
+    temperature: int
+    unit: str = "Celsius"
+
+    @property
+    def capability_type(self) -> str:
+        return CAPABILITY_TEMPERATURE_SETTING
+
+    @property
+    def instance(self) -> str:
+        return INSTANCE_TARGET_TEMPERATURE
+
+    def get_value(self) -> dict[str, Any]:
+        return {"temperature": self.temperature, "unit": self.unit}

custom_components/govee/models/device.py

@@ -22,6 +22,7 @@
 CAPABILITY_WORK_MODE = "devices.capabilities.work_mode"
 CAPABILITY_PROPERTY = "devices.capabilities.property"
 CAPABILITY_MODE = "devices.capabilities.mode"
+CAPABILITY_TEMPERATURE_SETTING = "devices.capabilities.temperature_setting"
 
 # Device type constants
 DEVICE_TYPE_LIGHT = "devices.types.light"
@@ -48,6 +49,7 @@
 INSTANCE_MUSIC_MODE = "musicMode"
 INSTANCE_DREAMVIEW = "dreamViewToggle"
 INSTANCE_TEMPERATURE = "temperature"
+INSTANCE_TARGET_TEMPERATURE = "targetTemperature"
 INSTANCE_FAN_SPEED = "fanSpeed"
 INSTANCE_PURIFIER_MODE = "purifierMode"
 
@@ -381,26 +383,39 @@ def get_music_sensitivity_range(self) -> tuple[int, int]:
     def get_temperature_range(self) -> tuple[int, int]:
         """Extract temperature range from capability.
 
+        Parses STRUCT-based temperature_setting capability where the range
+        is nested inside the fields array under the 'temperature' field.
+
         Returns (min, max) tuple, defaulting to (16, 35) Celsius.
         """
         for cap in self.capabilities:
-            if cap.type == CAPABILITY_RANGE and cap.instance == INSTANCE_TEMPERATURE:
-                range_data = cap.parameters.get("range", {})
-                return (
-                    int(range_data.get("min", 16)),
-                    int(range_data.get("max", 35)),
-                )
+            if (
+                cap.type == CAPABILITY_TEMPERATURE_SETTING
+                and cap.instance == INSTANCE_TARGET_TEMPERATURE
+            ):
+                for f in cap.parameters.get("fields", []):
+                    if f.get("fieldName") == "temperature":
+                        range_data = f.get("range", {})
+                        return (
+                            int(range_data.get("min", 16)),
+                            int(range_data.get("max", 35)),
+                        )
         return (16, 35)
 
     def get_fan_speed_options(self) -> list[dict[str, Any]]:
-        """Extract fan speed mode options from capability.
+        """Extract fan speed options from work_mode capability.
+
+        Heater fan speed options are inside the STRUCT-based work_mode
+        capability, in the 'workMode' field's options array.
 
         Returns list of {"name": "Low", "value": 1} dicts.
         """
         for cap in self.capabilities:
-            if cap.type == CAPABILITY_MODE and cap.instance == INSTANCE_FAN_SPEED:
-                options: list[dict[str, Any]] = cap.parameters.get("options", [])
-                return options
+            if cap.type == CAPABILITY_WORK_MODE and cap.instance == INSTANCE_WORK_MODE:
+                for f in cap.parameters.get("fields", []):
+                    if f.get("fieldName") == "workMode":
+                        options: list[dict[str, Any]] = f.get("options", [])
+                        return options
         return []
 
     def get_purifier_mode_options(self) -> list[dict[str, Any]]:

custom_components/govee/number.py

@@ -18,7 +18,7 @@
 
 from .const import DOMAIN, SUFFIX_HEATER_TEMPERATURE, SUFFIX_MUSIC_SENSITIVITY
 from .coordinator import GoveeCoordinator
-from .models import GoveeDevice, MusicModeCommand, RangeCommand
+from .models import GoveeDevice, MusicModeCommand, TemperatureSettingCommand
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -306,9 +306,8 @@ async def async_set_native_value(self, value: float) -> None:
         """
         temperature = int(value)
 
-        command = RangeCommand(
-            range_instance="temperature",
-            value=temperature,
+        command = TemperatureSettingCommand(
+            temperature=temperature,
         )
 
         success = await self.coordinator.async_control_device(

custom_components/govee/select.py

@@ -36,8 +36,9 @@
     ModeCommand,
     MusicModeCommand,
     SceneCommand,
+    WorkModeCommand,
 )
-from .models.device import INSTANCE_FAN_SPEED, INSTANCE_HDMI_SOURCE, INSTANCE_PURIFIER_MODE
+from .models.device import INSTANCE_HDMI_SOURCE, INSTANCE_PURIFIER_MODE
 
 # DIY Style options for select entity
 DIY_STYLE_OPTIONS = list(DIY_STYLE_NAMES.keys())
@@ -773,10 +774,9 @@ def __init__(
     def current_option(self) -> str | None:
         """Return current selected option from state."""
         state = self.coordinator.get_state(self._device_id)
-        if state and state.fan_speed is not None:
-            # Find option name matching the current value
+        if state and state.work_mode is not None:
             for name, value in self._option_map.items():
-                if value == state.fan_speed:
+                if value == state.work_mode:
                     return name
         # Return first option as default if available
         return self._attr_options[0] if self._attr_options else None
@@ -788,9 +788,9 @@ async def async_select_option(self, option: str) -> None:
             _LOGGER.warning("Unknown fan speed option: %s", option)
             return
 
-        command = ModeCommand(
-            mode_instance=INSTANCE_FAN_SPEED,
-            value=value,
+        command = WorkModeCommand(
+            work_mode=value,
+            mode_value=0,
         )
 
         success = await self.coordinator.async_control_device(