import { Server } from "@modelcontextprotocol/sdk/server/index.js";
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import { z } from "zod";

const AddToolSchema = {
  name: "add",
  description: "Add two numbers together",
  parameters: z.object({
    a: z.number(),
    b: z.number(),
  }),
};

const server = new Server({ name: "sdk-example-server", version: "1.0.0" }, {
  capabilities: { tools: {} } 
});

server.setToolHandler(AddToolSchema, async (params) => {
  const { a, b } = params;
  const sum = a + b;
  return {
    content: [{ type: "text", text: String(sum) }]
  };
});

const transport = new StdioServerTransport();
await server.connect(transport);
console.log("SDK Server connected via stdio");