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

Update config_flow.py

Browse Source
This commit is contained in:
smkrv
2024-11-15 01:11:24 +03:00
committed by GitHub
parent a6218c1cac
commit 37e89dacc1
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_components/ha_text_ai/config_flow.py
+3 -3
View File
@@ -6,7 +6,7 @@ from typing import Any
import voluptuous as vol
from openai import AsyncOpenAI
from openai.types.error import APIError
from openai import OpenAIError # Updated import
from openai.exceptions import AuthenticationError, APIConnectionError
from homeassistant import config_entries
@@ -86,8 +86,8 @@ class TextAIConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
raise ApiKeyError from err
except APIConnectionError as err:
raise ApiConnectionError from err
except APIError as err:
if err.status_code == 401:
except OpenAIError as err: # Changed from APIError
if getattr(err, 'status_code', None) == 401: # Safe status_code check
raise ApiKeyError from err
raise ApiConnectionError from err