NSQL-Text-To-SQL / example_strings.py
Johannes
add 6B model back in
b273a98
example1 = ["""CREATE TABLE stadium (
stadium_id number,
location text,
name text,
capacity number,
highest number,
lowest number,
average number
)
CREATE TABLE singer (
singer_id number,
name text,
country text,
song_name text,
song_release_year text,
age number,
is_male others
)
CREATE TABLE concert (
concert_id number,
concert_name text,
theme text,
stadium_id text,
year text
)
CREATE TABLE singer_in_concert (
concert_id number,
singer_id text
)""",
"Using valid SQLite, answer the following questions for the tables provided above.",
"What is the maximum, the average, and the minimum capacity of stadiums ?",
"nsql-6B"]
example2 = ["""CREATE TABLE stadium (
stadium_id number,
location text,
name text,
capacity number,
)""",
"Using valid SQLite, answer the following questions for the tables provided above.",
"how many stadiums in total?",
"nsql-2B"]
example3 = ["""CREATE TABLE work_orders (
ID NUMBER,
CREATED_AT TEXT,
COST FLOAT,
INVOICE_AMOUNT FLOAT,
IS_DUE BOOLEAN,
IS_OPEN BOOLEAN,
IS_OVERDUE BOOLEAN,
COUNTRY_NAME TEXT,
)""",
"Using valid SQLite, answer the following questions for the tables provided above.",
"how many work orders are open?",
"nsql-350M"]