How to Use AzureOpenAI with llm_classify in Python Code

I'm trying to use AzureOpenAI instead of OpenAi model in following code excerpt: with suppress_tracing(): clarity_eval = llm_classify( dataframe = clarity_df, template = CLARITY_LLM_JUDGE_PROMPT, rails = ['clear', 'unclear'], model=OpenAIModel(model="gpt-4o"), provide_explanation=True ) I am actually going through the Arize course on the Deeplearning.AI platform Struggling with this simple issue...is there any example how to use AzureOpenAI (from openai python library) with llm_classify?