import asyncio
import orjson
import logging
import sys
from mcp.server import Server
from mcp.server.models import InitializationOptions
from mcp.types import TextContent, Tool
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s - %(levelname)s - %(message)s",
handlers=[logging.StreamHandler(sys.stderr)],
)
log = logging.getLogger("{{ package_name }}")
server = Server("{{ dist_name }}")
@server.list_tools()
async def list_tools() -> list[Tool]:
return [
Tool(
name="echo",
description="Return the provided text.",
inputSchema={
"type": "object",
"properties": {"text": {"type": "string"}},
"required": ["text"],
},
)
]
@server.call_tool()
async def call_tool(name: str, arguments: dict) -> list[TextContent]:
if name == "echo":
return [TextContent(type="text", text=orjson.dumps({"ok": True, "echo": arguments["text"]}).decode())]
return [TextContent(type="text", text=orjson.dumps({"ok": False, "error": f"unknown tool: {name}"}).decode())]
async def main() -> None:
log.info("Starting {{ project_name }} (stdio)...")
from mcp.server.stdio import stdio_server
async with stdio_server() as (read_stream, write_stream):
await server.run(
read_stream,
write_stream,
InitializationOptions(
server_name="{{ dist_name }}",
server_version="{{ version }}",
capabilities={"tools": {}, "logging": {}},
),
)
if __name__ == "__main__": # pragma: no cover
asyncio.run(main())
