From 87b45df180f2fd8ffcf2109659ec8873edc1b7f6 Mon Sep 17 00:00:00 2001
From: SMKRV <smkrv@github.com>
Date: Fri, 22 Nov 2024 15:44:45 +0300
Subject: [PATCH] Release v2.0.0

---
 custom_components/ha_text_ai/sensor.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/custom_components/ha_text_ai/sensor.py b/custom_components/ha_text_ai/sensor.py
index 85f70bf..f036b37 100644
--- a/custom_components/ha_text_ai/sensor.py
+++ b/custom_components/ha_text_ai/sensor.py
@@ -59,12 +59,13 @@ async def async_setup_entry(
     coordinator = hass.data[DOMAIN][entry.entry_id]
     async_add_entities([HATextAISensor(coordinator, entry)], True)
 
+from homeassistant.components.sensor import SensorDeviceClass
 
 class HATextAISensor(CoordinatorEntity, SensorEntity):
     """HA text AI Sensor."""
 
     _attr_has_entity_name = True
-    _attr_device_class = SensorDeviceClass.TEXT
+    _attr_device_class = SensorDeviceClass.TEXT  # Оставляем как есть
 
     def __init__(
         self,