| 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 |
|
|
