Newer
Older
navi-1 / tests / _mcp_test_server.py
"""Minimal MCP server used only by integration tests."""

import os

from mcp.server.fastmcp import FastMCP

mcp = FastMCP("test-server")


@mcp.tool()
def hello(name: str) -> str:
    """Say hello to someone."""
    return f"Hello, {name}!"


@mcp.tool()
def add(a: int, b: int) -> int:
    """Add two numbers."""
    return a + b


@mcp.tool()
def echo_env(name: str) -> str:
    """Return the value of an environment variable this server was launched with."""
    return os.environ.get(name, "")


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