import { AgentMailClient } from 'agentmail'
import { AgentMailToolkit } from 'agentmail-toolkit/mcp'
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js'
import { z } from 'zod'
export const configSchema = z.object({
apiKey: z.string().describe('Your API key from the [AgentMail Console](https://console.agentmail.to)'),
})
export default function createServer({ config }: { config: z.infer<typeof configSchema> }) {
const server = new McpServer({ name: 'AgentMail', version: '1.0.0' })
const client = new AgentMailClient({ apiKey: config.apiKey })
const toolkit = new AgentMailToolkit(client)
for (const tool of toolkit.getTools()) server.registerTool(tool.name, tool, tool.callback)
return server.server
}
