server.py

import httpx
from mcp.server.fastmcp import FastMCP
import json

# Initialize FastMCP server
mcp = FastMCP("Pokemon")

@mcp.tool()
async def get_pokemon_details(pokemon_name: str) -> str:
    """
    Retrieve detailed information about a Pokémon.

    Args:
        pokemon_name (str): The name of the Pokémon to retrieve information for.
    """

    async with httpx.AsyncClient() as client:
        try:
            response = await client.get(f"https://pokeapi.co/api/v2/pokemon/{pokemon_name}", timeout=30.0)
            response.raise_for_status()

            pokemon = response.json()
            pokemon_details = {
                "name": pokemon_name,
                "abilities": [a["ability"]["name"] for a in pokemon["abilities"]],
                "weight": pokemon["weight"],
                "height": pokemon["height"],
                "moves": [move["move"]["name"] for move in pokemon["moves"]],
                "pokemon_type": [t["type"]["name"] for t in pokemon["types"]],
                "species": pokemon["species"]["name"],
                "stats": {s["stat"]["name"]: s["base_stat"] for s in pokemon["stats"]}
            }

            return json.dumps(pokemon_details)
        except Exception:
            return "Invalid Pokémon Name"

def main():
    print("Running pokemon-mcp-server!")
    mcp.run(transport='stdio')


if __name__ == "__main__":
    main()