From 13a9e1a5d73977532baf3967eafeadd8a3039a18 Mon Sep 17 00:00:00 2001
From: SMKRV <smkrv@github.com>
Date: Fri, 22 Nov 2024 19:18:10 +0300
Subject: [PATCH] Release v2.0.0

---
 custom_components/ha_text_ai/coordinator.py | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/custom_components/ha_text_ai/coordinator.py b/custom_components/ha_text_ai/coordinator.py
index de3549e..c2a1c00 100644
--- a/custom_components/ha_text_ai/coordinator.py
+++ b/custom_components/ha_text_ai/coordinator.py
@@ -49,7 +49,7 @@ class HATextAICoordinator(DataUpdateCoordinator):
         )
 
         self._validate_params(api_key, temperature, max_tokens)
-
+        self._entity_id = None
         self.api_key = api_key
         self.endpoint = endpoint
         self.model = model
@@ -211,6 +211,7 @@ class HATextAICoordinator(DataUpdateCoordinator):
 
         # Fire event for successful response
         self.hass.bus.async_fire(f"{DOMAIN}_response_received", {
+            "entity_id": self._entity_id,
             "question": question,
             "model": response_data["model"],
             "tokens": response_data.get("tokens", 0)
@@ -254,8 +255,9 @@ class HATextAICoordinator(DataUpdateCoordinator):
 
         # Fire error event
         self.hass.bus.async_fire(f"{DOMAIN}_error_occurred", {
+            "entity_id": self._entity_id,
             "error_type": type(error).__name__,
-            "error_message": error_msg,
+            "error_message": str(error),
             "question": question
         })
 
@@ -500,6 +502,16 @@ class HATextAICoordinator(DataUpdateCoordinator):
         self._performance_metrics["total_requests"] = self._request_count
         self._performance_metrics["total_tokens"] = self._tokens_used
 
+    @property
+    def entity_id(self) -> Optional[str]:
+        """Get entity ID."""
+        return self._entity_id
+
+    @entity_id.setter
+    def entity_id(self, value: str) -> None:
+        """Set entity ID."""
+        self._entity_id = value
+
     @property
     def performance_metrics(self) -> Dict[str, Any]:
         """Return current performance metrics."""
@@ -668,6 +680,7 @@ class HATextAICoordinator(DataUpdateCoordinator):
     def export_metrics(self) -> Dict[str, Any]:
         """Export all metrics and statistics."""
         return {
+            "entity_id": self._entity_id,
             "performance": self._performance_metrics,
             "requests": {
                 "total": self._request_count,