diff --git a/custom_components/ha_text_ai/__init__.py b/custom_components/ha_text_ai/__init__.py
index 2abdf3e..4b99b12 100644
--- a/custom_components/ha_text_ai/__init__.py
+++ b/custom_components/ha_text_ai/__init__.py
@@ -43,6 +43,8 @@ from .const import (
     SERVICE_SET_SYSTEM_PROMPT,
 )
 
+DOMAIN = "ha_text_ai"
+
 _LOGGER = logging.getLogger(__name__)
 
 CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)