Error when run from vanna.openai import AzureOpenAI_Chat

[meiyahan]

ModuleNotFoundError: No module named 'vanna.openai'

My openai package version is 2.16.0,and my vanna version is 2.0.1

[fynrup]

Either you want to import AzureOpenAI or OpenAI directly. Both options can be found below

from vanna.integrations.azureopenai import AzureOpenAILlmService
from vanna.integrations.openai import OpenAILlmService