|
team_table_document = '''team Table |
|
Stores information about NBA teams. |
|
CREATE TABLE IF NOT EXISTS "team" ( |
|
"id" TEXT PRIMARY KEY, -- Unique identifier for the team |
|
"full_name" TEXT, -- Full official name of the team (e.g., "Los Angeles Lakers") |
|
"abbreviation" TEXT, -- Shortened team name (e.g., "LAL") |
|
"nickname" TEXT, -- Commonly used nickname for the team (e.g., "Lakers") |
|
"city" TEXT, -- City where the team is based |
|
"state" TEXT, -- State where the team is located |
|
"year_founded" REAL -- Year the team was established |
|
);''' |
|
|
|
game_table_document = '''game Table |
|
Contains detailed statistics for each NBA game, including home and away team performance. |
|
CREATE TABLE IF NOT EXISTS "game" ( |
|
"season_id" TEXT, -- Season identifier, formatted as "2YYYY" (e.g., "21970" for the 1970 season) |
|
"team_id_home" TEXT, -- ID of the home team (matches "id" in team table) |
|
"team_abbreviation_home" TEXT, -- Abbreviation of the home team |
|
"team_name_home" TEXT, -- Full name of the home team |
|
"game_id" TEXT PRIMARY KEY, -- Unique identifier for the game |
|
"game_date" TIMESTAMP, -- Date the game was played (YYYY-MM-DD format) |
|
"matchup_home" TEXT, -- Matchup details including opponent (e.g., "LAL vs. BOS") |
|
"wl_home" TEXT, -- "W" if the home team won, "L" if they lost |
|
"min" INTEGER, -- Total minutes played in the game |
|
"fgm_home" REAL, -- Field goals made by the home team |
|
"fga_home" REAL, -- Field goals attempted by the home team |
|
"fg_pct_home" REAL, -- Field goal percentage of the home team |
|
"fg3m_home" REAL, -- Three-point field goals made by the home team |
|
"fg3a_home" REAL, -- Three-point attempts by the home team |
|
"fg3_pct_home" REAL, -- Three-point field goal percentage of the home team |
|
"ftm_home" REAL, -- Free throws made by the home team |
|
"fta_home" REAL, -- Free throws attempted by the home team |
|
"ft_pct_home" REAL, -- Free throw percentage of the home team |
|
"oreb_home" REAL, -- Offensive rebounds by the home team |
|
"dreb_home" REAL, -- Defensive rebounds by the home team |
|
"reb_home" REAL, -- Total rebounds by the home team |
|
"ast_home" REAL, -- Assists by the home team |
|
"stl_home" REAL, -- Steals by the home team |
|
"blk_home" REAL, -- Blocks by the home team |
|
"tov_home" REAL, -- Turnovers by the home team |
|
"pf_home" REAL, -- Personal fouls by the home team |
|
"pts_home" REAL, -- Total points scored by the home team |
|
"plus_minus_home" INTEGER, -- Plus/minus rating for the home team |
|
"video_available_home" INTEGER, -- Indicates whether video is available (1 = Yes, 0 = No) |
|
"team_id_away" TEXT, -- ID of the away team |
|
"team_abbreviation_away" TEXT, -- Abbreviation of the away team |
|
"team_name_away" TEXT, -- Full name of the away team |
|
"matchup_away" TEXT, -- Matchup details from the away team’s perspective |
|
"wl_away" TEXT, -- "W" if the away team won, "L" if they lost |
|
"fgm_away" REAL, -- Field goals made by the away team |
|
"fga_away" REAL, -- Field goals attempted by the away team |
|
"fg_pct_away" REAL, -- Field goal percentage of the away team |
|
"fg3m_away" REAL, -- Three-point field goals made by the away team |
|
"fg3a_away" REAL, -- Three-point attempts by the away team |
|
"fg3_pct_away" REAL, -- Three-point field goal percentage of the away team |
|
"ftm_away" REAL, -- Free throws made by the away team |
|
"fta_away" REAL, -- Free throws attempted by the away team |
|
"ft_pct_away" REAL, -- Free throw percentage of the away team |
|
"oreb_away" REAL, -- Offensive rebounds by the away team |
|
"dreb_away" REAL, -- Defensive rebounds by the away team |
|
"reb_away" REAL, -- Total rebounds by the away team |
|
"ast_away" REAL, -- Assists by the away team |
|
"stl_away" REAL, -- Steals by the away team |
|
"blk_away" REAL, -- Blocks by the away team |
|
"tov_away" REAL, -- Turnovers by the away team |
|
"pf_away" REAL, -- Personal fouls by the away team |
|
"pts_away" REAL, -- Total points scored by the away team |
|
"plus_minus_away" INTEGER, -- Plus/minus rating for the away team |
|
"video_available_away" INTEGER, -- Indicates whether video is available (1 = Yes, 0 = No) |
|
"season_type" TEXT -- Regular season or playoffs |
|
); |
|
''' |
|
|
|
other_stats_table_document = '''other_stats Table |
|
Stores additional statistics, linked to the game table via game_id. |
|
CREATE TABLE IF NOT EXISTS "other_stats" ( |
|
"game_id" TEXT, -- Unique game identifier, matches id column from game table |
|
"league_id" TEXT, -- League identifier |
|
"team_id_home" TEXT, -- Home team identifier |
|
"team_abbreviation_home" TEXT, -- Home team abbreviation |
|
"team_city_home" TEXT, -- Home team city |
|
"pts_paint_home" INTEGER, -- Points in the paint by the home team |
|
"pts_2nd_chance_home" INTEGER, -- Second chance points by the home team |
|
"pts_fb_home" INTEGER, -- Fast break points by the home team |
|
"largest_lead_home" INTEGER,-- Largest lead by the home team |
|
"lead_changes" INTEGER, -- Number of lead changes |
|
"times_tied" INTEGER, -- Number of times the score was tied |
|
"team_turnovers_home" INTEGER, -- Home team turnovers |
|
"total_turnovers_home" INTEGER, -- Total turnovers by the home team |
|
"team_rebounds_home" INTEGER, -- Home team rebounds |
|
"pts_off_to_home" INTEGER, -- Points off turnovers by the home team |
|
"team_id_away" TEXT, -- Away team identifier |
|
"team_abbreviation_away" TEXT, -- Away team abbreviation |
|
"pts_paint_away" INTEGER, -- Points in the paint by the away team |
|
"pts_2nd_chance_away" INTEGER, -- Second chance points by the away team |
|
"pts_fb_away" INTEGER, -- Fast break points by the away team |
|
"largest_lead_away" INTEGER,-- Largest lead by the away team |
|
"team_turnovers_away" INTEGER, -- Away team turnovers |
|
"total_turnovers_away" INTEGER, -- Total turnovers by the away team |
|
"team_rebounds_away" INTEGER, -- Away team rebounds |
|
"pts_off_to_away" INTEGER -- Points off turnovers by the away team |
|
); |
|
''' |
|
|
|
team_name_document = '''Team Name Information |
|
In plaintext user questions, only the full team names will be used, but in the queries you may use either full names or abbreviations. |
|
Full names are used with the game table, while abbreviations should be used with the other_stats table. |
|
Team names and abbreviations (separated by |): |
|
Atlanta Hawks|ATL, Boston Celtics|BOS, Cleveland Cavaliers|CLE, New Orleans Pelicans|NOP, |
|
Chicago Bulls|CHI, Dallas Mavericks|DAL, Denver Nuggets|DEN, Golden State Warriors|GSW, |
|
Houston Rockets|HOU, Los Angeles Clippers|LAC, Los Angeles Lakers|LAL, Miami Heat|MIA, |
|
Milwaukee Bucks|MIL, Minnesota Timberwolves|MIN, Brooklyn Nets|BKN, New York Knicks|NYK, |
|
Orlando Magic|ORL, Indiana Pacers|IND, Philadelphia 76ers|PHI, Phoenix Suns|PHX, |
|
Portland Trail Blazers|POR, Sacramento Kings|SAC, San Antonio Spurs|SAS, |
|
Oklahoma City Thunder|OKC, Toronto Raptors|TOR, Utah Jazz|UTA, Memphis Grizzlies|MEM, |
|
Washington Wizards|WAS, Detroit Pistons|DET, Charlotte Hornets|CHA |
|
''' |
|
|
|
def retrieve_doc(has_team_schema, has_game_schema, has_other_stats_schema, has_team_names = True): |
|
documents = [] |
|
|
|
if has_team_schema: |
|
documents.append(team_table_document) |
|
if has_game_schema: |
|
documents.append(game_table_document) |
|
if has_other_stats_schema: |
|
documents.append(other_stats_table_document) |
|
if has_team_names: |
|
documents.append(team_name_document) |
|
|
|
return documents |
|
|
