🤖AI-Powered Data Queries: Ask Questions in Plain English, Get Instant Results!🚀
Prashant Iyer
Posted on June 11, 2024
You've might have heard of SQL. It's a widely used programming language for storing and processing information in relational databases - simply put, relational databases store data in tables, where each row stores an entity and each column stores an attribute for that entity.
Let's say we have a table called customers
in a relational database. If I wanted to access the names of all customers (customer_names
) that have an annual_spend
of at least $1000, I would have to formulate an SQL query like this:
SELECT customer_names FROM customers WHERE annual_spend >= 1000
I would then run this query against the database to access my results.
But what if AI 🤖 could do all this for us?
LLMWare allows us to do just that, making use of small language models such as slim-sql-1b-v0, which is only 1 billion parameters in size.
SLIM SQL Tool
We'll be making use of the SLIM (Structured Language Instruction Model) SQL Tool, which is a GGUF quantized version of the slim-sql-1b-v0 model. This essentially means that our Tool is of a smaller scale than the original model. To our advantage, it doesn't require much computational power to run, so it can run locally on a CPU without an internet connection or a GPU!
This Tool is specialized in small, fast, local prototyping and is effective for SQL operations that involve a single table. The Tool enables us to ask our questions about data entirely in natural language and still get accurate results! Let's look at an example of how to do this from start to finish.
Step 1: Loading our Model 🪫🔋
We'll start off by loading in our SLIM SQL Tool. Here, we check to see if the model is already downloaded locally, if not, we download it using the ModelCatalog
class.
sql_tool_repo_path = os.path.join(LLMWareConfig().get_model_repo_path(), "slim-sql-tool")
if not os.path.exists(sql_tool_repo_path):
ModelCatalog().load_model("llmware/slim-sql-tool")
Step 2: Loading our Data 📊
We then load in the sample customer_table.csv
file containing our data. This sample file is provided by the llmware
library!
files = os.listdir(sql_tool_repo_path)
csv_file = "customer_table.csv"
Next, we create a new SQL table called customer1
from our CSV file using the SQLTables
class provided by the llmware
library. This is important because we can run SQL queries on an SQL table, but not on a CSV file!
sql_db = SQLTables(experimental=True)
sql_db.create_new_table_from_csv(sql_tool_repo_path, csv_file, table_name="customer1")
print("update: successfully created new db table")
Setting experimental=True
will use a provided testing database to create the table in.
Step 3: Querying with AI 🤖
We're finally getting to the good stuff! Now that we have our model and data, we can begin to ask our questions and get back some results.
We'll first load our agent, which is an instance of the LLMfx
class. This class provides us a way to interact with various models through function calls. Our agent will take a natural language input from the user, communicate it to the appropriate model, generate an SQL query, run that query against the database, and then return the results of the query to us. Essentially, this is where the magic happens!
agent = LLMfx()
agent.load_tool("sql", sample=False, get_logits=True, temperature=0.0)
Next, we create a list of natural language questions we're going to be asking given our customer data. Let's see if our agent can answer them!
query_list = ["What is the highest annual spend of any customer?",
"Which customer has account number 1234953",
"Which customer has the lowest annual spend?"]
We can loop through each of the queries, and let the query_db()
function do all the work for us. All our results will be stored in the agent object.
for i, query in enumerate(query_list):
response = agent.query_db(query, table=table_name)
Results! ✅
Now that we have our results in the agent's research_list
, we can print them out.
for x in range(0,len(agent.research_list)):
print("research: ", x, agent.research_list[x])
For the example we have seen so far, this is what the output would look like for the first question ("What is the highest annual spend of any customer?").
The output is a dictionary containing a lot of detailed information about the steps carried out by the agent, but here are some of the more interesting parts of it:
-
db_response
gives us what we want, the answer to the question! In this case, the response is 93540, meaning that the highest annual spend of any customer was $93540! -
sql_query
shows us the SQL query that was generated from our natural language question using the SLIM SQL tool. In this case, the query generated was: ```SQL
SELECT MAX(annual_spend) FROM customer1
---
## Conclusion
**And just like that, we've done it!** All we gave the program was a list of natural language questions and a CSV file with data. Behind the scenes, the `llmware` library:
1. created a table in a database with our data,
2. passed our questions into an AI model to get SQL queries,
3. ran the queries against the database, and
4. returned the results of the queries!
And if you're still not impressed, **remember that we can run this example locally on just a CPU 💻**!
Check out our YouTube video on this topic to see us explain the source code and analyze the results!
If you made it this far, thank you for taking the time to go through this topic with us ❤️! For more content like this, make sure to [visit our dev.to page](https://dev.to/llmware).
The source code for many more examples like this one are on [our GitHub](https://github.com/llmware-ai/llmware). Find this example [here](https://github.com/llmware-ai/llmware/blob/main/examples/SLIM-Agents/text2sql-end-to-end-2.py).
Our repository also contains a [notebook for this example](https://github.com/llmware-ai/llmware/blob/main/examples/Notebooks/NoteBook_Examples/text2sql-end-to-end-2-notebook.ipynb) that you can run yourself using Google Colab, Jupyter or any other platform that supports .ipynb notebooks.
[Join our Discord](https://discord.gg/fCztJQeV7J) to interact with a growing community of AI enthusiasts of all levels of experience!
Please be sure to visit our website [llmware.ai](https://llmware.ai/) for more information and updates.
Posted on June 11, 2024
Join Our Newsletter. No Spam, Only the good stuff.
Sign up to receive the latest update from our blog.
Related
June 11, 2024