Vanna使用ollama分析本地MySQL数据库

原创

已于 2024-11-18 09:21:54 修改

· 6.2k 阅读

40 ·

版权

文章标签：

#Vanna #大模型 #Text2SQL

于 2024-05-28 17:54:27 首次发布

上一章节中已经实现了vanna的本地运行，但是大模型和数据库都还是远程的，因为也就没办法去训练，这节一起来实现vanna分析本地mysql数据库，因为要使用本地大模型，所以开始之前需要给本地安装好大模型，我这里用ollama下载了llama3，这节就用llama3作为大模型分析，如果本地没有ollama的可以参考我之前的章节安装一下，同时本地还需要安装mysql数据库，并创建一个表作为接下来用大模型分析的对象。

我这里的数据表如下：

因为需要连接mysql数据库，因此需要安装mysql-connecter-python，同时我们将向量数据库换为ChromDB，在pycharm的命令提示符执行

pip install vanna
pip install ollama
pip install ChromaDB
pip install mysql-connector-python
pip install 'vanna[chromadb,ollama,mysql]'

安装完成后，按照vanna的官方文档，实现MyVanna的方法

class MyVanna(ChromaDB_VectorStore, Ollama):
    def __init__(self, config=None):
        ChromaDB_VectorStore.__init__(self, config=config)
        Ollama.__init__(self, config=config