Quickstart with your own data
This guide helps you learn how to:
- Add your own dataset to Wren AI
- Define relationships between tables
Add your own data
Wren AI can establish connections to your dataset by either linking to a database or reading from files. For more information about connecting to a data source, please refer to Connect Data Sources Overview.
In this tutorial, we will use DuckDB to illustrate how to add datasets from CSV files in the following steps.
1. Prepare your Dataset
Prepare your dataset files. Here we use the e-commerce dataset files for example.
See the DuckDB documentation for more information about our supported file formats.
Download our sample data to try out: e-commerce.tar
2. Create a folder in container
docker exec wrenai-wren-engine-1 bash -c "mkdir -p /usr/src/app/etc/data"
3. Copy the archived file into container
docker cp ./e-commerce.tar wrenai-wren-engine-1:/usr/src/app/etc/data/
4. Extracting the archived file
docker exec wrenai-wren-engine-1 bash -c "tar xvf /usr/src/app/etc/data/e-commerce.tar -C /usr/src/app/etc/data"
5. Read files via DuckDB connection
Click on the DuckDB option
1. Fill in the connnection properties
- Display name: Name this connection, here we fill in 'e-commerce'.
- initial SQL statements: Use SQL statement to read your dataset.
CREATE TABLE reviews AS SELECT * FROM read_csv('./etc/data/e-commerce/reviews.csv', header = true);
CREATE TABLE customers AS SELECT * FROM read_csv('./etc/data/e-commerce/customers.csv', header = true);
CREATE TABLE orders AS SELECT * FROM read_csv('./etc/data/e-commerce/orders.csv', header = true);
Click Next button to start connection.
Select tables
Each CSV file will be created as a table, select all tables and click Next.
Define Relationship
Click the +Add button to define relationships between tables in order to make the data model visible.
1. Add the following relationship in customer table:
| From | To | Type |
|---|---|---|
| customers.Id | orders.CustomerId | One-to-Many |
2. Add the following relationship in orders table:
| From | To | Type |
|---|---|---|
| orders.OrderId | reviews.OrderId | One-to-Many |
Click Finish to submit your settings.
Understand your data in Modeling page
Ask questions in the Home page
Feel free to ask Wren AI any question.
