From ef78a9da0d3042019f6e92e3c332136851956208 Mon Sep 17 00:00:00 2001
From: smkrv <17809065+smkrv@users.noreply.github.com>
Date: Fri, 15 Nov 2024 00:55:48 +0300
Subject: [PATCH] Update config_flow.py

---
 custom_components/ha_text_ai/config_flow.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/custom_components/ha_text_ai/config_flow.py b/custom_components/ha_text_ai/config_flow.py
index 9996c96..1e0815f 100644
--- a/custom_components/ha_text_ai/config_flow.py
+++ b/custom_components/ha_text_ai/config_flow.py
@@ -25,10 +25,16 @@ from .const import (
 
 _LOGGER = logging.getLogger(__name__)  
 
+@config_entries.HANDLERS.register(DOMAIN)  
 class TextAIConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):  
     """Handle a config flow for HA Text AI Integration."""  
 
     VERSION = 1  
+    CONNECTION_CLASS = config_entries.CONN_CLASS_CLOUD_POLL  
+
+    def __init__(self) -> None:  
+        """Initialize the config flow."""  
+        super().__init__()  
 
     async def async_step_user(  
         self, user_input: dict[str, Any] | None = None