Hass/ha-text-ai
1
0
Fork 0
mirror of https://github.com/smkrv/ha-text-ai.git synced 2026-07-30 11:03:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Release v2.0.0

Browse Source
This commit is contained in:
SMKRV
2024-11-22 12:23:28 +03:00
parent 80c91039e1
commit 50b414a904
1 changed files with 20 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_components/ha_text_ai/coordinator.py
+20 -17
View File
@@ -7,6 +7,7 @@ import time
import ssl import ssl
import certifi import certifi
import aiohttp import aiohttp
import httpx
from homeassistant.helpers import aiohttp_client from homeassistant.helpers import aiohttp_client
from openai import AsyncOpenAI, APIError, AuthenticationError, RateLimitError from openai import AsyncOpenAI, APIError, AuthenticationError, RateLimitError
@@ -97,21 +98,21 @@ class HATextAICoordinator(DataUpdateCoordinator):
async def _init_client(self): async def _init_client(self):
"""Initialize API client with proper SSL context.""" """Initialize API client with proper SSL context."""
try: try:
ssl_context = await self._create_ssl_context()
connector = aiohttp.TCPConnector(ssl=ssl_context)
client_session = aiohttp.ClientSession(connector=connector)
if self._is_anthropic: if self._is_anthropic:
self.client = AsyncAnthropic( self.client = AsyncAnthropic(
api_key=self.api_key, api_key=self.api_key
http_client=client_session )
) else: # OpenAI
else: async_client = httpx.AsyncClient(
self.client = AsyncOpenAI( base_url=self.endpoint,
api_key=self.api_key, timeout=httpx.Timeout(30.0)
base_url=self.endpoint, )
http_client=client_session,
) self.client = AsyncOpenAI(
api_key=self.api_key,
base_url=self.endpoint,
http_client=async_client
)
except Exception as e: except Exception as e:
_LOGGER.error("Error initializing API client: %s", str(e)) _LOGGER.error("Error initializing API client: %s", str(e))
raise raise
@@ -468,8 +469,10 @@ class HATextAICoordinator(DataUpdateCoordinator):
except asyncio.QueueEmpty: except asyncio.QueueEmpty:
break break
if hasattr(self.client, 'close'): if hasattr(self.client, 'close'):
await self.client.close() await self.client.close()
elif hasattr(self.client, '_client') and hasattr(self.client._client, 'aclose'):
await self.client._client.aclose()
self._is_ready = False self._is_ready = False
self._endpoint_status = "disconnected" self._endpoint_status = "disconnected"
@@ -478,7 +481,7 @@ class HATextAICoordinator(DataUpdateCoordinator):
self._update_final_metrics() self._update_final_metrics()
except Exception as err: except Exception as err:
_LOGGER.error("Error during shutdown: %s", err) _LOGGER.error("Error during shutdown: %s", err)
def _update_final_metrics(self) -> None: def _update_final_metrics(self) -> None:
"""Update final metrics before shutdown.""" """Update final metrics before shutdown."""