mirror of
https://github.com/smkrv/ha-text-ai.git
synced 2026-07-21 22:54:00 +08:00
Update config_flow.py
This commit is contained in:
@@ -6,7 +6,7 @@ from typing import Any
|
|||||||
|
|
||||||
import voluptuous as vol
|
import voluptuous as vol
|
||||||
from openai import AsyncOpenAI
|
from openai import AsyncOpenAI
|
||||||
from openai.types.error import APIError
|
from openai import OpenAIError # Updated import
|
||||||
from openai.exceptions import AuthenticationError, APIConnectionError
|
from openai.exceptions import AuthenticationError, APIConnectionError
|
||||||
|
|
||||||
from homeassistant import config_entries
|
from homeassistant import config_entries
|
||||||
@@ -86,8 +86,8 @@ class TextAIConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
|
|||||||
raise ApiKeyError from err
|
raise ApiKeyError from err
|
||||||
except APIConnectionError as err:
|
except APIConnectionError as err:
|
||||||
raise ApiConnectionError from err
|
raise ApiConnectionError from err
|
||||||
except APIError as err:
|
except OpenAIError as err: # Changed from APIError
|
||||||
if err.status_code == 401:
|
if getattr(err, 'status_code', None) == 401: # Safe status_code check
|
||||||
raise ApiKeyError from err
|
raise ApiKeyError from err
|
||||||
raise ApiConnectionError from err
|
raise ApiConnectionError from err
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user