F1-MCP/server.py at main · devangpratap/F1-MCP · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
from mcp.server.fastmcp import FastMCP
import api

mcp = FastMCP("f1-mcp")


@mcp.tool()
def get_driver_standings() -> list:
    """Get the current F1 driver championship standings."""
    return api.get_current_season_standings()


@mcp.tool()
def get_constructor_standings() -> list:
    """Get the current F1 constructor (team) championship standings."""
    return api.get_constructor_standings()


@mcp.tool()
def get_season_schedule(season: str = "current") -> list:
    """Get the full race schedule for a season. Use 'current' for the ongoing season."""
    return api.get_season_schedule(season)


@mcp.tool()
def get_race_results(season: str, round_number: str) -> list:
    """Get race results for a specific round. Use season='current' for the ongoing season."""
    return api.get_race_results(season, round_number)


@mcp.tool()
def get_qualifying_results(season: str, round_number: str) -> list:
    """Get qualifying results for a specific round."""
    return api.get_qualifying_results(season, round_number)


@mcp.tool()
def get_lap_times(season: str, round_number: str, lap: str = None) -> list:
    """Get lap times for a race. Optionally filter to a specific lap number."""
    return api.get_lap_times(season, round_number, lap)


@mcp.tool()
def get_pit_stops(season: str, round_number: str) -> list:
    """Get pit stop data for a specific race."""
    return api.get_pit_stops(season, round_number)


@mcp.tool()
def get_sprint_results(season: str, round_number: str) -> list:
    """Get sprint race results for a specific round."""
    return api.get_sprint_results(season, round_number)


@mcp.tool()
def get_driver_season_results(season: str, driver_id: str) -> list:
    """Get all race results for a driver in a season. driver_id is lowercase e.g. 'leclerc', 'norris'."""
    return api.get_driver_season_results(season, driver_id)


@mcp.tool()
def get_circuits(season: str = "current") -> list:
    """Get all circuits for a season."""
    return api.get_circuits(season)


@mcp.tool()
def get_drivers(season: str = "current") -> list:
    """Get all drivers for a season."""
    return api.get_drivers(season)


@mcp.tool()
def get_finish_statuses() -> list:
    """Get all possible finish status codes e.g. Finished, DNF, DNS."""
    return api.get_finish_statuses()


@mcp.resource("f1://standings/drivers")
def standings_resource() -> str:
    """Current driver standings as a readable resource."""
    standings = api.get_current_season_standings()
    lines = [f"{s['position']}. {s['driver']} — {s['points']} pts ({s['team']})" for s in standings]
    return "\n".join(lines)


@mcp.resource("f1://standings/constructors")
def constructor_standings_resource() -> str:
    """Current constructor standings as a readable resource."""
    standings = api.get_constructor_standings()
    lines = [f"{s['position']}. {s['team']} — {s['points']} pts" for s in standings]
    return "\n".join(lines)


if __name__ == "__main__":
    mcp.run()