1 files changed, 22 insertions, 16 deletions

diff --git a/examples/temperature.rs b/examples/sensor.rs
index d449cd4..69bd87a 100644
--- a/examples/temperature.rs
+++ b/examples/sensor.rs
@@ -21,46 +21,52 @@ async fn main_task(spawner: Spawner) {
         },
     );
 
-    let constant_temperature_sensor = device.create_temperature_sensor(
-        "constant-temperature-sensor-id",
-        embassy_ha::TemperatureSensorConfig {
+    let temperature_sensor = device.create_sensor(
+        "random-temperature-sensor-id",
+        embassy_ha::SensorConfig {
             common: embassy_ha::EntityCommonConfig {
-                name: Some("Constant Temperature Sensor"),
+                name: Some("Random Temperature Sensor"),
                 ..Default::default()
             },
-            unit: embassy_ha::TemperatureUnit::Celcius,
+            class: embassy_ha::SensorClass::Temperature,
+            state_class: embassy_ha::StateClass::Measurement,
+            unit: Some(embassy_ha::constants::HA_UNIT_TEMPERATURE_CELSIUS),
+            suggested_display_precision: Some(1),
         },
     );
 
-    let random_temperature_sensor = device.create_temperature_sensor(
-        "random-temperature-sensor-id",
-        embassy_ha::TemperatureSensorConfig {
+    let humidity_sensor = device.create_sensor(
+        "random-humidity-sensor-id",
+        embassy_ha::SensorConfig {
             common: embassy_ha::EntityCommonConfig {
-                name: Some("Random Temperature Sensor"),
+                name: Some("Random Humidity Sensor"),
                 ..Default::default()
             },
-            unit: embassy_ha::TemperatureUnit::Celcius,
+            class: embassy_ha::SensorClass::Humidity,
+            state_class: embassy_ha::StateClass::Measurement,
+            unit: Some(embassy_ha::constants::HA_UNIT_PERCENTAGE),
+            suggested_display_precision: Some(0),
         },
     );
 
-    spawner.must_spawn(constant_temperature_task(constant_temperature_sensor));
-    spawner.must_spawn(random_temperature_task(random_temperature_sensor));
+    spawner.must_spawn(random_temperature_task(temperature_sensor));
+    spawner.must_spawn(random_humidity_task(humidity_sensor));
 
     device.run(&mut stream).await.unwrap();
 }
 
 #[embassy_executor::task]
-async fn constant_temperature_task(mut sensor: embassy_ha::TemperatureSensor<'static>) {
+async fn random_temperature_task(mut sensor: embassy_ha::Sensor<'static>) {
     loop {
-        sensor.publish(42.0);
+        sensor.publish(rand::random_range(0.0..50.0));
         Timer::after_secs(1).await;
     }
 }
 
 #[embassy_executor::task]
-async fn random_temperature_task(mut sensor: embassy_ha::TemperatureSensor<'static>) {
+async fn random_humidity_task(mut sensor: embassy_ha::Sensor<'static>) {
     loop {
-        sensor.publish(rand::random_range(0.0..50.0));
+        sensor.publish(rand::random_range(0.0..100.0));
         Timer::after_secs(1).await;
     }
 }