1313from bs4 import BeautifulSoup
1414
1515from openai import OpenAI
16- from openai import error as openai_error
16+ from openai import OpenAIError
1717
1818import random
1919
@@ -208,7 +208,7 @@ def do_call():
208208 )
209209 # Return the textual reply
210210 return resp .choices [0 ].message ["content" ]
211- except openai_error . OpenAIError as e :
211+ except OpenAIError as e :
212212 logger .exception ("OpenAI error: %s" , e )
213213 raise
214214
@@ -229,7 +229,7 @@ def do_call():
229229 try :
230230 resp = client .images .generate (model = model , prompt = prompt , size = size , n = 1 )
231231 return resp .data [0 ].url
232- except openai_error . OpenAIError as e :
232+ except OpenAIError as e :
233233 logger .exception ("OpenAI images error: %s" , e )
234234 raise
235235
@@ -253,7 +253,7 @@ def do_call():
253253 try :
254254 resp = client .images .edit (model = model , image = image_content , prompt = prompt , n = 1 , size = "1024x1024" )
255255 return resp .data [0 ].url
256- except openai_error . OpenAIError as e :
256+ except OpenAIError as e :
257257 logger .exception ("OpenAI images.edit error: %s" , e )
258258 raise
259259
@@ -272,7 +272,7 @@ def do_call():
272272 try :
273273 resp = client .images .variations (model = model , image = bc , n = 1 , size = "1024x1024" )
274274 return resp .data [0 ].url
275- except openai_error . OpenAIError as e :
275+ except OpenAIError as e :
276276 logger .exception ("OpenAI images.variations error: %s" , e )
277277 raise
278278
0 commit comments