🧠 MindSQL

MindSQL is a Python RAG (Retrieval-Augmented Generation) Library designed to streamline the interaction between users and their databases using just a few lines of code. With seamless integration for renowned databases such as PostgreSQL, MySQL, and SQLite, MindSQL also extends its capabilities to major databases like Snowflake and BigQuery by extending the IDatabase Interface. This library utilizes large language models (LLM) like GPT-4, Llama 2, Google Gemini, and supports knowledge bases like ChromaDB and Faiss.

🚀 Installation

To install MindSQL, you can use pip:

pip install mindsql

MindSQL requires Python 3.10 or higher.

💡 Usage

# !pip install mindsql

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

# Add Your Configurations
config = {"api_key": "YOUR-API-KEY"}

# Choose the Vector Store. LLM and DB You Want to Work With And
# Create MindSQLCore Instance With Configured Llm, Vectorstore, And Database
minds = MindSQLCore(
    llm=GoogleGenAi(config=config),
    vectorstore=ChromaDB(),
    database=Sqlite()
)

# Create a Database Connection Using The Specified URL
connection = minds.database.create_connection(url="YOUR_DATABASE_CONNECTION_URL")

# Index All Data Definition Language (DDL) Statements in The Specified Database Into The Vectorstore
minds.index_all_ddls(connection=connection, db_name='NAME_OF_THE_DB')

# Index Question-Sql Pair in Bulk From the Specified Example Path
minds.index(bulk=True, path="your-qsn-sql-example.json")

# Ask a Question to The Database And Visualize The Result
response = minds.ask_db(
    question="YOUR_QUESTION",
    connection=connection,
    visualize=True
)

# Extract And Display The Chart From The Response
chart = response["chart"]
chart.show()

# Close The Connection to Your DB
connection.close()

📁 Code Structure

_utils: Utility modules containing constants and a logger.
_helper: The helper module.
core: The main core module, minds_core.py.
databases: Database-related modules.
llms: Modules related to Language Models.
testing: Testing scripts.
vectorstores: Modules related to vector stores.
poetry.lock and pyproject.toml: Poetry dependencies and configuration files.
tests: Testcases.

🤝 Contributing Guidelines

Thank you for considering contributing to our project! Please follow these guidelines for smooth collaboration:

Fork the repository and create your branch from master.
Ensure your code adheres to our coding standards and conventions.
Test your changes thoroughly and add a test case in the tests folder.
Submit a pull request with a clear description of the problem and solution.

🐛 Bug Reports

If you encounter a bug while using MindSQL, help us resolve it by following these steps:

Check existing issues to see if the bug has been reported.
If not, open a new issue with a detailed description, including steps to reproduce and relevant screenshots or error messages.

🚀 Feature Requests

We welcome suggestions for new features or improvements to MindSQL. Here's how you can request a new feature:

Check existing feature requests to avoid duplication.
If your feature request is unique, open a new issue and describe the feature you would like to see.
Provide as much context and detail as possible to help us understand your request.

📣 Feedback

We value your feedback and strive to improve MindSQL. Here's how you can share your thoughts with us:

Open an issue to provide general feedback, suggestions, or comments.
Be constructive and specific in your feedback to help us understand your perspective better.

Thank you for your interest in contributing to our project! We appreciate your support and look forward to working with you. 🚀