@langchain/community
Version:
Third-party integrations for LangChain.js
89 lines (88 loc) • 2.63 kB
JavaScript
import { GmailBaseTool } from "./base.js";
import { SEARCH_DESCRIPTION } from "./descriptions.js";
import { z } from "zod/v3";
//#region src/tools/gmail/search.ts
const searchSchema = z.object({
query: z.string(),
maxResults: z.number().optional(),
resource: z.enum(["messages", "threads"]).optional()
});
var GmailSearch = class extends GmailBaseTool {
name = "search_gmail";
schema = searchSchema;
description = SEARCH_DESCRIPTION;
constructor(fields) {
super(fields);
}
async _call(arg) {
const { query, maxResults = 10, resource = "messages" } = arg;
try {
const gmail = await this.getGmailClient();
const { data } = await gmail.users.messages.list({
userId: "me",
q: query,
maxResults
});
if (!data) throw new Error("No data returned from Gmail");
const { messages } = data;
if (!messages) throw new Error("No messages returned from Gmail");
if (resource === "messages") {
const parsedMessages = await this.parseMessages(gmail, messages);
return `Result for the query ${query}:\n${JSON.stringify(parsedMessages)}`;
} else if (resource === "threads") {
const parsedThreads = await this.parseThreads(gmail, messages);
return `Result for the query ${query}:\n${JSON.stringify(parsedThreads)}`;
}
throw new Error(`Invalid resource: ${resource}`);
} catch (error) {
throw new Error(`Error while searching Gmail: ${error}`);
}
}
async parseMessages(gmail, messages) {
return await Promise.all(messages.map(async (message) => {
try {
const { data } = await gmail.users.messages.get({
userId: "me",
format: "full",
id: message.id ?? ""
});
const { payload } = data;
const { subject, sender, body } = this.parseHeaderAndBody(payload);
return {
id: message.id,
threadId: message.threadId,
snippet: data.snippet,
body,
subject,
sender
};
} catch (error) {
throw new Error(`Error while fetching message: ${error}`);
}
}));
}
async parseThreads(gmail, messages) {
return await Promise.all(messages.map(async (message) => {
try {
const { data: { messages } } = await gmail.users.threads.get({
userId: "me",
format: "full",
id: message.threadId ?? ""
});
const { subject, sender, body } = this.parseHeaderAndBody(messages?.[0]?.payload);
return {
id: message.threadId,
snippet: messages?.[0]?.snippet,
body,
subject,
sender
};
} catch (error) {
throw new Error(`Error while fetching thread: ${error}`);
}
}));
}
};
//#endregion
export { GmailSearch };
//# sourceMappingURL=search.js.map