@langchain/community
Version:
Third-party integrations for LangChain.js
76 lines (75 loc) • 2.81 kB
JavaScript
require("../../_virtual/_rolldown/runtime.cjs");
let _langchain_core_tools = require("@langchain/core/tools");
let _langchain_core_utils_env = require("@langchain/core/utils/env");
let googleapis = require("googleapis");
//#region src/tools/gmail/base.ts
var GmailBaseTool = class extends _langchain_core_tools.StructuredTool {
name = "Gmail";
description = "A tool to send and view emails through Gmail";
params;
gmail;
constructor({ credentials, scopes } = {
credentials: {
clientEmail: (0, _langchain_core_utils_env.getEnvironmentVariable)("GMAIL_CLIENT_EMAIL"),
privateKey: (0, _langchain_core_utils_env.getEnvironmentVariable)("GMAIL_PRIVATE_KEY"),
keyfile: (0, _langchain_core_utils_env.getEnvironmentVariable)("GMAIL_KEYFILE"),
subject: (0, _langchain_core_utils_env.getEnvironmentVariable)("GMAIL_SUBJECT")
},
scopes: ["https://mail.google.com/"]
}) {
super(...arguments);
if (!credentials) throw new Error("Missing credentials to authenticate to Gmail");
if (!credentials.accessToken) {
if (!credentials.clientEmail) throw new Error("Missing GMAIL_CLIENT_EMAIL to interact with Gmail");
if (!credentials.privateKey && !credentials.keyfile) throw new Error("Missing GMAIL_PRIVATE_KEY or GMAIL_KEYFILE or accessToken to interact with Gmail");
}
this.params = {
credentials,
scopes
};
}
async getGmailClient() {
const { credentials, scopes } = this.params;
if (credentials?.accessToken) {
const auth = new googleapis.google.auth.OAuth2();
const accessToken = typeof credentials.accessToken === "function" ? await credentials.accessToken() : credentials.accessToken;
auth.setCredentials({ access_token: accessToken });
return googleapis.google.gmail({
version: "v1",
auth
});
}
if (this.gmail) return this.gmail;
const auth = new googleapis.google.auth.JWT(credentials?.clientEmail, credentials?.keyfile, credentials?.privateKey, scopes, credentials?.subject);
this.gmail = googleapis.google.gmail({
version: "v1",
auth
});
return this.gmail;
}
parseHeaderAndBody(payload) {
if (!payload) return { body: "" };
const headers = payload.headers || [];
const subject = headers.find((header) => header.name === "Subject");
const sender = headers.find((header) => header.name === "From");
let body = "";
if (payload.parts) body = payload.parts.map((part) => part.mimeType === "text/plain" ? this.decodeBody(part.body?.data ?? "") : "").join("");
else if (payload.body?.data) body = this.decodeBody(payload.body.data);
return {
subject,
sender,
body
};
}
decodeBody(body) {
if (body) try {
return atob(body.replace(/-/g, "+").replace(/_/g, "/"));
} catch {
return body;
}
return "";
}
};
//#endregion
exports.GmailBaseTool = GmailBaseTool;
//# sourceMappingURL=base.cjs.map