# team_server.py — a tiny MCP SERVER you can reuse in Station 7 (Langflow) or Station 8 (code).
# It exposes a private "team database" as MCP tools. Run it, and any MCP-aware agent can call it.
#
# Setup (once):   pip install mcp
# Run standalone: python3 team_server.py       (talks MCP over stdin/stdout)
# In Langflow:    MCP Tools component → Stdio → command: python3   args: team_server.py
# In code (ADK):  MCPToolset(StdioConnectionParams(StdioServerParameters(
#                     command="python3", args=["team_server.py"])))

from mcp.server.fastmcp import FastMCP

mcp = FastMCP("team")

ROLES = {"maya": "Team Lead", "leo": "Data Analyst", "ada": "Designer"}


@mcp.tool()
def get_member_role(name: str) -> str:
    """Look up a team member's role by first name."""
    return ROLES.get(name.lower(), "Unknown member")


@mcp.tool()
def list_members() -> list:
    """List every team member's first name."""
    return list(ROLES.keys())


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