🧠 MindSQL

MindSQL 是一个 Python RAG（检索增强生成）库，旨在通过几行代码简化用户与数据库的交互。它可以无缝集成著名的数据库，如 PostgreSQL、MySQL 和 SQLite，同时通过扩展 IDatabase 接口，还支持 Snowflake 和 BigQuery 等主要数据库。该库利用大型语言模型（LLM）如 GPT-4、Llama 2、Google Gemini，并支持 ChromaDB 和 Faiss 等知识库。

MindSQL 图表

🚀 安装

要安装 MindSQL，你可以使用 pip：

pip install mindsql

MindSQL 需要 Python 3.10 或更高版本。

💡 使用方法

# !pip install mindsql

from mindsql.core import MindSQLCore
from mindsql.databases import Sqlite
from mindsql.llms import GoogleGenAi
from mindsql.vectorstores import ChromaDB

# 添加你的配置
config = {"api_key": "你的API密钥"}

# 选择你想要使用的向量存储、LLM 和数据库
# 使用配置的 Llm、向量存储和数据库创建 MindSQLCore 实例
minds = MindSQLCore(
    llm=GoogleGenAi(config=config),
    vectorstore=ChromaDB(),
    database=Sqlite()
)

# 使用指定的 URL 创建数据库连接
connection = minds.database.create_connection(url="你的数据库连接URL")

# 将指定数据库中的所有数据定义语言（DDL）语句索引到向量存储中
minds.index_all_ddls(connection=connection, db_name='数据库名称')

# 从指定的示例路径批量索引问题-SQL 对
minds.index(bulk=True, path="你的问题-SQL示例.json")

# 向数据库提问并可视化结果
response = minds.ask_db(
    question="你的问题",
    connection=connection,
    visualize=True
)

# 从响应中提取并显示图表
chart = response["chart"]
chart.show()

# 关闭数据库连接
connection.close()