From b17e5c3db21f8745afac53e70cb782a447be642a Mon Sep 17 00:00:00 2001
From: SMKRV <smkrv@github.com>
Date: Fri, 22 Nov 2024 16:29:26 +0300
Subject: [PATCH] Release v2.0.0

---
 custom_components/ha_text_ai/sensor.py | 50 +++++++++++++-------------
 1 file changed, 24 insertions(+), 26 deletions(-)

diff --git a/custom_components/ha_text_ai/sensor.py b/custom_components/ha_text_ai/sensor.py
index 1de63c5..36383b0 100644
--- a/custom_components/ha_text_ai/sensor.py
+++ b/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: