v.1.0.10

added: from .coordinator import HATextAICoordinator

custom_components/ha_text_ai/__init__.py

@@ -9,6 +9,8 @@ from homeassistant.exceptions import ConfigEntryNotReady
 from homeassistant.helpers import aiohttp_client
 from homeassistant.helpers import config_validation as cv
 
+from .coordinator import HATextAICoordinator  
+
 from .const import (
     DOMAIN,
     PLATFORMS,

custom_components/ha_text_ai/config_flow.py

@@ -62,24 +62,6 @@ STEP_USER_DATA_SCHEMA = vol.Schema({
     ),
 })
 
-async def validate_endpoint(endpoint: str) -> Tuple[bool, str]:
-    """Validate API endpoint accessibility."""
-    try:
-        parsed_url = urlparse(endpoint)
-        if parsed_url.scheme not in ('http', 'https'):
-            return False, "invalid_endpoint_scheme"
-
-        connector = aiohttp.TCPConnector(ssl=SSL_CONTEXT)
-        async with timeout(5):
-            async with aiohttp.ClientSession(connector=connector) as session:
-                async with session.get(endpoint) as response:
-                    if response.status != 200:
-                        return False, "endpoint_not_available"
-        return True, ""
-    except Exception as e:
-        _LOGGER.error("Error validating endpoint: %s", str(e))
-        return False, "endpoint_error"
-
 async def validate_api_connection(
     api_key: str,
     endpoint: str,
@@ -87,64 +69,56 @@ async def validate_api_connection(
     retry_count: int = 3,
     retry_delay: float = 1.0
 ) -> Tuple[bool, str, list]:
-    """Validate API connection with improved retry logic."""
-    # Validate endpoint first
-    endpoint_valid, endpoint_error = await validate_endpoint(endpoint)
-    if not endpoint_valid:
-        return False, endpoint_error, []
-
-    connector = aiohttp.TCPConnector(ssl=SSL_CONTEXT)
-    async with aiohttp.ClientSession(connector=connector) as session:
-        for attempt in range(retry_count):
-            try:
-                async with timeout(10):
-                    client = AsyncOpenAI(
-                        api_key=api_key,
-                        base_url=endpoint,
-                        http_client=session
-                    )
-
-                    models = await client.models.list()
-                    model_ids = [model.id for model in models.data]
-
-                    if model not in model_ids:
-                        _LOGGER.warning(
-                            "Model %s not found in available models: %s",
-                            model,
-                            ", ".join(model_ids)
-                        )
-                        return False, "invalid_model", model_ids
-                    return True, "", model_ids
-
-            except asyncio.TimeoutError:
-                _LOGGER.warning(
-                    "Timeout during API validation (attempt %d/%d)",
-                    attempt + 1,
-                    retry_count
+    """Validate API connection with retry logic."""
+    for attempt in range(retry_count):
+        try:
+            async with timeout(10):
+                client = AsyncOpenAI(
+                    api_key=api_key,
+                    base_url=endpoint,
                 )
-                if attempt == retry_count - 1:
-                    return False, "timeout", []
-                await asyncio.sleep(retry_delay)
 
-            except AuthenticationError as err:
-                _LOGGER.error("Authentication error: %s", str(err))
-                return False, "invalid_auth", []
+                models = await client.models.list()
+                model_ids = [model.id for model in models.data]
 
-            except RateLimitError as err:
-                _LOGGER.error("Rate limit exceeded: %s", str(err))
-                return False, "rate_limit", []
+                if model not in model_ids:
+                    _LOGGER.warning(
+                        "Model %s not found in available models: %s",
+                        model,
+                        ", ".join(model_ids)
+                    )
+                    return False, "invalid_model", model_ids
+                return True, "", model_ids
 
-            except APIConnectionError as err:
-                _LOGGER.error("API connection error: %s", str(err))
-                return False, "cannot_connect", []
+        except asyncio.TimeoutError:
+            _LOGGER.warning(
+                "Timeout during API validation (attempt %d/%d)",
+                attempt + 1,
+                retry_count
+            )
+            if attempt == retry_count - 1:
+                return False, "timeout", []
+            await asyncio.sleep(retry_delay)
 
-            except APIError as err:
-                _LOGGER.error("API error: %s", str(err))
-                return False, "api_error", []
+        except AuthenticationError as err:
+            _LOGGER.error("Authentication error: %s", str(err))
+            return False, "invalid_auth", []
 
-            except Exception as err:
-                _LOGGER.exception("Unexpected error during validation: %s", str(err))
-                return False, "unknown", []  
+        except RateLimitError as err:
+            _LOGGER.error("Rate limit exceeded: %s", str(err))
+            return False, "rate_limit", []
+
+        except APIConnectionError as err:
+            _LOGGER.error("API connection error: %s", str(err))
+            return False, "cannot_connect", []
+
+        except APIError as err:
+            _LOGGER.error("API error: %s", str(err))
+            return False, "api_error", []
+
+        except Exception as err:
+            _LOGGER.exception("Unexpected error during validation: %s", str(err))
+            return False, "unknown", []
 
 class HATextAIConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
     """Handle a config flow for HA text AI."""

custom_components/ha_text_ai/manifest.json

@@ -9,6 +9,6 @@
   "issue_tracker": "https://github.com/smkrv/ha-text-ai/issues",
   "requirements": ["openai>=1.0.0"],
   "ssdp": [],
-  "version": "1.0.8",
+  "version": "1.0.10",
   "zeroconf": []
 }

hacs.json

@@ -4,6 +4,6 @@
   "domains": ["sensor"],
   "homeassistant": "2024.11.0",
   "icon": "mdi:brain",
-  "version": "1.0.8",
+  "version": "1.0.10",
   "documentation": "https://github.com/smkrv/ha-text-ai"
 }

structure.md

@@ -0,0 +1,17 @@
+```
+ha-text-ai/
+│
+├── custom_components/
+│   └── ha_text_ai/
+│       ├── __init__.py
+│       ├── config_flow.py
+│       ├── coordinator.py
+│       ├── manifest.json
+│       ├── sensor.py
+│       ├── services.yaml
+│       └── const.py
+│
+└── strings/
+    ├── en.json
+    └── ru.json
+```