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

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

diff --git a/custom_components/ha_text_ai/config_flow.py b/custom_components/ha_text_ai/config_flow.py
index 1e0815f..f46bfda 100644
--- a/custom_components/ha_text_ai/config_flow.py
+++ b/custom_components/ha_text_ai/config_flow.py
@@ -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