1823aae891Co-authored-by: Matt <77928207+mattzcarey@users.noreply.github.com> Co-authored-by: Konstantin Konstantinov <KKonstantinov@users.noreply.github.com> Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
65 lines
2.7 KiB
TypeScript
65 lines
2.7 KiB
TypeScript
/**
|
|
* Tools primitive — start here.
|
|
*
|
|
* Register tools with `McpServer.registerTool`: typed input via any
|
|
* Standard-Schema-with-JSON library (Zod here), inferred output schema +
|
|
* `structuredContent` from `outputSchema`, `annotations` for behavioral hints
|
|
* (`readOnlyHint`, `destructiveHint`). One binary, either transport.
|
|
*/
|
|
import { createServer } from 'node:http';
|
|
|
|
import { parseExampleArgs } from '@mcp-examples/shared';
|
|
import { toNodeHandler } from '@modelcontextprotocol/node';
|
|
import type { CallToolResult } from '@modelcontextprotocol/server';
|
|
import { createMcpHandler, McpServer } from '@modelcontextprotocol/server';
|
|
import { serveStdio } from '@modelcontextprotocol/server/stdio';
|
|
import * as z from 'zod/v4';
|
|
|
|
function buildServer(): McpServer {
|
|
const server = new McpServer({ name: 'tools-example', version: '1.0.0' });
|
|
|
|
// A read-only tool with typed input and inferred structured output.
|
|
server.registerTool(
|
|
'calc',
|
|
{
|
|
title: 'Calculator',
|
|
description: 'Apply an arithmetic operation to two numbers',
|
|
inputSchema: z.object({
|
|
op: z.enum(['add', 'sub', 'mul']).describe('the operation to apply'),
|
|
a: z.number().describe('left operand'),
|
|
b: z.number().describe('right operand')
|
|
}),
|
|
outputSchema: z.object({ op: z.string(), result: z.number() }),
|
|
annotations: { readOnlyHint: true, idempotentHint: true },
|
|
// Icons a client may render in its UI. `src` is required;
|
|
// `mimeType`, `sizes`, and `theme` are optional hints.
|
|
icons: [{ src: 'https://example.test/calc.svg', mimeType: 'image/svg+xml', sizes: ['any'] }]
|
|
},
|
|
async ({ op, a, b }) => {
|
|
const result = op === 'add' ? a + b : op === 'sub' ? a - b : a * b;
|
|
const structuredContent = { op, result };
|
|
return { content: [{ type: 'text', text: `${a} ${op} ${b} = ${result}` }], structuredContent };
|
|
}
|
|
);
|
|
|
|
// A plain string-returning tool (no structuredContent).
|
|
server.registerTool(
|
|
'echo',
|
|
{ description: 'Echoes the input', inputSchema: z.object({ text: z.string() }) },
|
|
async ({ text }): Promise<CallToolResult> => ({ content: [{ type: 'text', text }] })
|
|
);
|
|
|
|
return server;
|
|
}
|
|
|
|
const { transport, port } = parseExampleArgs();
|
|
|
|
if (transport === 'stdio') {
|
|
void serveStdio(buildServer);
|
|
console.error('[server] serving over stdio');
|
|
} else {
|
|
const handler = createMcpHandler(buildServer);
|
|
createServer(toNodeHandler(handler)).listen(port, () => {
|
|
console.error(`[server] listening on http://127.0.0.1:${port}/mcp`);
|
|
});
|
|
}
|