AgentMail

import { AgentMailClient } from 'agentmail' import { CanvasFactory } from 'pdf-parse/worker' import { PDFParse } from 'pdf-parse' import mammoth from 'mammoth' import { fileTypeFromBuffer } from 'file-type' export type Args = Record<string, any> interface Attachment { text?: string error?: string fileType?: string } export async function listInboxes(client: AgentMailClient, args: Args) { return client.inboxes.list(args) } export async function getInbox(client: AgentMailClient, args: Args) { const { inboxId, ...options } = args return client.inboxes.get(inboxId, options) } export async function createInbox(client: AgentMailClient, args: Args) { return client.inboxes.create(args) } export async function deleteInbox(client: AgentMailClient, args: Args) { const { inboxId } = args return client.inboxes.delete(inboxId) } export async function listThreads(client: AgentMailClient, args: Args) { const { inboxId, ...options } = args return client.inboxes.threads.list(inboxId, options) } export async function getThread(client: AgentMailClient, args: Args) { const { inboxId, threadId, ...options } = args return client.inboxes.threads.get(inboxId, threadId, options) } export async function getAttachment(client: AgentMailClient, args: Args): Promise<Attachment> { const { threadId, attachmentId } = args const response = await client.threads.getAttachment(threadId, attachmentId) const fileBytes = Buffer.from(await response.arrayBuffer()) const fileKind = await fileTypeFromBuffer(fileBytes) const fileType = fileKind?.mime let text = undefined if (fileType === 'application/pdf') { const parser = new PDFParse({ data: fileBytes, CanvasFactory }) const pdfData = await parser.getText() text = pdfData.text } else if (fileType === 'application/vnd.openxmlformats-officedocument.wordprocessingml.document') { const result = await mammoth.extractRawText({ buffer: fileBytes }) text = result.value } else { return { error: `Unsupported file type: ${fileType || 'unknown'}`, fileType } } return { text, fileType } } export async function sendMessage(client: AgentMailClient, args: Args) { const { inboxId, ...options } = args return client.inboxes.messages.send(inboxId, options) } export async function replyToMessage(client: AgentMailClient, args: Args) { const { inboxId, messageId, ...options } = args return client.inboxes.messages.reply(inboxId, messageId, options) } export async function updateMessage(client: AgentMailClient, args: Args) { const { inboxId, messageId, ...options } = args return client.inboxes.messages.update(inboxId, messageId, options) }