Release v2.0.0

custom_components/ha_text_ai/sensor.py

@@ -59,33 +59,31 @@ 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."""
+def __init__(
+    self,
+    coordinator: HATextAICoordinator,
+    config_entry: ConfigEntry,
+) -> None:
+    """Initialize the sensor."""
+    super().__init__(coordinator)
+    self._config_entry = config_entry
+    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",
+        "manufacturer": "Community",
+        "model": f"{coordinator.model} ({self._config_entry.data.get(CONF_API_PROVIDER, 'Unknown')} provider)",
+        "sw_version": coordinator.api_version,
+    }
 
-    _attr_has_entity_name = True
-
-    def __init__(
-        self,
-        coordinator: HATextAICoordinator,
-        config_entry: ConfigEntry,
-    ) -> None:
-        """Initialize the sensor."""
-        super().__init__(coordinator)
-        self._config_entry = config_entry
-        self._attr_unique_id = f"{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",
-            "manufacturer": "Community",
-            "model": f"{coordinator.model} ({self._config_entry.data.get(CONF_API_PROVIDER, 'Unknown')} provider)",
-            "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: