SQL-Generation / src /rag /table_documents.py
licesma's picture
Add second RAG notebook
1860a94
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
);
'''