AI大模型之旅--Vanna的安装与部署

最新推荐文章于 2025-06-17 09:37:29 发布

chenkangck50

最新推荐文章于 2025-06-17 09:37:29 发布

阅读量3.6k

点赞数 6

CC 4.0 BY-SA版权

分类专栏： AI大模型 python 文章标签： AIGC

本文链接：https://blog.youkuaiyun.com/weixin_43866043/article/details/141645916

根据前章，我们部署了ollama，此章刚好用来测试text2sql -----vanna
安装部署

1.安装部署

https://vanna.ai/docs/mysql-ollama-chromadb/

根据官网安装部署就可以了，官网很详细，我这次选择Ollama、ChromaDB和MySQL
在这里插入图片描述

在这里插入图片描述

from vanna.ollama import Ollama
from vanna.chromadb import ChromaDB_VectorStore

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

vn = MyVanna(config={
   
   'model': 'mistral'})

vn.connect_to_mysql(host='my-host', dbname='my-db', user='my-user', password='my-password', port=123)

以上就是官网代码，根据需要，改成自己的配置

以下是我的配置

from vanna.ollama import Ollama
from vanna.chromadb import ChromaDB_VectorStore
class MyVanna(ChromaDB_VectorStore, Ollama):
    def __init__(self, config=None