agentmail-toolkit 0.1.20 → 0.1.22

package/dist/langchain.js

@@ -28,9 +28,8 @@ var import_tools2 = require("@langchain/core/tools");
 // src/wrapper.ts
 var import_agentmail = require("agentmail");
 var Wrapper = class {
-  constructor(client) {
+  constructor(client = new import_agentmail.AgentMailClient()) {
     this.client = client;
-    if (!this.client) this.client = new import_agentmail.AgentMailClient();
   }
   async call(method, args) {
     const parts = method.split(".");
@@ -55,52 +54,47 @@ var Wrapper = class {
 // src/schemas.ts
 var import_zod = require("zod");
 var ListItemsParams = import_zod.z.object({
-  limit: import_zod.z.number().optional().describe("The maximum number of items to return"),
-  last_key: import_zod.z.string().optional().describe("The last key to use for pagination")
+  limit: import_zod.z.number().optional().describe("Max number of items to return"),
+  page_token: import_zod.z.string().optional().describe("Page token for pagination")
 });
 var GetInboxParams = import_zod.z.object({
-  inbox_id: import_zod.z.string().describe("The ID of the inbox to get")
+  inbox_id: import_zod.z.string().describe("ID of inbox to get")
 });
 var CreateInboxParams = import_zod.z.object({
-  username: import_zod.z.string().optional().describe("The username of the inbox to create"),
-  domain: import_zod.z.string().optional().describe("The domain of the inbox to create"),
-  display_name: import_zod.z.string().optional().describe("The display name of the inbox to create")
+  username: import_zod.z.string().optional().describe("Username of inbox to create"),
+  domain: import_zod.z.string().optional().describe("Domain of inbox to create"),
+  display_name: import_zod.z.string().optional().describe("Display name of inbox to create")
 });
 var ListThreadsParams = ListItemsParams.extend({
-  inbox_id: import_zod.z.string().describe("The ID of the inbox to list threads for"),
-  labels: import_zod.z.array(import_zod.z.string()).optional().describe("The labels to filter threads by")
+  inbox_id: import_zod.z.string().describe("ID of inbox to list threads from"),
+  labels: import_zod.z.array(import_zod.z.string()).optional().describe("Labels to filter threads by")
 });
 var GetThreadParams = import_zod.z.object({
-  inbox_id: import_zod.z.string().describe("The ID of the inbox to get the thread for"),
-  thread_id: import_zod.z.string().describe("The ID of the thread to get")
+  inbox_id: import_zod.z.string().describe("ID of inbox to get thread from"),
+  thread_id: import_zod.z.string().describe("ID of thread to get")
 });
 var ListMessagesParams = ListItemsParams.extend({
-  inbox_id: import_zod.z.string().describe("The ID of the inbox to list messages for"),
-  labels: import_zod.z.array(import_zod.z.string()).optional().describe("The labels to filter messages by")
+  inbox_id: import_zod.z.string().describe("ID of inbox to list messages from"),
+  labels: import_zod.z.array(import_zod.z.string()).optional().describe("Labels to filter messages by")
 });
 var GetMessageParams = import_zod.z.object({
-  inbox_id: import_zod.z.string().describe("The ID of the inbox to get the message for"),
-  message_id: import_zod.z.string().describe("The ID of the message to get")
-});
-var GetAttachmentParams = import_zod.z.object({
-  inbox_id: import_zod.z.string().describe("The ID of the inbox to get the attachment for"),
-  message_id: import_zod.z.string().describe("The ID of the message to get the attachment for"),
-  attachment_id: import_zod.z.string().describe("The ID of the attachment to get")
+  inbox_id: import_zod.z.string().describe("ID of inbox to get message from"),
+  message_id: import_zod.z.string().describe("ID of message to get")
 });
 var SendMessageParams = import_zod.z.object({
-  inbox_id: import_zod.z.string().describe("The ID of the inbox to send the message from"),
-  to: import_zod.z.array(import_zod.z.string()).describe("The list of recipients"),
-  cc: import_zod.z.array(import_zod.z.string()).optional().describe("The list of CC recipients"),
-  bcc: import_zod.z.array(import_zod.z.string()).optional().describe("The list of BCC recipients"),
-  subject: import_zod.z.string().optional().describe("The subject of the message"),
-  text: import_zod.z.string().optional().describe("The plain text body of the message"),
-  html: import_zod.z.string().optional().describe("The HTML body of the message")
+  inbox_id: import_zod.z.string().describe("ID of inbox to send message from"),
+  to: import_zod.z.array(import_zod.z.string()).describe("Recipients of message"),
+  cc: import_zod.z.array(import_zod.z.string()).optional().describe("CC recipients of message"),
+  bcc: import_zod.z.array(import_zod.z.string()).optional().describe("BCC recipients of message"),
+  subject: import_zod.z.string().optional().describe("Subject of message"),
+  text: import_zod.z.string().optional().describe("Plain text body of message"),
+  html: import_zod.z.string().optional().describe("HTML body of message")
 });
 var ReplyToMessageParams = import_zod.z.object({
-  inbox_id: import_zod.z.string().describe("The inbox ID of the inbox to reply to the message from"),
-  message_id: import_zod.z.string().describe("The message ID of the message you wish to reply to"),
-  text: import_zod.z.string().optional().describe("The plain text body of the reply"),
-  html: import_zod.z.string().optional().describe("The HTML body of the reply")
+  inbox_id: import_zod.z.string().describe("ID of inbox to reply to message from"),
+  message_id: import_zod.z.string().describe("ID of message to reply to"),
+  text: import_zod.z.string().optional().describe("Plain text body of reply"),
+  html: import_zod.z.string().optional().describe("HTML body of reply")
 });
 
 // src/tools.ts
@@ -108,82 +102,85 @@ var tools = [
   {
     name: "list_inboxes",
     method: "inboxes.list",
-    description: "List all inboxes",
+    description: "List inboxes",
     schema: ListItemsParams
   },
   {
     name: "get_inbox",
     method: "inboxes.get",
-    description: "Get an inbox by ID",
+    description: "Get inbox",
     schema: GetInboxParams
   },
   {
     name: "create_inbox",
     method: "inboxes.create",
-    description: "Create a new inbox",
+    description: "Create inbox. Use default username, domain, and display name unless otherwise specified.",
     schema: CreateInboxParams
   },
   {
     name: "list_threads",
-    method: "threads.list",
-    description: "List all threads",
+    method: "inboxes.threads.list",
+    description: "List threads in inbox",
     schema: ListThreadsParams
   },
   {
     name: "get_thread",
-    method: "threads.get",
-    description: "Get a thread by ID",
+    method: "inboxes.threads.get",
+    description: "Get thread",
     schema: GetThreadParams
   },
   {
     name: "list_messages",
-    method: "messages.list",
-    description: "List all messages",
+    method: "inboxes.messages.list",
+    description: "List messages in inbox",
     schema: ListMessagesParams
   },
   {
     name: "get_message",
-    method: "messages.get",
-    description: "Get a message by ID",
+    method: "inboxes.messages.get",
+    description: "Get message",
     schema: GetMessageParams
   },
-  {
-    name: "get_attachment",
-    method: "attachments.get",
-    description: "Get an attachment by ID",
-    schema: GetAttachmentParams
-  },
   {
     name: "send_message",
-    method: "messages.send",
-    description: "Send a message",
+    method: "inboxes.messages.send",
+    description: "Send message",
     schema: SendMessageParams
   },
   {
     name: "reply_to_message",
-    method: "messages.reply",
-    description: "Reply to a message",
+    method: "inboxes.messages.reply",
+    description: "Reply to message",
     schema: ReplyToMessageParams
   }
 ];
 
 // src/toolkit.ts
-var ListToolkit = class extends Wrapper {
-  tools = [];
+var BaseToolkit = class extends Wrapper {
+  tools = {};
   constructor(client) {
     super(client);
-    this.tools = tools.map((tool) => this.buildTool(tool));
+    this.tools = tools.reduce(
+      (acc, tool) => {
+        acc[tool.name] = this.buildTool(tool);
+        return acc;
+      },
+      {}
+    );
   }
-  getTools() {
-    return this.tools;
+};
+var ListToolkit = class extends BaseToolkit {
+  getTools(names) {
+    if (!names) return Object.values(this.tools);
+    return names.reduce((acc, name) => {
+      if (name in this.tools) acc.push(this.tools[name]);
+      return acc;
+    }, []);
   }
 };
 
 // src/langchain.ts
 var AgentMailToolkit = class extends ListToolkit {
-  constructor(client) {
-    super(client);
-  }
   buildTool(tool) {
     return (0, import_tools2.tool)(async (args) => JSON.stringify((await this.safeCall(tool.method, args)).result, null, 2), {
       name: tool.name,

package/dist/langchain.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/langchain.ts","../src/wrapper.ts","../src/schemas.ts","../src/tools.ts","../src/toolkit.ts"],"sourcesContent":["import { type StructuredTool, tool as langchainTool } from '@langchain/core/tools'\nimport { AgentMailClient } from 'agentmail'\n\nimport { ListToolkit } from './toolkit'\nimport { type Tool } from './tools'\n\nexport class AgentMailToolkit extends ListToolkit<StructuredTool> {\n    constructor(client?: AgentMailClient) {\n        super(client)\n    }\n\n    protected buildTool(tool: Tool): StructuredTool {\n        return langchainTool(async (args) => JSON.stringify((await this.safeCall(tool.method, args)).result, null, 2), {\n            name: tool.name,\n            description: tool.description,\n            schema: tool.schema,\n        })\n    }\n}\n","import { AgentMailClient } from 'agentmail'\n\nexport class Wrapper {\n    constructor(private readonly client?: AgentMailClient) {\n        if (!this.client) this.client = new AgentMailClient()\n    }\n\n    public async call(method: string, args: any) {\n        const parts = method.split('.')\n        const methodKey = parts.pop()\n\n        if (!methodKey) throw new Error('Method name empty')\n\n        let parent: any = this.client\n        for (const part of parts) parent = parent[part]\n\n        const { inboxId, ...options } = args\n\n        if (inboxId) return await parent[methodKey].call(parent, inboxId, options)\n        else return await parent[methodKey].call(parent, options)\n    }\n\n    public async safeCall(method: string, args: any) {\n        try {\n            return { isError: false, result: await this.call(method, args) }\n        } catch (error) {\n            if (error instanceof Error) return { isError: true, result: error.message }\n            else return { isError: true, result: 'Unknown error' }\n        }\n    }\n}\n","import { z } from 'zod'\n\nexport const ListItemsParams = z.object({\n    limit: z.number().optional().describe('The maximum number of items to return'),\n    last_key: z.string().optional().describe('The last key to use for pagination'),\n})\n\nexport const GetInboxParams = z.object({\n    inbox_id: z.string().describe('The ID of the inbox to get'),\n})\n\nexport const CreateInboxParams = z.object({\n    username: z.string().optional().describe('The username of the inbox to create'),\n    domain: z.string().optional().describe('The domain of the inbox to create'),\n    display_name: z.string().optional().describe('The display name of the inbox to create'),\n})\n\nexport const ListThreadsParams = ListItemsParams.extend({\n    inbox_id: z.string().describe('The ID of the inbox to list threads for'),\n    labels: z.array(z.string()).optional().describe('The labels to filter threads by'),\n})\n\nexport const GetThreadParams = z.object({\n    inbox_id: z.string().describe('The ID of the inbox to get the thread for'),\n    thread_id: z.string().describe('The ID of the thread to get'),\n})\n\nexport const ListMessagesParams = ListItemsParams.extend({\n    inbox_id: z.string().describe('The ID of the inbox to list messages for'),\n    labels: z.array(z.string()).optional().describe('The labels to filter messages by'),\n})\n\nexport const GetMessageParams = z.object({\n    inbox_id: z.string().describe('The ID of the inbox to get the message for'),\n    message_id: z.string().describe('The ID of the message to get'),\n})\n\nexport const GetAttachmentParams = z.object({\n    inbox_id: z.string().describe('The ID of the inbox to get the attachment for'),\n    message_id: z.string().describe('The ID of the message to get the attachment for'),\n    attachment_id: z.string().describe('The ID of the attachment to get'),\n})\n\nexport const SendMessageParams = z.object({\n    inbox_id: z.string().describe('The ID of the inbox to send the message from'),\n    to: z.array(z.string()).describe('The list of recipients'),\n    cc: z.array(z.string()).optional().describe('The list of CC recipients'),\n    bcc: z.array(z.string()).optional().describe('The list of BCC recipients'),\n    subject: z.string().optional().describe('The subject of the message'),\n    text: z.string().optional().describe('The plain text body of the message'),\n    html: z.string().optional().describe('The HTML body of the message'),\n})\n\nexport const ReplyToMessageParams = z.object({\n    inbox_id: z.string().describe('The inbox ID of the inbox to reply to the message from'),\n    message_id: z.string().describe('The message ID of the message you wish to reply to'),\n    text: z.string().optional().describe('The plain text body of the reply'),\n    html: z.string().optional().describe('The HTML body of the reply'),\n})\n","import { AnyZodObject } from 'zod'\n\nimport {\n    ListItemsParams,\n    GetInboxParams,\n    CreateInboxParams,\n    ListThreadsParams,\n    GetThreadParams,\n    ListMessagesParams,\n    GetMessageParams,\n    GetAttachmentParams,\n    SendMessageParams,\n    ReplyToMessageParams,\n} from './schemas'\n\nexport type Tool = {\n    name: string\n    method: string\n    description: string\n    schema: AnyZodObject\n}\n\nexport const tools: Tool[] = [\n    {\n        name: 'list_inboxes',\n        method: 'inboxes.list',\n        description: 'List all inboxes',\n        schema: ListItemsParams,\n    },\n    {\n        name: 'get_inbox',\n        method: 'inboxes.get',\n        description: 'Get an inbox by ID',\n        schema: GetInboxParams,\n    },\n    {\n        name: 'create_inbox',\n        method: 'inboxes.create',\n        description: 'Create a new inbox',\n        schema: CreateInboxParams,\n    },\n    {\n        name: 'list_threads',\n        method: 'threads.list',\n        description: 'List all threads',\n        schema: ListThreadsParams,\n    },\n    {\n        name: 'get_thread',\n        method: 'threads.get',\n        description: 'Get a thread by ID',\n        schema: GetThreadParams,\n    },\n    {\n        name: 'list_messages',\n        method: 'messages.list',\n        description: 'List all messages',\n        schema: ListMessagesParams,\n    },\n    {\n        name: 'get_message',\n        method: 'messages.get',\n        description: 'Get a message by ID',\n        schema: GetMessageParams,\n    },\n    {\n        name: 'get_attachment',\n        method: 'attachments.get',\n        description: 'Get an attachment by ID',\n        schema: GetAttachmentParams,\n    },\n    {\n        name: 'send_message',\n        method: 'messages.send',\n        description: 'Send a message',\n        schema: SendMessageParams,\n    },\n    {\n        name: 'reply_to_message',\n        method: 'messages.reply',\n        description: 'Reply to a message',\n        schema: ReplyToMessageParams,\n    },\n]\n","import { AgentMailClient } from 'agentmail'\n\nimport { Wrapper } from './wrapper'\nimport { type Tool, tools } from './tools'\n\nexport abstract class ListToolkit<T> extends Wrapper {\n    protected readonly tools: T[] = []\n\n    constructor(client?: AgentMailClient) {\n        super(client)\n\n        this.tools = tools.map((tool) => this.buildTool(tool))\n    }\n\n    protected abstract buildTool(tool: Tool): T\n\n    public getTools() {\n        return this.tools\n    }\n}\n\nexport abstract class MapToolkit<T> extends Wrapper {\n    protected readonly tools: Record<string, T> = {}\n\n    constructor(client?: AgentMailClient) {\n        super(client)\n\n        this.tools = tools.reduce(\n            (acc, tool) => {\n                acc[tool.name] = this.buildTool(tool)\n                return acc\n            },\n            {} as Record<string, T>\n        )\n    }\n\n    protected abstract buildTool(tool: Tool): T\n\n    public getTools() {\n        return this.tools\n    }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,IAAAA,gBAA2D;;;ACA3D,uBAAgC;AAEzB,IAAM,UAAN,MAAc;AAAA,EACjB,YAA6B,QAA0B;AAA1B;AACzB,QAAI,CAAC,KAAK,OAAQ,MAAK,SAAS,IAAI,iCAAgB;AAAA,EACxD;AAAA,EAEA,MAAa,KAAK,QAAgB,MAAW;AACzC,UAAM,QAAQ,OAAO,MAAM,GAAG;AAC9B,UAAM,YAAY,MAAM,IAAI;AAE5B,QAAI,CAAC,UAAW,OAAM,IAAI,MAAM,mBAAmB;AAEnD,QAAI,SAAc,KAAK;AACvB,eAAW,QAAQ,MAAO,UAAS,OAAO,IAAI;AAE9C,UAAM,EAAE,SAAS,GAAG,QAAQ,IAAI;AAEhC,QAAI,QAAS,QAAO,MAAM,OAAO,SAAS,EAAE,KAAK,QAAQ,SAAS,OAAO;AAAA,QACpE,QAAO,MAAM,OAAO,SAAS,EAAE,KAAK,QAAQ,OAAO;AAAA,EAC5D;AAAA,EAEA,MAAa,SAAS,QAAgB,MAAW;AAC7C,QAAI;AACA,aAAO,EAAE,SAAS,OAAO,QAAQ,MAAM,KAAK,KAAK,QAAQ,IAAI,EAAE;AAAA,IACnE,SAAS,OAAO;AACZ,UAAI,iBAAiB,MAAO,QAAO,EAAE,SAAS,MAAM,QAAQ,MAAM,QAAQ;AAAA,UACrE,QAAO,EAAE,SAAS,MAAM,QAAQ,gBAAgB;AAAA,IACzD;AAAA,EACJ;AACJ;;;AC9BA,iBAAkB;AAEX,IAAM,kBAAkB,aAAE,OAAO;AAAA,EACpC,OAAO,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,uCAAuC;AAAA,EAC7E,UAAU,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,oCAAoC;AACjF,CAAC;AAEM,IAAM,iBAAiB,aAAE,OAAO;AAAA,EACnC,UAAU,aAAE,OAAO,EAAE,SAAS,4BAA4B;AAC9D,CAAC;AAEM,IAAM,oBAAoB,aAAE,OAAO;AAAA,EACtC,UAAU,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,qCAAqC;AAAA,EAC9E,QAAQ,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,mCAAmC;AAAA,EAC1E,cAAc,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,yCAAyC;AAC1F,CAAC;AAEM,IAAM,oBAAoB,gBAAgB,OAAO;AAAA,EACpD,UAAU,aAAE,OAAO,EAAE,SAAS,yCAAyC;AAAA,EACvE,QAAQ,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,iCAAiC;AACrF,CAAC;AAEM,IAAM,kBAAkB,aAAE,OAAO;AAAA,EACpC,UAAU,aAAE,OAAO,EAAE,SAAS,2CAA2C;AAAA,EACzE,WAAW,aAAE,OAAO,EAAE,SAAS,6BAA6B;AAChE,CAAC;AAEM,IAAM,qBAAqB,gBAAgB,OAAO;AAAA,EACrD,UAAU,aAAE,OAAO,EAAE,SAAS,0CAA0C;AAAA,EACxE,QAAQ,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,kCAAkC;AACtF,CAAC;AAEM,IAAM,mBAAmB,aAAE,OAAO;AAAA,EACrC,UAAU,aAAE,OAAO,EAAE,SAAS,4CAA4C;AAAA,EAC1E,YAAY,aAAE,OAAO,EAAE,SAAS,8BAA8B;AAClE,CAAC;AAEM,IAAM,sBAAsB,aAAE,OAAO;AAAA,EACxC,UAAU,aAAE,OAAO,EAAE,SAAS,+CAA+C;AAAA,EAC7E,YAAY,aAAE,OAAO,EAAE,SAAS,iDAAiD;AAAA,EACjF,eAAe,aAAE,OAAO,EAAE,SAAS,iCAAiC;AACxE,CAAC;AAEM,IAAM,oBAAoB,aAAE,OAAO;AAAA,EACtC,UAAU,aAAE,OAAO,EAAE,SAAS,8CAA8C;AAAA,EAC5E,IAAI,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,wBAAwB;AAAA,EACzD,IAAI,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,2BAA2B;AAAA,EACvE,KAAK,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,4BAA4B;AAAA,EACzE,SAAS,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,4BAA4B;AAAA,EACpE,MAAM,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,oCAAoC;AAAA,EACzE,MAAM,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,8BAA8B;AACvE,CAAC;AAEM,IAAM,uBAAuB,aAAE,OAAO;AAAA,EACzC,UAAU,aAAE,OAAO,EAAE,SAAS,wDAAwD;AAAA,EACtF,YAAY,aAAE,OAAO,EAAE,SAAS,oDAAoD;AAAA,EACpF,MAAM,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,kCAAkC;AAAA,EACvE,MAAM,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,4BAA4B;AACrE,CAAC;;;ACpCM,IAAM,QAAgB;AAAA,EACzB;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AACJ;;;AC9EO,IAAe,cAAf,cAAsC,QAAQ;AAAA,EAC9B,QAAa,CAAC;AAAA,EAEjC,YAAY,QAA0B;AAClC,UAAM,MAAM;AAEZ,SAAK,QAAQ,MAAM,IAAI,CAAC,SAAS,KAAK,UAAU,IAAI,CAAC;AAAA,EACzD;AAAA,EAIO,WAAW;AACd,WAAO,KAAK;AAAA,EAChB;AACJ;;;AJbO,IAAM,mBAAN,cAA+B,YAA4B;AAAA,EAC9D,YAAY,QAA0B;AAClC,UAAM,MAAM;AAAA,EAChB;AAAA,EAEU,UAAU,MAA4B;AAC5C,eAAO,cAAAC,MAAc,OAAO,SAAS,KAAK,WAAW,MAAM,KAAK,SAAS,KAAK,QAAQ,IAAI,GAAG,QAAQ,MAAM,CAAC,GAAG;AAAA,MAC3G,MAAM,KAAK;AAAA,MACX,aAAa,KAAK;AAAA,MAClB,QAAQ,KAAK;AAAA,IACjB,CAAC;AAAA,EACL;AACJ;","names":["import_tools","langchainTool"]}
+{"version":3,"sources":["../src/langchain.ts","../src/wrapper.ts","../src/schemas.ts","../src/tools.ts","../src/toolkit.ts"],"sourcesContent":["import { type StructuredTool, tool as langchainTool } from '@langchain/core/tools'\n\nimport { ListToolkit } from './toolkit'\nimport { type Tool } from './tools'\n\nexport class AgentMailToolkit extends ListToolkit<StructuredTool> {\n    protected buildTool(tool: Tool): StructuredTool {\n        return langchainTool(async (args) => JSON.stringify((await this.safeCall(tool.method, args)).result, null, 2), {\n            name: tool.name,\n            description: tool.description,\n            schema: tool.schema,\n        })\n    }\n}\n","import { AgentMailClient } from 'agentmail'\n\nexport class Wrapper {\n    constructor(private readonly client = new AgentMailClient()) {}\n\n    public async call(method: string, args: any) {\n        const parts = method.split('.')\n        const methodKey = parts.pop()\n\n        if (!methodKey) throw new Error('Method name empty')\n\n        let parent: any = this.client\n        for (const part of parts) parent = parent[part]\n\n        const { inboxId, ...options } = args\n\n        if (inboxId) return await parent[methodKey].call(parent, inboxId, options)\n        else return await parent[methodKey].call(parent, options)\n    }\n\n    public async safeCall(method: string, args: any) {\n        try {\n            return { isError: false, result: await this.call(method, args) }\n        } catch (error) {\n            if (error instanceof Error) return { isError: true, result: error.message }\n            else return { isError: true, result: 'Unknown error' }\n        }\n    }\n}\n","import { z } from 'zod'\n\nexport const ListItemsParams = z.object({\n    limit: z.number().optional().describe('Max number of items to return'),\n    page_token: z.string().optional().describe('Page token for pagination'),\n})\n\nexport const GetInboxParams = z.object({\n    inbox_id: z.string().describe('ID of inbox to get'),\n})\n\nexport const CreateInboxParams = z.object({\n    username: z.string().optional().describe('Username of inbox to create'),\n    domain: z.string().optional().describe('Domain of inbox to create'),\n    display_name: z.string().optional().describe('Display name of inbox to create'),\n})\n\nexport const ListThreadsParams = ListItemsParams.extend({\n    inbox_id: z.string().describe('ID of inbox to list threads from'),\n    labels: z.array(z.string()).optional().describe('Labels to filter threads by'),\n})\n\nexport const GetThreadParams = z.object({\n    inbox_id: z.string().describe('ID of inbox to get thread from'),\n    thread_id: z.string().describe('ID of thread to get'),\n})\n\nexport const ListMessagesParams = ListItemsParams.extend({\n    inbox_id: z.string().describe('ID of inbox to list messages from'),\n    labels: z.array(z.string()).optional().describe('Labels to filter messages by'),\n})\n\nexport const GetMessageParams = z.object({\n    inbox_id: z.string().describe('ID of inbox to get message from'),\n    message_id: z.string().describe('ID of message to get'),\n})\n\nexport const SendMessageParams = z.object({\n    inbox_id: z.string().describe('ID of inbox to send message from'),\n    to: z.array(z.string()).describe('Recipients of message'),\n    cc: z.array(z.string()).optional().describe('CC recipients of message'),\n    bcc: z.array(z.string()).optional().describe('BCC recipients of message'),\n    subject: z.string().optional().describe('Subject of message'),\n    text: z.string().optional().describe('Plain text body of message'),\n    html: z.string().optional().describe('HTML body of message'),\n})\n\nexport const ReplyToMessageParams = z.object({\n    inbox_id: z.string().describe('ID of inbox to reply to message from'),\n    message_id: z.string().describe('ID of message to reply to'),\n    text: z.string().optional().describe('Plain text body of reply'),\n    html: z.string().optional().describe('HTML body of reply'),\n})\n","import { AnyZodObject } from 'zod'\n\nimport {\n    ListItemsParams,\n    GetInboxParams,\n    CreateInboxParams,\n    ListThreadsParams,\n    GetThreadParams,\n    ListMessagesParams,\n    GetMessageParams,\n    SendMessageParams,\n    ReplyToMessageParams,\n} from './schemas'\n\nexport interface Tool {\n    name: string\n    method: string\n    description: string\n    schema: AnyZodObject\n}\n\nexport const tools: Tool[] = [\n    {\n        name: 'list_inboxes',\n        method: 'inboxes.list',\n        description: 'List inboxes',\n        schema: ListItemsParams,\n    },\n    {\n        name: 'get_inbox',\n        method: 'inboxes.get',\n        description: 'Get inbox',\n        schema: GetInboxParams,\n    },\n    {\n        name: 'create_inbox',\n        method: 'inboxes.create',\n        description: 'Create inbox. Use default username, domain, and display name unless otherwise specified.',\n        schema: CreateInboxParams,\n    },\n    {\n        name: 'list_threads',\n        method: 'inboxes.threads.list',\n        description: 'List threads in inbox',\n        schema: ListThreadsParams,\n    },\n    {\n        name: 'get_thread',\n        method: 'inboxes.threads.get',\n        description: 'Get thread',\n        schema: GetThreadParams,\n    },\n    {\n        name: 'list_messages',\n        method: 'inboxes.messages.list',\n        description: 'List messages in inbox',\n        schema: ListMessagesParams,\n    },\n    {\n        name: 'get_message',\n        method: 'inboxes.messages.get',\n        description: 'Get message',\n        schema: GetMessageParams,\n    },\n    {\n        name: 'send_message',\n        method: 'inboxes.messages.send',\n        description: 'Send message',\n        schema: SendMessageParams,\n    },\n    {\n        name: 'reply_to_message',\n        method: 'inboxes.messages.reply',\n        description: 'Reply to message',\n        schema: ReplyToMessageParams,\n    },\n]\n","import { AgentMailClient } from 'agentmail'\n\nimport { Wrapper } from './wrapper'\nimport { type Tool, tools } from './tools'\n\nexport abstract class BaseToolkit<T> extends Wrapper {\n    protected readonly tools: Record<string, T> = {}\n\n    constructor(client?: AgentMailClient) {\n        super(client)\n\n        this.tools = tools.reduce(\n            (acc, tool) => {\n                acc[tool.name] = this.buildTool(tool)\n                return acc\n            },\n            {} as Record<string, T>\n        )\n    }\n\n    protected abstract buildTool(tool: Tool): T\n}\n\nexport abstract class ListToolkit<T> extends BaseToolkit<T> {\n    public getTools(names?: string[]) {\n        if (!names) return Object.values(this.tools)\n\n        return names.reduce((acc, name) => {\n            if (name in this.tools) acc.push(this.tools[name])\n            return acc\n        }, [] as T[])\n    }\n}\n\nexport abstract class MapToolkit<T> extends BaseToolkit<T> {\n    public getTools(names?: string[]) {\n        if (!names) return this.tools\n\n        return names.reduce(\n            (acc, name) => {\n                if (name in this.tools) acc[name] = this.tools[name]\n                return acc\n            },\n            {} as Record<string, T>\n        )\n    }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,IAAAA,gBAA2D;;;ACA3D,uBAAgC;AAEzB,IAAM,UAAN,MAAc;AAAA,EACjB,YAA6B,SAAS,IAAI,iCAAgB,GAAG;AAAhC;AAAA,EAAiC;AAAA,EAE9D,MAAa,KAAK,QAAgB,MAAW;AACzC,UAAM,QAAQ,OAAO,MAAM,GAAG;AAC9B,UAAM,YAAY,MAAM,IAAI;AAE5B,QAAI,CAAC,UAAW,OAAM,IAAI,MAAM,mBAAmB;AAEnD,QAAI,SAAc,KAAK;AACvB,eAAW,QAAQ,MAAO,UAAS,OAAO,IAAI;AAE9C,UAAM,EAAE,SAAS,GAAG,QAAQ,IAAI;AAEhC,QAAI,QAAS,QAAO,MAAM,OAAO,SAAS,EAAE,KAAK,QAAQ,SAAS,OAAO;AAAA,QACpE,QAAO,MAAM,OAAO,SAAS,EAAE,KAAK,QAAQ,OAAO;AAAA,EAC5D;AAAA,EAEA,MAAa,SAAS,QAAgB,MAAW;AAC7C,QAAI;AACA,aAAO,EAAE,SAAS,OAAO,QAAQ,MAAM,KAAK,KAAK,QAAQ,IAAI,EAAE;AAAA,IACnE,SAAS,OAAO;AACZ,UAAI,iBAAiB,MAAO,QAAO,EAAE,SAAS,MAAM,QAAQ,MAAM,QAAQ;AAAA,UACrE,QAAO,EAAE,SAAS,MAAM,QAAQ,gBAAgB;AAAA,IACzD;AAAA,EACJ;AACJ;;;AC5BA,iBAAkB;AAEX,IAAM,kBAAkB,aAAE,OAAO;AAAA,EACpC,OAAO,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,+BAA+B;AAAA,EACrE,YAAY,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,2BAA2B;AAC1E,CAAC;AAEM,IAAM,iBAAiB,aAAE,OAAO;AAAA,EACnC,UAAU,aAAE,OAAO,EAAE,SAAS,oBAAoB;AACtD,CAAC;AAEM,IAAM,oBAAoB,aAAE,OAAO;AAAA,EACtC,UAAU,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,6BAA6B;AAAA,EACtE,QAAQ,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,2BAA2B;AAAA,EAClE,cAAc,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,iCAAiC;AAClF,CAAC;AAEM,IAAM,oBAAoB,gBAAgB,OAAO;AAAA,EACpD,UAAU,aAAE,OAAO,EAAE,SAAS,kCAAkC;AAAA,EAChE,QAAQ,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,6BAA6B;AACjF,CAAC;AAEM,IAAM,kBAAkB,aAAE,OAAO;AAAA,EACpC,UAAU,aAAE,OAAO,EAAE,SAAS,gCAAgC;AAAA,EAC9D,WAAW,aAAE,OAAO,EAAE,SAAS,qBAAqB;AACxD,CAAC;AAEM,IAAM,qBAAqB,gBAAgB,OAAO;AAAA,EACrD,UAAU,aAAE,OAAO,EAAE,SAAS,mCAAmC;AAAA,EACjE,QAAQ,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,8BAA8B;AAClF,CAAC;AAEM,IAAM,mBAAmB,aAAE,OAAO;AAAA,EACrC,UAAU,aAAE,OAAO,EAAE,SAAS,iCAAiC;AAAA,EAC/D,YAAY,aAAE,OAAO,EAAE,SAAS,sBAAsB;AAC1D,CAAC;AAEM,IAAM,oBAAoB,aAAE,OAAO;AAAA,EACtC,UAAU,aAAE,OAAO,EAAE,SAAS,kCAAkC;AAAA,EAChE,IAAI,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,uBAAuB;AAAA,EACxD,IAAI,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,0BAA0B;AAAA,EACtE,KAAK,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,2BAA2B;AAAA,EACxE,SAAS,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,oBAAoB;AAAA,EAC5D,MAAM,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,4BAA4B;AAAA,EACjE,MAAM,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,sBAAsB;AAC/D,CAAC;AAEM,IAAM,uBAAuB,aAAE,OAAO;AAAA,EACzC,UAAU,aAAE,OAAO,EAAE,SAAS,sCAAsC;AAAA,EACpE,YAAY,aAAE,OAAO,EAAE,SAAS,2BAA2B;AAAA,EAC3D,MAAM,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,0BAA0B;AAAA,EAC/D,MAAM,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,oBAAoB;AAC7D,CAAC;;;AC/BM,IAAM,QAAgB;AAAA,EACzB;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AACJ;;;ACvEO,IAAe,cAAf,cAAsC,QAAQ;AAAA,EAC9B,QAA2B,CAAC;AAAA,EAE/C,YAAY,QAA0B;AAClC,UAAM,MAAM;AAEZ,SAAK,QAAQ,MAAM;AAAA,MACf,CAAC,KAAK,SAAS;AACX,YAAI,KAAK,IAAI,IAAI,KAAK,UAAU,IAAI;AACpC,eAAO;AAAA,MACX;AAAA,MACA,CAAC;AAAA,IACL;AAAA,EACJ;AAGJ;AAEO,IAAe,cAAf,cAAsC,YAAe;AAAA,EACjD,SAAS,OAAkB;AAC9B,QAAI,CAAC,MAAO,QAAO,OAAO,OAAO,KAAK,KAAK;AAE3C,WAAO,MAAM,OAAO,CAAC,KAAK,SAAS;AAC/B,UAAI,QAAQ,KAAK,MAAO,KAAI,KAAK,KAAK,MAAM,IAAI,CAAC;AACjD,aAAO;AAAA,IACX,GAAG,CAAC,CAAQ;AAAA,EAChB;AACJ;;;AJ3BO,IAAM,mBAAN,cAA+B,YAA4B;AAAA,EACpD,UAAU,MAA4B;AAC5C,eAAO,cAAAC,MAAc,OAAO,SAAS,KAAK,WAAW,MAAM,KAAK,SAAS,KAAK,QAAQ,IAAI,GAAG,QAAQ,MAAM,CAAC,GAAG;AAAA,MAC3G,MAAM,KAAK;AAAA,MACX,aAAa,KAAK;AAAA,MAClB,QAAQ,KAAK;AAAA,IACjB,CAAC;AAAA,EACL;AACJ;","names":["import_tools","langchainTool"]}

package/dist/langchain.mjs

@@ -1,15 +1,11 @@
 import {
   ListToolkit
-} from "./chunk-4F4JVYOE.mjs";
-import "./chunk-XAJA2AYA.mjs";
-import "./chunk-QBOE34GE.mjs";
+} from "./chunk-W2LJOBDV.mjs";
+import "./chunk-CIU3KR25.mjs";
 
 // src/langchain.ts
 import { tool as langchainTool } from "@langchain/core/tools";
 var AgentMailToolkit = class extends ListToolkit {
-  constructor(client) {
-    super(client);
-  }
   buildTool(tool) {
     return langchainTool(async (args) => JSON.stringify((await this.safeCall(tool.method, args)).result, null, 2), {
       name: tool.name,

package/dist/langchain.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/langchain.ts"],"sourcesContent":["import { type StructuredTool, tool as langchainTool } from '@langchain/core/tools'\nimport { AgentMailClient } from 'agentmail'\n\nimport { ListToolkit } from './toolkit'\nimport { type Tool } from './tools'\n\nexport class AgentMailToolkit extends ListToolkit<StructuredTool> {\n    constructor(client?: AgentMailClient) {\n        super(client)\n    }\n\n    protected buildTool(tool: Tool): StructuredTool {\n        return langchainTool(async (args) => JSON.stringify((await this.safeCall(tool.method, args)).result, null, 2), {\n            name: tool.name,\n            description: tool.description,\n            schema: tool.schema,\n        })\n    }\n}\n"],"mappings":";;;;;;;AAAA,SAA8B,QAAQ,qBAAqB;AAMpD,IAAM,mBAAN,cAA+B,YAA4B;AAAA,EAC9D,YAAY,QAA0B;AAClC,UAAM,MAAM;AAAA,EAChB;AAAA,EAEU,UAAU,MAA4B;AAC5C,WAAO,cAAc,OAAO,SAAS,KAAK,WAAW,MAAM,KAAK,SAAS,KAAK,QAAQ,IAAI,GAAG,QAAQ,MAAM,CAAC,GAAG;AAAA,MAC3G,MAAM,KAAK;AAAA,MACX,aAAa,KAAK;AAAA,MAClB,QAAQ,KAAK;AAAA,IACjB,CAAC;AAAA,EACL;AACJ;","names":[]}
+{"version":3,"sources":["../src/langchain.ts"],"sourcesContent":["import { type StructuredTool, tool as langchainTool } from '@langchain/core/tools'\n\nimport { ListToolkit } from './toolkit'\nimport { type Tool } from './tools'\n\nexport class AgentMailToolkit extends ListToolkit<StructuredTool> {\n    protected buildTool(tool: Tool): StructuredTool {\n        return langchainTool(async (args) => JSON.stringify((await this.safeCall(tool.method, args)).result, null, 2), {\n            name: tool.name,\n            description: tool.description,\n            schema: tool.schema,\n        })\n    }\n}\n"],"mappings":";;;;;;AAAA,SAA8B,QAAQ,qBAAqB;AAKpD,IAAM,mBAAN,cAA+B,YAA4B;AAAA,EACpD,UAAU,MAA4B;AAC5C,WAAO,cAAc,OAAO,SAAS,KAAK,WAAW,MAAM,KAAK,SAAS,KAAK,QAAQ,IAAI,GAAG,QAAQ,MAAM,CAAC,GAAG;AAAA,MAC3G,MAAM,KAAK;AAAA,MACX,aAAa,KAAK;AAAA,MAClB,QAAQ,KAAK;AAAA,IACjB,CAAC;AAAA,EACL;AACJ;","names":[]}

package/dist/mcp.d.mts

@@ -1,9 +1,17 @@
-import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
-import { AgentMailClient } from 'agentmail';
+import { ToolCallback } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { ZodRawShape } from 'zod';
+import { L as ListToolkit } from './toolkit-BLAprE02.mjs';
+import { T as Tool } from './tools-Dzp_FZN_.mjs';
+import 'agentmail';
 
-declare class AgentMailMcpServer extends McpServer {
-    private wrapper;
-    constructor(client?: AgentMailClient);
+interface McpTool {
+    name: string;
+    description: string;
+    paramsSchema: ZodRawShape;
+    callback: ToolCallback<ZodRawShape>;
+}
+declare class AgentMailToolkit extends ListToolkit<McpTool> {
+    protected buildTool(tool: Tool): McpTool;
 }
 
-export { AgentMailMcpServer };
+export { AgentMailToolkit };

package/dist/mcp.d.ts

@@ -1,9 +1,17 @@
-import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
-import { AgentMailClient } from 'agentmail';
+import { ToolCallback } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { ZodRawShape } from 'zod';
+import { L as ListToolkit } from './toolkit-3F468CU6.js';
+import { T as Tool } from './tools-Dzp_FZN_.js';
+import 'agentmail';
 
-declare class AgentMailMcpServer extends McpServer {
-    private wrapper;
-    constructor(client?: AgentMailClient);
+interface McpTool {
+    name: string;
+    description: string;
+    paramsSchema: ZodRawShape;
+    callback: ToolCallback<ZodRawShape>;
+}
+declare class AgentMailToolkit extends ListToolkit<McpTool> {
+    protected buildTool(tool: Tool): McpTool;
 }
 
-export { AgentMailMcpServer };
+export { AgentMailToolkit };

package/dist/mcp.js

@@ -20,17 +20,15 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/mcp.ts
 var mcp_exports = {};
 __export(mcp_exports, {
-  AgentMailMcpServer: () => AgentMailMcpServer
+  AgentMailToolkit: () => AgentMailToolkit
 });
 module.exports = __toCommonJS(mcp_exports);
-var import_mcp = require("@modelcontextprotocol/sdk/server/mcp.js");
 
 // src/wrapper.ts
 var import_agentmail = require("agentmail");
 var Wrapper = class {
-  constructor(client) {
+  constructor(client = new import_agentmail.AgentMailClient()) {
     this.client = client;
-    if (!this.client) this.client = new import_agentmail.AgentMailClient();
   }
   async call(method, args) {
     const parts = method.split(".");
@@ -55,52 +53,47 @@ var Wrapper = class {
 // src/schemas.ts
 var import_zod = require("zod");
 var ListItemsParams = import_zod.z.object({
-  limit: import_zod.z.number().optional().describe("The maximum number of items to return"),
-  last_key: import_zod.z.string().optional().describe("The last key to use for pagination")
+  limit: import_zod.z.number().optional().describe("Max number of items to return"),
+  page_token: import_zod.z.string().optional().describe("Page token for pagination")
 });
 var GetInboxParams = import_zod.z.object({
-  inbox_id: import_zod.z.string().describe("The ID of the inbox to get")
+  inbox_id: import_zod.z.string().describe("ID of inbox to get")
 });
 var CreateInboxParams = import_zod.z.object({
-  username: import_zod.z.string().optional().describe("The username of the inbox to create"),
-  domain: import_zod.z.string().optional().describe("The domain of the inbox to create"),
-  display_name: import_zod.z.string().optional().describe("The display name of the inbox to create")
+  username: import_zod.z.string().optional().describe("Username of inbox to create"),
+  domain: import_zod.z.string().optional().describe("Domain of inbox to create"),
+  display_name: import_zod.z.string().optional().describe("Display name of inbox to create")
 });
 var ListThreadsParams = ListItemsParams.extend({
-  inbox_id: import_zod.z.string().describe("The ID of the inbox to list threads for"),
-  labels: import_zod.z.array(import_zod.z.string()).optional().describe("The labels to filter threads by")
+  inbox_id: import_zod.z.string().describe("ID of inbox to list threads from"),
+  labels: import_zod.z.array(import_zod.z.string()).optional().describe("Labels to filter threads by")
 });
 var GetThreadParams = import_zod.z.object({
-  inbox_id: import_zod.z.string().describe("The ID of the inbox to get the thread for"),
-  thread_id: import_zod.z.string().describe("The ID of the thread to get")
+  inbox_id: import_zod.z.string().describe("ID of inbox to get thread from"),
+  thread_id: import_zod.z.string().describe("ID of thread to get")
 });
 var ListMessagesParams = ListItemsParams.extend({
-  inbox_id: import_zod.z.string().describe("The ID of the inbox to list messages for"),
-  labels: import_zod.z.array(import_zod.z.string()).optional().describe("The labels to filter messages by")
+  inbox_id: import_zod.z.string().describe("ID of inbox to list messages from"),
+  labels: import_zod.z.array(import_zod.z.string()).optional().describe("Labels to filter messages by")
 });
 var GetMessageParams = import_zod.z.object({
-  inbox_id: import_zod.z.string().describe("The ID of the inbox to get the message for"),
-  message_id: import_zod.z.string().describe("The ID of the message to get")
-});
-var GetAttachmentParams = import_zod.z.object({
-  inbox_id: import_zod.z.string().describe("The ID of the inbox to get the attachment for"),
-  message_id: import_zod.z.string().describe("The ID of the message to get the attachment for"),
-  attachment_id: import_zod.z.string().describe("The ID of the attachment to get")
+  inbox_id: import_zod.z.string().describe("ID of inbox to get message from"),
+  message_id: import_zod.z.string().describe("ID of message to get")
 });
 var SendMessageParams = import_zod.z.object({
-  inbox_id: import_zod.z.string().describe("The ID of the inbox to send the message from"),
-  to: import_zod.z.array(import_zod.z.string()).describe("The list of recipients"),
-  cc: import_zod.z.array(import_zod.z.string()).optional().describe("The list of CC recipients"),
-  bcc: import_zod.z.array(import_zod.z.string()).optional().describe("The list of BCC recipients"),
-  subject: import_zod.z.string().optional().describe("The subject of the message"),
-  text: import_zod.z.string().optional().describe("The plain text body of the message"),
-  html: import_zod.z.string().optional().describe("The HTML body of the message")
+  inbox_id: import_zod.z.string().describe("ID of inbox to send message from"),
+  to: import_zod.z.array(import_zod.z.string()).describe("Recipients of message"),
+  cc: import_zod.z.array(import_zod.z.string()).optional().describe("CC recipients of message"),
+  bcc: import_zod.z.array(import_zod.z.string()).optional().describe("BCC recipients of message"),
+  subject: import_zod.z.string().optional().describe("Subject of message"),
+  text: import_zod.z.string().optional().describe("Plain text body of message"),
+  html: import_zod.z.string().optional().describe("HTML body of message")
 });
 var ReplyToMessageParams = import_zod.z.object({
-  inbox_id: import_zod.z.string().describe("The inbox ID of the inbox to reply to the message from"),
-  message_id: import_zod.z.string().describe("The message ID of the message you wish to reply to"),
-  text: import_zod.z.string().optional().describe("The plain text body of the reply"),
-  html: import_zod.z.string().optional().describe("The HTML body of the reply")
+  inbox_id: import_zod.z.string().describe("ID of inbox to reply to message from"),
+  message_id: import_zod.z.string().describe("ID of message to reply to"),
+  text: import_zod.z.string().optional().describe("Plain text body of reply"),
+  html: import_zod.z.string().optional().describe("HTML body of reply")
 });
 
 // src/tools.ts
@@ -108,84 +101,99 @@ var tools = [
   {
     name: "list_inboxes",
     method: "inboxes.list",
-    description: "List all inboxes",
+    description: "List inboxes",
     schema: ListItemsParams
   },
   {
     name: "get_inbox",
     method: "inboxes.get",
-    description: "Get an inbox by ID",
+    description: "Get inbox",
     schema: GetInboxParams
   },
   {
     name: "create_inbox",
     method: "inboxes.create",
-    description: "Create a new inbox",
+    description: "Create inbox. Use default username, domain, and display name unless otherwise specified.",
     schema: CreateInboxParams
   },
   {
     name: "list_threads",
-    method: "threads.list",
-    description: "List all threads",
+    method: "inboxes.threads.list",
+    description: "List threads in inbox",
     schema: ListThreadsParams
   },
   {
     name: "get_thread",
-    method: "threads.get",
-    description: "Get a thread by ID",
+    method: "inboxes.threads.get",
+    description: "Get thread",
     schema: GetThreadParams
   },
   {
     name: "list_messages",
-    method: "messages.list",
-    description: "List all messages",
+    method: "inboxes.messages.list",
+    description: "List messages in inbox",
     schema: ListMessagesParams
   },
   {
     name: "get_message",
-    method: "messages.get",
-    description: "Get a message by ID",
+    method: "inboxes.messages.get",
+    description: "Get message",
     schema: GetMessageParams
   },
-  {
-    name: "get_attachment",
-    method: "attachments.get",
-    description: "Get an attachment by ID",
-    schema: GetAttachmentParams
-  },
   {
     name: "send_message",
-    method: "messages.send",
-    description: "Send a message",
+    method: "inboxes.messages.send",
+    description: "Send message",
     schema: SendMessageParams
   },
   {
     name: "reply_to_message",
-    method: "messages.reply",
-    description: "Reply to a message",
+    method: "inboxes.messages.reply",
+    description: "Reply to message",
     schema: ReplyToMessageParams
   }
 ];
 
-// src/mcp.ts
-var AgentMailMcpServer = class extends import_mcp.McpServer {
-  wrapper;
+// src/toolkit.ts
+var BaseToolkit = class extends Wrapper {
+  tools = {};
   constructor(client) {
-    super({
-      name: "AgentMail",
-      version: "0.1.0"
-    });
-    this.wrapper = new Wrapper(client);
-    for (const tool of tools) {
-      this.tool(tool.name, tool.description, tool.schema.shape, async (args) => {
-        const { isError, result } = await this.wrapper.safeCall(tool.method, args);
+    super(client);
+    this.tools = tools.reduce(
+      (acc, tool) => {
+        acc[tool.name] = this.buildTool(tool);
+        return acc;
+      },
+      {}
+    );
+  }
+};
+var ListToolkit = class extends BaseToolkit {
+  getTools(names) {
+    if (!names) return Object.values(this.tools);
+    return names.reduce((acc, name) => {
+      if (name in this.tools) acc.push(this.tools[name]);
+      return acc;
+    }, []);
+  }
+};
+
+// src/mcp.ts
+var AgentMailToolkit = class extends ListToolkit {
+  buildTool(tool) {
+    return {
+      name: tool.name,
+      description: tool.description,
+      paramsSchema: tool.schema.shape,
+      callback: async (args) => {
+        const { isError, result } = await this.safeCall(tool.method, args);
         return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }], isError };
-      });
-    }
+      }
+    };
   }
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  AgentMailMcpServer
+  AgentMailToolkit
 });
 //# sourceMappingURL=mcp.js.map

package/dist/mcp.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/mcp.ts","../src/wrapper.ts","../src/schemas.ts","../src/tools.ts"],"sourcesContent":["import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js'\nimport { AgentMailClient } from 'agentmail'\n\nimport { Wrapper } from './wrapper'\nimport { tools } from './tools'\n\nexport class AgentMailMcpServer extends McpServer {\n    private wrapper: Wrapper\n\n    constructor(client?: AgentMailClient) {\n        super({\n            name: 'AgentMail',\n            version: '0.1.0',\n        })\n\n        this.wrapper = new Wrapper(client)\n\n        for (const tool of tools) {\n            this.tool(tool.name, tool.description, tool.schema.shape, async (args: unknown) => {\n                const { isError, result } = await this.wrapper.safeCall(tool.method, args)\n                return { content: [{ type: 'text' as const, text: JSON.stringify(result, null, 2) }], isError }\n            })\n        }\n    }\n}\n","import { AgentMailClient } from 'agentmail'\n\nexport class Wrapper {\n    constructor(private readonly client?: AgentMailClient) {\n        if (!this.client) this.client = new AgentMailClient()\n    }\n\n    public async call(method: string, args: any) {\n        const parts = method.split('.')\n        const methodKey = parts.pop()\n\n        if (!methodKey) throw new Error('Method name empty')\n\n        let parent: any = this.client\n        for (const part of parts) parent = parent[part]\n\n        const { inboxId, ...options } = args\n\n        if (inboxId) return await parent[methodKey].call(parent, inboxId, options)\n        else return await parent[methodKey].call(parent, options)\n    }\n\n    public async safeCall(method: string, args: any) {\n        try {\n            return { isError: false, result: await this.call(method, args) }\n        } catch (error) {\n            if (error instanceof Error) return { isError: true, result: error.message }\n            else return { isError: true, result: 'Unknown error' }\n        }\n    }\n}\n","import { z } from 'zod'\n\nexport const ListItemsParams = z.object({\n    limit: z.number().optional().describe('The maximum number of items to return'),\n    last_key: z.string().optional().describe('The last key to use for pagination'),\n})\n\nexport const GetInboxParams = z.object({\n    inbox_id: z.string().describe('The ID of the inbox to get'),\n})\n\nexport const CreateInboxParams = z.object({\n    username: z.string().optional().describe('The username of the inbox to create'),\n    domain: z.string().optional().describe('The domain of the inbox to create'),\n    display_name: z.string().optional().describe('The display name of the inbox to create'),\n})\n\nexport const ListThreadsParams = ListItemsParams.extend({\n    inbox_id: z.string().describe('The ID of the inbox to list threads for'),\n    labels: z.array(z.string()).optional().describe('The labels to filter threads by'),\n})\n\nexport const GetThreadParams = z.object({\n    inbox_id: z.string().describe('The ID of the inbox to get the thread for'),\n    thread_id: z.string().describe('The ID of the thread to get'),\n})\n\nexport const ListMessagesParams = ListItemsParams.extend({\n    inbox_id: z.string().describe('The ID of the inbox to list messages for'),\n    labels: z.array(z.string()).optional().describe('The labels to filter messages by'),\n})\n\nexport const GetMessageParams = z.object({\n    inbox_id: z.string().describe('The ID of the inbox to get the message for'),\n    message_id: z.string().describe('The ID of the message to get'),\n})\n\nexport const GetAttachmentParams = z.object({\n    inbox_id: z.string().describe('The ID of the inbox to get the attachment for'),\n    message_id: z.string().describe('The ID of the message to get the attachment for'),\n    attachment_id: z.string().describe('The ID of the attachment to get'),\n})\n\nexport const SendMessageParams = z.object({\n    inbox_id: z.string().describe('The ID of the inbox to send the message from'),\n    to: z.array(z.string()).describe('The list of recipients'),\n    cc: z.array(z.string()).optional().describe('The list of CC recipients'),\n    bcc: z.array(z.string()).optional().describe('The list of BCC recipients'),\n    subject: z.string().optional().describe('The subject of the message'),\n    text: z.string().optional().describe('The plain text body of the message'),\n    html: z.string().optional().describe('The HTML body of the message'),\n})\n\nexport const ReplyToMessageParams = z.object({\n    inbox_id: z.string().describe('The inbox ID of the inbox to reply to the message from'),\n    message_id: z.string().describe('The message ID of the message you wish to reply to'),\n    text: z.string().optional().describe('The plain text body of the reply'),\n    html: z.string().optional().describe('The HTML body of the reply'),\n})\n","import { AnyZodObject } from 'zod'\n\nimport {\n    ListItemsParams,\n    GetInboxParams,\n    CreateInboxParams,\n    ListThreadsParams,\n    GetThreadParams,\n    ListMessagesParams,\n    GetMessageParams,\n    GetAttachmentParams,\n    SendMessageParams,\n    ReplyToMessageParams,\n} from './schemas'\n\nexport type Tool = {\n    name: string\n    method: string\n    description: string\n    schema: AnyZodObject\n}\n\nexport const tools: Tool[] = [\n    {\n        name: 'list_inboxes',\n        method: 'inboxes.list',\n        description: 'List all inboxes',\n        schema: ListItemsParams,\n    },\n    {\n        name: 'get_inbox',\n        method: 'inboxes.get',\n        description: 'Get an inbox by ID',\n        schema: GetInboxParams,\n    },\n    {\n        name: 'create_inbox',\n        method: 'inboxes.create',\n        description: 'Create a new inbox',\n        schema: CreateInboxParams,\n    },\n    {\n        name: 'list_threads',\n        method: 'threads.list',\n        description: 'List all threads',\n        schema: ListThreadsParams,\n    },\n    {\n        name: 'get_thread',\n        method: 'threads.get',\n        description: 'Get a thread by ID',\n        schema: GetThreadParams,\n    },\n    {\n        name: 'list_messages',\n        method: 'messages.list',\n        description: 'List all messages',\n        schema: ListMessagesParams,\n    },\n    {\n        name: 'get_message',\n        method: 'messages.get',\n        description: 'Get a message by ID',\n        schema: GetMessageParams,\n    },\n    {\n        name: 'get_attachment',\n        method: 'attachments.get',\n        description: 'Get an attachment by ID',\n        schema: GetAttachmentParams,\n    },\n    {\n        name: 'send_message',\n        method: 'messages.send',\n        description: 'Send a message',\n        schema: SendMessageParams,\n    },\n    {\n        name: 'reply_to_message',\n        method: 'messages.reply',\n        description: 'Reply to a message',\n        schema: ReplyToMessageParams,\n    },\n]\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,iBAA0B;;;ACA1B,uBAAgC;AAEzB,IAAM,UAAN,MAAc;AAAA,EACjB,YAA6B,QAA0B;AAA1B;AACzB,QAAI,CAAC,KAAK,OAAQ,MAAK,SAAS,IAAI,iCAAgB;AAAA,EACxD;AAAA,EAEA,MAAa,KAAK,QAAgB,MAAW;AACzC,UAAM,QAAQ,OAAO,MAAM,GAAG;AAC9B,UAAM,YAAY,MAAM,IAAI;AAE5B,QAAI,CAAC,UAAW,OAAM,IAAI,MAAM,mBAAmB;AAEnD,QAAI,SAAc,KAAK;AACvB,eAAW,QAAQ,MAAO,UAAS,OAAO,IAAI;AAE9C,UAAM,EAAE,SAAS,GAAG,QAAQ,IAAI;AAEhC,QAAI,QAAS,QAAO,MAAM,OAAO,SAAS,EAAE,KAAK,QAAQ,SAAS,OAAO;AAAA,QACpE,QAAO,MAAM,OAAO,SAAS,EAAE,KAAK,QAAQ,OAAO;AAAA,EAC5D;AAAA,EAEA,MAAa,SAAS,QAAgB,MAAW;AAC7C,QAAI;AACA,aAAO,EAAE,SAAS,OAAO,QAAQ,MAAM,KAAK,KAAK,QAAQ,IAAI,EAAE;AAAA,IACnE,SAAS,OAAO;AACZ,UAAI,iBAAiB,MAAO,QAAO,EAAE,SAAS,MAAM,QAAQ,MAAM,QAAQ;AAAA,UACrE,QAAO,EAAE,SAAS,MAAM,QAAQ,gBAAgB;AAAA,IACzD;AAAA,EACJ;AACJ;;;AC9BA,iBAAkB;AAEX,IAAM,kBAAkB,aAAE,OAAO;AAAA,EACpC,OAAO,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,uCAAuC;AAAA,EAC7E,UAAU,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,oCAAoC;AACjF,CAAC;AAEM,IAAM,iBAAiB,aAAE,OAAO;AAAA,EACnC,UAAU,aAAE,OAAO,EAAE,SAAS,4BAA4B;AAC9D,CAAC;AAEM,IAAM,oBAAoB,aAAE,OAAO;AAAA,EACtC,UAAU,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,qCAAqC;AAAA,EAC9E,QAAQ,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,mCAAmC;AAAA,EAC1E,cAAc,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,yCAAyC;AAC1F,CAAC;AAEM,IAAM,oBAAoB,gBAAgB,OAAO;AAAA,EACpD,UAAU,aAAE,OAAO,EAAE,SAAS,yCAAyC;AAAA,EACvE,QAAQ,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,iCAAiC;AACrF,CAAC;AAEM,IAAM,kBAAkB,aAAE,OAAO;AAAA,EACpC,UAAU,aAAE,OAAO,EAAE,SAAS,2CAA2C;AAAA,EACzE,WAAW,aAAE,OAAO,EAAE,SAAS,6BAA6B;AAChE,CAAC;AAEM,IAAM,qBAAqB,gBAAgB,OAAO;AAAA,EACrD,UAAU,aAAE,OAAO,EAAE,SAAS,0CAA0C;AAAA,EACxE,QAAQ,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,kCAAkC;AACtF,CAAC;AAEM,IAAM,mBAAmB,aAAE,OAAO;AAAA,EACrC,UAAU,aAAE,OAAO,EAAE,SAAS,4CAA4C;AAAA,EAC1E,YAAY,aAAE,OAAO,EAAE,SAAS,8BAA8B;AAClE,CAAC;AAEM,IAAM,sBAAsB,aAAE,OAAO;AAAA,EACxC,UAAU,aAAE,OAAO,EAAE,SAAS,+CAA+C;AAAA,EAC7E,YAAY,aAAE,OAAO,EAAE,SAAS,iDAAiD;AAAA,EACjF,eAAe,aAAE,OAAO,EAAE,SAAS,iCAAiC;AACxE,CAAC;AAEM,IAAM,oBAAoB,aAAE,OAAO;AAAA,EACtC,UAAU,aAAE,OAAO,EAAE,SAAS,8CAA8C;AAAA,EAC5E,IAAI,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,wBAAwB;AAAA,EACzD,IAAI,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,2BAA2B;AAAA,EACvE,KAAK,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,4BAA4B;AAAA,EACzE,SAAS,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,4BAA4B;AAAA,EACpE,MAAM,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,oCAAoC;AAAA,EACzE,MAAM,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,8BAA8B;AACvE,CAAC;AAEM,IAAM,uBAAuB,aAAE,OAAO;AAAA,EACzC,UAAU,aAAE,OAAO,EAAE,SAAS,wDAAwD;AAAA,EACtF,YAAY,aAAE,OAAO,EAAE,SAAS,oDAAoD;AAAA,EACpF,MAAM,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,kCAAkC;AAAA,EACvE,MAAM,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,4BAA4B;AACrE,CAAC;;;ACpCM,IAAM,QAAgB;AAAA,EACzB;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AACJ;;;AH7EO,IAAM,qBAAN,cAAiC,qBAAU;AAAA,EACtC;AAAA,EAER,YAAY,QAA0B;AAClC,UAAM;AAAA,MACF,MAAM;AAAA,MACN,SAAS;AAAA,IACb,CAAC;AAED,SAAK,UAAU,IAAI,QAAQ,MAAM;AAEjC,eAAW,QAAQ,OAAO;AACtB,WAAK,KAAK,KAAK,MAAM,KAAK,aAAa,KAAK,OAAO,OAAO,OAAO,SAAkB;AAC/E,cAAM,EAAE,SAAS,OAAO,IAAI,MAAM,KAAK,QAAQ,SAAS,KAAK,QAAQ,IAAI;AACzE,eAAO,EAAE,SAAS,CAAC,EAAE,MAAM,QAAiB,MAAM,KAAK,UAAU,QAAQ,MAAM,CAAC,EAAE,CAAC,GAAG,QAAQ;AAAA,MAClG,CAAC;AAAA,IACL;AAAA,EACJ;AACJ;","names":[]}
+{"version":3,"sources":["../src/mcp.ts","../src/wrapper.ts","../src/schemas.ts","../src/tools.ts","../src/toolkit.ts"],"sourcesContent":["import { type ToolCallback } from '@modelcontextprotocol/sdk/server/mcp.js'\nimport { type ZodRawShape } from 'zod'\n\nimport { ListToolkit } from './toolkit'\nimport { type Tool } from './tools'\n\ninterface McpTool {\n    name: string\n    description: string\n    paramsSchema: ZodRawShape\n    callback: ToolCallback<ZodRawShape>\n}\n\nexport class AgentMailToolkit extends ListToolkit<McpTool> {\n    protected buildTool(tool: Tool): McpTool {\n        return {\n            name: tool.name,\n            description: tool.description,\n            paramsSchema: tool.schema.shape,\n            callback: async (args) => {\n                const { isError, result } = await this.safeCall(tool.method, args)\n                return { content: [{ type: 'text' as const, text: JSON.stringify(result, null, 2) }], isError }\n            },\n        }\n    }\n}\n","import { AgentMailClient } from 'agentmail'\n\nexport class Wrapper {\n    constructor(private readonly client = new AgentMailClient()) {}\n\n    public async call(method: string, args: any) {\n        const parts = method.split('.')\n        const methodKey = parts.pop()\n\n        if (!methodKey) throw new Error('Method name empty')\n\n        let parent: any = this.client\n        for (const part of parts) parent = parent[part]\n\n        const { inboxId, ...options } = args\n\n        if (inboxId) return await parent[methodKey].call(parent, inboxId, options)\n        else return await parent[methodKey].call(parent, options)\n    }\n\n    public async safeCall(method: string, args: any) {\n        try {\n            return { isError: false, result: await this.call(method, args) }\n        } catch (error) {\n            if (error instanceof Error) return { isError: true, result: error.message }\n            else return { isError: true, result: 'Unknown error' }\n        }\n    }\n}\n","import { z } from 'zod'\n\nexport const ListItemsParams = z.object({\n    limit: z.number().optional().describe('Max number of items to return'),\n    page_token: z.string().optional().describe('Page token for pagination'),\n})\n\nexport const GetInboxParams = z.object({\n    inbox_id: z.string().describe('ID of inbox to get'),\n})\n\nexport const CreateInboxParams = z.object({\n    username: z.string().optional().describe('Username of inbox to create'),\n    domain: z.string().optional().describe('Domain of inbox to create'),\n    display_name: z.string().optional().describe('Display name of inbox to create'),\n})\n\nexport const ListThreadsParams = ListItemsParams.extend({\n    inbox_id: z.string().describe('ID of inbox to list threads from'),\n    labels: z.array(z.string()).optional().describe('Labels to filter threads by'),\n})\n\nexport const GetThreadParams = z.object({\n    inbox_id: z.string().describe('ID of inbox to get thread from'),\n    thread_id: z.string().describe('ID of thread to get'),\n})\n\nexport const ListMessagesParams = ListItemsParams.extend({\n    inbox_id: z.string().describe('ID of inbox to list messages from'),\n    labels: z.array(z.string()).optional().describe('Labels to filter messages by'),\n})\n\nexport const GetMessageParams = z.object({\n    inbox_id: z.string().describe('ID of inbox to get message from'),\n    message_id: z.string().describe('ID of message to get'),\n})\n\nexport const SendMessageParams = z.object({\n    inbox_id: z.string().describe('ID of inbox to send message from'),\n    to: z.array(z.string()).describe('Recipients of message'),\n    cc: z.array(z.string()).optional().describe('CC recipients of message'),\n    bcc: z.array(z.string()).optional().describe('BCC recipients of message'),\n    subject: z.string().optional().describe('Subject of message'),\n    text: z.string().optional().describe('Plain text body of message'),\n    html: z.string().optional().describe('HTML body of message'),\n})\n\nexport const ReplyToMessageParams = z.object({\n    inbox_id: z.string().describe('ID of inbox to reply to message from'),\n    message_id: z.string().describe('ID of message to reply to'),\n    text: z.string().optional().describe('Plain text body of reply'),\n    html: z.string().optional().describe('HTML body of reply'),\n})\n","import { AnyZodObject } from 'zod'\n\nimport {\n    ListItemsParams,\n    GetInboxParams,\n    CreateInboxParams,\n    ListThreadsParams,\n    GetThreadParams,\n    ListMessagesParams,\n    GetMessageParams,\n    SendMessageParams,\n    ReplyToMessageParams,\n} from './schemas'\n\nexport interface Tool {\n    name: string\n    method: string\n    description: string\n    schema: AnyZodObject\n}\n\nexport const tools: Tool[] = [\n    {\n        name: 'list_inboxes',\n        method: 'inboxes.list',\n        description: 'List inboxes',\n        schema: ListItemsParams,\n    },\n    {\n        name: 'get_inbox',\n        method: 'inboxes.get',\n        description: 'Get inbox',\n        schema: GetInboxParams,\n    },\n    {\n        name: 'create_inbox',\n        method: 'inboxes.create',\n        description: 'Create inbox. Use default username, domain, and display name unless otherwise specified.',\n        schema: CreateInboxParams,\n    },\n    {\n        name: 'list_threads',\n        method: 'inboxes.threads.list',\n        description: 'List threads in inbox',\n        schema: ListThreadsParams,\n    },\n    {\n        name: 'get_thread',\n        method: 'inboxes.threads.get',\n        description: 'Get thread',\n        schema: GetThreadParams,\n    },\n    {\n        name: 'list_messages',\n        method: 'inboxes.messages.list',\n        description: 'List messages in inbox',\n        schema: ListMessagesParams,\n    },\n    {\n        name: 'get_message',\n        method: 'inboxes.messages.get',\n        description: 'Get message',\n        schema: GetMessageParams,\n    },\n    {\n        name: 'send_message',\n        method: 'inboxes.messages.send',\n        description: 'Send message',\n        schema: SendMessageParams,\n    },\n    {\n        name: 'reply_to_message',\n        method: 'inboxes.messages.reply',\n        description: 'Reply to message',\n        schema: ReplyToMessageParams,\n    },\n]\n","import { AgentMailClient } from 'agentmail'\n\nimport { Wrapper } from './wrapper'\nimport { type Tool, tools } from './tools'\n\nexport abstract class BaseToolkit<T> extends Wrapper {\n    protected readonly tools: Record<string, T> = {}\n\n    constructor(client?: AgentMailClient) {\n        super(client)\n\n        this.tools = tools.reduce(\n            (acc, tool) => {\n                acc[tool.name] = this.buildTool(tool)\n                return acc\n            },\n            {} as Record<string, T>\n        )\n    }\n\n    protected abstract buildTool(tool: Tool): T\n}\n\nexport abstract class ListToolkit<T> extends BaseToolkit<T> {\n    public getTools(names?: string[]) {\n        if (!names) return Object.values(this.tools)\n\n        return names.reduce((acc, name) => {\n            if (name in this.tools) acc.push(this.tools[name])\n            return acc\n        }, [] as T[])\n    }\n}\n\nexport abstract class MapToolkit<T> extends BaseToolkit<T> {\n    public getTools(names?: string[]) {\n        if (!names) return this.tools\n\n        return names.reduce(\n            (acc, name) => {\n                if (name in this.tools) acc[name] = this.tools[name]\n                return acc\n            },\n            {} as Record<string, T>\n        )\n    }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,uBAAgC;AAEzB,IAAM,UAAN,MAAc;AAAA,EACjB,YAA6B,SAAS,IAAI,iCAAgB,GAAG;AAAhC;AAAA,EAAiC;AAAA,EAE9D,MAAa,KAAK,QAAgB,MAAW;AACzC,UAAM,QAAQ,OAAO,MAAM,GAAG;AAC9B,UAAM,YAAY,MAAM,IAAI;AAE5B,QAAI,CAAC,UAAW,OAAM,IAAI,MAAM,mBAAmB;AAEnD,QAAI,SAAc,KAAK;AACvB,eAAW,QAAQ,MAAO,UAAS,OAAO,IAAI;AAE9C,UAAM,EAAE,SAAS,GAAG,QAAQ,IAAI;AAEhC,QAAI,QAAS,QAAO,MAAM,OAAO,SAAS,EAAE,KAAK,QAAQ,SAAS,OAAO;AAAA,QACpE,QAAO,MAAM,OAAO,SAAS,EAAE,KAAK,QAAQ,OAAO;AAAA,EAC5D;AAAA,EAEA,MAAa,SAAS,QAAgB,MAAW;AAC7C,QAAI;AACA,aAAO,EAAE,SAAS,OAAO,QAAQ,MAAM,KAAK,KAAK,QAAQ,IAAI,EAAE;AAAA,IACnE,SAAS,OAAO;AACZ,UAAI,iBAAiB,MAAO,QAAO,EAAE,SAAS,MAAM,QAAQ,MAAM,QAAQ;AAAA,UACrE,QAAO,EAAE,SAAS,MAAM,QAAQ,gBAAgB;AAAA,IACzD;AAAA,EACJ;AACJ;;;AC5BA,iBAAkB;AAEX,IAAM,kBAAkB,aAAE,OAAO;AAAA,EACpC,OAAO,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,+BAA+B;AAAA,EACrE,YAAY,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,2BAA2B;AAC1E,CAAC;AAEM,IAAM,iBAAiB,aAAE,OAAO;AAAA,EACnC,UAAU,aAAE,OAAO,EAAE,SAAS,oBAAoB;AACtD,CAAC;AAEM,IAAM,oBAAoB,aAAE,OAAO;AAAA,EACtC,UAAU,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,6BAA6B;AAAA,EACtE,QAAQ,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,2BAA2B;AAAA,EAClE,cAAc,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,iCAAiC;AAClF,CAAC;AAEM,IAAM,oBAAoB,gBAAgB,OAAO;AAAA,EACpD,UAAU,aAAE,OAAO,EAAE,SAAS,kCAAkC;AAAA,EAChE,QAAQ,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,6BAA6B;AACjF,CAAC;AAEM,IAAM,kBAAkB,aAAE,OAAO;AAAA,EACpC,UAAU,aAAE,OAAO,EAAE,SAAS,gCAAgC;AAAA,EAC9D,WAAW,aAAE,OAAO,EAAE,SAAS,qBAAqB;AACxD,CAAC;AAEM,IAAM,qBAAqB,gBAAgB,OAAO;AAAA,EACrD,UAAU,aAAE,OAAO,EAAE,SAAS,mCAAmC;AAAA,EACjE,QAAQ,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,8BAA8B;AAClF,CAAC;AAEM,IAAM,mBAAmB,aAAE,OAAO;AAAA,EACrC,UAAU,aAAE,OAAO,EAAE,SAAS,iCAAiC;AAAA,EAC/D,YAAY,aAAE,OAAO,EAAE,SAAS,sBAAsB;AAC1D,CAAC;AAEM,IAAM,oBAAoB,aAAE,OAAO;AAAA,EACtC,UAAU,aAAE,OAAO,EAAE,SAAS,kCAAkC;AAAA,EAChE,IAAI,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,uBAAuB;AAAA,EACxD,IAAI,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,0BAA0B;AAAA,EACtE,KAAK,aAAE,MAAM,aAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,2BAA2B;AAAA,EACxE,SAAS,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,oBAAoB;AAAA,EAC5D,MAAM,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,4BAA4B;AAAA,EACjE,MAAM,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,sBAAsB;AAC/D,CAAC;AAEM,IAAM,uBAAuB,aAAE,OAAO;AAAA,EACzC,UAAU,aAAE,OAAO,EAAE,SAAS,sCAAsC;AAAA,EACpE,YAAY,aAAE,OAAO,EAAE,SAAS,2BAA2B;AAAA,EAC3D,MAAM,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,0BAA0B;AAAA,EAC/D,MAAM,aAAE,OAAO,EAAE,SAAS,EAAE,SAAS,oBAAoB;AAC7D,CAAC;;;AC/BM,IAAM,QAAgB;AAAA,EACzB;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AAAA,EACA;AAAA,IACI,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,QAAQ;AAAA,EACZ;AACJ;;;ACvEO,IAAe,cAAf,cAAsC,QAAQ;AAAA,EAC9B,QAA2B,CAAC;AAAA,EAE/C,YAAY,QAA0B;AAClC,UAAM,MAAM;AAEZ,SAAK,QAAQ,MAAM;AAAA,MACf,CAAC,KAAK,SAAS;AACX,YAAI,KAAK,IAAI,IAAI,KAAK,UAAU,IAAI;AACpC,eAAO;AAAA,MACX;AAAA,MACA,CAAC;AAAA,IACL;AAAA,EACJ;AAGJ;AAEO,IAAe,cAAf,cAAsC,YAAe;AAAA,EACjD,SAAS,OAAkB;AAC9B,QAAI,CAAC,MAAO,QAAO,OAAO,OAAO,KAAK,KAAK;AAE3C,WAAO,MAAM,OAAO,CAAC,KAAK,SAAS;AAC/B,UAAI,QAAQ,KAAK,MAAO,KAAI,KAAK,KAAK,MAAM,IAAI,CAAC;AACjD,aAAO;AAAA,IACX,GAAG,CAAC,CAAQ;AAAA,EAChB;AACJ;;;AJnBO,IAAM,mBAAN,cAA+B,YAAqB;AAAA,EAC7C,UAAU,MAAqB;AACrC,WAAO;AAAA,MACH,MAAM,KAAK;AAAA,MACX,aAAa,KAAK;AAAA,MAClB,cAAc,KAAK,OAAO;AAAA,MAC1B,UAAU,OAAO,SAAS;AACtB,cAAM,EAAE,SAAS,OAAO,IAAI,MAAM,KAAK,SAAS,KAAK,QAAQ,IAAI;AACjE,eAAO,EAAE,SAAS,CAAC,EAAE,MAAM,QAAiB,MAAM,KAAK,UAAU,QAAQ,MAAM,CAAC,EAAE,CAAC,GAAG,QAAQ;AAAA,MAClG;AAAA,IACJ;AAAA,EACJ;AACJ;","names":[]}