agentmail-toolkit 0.3.0 → 0.4.0

package/dist/chunk-2WFL2XH4.js

@@ -0,0 +1,596 @@
+// src/util.ts
+import { getDocumentProxy, extractText } from "unpdf";
+import JSZip from "jszip";
+var safeFunc = async (func, client, args) => {
+  try {
+    return { isError: false, result: await func(client, args) };
+  } catch (error) {
+    if (error instanceof Error) return { isError: true, result: error.message };
+    else return { isError: true, result: "Unknown error" };
+  }
+};
+function detectFileType(bytes) {
+  if (bytes[0] === 37 && bytes[1] === 80 && bytes[2] === 68 && bytes[3] === 70) {
+    return "application/pdf";
+  }
+  if (bytes[0] === 80 && bytes[1] === 75 && bytes[2] === 3 && bytes[3] === 4) {
+    return "application/zip";
+  }
+  return void 0;
+}
+async function extractPdfText(bytes) {
+  const pdf = await getDocumentProxy(bytes);
+  const { text } = await extractText(pdf);
+  return Array.isArray(text) ? text.join("\n") : text;
+}
+async function extractDocxText(bytes) {
+  const zip = await JSZip.loadAsync(bytes);
+  const documentXml = await zip.file("word/document.xml")?.async("string");
+  if (!documentXml) return void 0;
+  return documentXml.replace(/<w:p[^>]*>/g, "\n").replace(/<[^>]+>/g, "").replace(/&lt;/g, "<").replace(/&gt;/g, ">").replace(/&amp;/g, "&").replace(/&quot;/g, '"').replace(/&apos;/g, "'").replace(/\n{3,}/g, "\n\n").trim();
+}
+
+// src/toolkit.ts
+import { AgentMailClient } from "agentmail";
+
+// src/schemas.ts
+import { z } from "zod";
+var InboxIdSchema = z.string().describe("ID of inbox");
+var ThreadIdSchema = z.string().describe("ID of thread");
+var MessageIdSchema = z.string().describe("ID of message");
+var AttachmentIdSchema = z.string().describe("ID of attachment");
+var DraftIdSchema = z.string().describe("ID of draft");
+var ListItemsParams = z.object({
+  limit: z.number().optional().default(10).describe("Max number of items to return"),
+  pageToken: z.string().optional().describe("Page token for pagination")
+});
+var GetInboxParams = z.object({
+  inboxId: InboxIdSchema
+});
+var CreateInboxParams = z.object({
+  username: z.string().optional().describe("Username"),
+  domain: z.string().optional().describe("Domain"),
+  displayName: z.string().optional().describe("Display name"),
+  clientId: z.string().optional().describe("Client-provided ID for idempotent creation"),
+  metadata: z.record(z.string(), z.union([z.string(), z.number(), z.boolean()])).optional().describe("Custom metadata key-value pairs to attach to the inbox")
+});
+var UpdateInboxParams = z.object({
+  inboxId: InboxIdSchema,
+  displayName: z.string().optional().describe("Display name"),
+  metadata: z.record(z.string(), z.union([z.string(), z.number(), z.boolean(), z.null()])).nullable().optional().describe("Metadata to merge into existing metadata. Set a key to null to remove it, or the whole field to null to clear all metadata")
+});
+var ListInboxItemsParams = ListItemsParams.extend({
+  inboxId: InboxIdSchema,
+  labels: z.array(z.string()).optional().describe("Labels to filter items by"),
+  before: z.string().pipe(z.coerce.date()).optional().describe("Filter items before datetime"),
+  after: z.string().pipe(z.coerce.date()).optional().describe("Filter items after datetime"),
+  ascending: z.boolean().optional().describe("Sort by oldest first instead of most recent first")
+});
+var ListThreadsParams = ListInboxItemsParams.extend({
+  senders: z.array(z.string()).optional().describe("Filter threads by senders (substring match; all values must match)"),
+  recipients: z.array(z.string()).optional().describe("Filter threads by recipients (substring match; all values must match)"),
+  subject: z.array(z.string()).optional().describe("Filter threads by subject (substring match; all values must match)"),
+  includeSpam: z.boolean().optional().describe("Include threads in spam"),
+  includeTrash: z.boolean().optional().describe("Include threads in trash")
+});
+var SearchInboxItemsParams = ListItemsParams.extend({
+  inboxId: InboxIdSchema,
+  q: z.string().describe("Full-text search query"),
+  before: z.string().pipe(z.coerce.date()).optional().describe("Filter items before datetime"),
+  after: z.string().pipe(z.coerce.date()).optional().describe("Filter items after datetime")
+});
+var GetThreadParams = z.object({
+  inboxId: InboxIdSchema,
+  threadId: ThreadIdSchema
+});
+var UpdateThreadParams = GetThreadParams.extend({
+  addLabels: z.array(z.string()).optional().describe("Labels to add"),
+  removeLabels: z.array(z.string()).optional().describe("Labels to remove")
+});
+var GetAttachmentParams = z.object({
+  inboxId: InboxIdSchema,
+  threadId: ThreadIdSchema,
+  attachmentId: AttachmentIdSchema
+});
+var AttachmentSchema = z.object({
+  filename: z.string().optional().describe("Filename"),
+  content_id: z.string().optional().describe("Content ID for inline attachments"),
+  content: z.string().optional().describe("Base64 encoded content"),
+  url: z.url().optional().describe("URL")
+});
+var BaseMessageParams = z.object({
+  inboxId: InboxIdSchema,
+  text: z.string().optional().describe("Plain text body"),
+  html: z.string().optional().describe("HTML body"),
+  labels: z.array(z.string()).optional().describe("Labels"),
+  attachments: z.array(AttachmentSchema).optional().describe("Attachments")
+});
+var SendMessageParams = BaseMessageParams.extend({
+  to: z.array(z.string()).describe("Recipients"),
+  cc: z.array(z.string()).optional().describe("CC recipients"),
+  bcc: z.array(z.string()).optional().describe("BCC recipients"),
+  subject: z.string().optional().describe("Subject"),
+  replyTo: z.array(z.string()).optional().describe("Reply-to addresses")
+});
+var ReplyToMessageParams = BaseMessageParams.extend({
+  messageId: MessageIdSchema,
+  replyAll: z.boolean().optional().describe("Reply to all recipients"),
+  to: z.array(z.string()).optional().describe("Override reply recipients"),
+  cc: z.array(z.string()).optional().describe("Override CC recipients"),
+  bcc: z.array(z.string()).optional().describe("Override BCC recipients"),
+  replyTo: z.array(z.string()).optional().describe("Reply-to addresses")
+});
+var ForwardMessageParams = SendMessageParams.extend({
+  messageId: MessageIdSchema
+});
+var UpdateMessageParams = z.object({
+  inboxId: InboxIdSchema,
+  messageId: MessageIdSchema,
+  addLabels: z.array(z.string()).optional().describe("Labels to add"),
+  removeLabels: z.array(z.string()).optional().describe("Labels to remove")
+});
+var ListMessagesParams = ListInboxItemsParams.extend({
+  from: z.array(z.string()).optional().describe("Filter messages by sender (substring match; all values must match)"),
+  to: z.array(z.string()).optional().describe("Filter messages by recipients (substring match; all values must match)"),
+  subject: z.array(z.string()).optional().describe("Filter messages by subject (substring match; all values must match)"),
+  includeSpam: z.boolean().optional().describe("Include messages in spam"),
+  includeTrash: z.boolean().optional().describe("Include messages in trash")
+});
+var CreateDraftParams = BaseMessageParams.extend({
+  to: z.array(z.string()).optional().describe("Recipients"),
+  cc: z.array(z.string()).optional().describe("CC recipients"),
+  bcc: z.array(z.string()).optional().describe("BCC recipients"),
+  subject: z.string().optional().describe("Subject"),
+  replyTo: z.array(z.string()).optional().describe("Reply-to addresses"),
+  inReplyTo: z.string().optional().describe("Message ID this draft is replying to"),
+  sendAt: z.string().pipe(z.coerce.date()).optional().describe("ISO 8601 datetime to schedule sending (e.g. 2026-04-01T09:00:00Z)"),
+  clientId: z.string().optional().describe("Client-provided ID for idempotent creation")
+});
+var ListDraftsParams = ListInboxItemsParams;
+var GetDraftParams = z.object({
+  inboxId: InboxIdSchema,
+  draftId: DraftIdSchema
+});
+var UpdateDraftParams = z.object({
+  inboxId: InboxIdSchema,
+  draftId: DraftIdSchema,
+  to: z.array(z.string()).optional().describe("Recipients"),
+  cc: z.array(z.string()).optional().describe("CC recipients"),
+  bcc: z.array(z.string()).optional().describe("BCC recipients"),
+  subject: z.string().optional().describe("Subject"),
+  text: z.string().optional().describe("Plain text body"),
+  html: z.string().optional().describe("HTML body"),
+  replyTo: z.array(z.string()).optional().describe("Reply-to addresses"),
+  sendAt: z.string().pipe(z.coerce.date()).optional().describe("ISO 8601 datetime to reschedule sending")
+});
+var SendDraftParams = z.object({
+  inboxId: InboxIdSchema,
+  draftId: DraftIdSchema
+});
+var DeleteDraftParams = z.object({
+  inboxId: InboxIdSchema,
+  draftId: DraftIdSchema
+});
+var AuthMeParams = z.object({});
+
+// src/functions.ts
+async function listInboxes(client, args) {
+  return client.inboxes.list(args);
+}
+async function getInbox(client, args) {
+  const { inboxId, ...options } = args;
+  return client.inboxes.get(inboxId, options);
+}
+async function createInbox(client, args) {
+  return client.inboxes.create(args);
+}
+async function updateInbox(client, args) {
+  const { inboxId, ...options } = args;
+  return client.inboxes.update(inboxId, options);
+}
+async function deleteInbox(client, args) {
+  const { inboxId } = args;
+  return client.inboxes.delete(inboxId);
+}
+async function listThreads(client, args) {
+  const { inboxId, ...options } = args;
+  return client.inboxes.threads.list(inboxId, options);
+}
+async function searchThreads(client, args) {
+  const { inboxId, q, ...options } = args;
+  return client.inboxes.threads.search(inboxId, { q, ...options });
+}
+async function getThread(client, args) {
+  const { inboxId, threadId, ...options } = args;
+  return client.inboxes.threads.get(inboxId, threadId, options);
+}
+async function updateThread(client, args) {
+  const { inboxId, threadId, ...options } = args;
+  return client.inboxes.threads.update(inboxId, threadId, options);
+}
+async function deleteThread(client, args) {
+  const { inboxId, threadId } = args;
+  return client.inboxes.threads.delete(inboxId, threadId);
+}
+async function getAttachment(client, args) {
+  const { inboxId, threadId, attachmentId } = args;
+  const attachment = await client.inboxes.threads.getAttachment(inboxId, threadId, attachmentId);
+  try {
+    const response = await fetch(attachment.downloadUrl);
+    const arrayBuffer = await response.arrayBuffer();
+    const fileBytes = new Uint8Array(arrayBuffer);
+    const detectedType = detectFileType(fileBytes);
+    if (detectedType === "application/pdf") {
+      return { ...attachment, text: await extractPdfText(fileBytes) };
+    } else if (detectedType === "application/zip") {
+      return { ...attachment, text: await extractDocxText(fileBytes) };
+    }
+  } catch {
+  }
+  return attachment;
+}
+async function listMessages(client, args) {
+  const { inboxId, ...options } = args;
+  return client.inboxes.messages.list(inboxId, options);
+}
+async function searchMessages(client, args) {
+  const { inboxId, q, ...options } = args;
+  return client.inboxes.messages.search(inboxId, { q, ...options });
+}
+async function sendMessage(client, args) {
+  const { inboxId, ...options } = args;
+  return client.inboxes.messages.send(inboxId, options);
+}
+async function replyToMessage(client, args) {
+  const { inboxId, messageId, ...options } = args;
+  return client.inboxes.messages.reply(inboxId, messageId, options);
+}
+async function forwardMessage(client, args) {
+  const { inboxId, messageId, ...options } = args;
+  return client.inboxes.messages.forward(inboxId, messageId, options);
+}
+async function updateMessage(client, args) {
+  const { inboxId, messageId, ...options } = args;
+  return client.inboxes.messages.update(inboxId, messageId, options);
+}
+async function createDraft(client, args) {
+  const { inboxId, ...options } = args;
+  return client.inboxes.drafts.create(inboxId, options);
+}
+async function listDrafts(client, args) {
+  const { inboxId, ...options } = args;
+  return client.inboxes.drafts.list(inboxId, options);
+}
+async function getDraft(client, args) {
+  const { inboxId, draftId } = args;
+  return client.inboxes.drafts.get(inboxId, draftId);
+}
+async function updateDraft(client, args) {
+  const { inboxId, draftId, ...options } = args;
+  return client.inboxes.drafts.update(inboxId, draftId, options);
+}
+async function sendDraft(client, args) {
+  const { inboxId, draftId } = args;
+  return client.inboxes.drafts.send(inboxId, draftId, {});
+}
+async function deleteDraft(client, args) {
+  const { inboxId, draftId } = args;
+  return client.inboxes.drafts.delete(inboxId, draftId);
+}
+async function authMe(client) {
+  return client.auth.me();
+}
+
+// src/tools.ts
+var tools = [
+  {
+    name: "list_inboxes",
+    description: "List email inboxes, paginated.",
+    paramsSchema: ListItemsParams,
+    func: listInboxes,
+    annotations: {
+      readOnlyHint: true,
+      openWorldHint: false
+    }
+  },
+  {
+    name: "get_inbox",
+    description: "Get an inbox by ID.",
+    paramsSchema: GetInboxParams,
+    func: getInbox,
+    annotations: {
+      readOnlyHint: true,
+      openWorldHint: false
+    }
+  },
+  {
+    name: "create_inbox",
+    description: "Create a new email inbox. Optionally specify username, domain, display name, and metadata.",
+    paramsSchema: CreateInboxParams,
+    func: createInbox,
+    annotations: {
+      readOnlyHint: false,
+      destructiveHint: false,
+      idempotentHint: false,
+      openWorldHint: false
+    }
+  },
+  {
+    name: "update_inbox",
+    description: "Update an inbox's display name or metadata. Metadata keys are merged; set a key to null to remove it, or set metadata to null to clear all.",
+    paramsSchema: UpdateInboxParams,
+    func: updateInbox,
+    annotations: {
+      readOnlyHint: false,
+      destructiveHint: true,
+      idempotentHint: true,
+      openWorldHint: false
+    }
+  },
+  {
+    name: "delete_inbox",
+    description: "Delete an inbox by ID.",
+    paramsSchema: GetInboxParams,
+    func: deleteInbox,
+    annotations: {
+      readOnlyHint: false,
+      destructiveHint: true,
+      idempotentHint: true,
+      openWorldHint: false
+    }
+  },
+  {
+    name: "list_threads",
+    description: "List email threads in an inbox. Filter by labels, sender, recipient, subject, or before/after datetime, paginated.",
+    paramsSchema: ListThreadsParams,
+    func: listThreads,
+    annotations: {
+      readOnlyHint: true,
+      openWorldHint: true
+    }
+  },
+  {
+    name: "search_threads",
+    description: "Search threads in an inbox with a full-text query, ranked by relevance. Matches senders, recipients, subject, and message body. Spam and trash are excluded.",
+    paramsSchema: SearchInboxItemsParams,
+    func: searchThreads,
+    annotations: {
+      readOnlyHint: true,
+      openWorldHint: true
+    }
+  },
+  {
+    name: "get_thread",
+    description: "Get a thread by ID, including its messages.",
+    paramsSchema: GetThreadParams,
+    func: getThread,
+    annotations: {
+      readOnlyHint: true,
+      openWorldHint: true
+    }
+  },
+  {
+    name: "get_attachment",
+    description: "Get an attachment from a thread. Returns metadata and a download URL, plus extracted text for PDF and DOCX files.",
+    paramsSchema: GetAttachmentParams,
+    func: getAttachment,
+    annotations: {
+      readOnlyHint: true,
+      openWorldHint: true
+    }
+  },
+  {
+    name: "update_thread",
+    description: "Update a thread's labels (add or remove). System labels cannot be modified.",
+    paramsSchema: UpdateThreadParams,
+    func: updateThread,
+    annotations: {
+      readOnlyHint: false,
+      destructiveHint: true,
+      idempotentHint: true,
+      openWorldHint: false
+    }
+  },
+  {
+    name: "delete_thread",
+    description: "Delete a thread from an inbox.",
+    paramsSchema: GetThreadParams,
+    func: deleteThread,
+    annotations: {
+      readOnlyHint: false,
+      destructiveHint: true,
+      idempotentHint: false,
+      openWorldHint: false
+    }
+  },
+  {
+    name: "list_messages",
+    description: "List messages in an inbox. Filter by labels, sender, recipient, subject, or before/after datetime, paginated.",
+    paramsSchema: ListMessagesParams,
+    func: listMessages,
+    annotations: {
+      readOnlyHint: true,
+      openWorldHint: true
+    }
+  },
+  {
+    name: "search_messages",
+    description: "Search messages in an inbox with a full-text query, ranked by relevance. Matches sender, recipients, subject, and message body. Spam and trash are excluded.",
+    paramsSchema: SearchInboxItemsParams,
+    func: searchMessages,
+    annotations: {
+      readOnlyHint: true,
+      openWorldHint: true
+    }
+  },
+  {
+    name: "send_message",
+    description: "Send an email from an inbox to one or more recipients.",
+    paramsSchema: SendMessageParams,
+    func: sendMessage,
+    annotations: {
+      readOnlyHint: false,
+      destructiveHint: true,
+      idempotentHint: false,
+      openWorldHint: true
+    }
+  },
+  {
+    name: "reply_to_message",
+    description: "Reply to a message in its thread. Set replyAll to include all original recipients.",
+    paramsSchema: ReplyToMessageParams,
+    func: replyToMessage,
+    annotations: {
+      readOnlyHint: false,
+      destructiveHint: true,
+      idempotentHint: false,
+      openWorldHint: true
+    }
+  },
+  {
+    name: "forward_message",
+    description: "Forward a message to new recipients.",
+    paramsSchema: ForwardMessageParams,
+    func: forwardMessage,
+    annotations: {
+      readOnlyHint: false,
+      destructiveHint: true,
+      idempotentHint: false,
+      openWorldHint: true
+    }
+  },
+  {
+    name: "update_message",
+    description: "Update a message's labels (add or remove).",
+    paramsSchema: UpdateMessageParams,
+    func: updateMessage,
+    annotations: {
+      readOnlyHint: false,
+      destructiveHint: true,
+      idempotentHint: true,
+      openWorldHint: false
+    }
+  },
+  {
+    name: "create_draft",
+    description: "Create a draft email. Use sendAt (ISO 8601 datetime) to schedule it for later sending.",
+    paramsSchema: CreateDraftParams,
+    func: createDraft,
+    annotations: {
+      readOnlyHint: false,
+      destructiveHint: false,
+      idempotentHint: false,
+      openWorldHint: false
+    }
+  },
+  {
+    name: "list_drafts",
+    description: 'List drafts in inbox. Filter by labels (e.g. "scheduled") to find scheduled drafts.',
+    paramsSchema: ListDraftsParams,
+    func: listDrafts,
+    annotations: {
+      readOnlyHint: true,
+      openWorldHint: false
+    }
+  },
+  {
+    name: "get_draft",
+    description: "Get a draft by ID, including its content, status, and scheduled send time.",
+    paramsSchema: GetDraftParams,
+    func: getDraft,
+    annotations: {
+      readOnlyHint: true,
+      openWorldHint: false
+    }
+  },
+  {
+    name: "update_draft",
+    description: "Update a draft. Use sendAt to reschedule a scheduled draft.",
+    paramsSchema: UpdateDraftParams,
+    func: updateDraft,
+    annotations: {
+      readOnlyHint: false,
+      destructiveHint: true,
+      idempotentHint: true,
+      openWorldHint: false
+    }
+  },
+  {
+    name: "send_draft",
+    description: "Send a draft immediately. The draft is converted to a sent message and deleted.",
+    paramsSchema: SendDraftParams,
+    func: sendDraft,
+    annotations: {
+      readOnlyHint: false,
+      destructiveHint: true,
+      idempotentHint: false,
+      openWorldHint: true
+    }
+  },
+  {
+    name: "delete_draft",
+    description: "Delete a draft. Also used to cancel a scheduled send.",
+    paramsSchema: DeleteDraftParams,
+    func: deleteDraft,
+    annotations: {
+      readOnlyHint: false,
+      destructiveHint: true,
+      idempotentHint: true,
+      openWorldHint: false
+    }
+  },
+  {
+    name: "auth_me",
+    description: "Get the identity and scope of the authenticated credential, including organization, pod, and inbox IDs.",
+    paramsSchema: AuthMeParams,
+    func: authMe,
+    annotations: {
+      readOnlyHint: true,
+      openWorldHint: false
+    }
+  }
+];
+
+// src/toolkit.ts
+var BaseToolkit = class {
+  client;
+  tools = {};
+  constructor(client) {
+    this.client = client ?? new AgentMailClient();
+    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;
+    }, []);
+  }
+};
+var MapToolkit = class extends BaseToolkit {
+  getTools(names) {
+    if (!names) return this.tools;
+    return names.reduce(
+      (acc, name) => {
+        if (name in this.tools) acc[name] = this.tools[name];
+        return acc;
+      },
+      {}
+    );
+  }
+};
+
+export {
+  safeFunc,
+  ListToolkit,
+  MapToolkit
+};
+//# sourceMappingURL=chunk-2WFL2XH4.js.map