VannaAI文档代码已弃用，运行时遇ImportError错误求助

问题情况 #

运行VannaAI官方文档中的代码时触发ImportError，无法从vanna包中导入OpenAI_Chat和ChromaDB_VectorStore。已通过pip install 'vanna[chromadb,openai]'完成依赖安装。

报错信息 #

line 1, in <module>
from vanna import OpenAI_Chat
ImportError: cannot import name 'OpenAI_Chat' from 'vanna'

原运行代码 #

from vanna import OpenAI_Chat
from openai import AzureOpenAI
from vanna import ChromaDB_VectorStore

client = AzureOpenAI(
  azure_endpoint = "https://example-gpt.openai.azure.com/", 
  api_key="api key",  
  api_version="2024-02-01"
)
class MyVanna(ChromaDB_VectorStore, OpenAI_Chat):
    def __init__(self, config=None):
        ChromaDB_VectorStore.__init__(self, config=config)
        OpenAI_Chat.__init__(self, client=client, config=config) 

解决方案 #

修改导入语句，直接从对应的子模块导入类即可正常运行：

from vanna.openai.openai_chat import OpenAI_Chat
from openai import AzureOpenAI
from vanna.chromadb.chromadb_vector import ChromaDB_VectorStore

感谢@Jared Smith提供的解决方案。

内容的提问来源于stack exchange，提问作者the programmer