get_player_stats, get_team_info)"Estadísticas de Lebron James en la temporada 2023-24"get_player_stats | player_name='LeBron James', season=2023-24
1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3tokenizer = AutoTokenizer.from_pretrained("joanmii/nba-agent-flan-t5")
4model = AutoModelForSeq2SeqLM.from_pretrained("joanmii/nba-agent-flan-t5")
5
6# Consulta de ejemplo
7query = "triples dobles de russell westbrook en playoffs"
8inputs = tokenizer(query, return_tensors="pt", truncation=True)
9outputs = model.generate(
10 **inputs,
11 max_new_tokens=64,
12 do_sample=False,
13 num_beams=1
14)
15result = tokenizer.decode(outputs[0], skip_special_tokens=True)
16print(result)
17# Salida: get_triple_doubles | player_name='Russell Westbrook', season_type='Playoffs'get_seasons - Temporadas disponibles de un jugadorget_player_info - Información básica de jugadores (altura, peso, equipo, etc.)get_player_stats - Estadísticas detalladas por temporada/carreraget_player_awards - Premios y reconocimientos (MVP, DPOY, etc.)get_player_games - Registro de partidos individualesget_high_low - Máximos y mínimos estadísticos de un jugadorget_best_game - Mejores (o peores) partidos de un jugadorcount_games - Contar partidos con condiciones específicasget_triple_doubles - Triples dobles y dobles-doblesget_all_time_leaders - Líderes históricos de la NBAget_league_leaders - Líderes de temporada específicaget_team_info - Información general de equiposget_franchise_leaders - Líderes históricos por franquiciaget_team_roster - Plantillas actuales por temporadaget_team_year_by_year_stats - Estadísticas históricas de equiposget_league_standings - Clasificaciones de ligaget_games - Búsqueda de partidos por criteriosget_game - Detalles de partidos específicosget_game_stats - Estadísticas detalladas de un partidoget_awards - Premios anuales (MVP, ROY, etc.)get_draft_history - Historial del draft NBAget_nba_champions - Campeones históricos de la NBAget_playoffs - Información de playoffs por año| Consulta | Función + Parámetros |
|---|---|
| "MVP de Stephen Curry" | get_player_awards | player_name='Stephen Curry', award_type='MVP' |
| "Roster de Lakers 2024" | get_team_roster | team_name='Los Angeles Lakers', season='2024-25' |
| "Estadísticas de Michael Jordan" | get_player_stats | player_name='Michael Jordan' |
| "Triples dobles de Westbrook playoffs" | get_triple_doubles | player_name='Russell Westbrook', season_type='Playoffs' |
| "Mejores anotadores 2023" | get_league_leaders | stat='PTS', season='2023-24' |
| "Draft de 2003" | get_draft_history | season='2003' |
| "Campeones NBA 2016" | get_nba_champions | year=2016 |
| "Partidos Lakers vs Warriors" | get_games | team1='Los Angeles Lakers', team2='Golden State Warriors' |
| "Líderes históricos en rebotes" | get_all_time_leaders | stat='REB' |
| "Playoffs 2024" | get_playoffs | year=2024 |
función | parámetro1=valor1, parámetro2=valor2