Release v2.0.0

custom_components/ha_text_ai/sensor.py

@@ -59,11 +59,6 @@ async def async_setup_entry(
     coordinator = hass.data[DOMAIN][entry.entry_id]
     async_add_entities([HATextAISensor(coordinator, entry)], True)
 
-class HATextAISensor(CoordinatorEntity, SensorEntity):
-    """HA text AI Sensor."""
-
-    _attr_has_entity_name = True
-
 def __init__(
     self,
     coordinator: HATextAICoordinator,
@@ -72,13 +67,11 @@ class HATextAISensor(CoordinatorEntity, SensorEntity):
     """Initialize the sensor."""
     super().__init__(coordinator)
     self._config_entry = config_entry
-        self._attr_unique_id = f"{config_entry.entry_id}"
+    self._attr_unique_id = config_entry.entry_id
-        
     self._attr_suggested_display_precision = 0
     self._error_count = 0
     self._last_error = None
     self._state = STATE_INITIALIZING
-
     self._attr_device_info = {
         "identifiers": {(DOMAIN, self._attr_unique_id)},
         "name": "HA Text AI",
@@ -87,6 +80,11 @@ class HATextAISensor(CoordinatorEntity, SensorEntity):
         "sw_version": coordinator.api_version,
     }
 
+    @property
+    def name(self) -> str:
+        """Return the name of the sensor."""
+        return "HA Text AI"
+
     @property
     def icon(self) -> str:
         """Return the icon based on the current state."""