Hi, I was able to resolve this when I restarted the python library in Databricks. 
Let me know if that helps

Code:

pip install openai

dbutils.library.restartPython()

from openai import AzureOpenAI