Added Outlook integration toolkit

langchain/.env.example

@@ -88,3 +88,8 @@ NEO4J_USERNAME=ADD_YOURS_HERE
 NEO4J_PASSWORD=ADD_YOURS_HERE
 CLOSEVECTOR_API_KEY=ADD_YOURS_HERE
 CLOSEVECTOR_API_SECRET=ADD_YOURS_HERE
+OUTLOOK_CLIENT_ID=ADD_YOURS_HERE
+OUTLOOK_CLIENT_SECRET=ADD_YOURS_HERE
+OUTLOOK_REDIRECT_URI=ADD_YOURS_HERE
+OUTLOOK_REFRESH_TOKEN=ADD_YOURS_HERE
+OUTLOOK_ACCESS_TOKEN=ADD_YOURS_HERE

langchain/package.json

@@ -903,6 +903,7 @@
     "@types/jsdom": "^21.1.1",
     "@types/lodash": "^4",
     "@types/mozilla-readability": "^0.2.1",
+    "@types/openurl": "^1",
     "@types/pdf-parse": "^1.1.1",
     "@types/pg": "^8",
     "@types/pg-copy-streams": "^1.2.2",
@@ -1410,6 +1411,8 @@
     "ml-distance": "^4.0.0",
     "openai": "^4.19.0",
     "openapi-types": "^12.1.3",
+    "openurl": "^1.1.1",
+    "p-queue": "^6.6.2",
     "p-retry": "4",
     "uuid": "^9.0.0",
     "yaml": "^2.2.1",

langchain/scripts/create-entrypoints.js

@@ -36,6 +36,7 @@ const entrypoints = {
   "tools/sql": "tools/sql",
   "tools/webbrowser": "tools/webbrowser",
   "tools/google_calendar": "tools/google_calendar/index",
+  "tools/outlook": "tools/outlook/index",
   // chains
   chains: "chains/index",
   "chains/combine_documents/reduce": "chains/combine_documents/reduce",

langchain/src/load/import_map.ts

@@ -14,6 +14,7 @@ export * as agents__openai__output_parser from "../agents/openai/output_parser.j
 export * as base_language from "../base_language/index.js";
 export * as tools from "../tools/index.js";
 export * as tools__render from "../tools/render.js";
+export * as tools__outlook from "../tools/outlook/index.js";
 export * as chains from "../chains/index.js";
 export * as chains__combine_documents__reduce from "../chains/combine_documents/reduce.js";
 export * as chains__openai_functions from "../chains/openai_functions/index.js";

langchain/src/tools/index.ts

@@ -47,3 +47,5 @@ export {
   formatToOpenAIFunction,
   formatToOpenAITool,
 } from "./convert_to_openai.js";
+// export { OutlookIntegration } from "./outlook/outlookIntegration.js";
+export { OutlookSendMailTool, OutlookReadMailTool } from "./outlook/index.js";

langchain/src/tools/outlook/authFlowBase.ts

@@ -0,0 +1,13 @@
+export abstract class AuthFlowBase {
+  protected clientId: string;
+
+  protected accessToken = "";
+
+  constructor(clientId: string) {
+    this.clientId = clientId;
+  }
+
+  public abstract getAccessToken(): Promise<string>;
+
+  public abstract refreshAccessToken(): Promise<string>;
+}

langchain/src/tools/outlook/authFlowREST.ts

@@ -0,0 +1,189 @@
+import * as http from "http";
+import * as url from "url";
+import * as openurl from "openurl";
+import { AuthFlowBase } from "./authFlowBase.js";
+import { getEnvironmentVariable } from "../../util/env.js";
+
+interface AccessTokenResponse {
+  access_token: string;
+  refresh_token: string;
+}
+
+export class AuthFlowREST extends AuthFlowBase {
+  private clientSecret: string;
+
+  private redirectUri: string;
+
+  private port: number;
+
+  private pathname: string;
+
+  private refreshToken = "";
+
+  constructor({ clientId, clientSecret, redirectUri }: { clientId?: string, clientSecret?: string, redirectUri?: string } = {}) {
+    let id = clientId;
+    let secret = clientSecret;
+    let uri = redirectUri;
+    if (!id || !secret || !uri) {
+      id = getEnvironmentVariable("OUTLOOK_CLIENT_ID");
+      secret = getEnvironmentVariable("OUTLOOK_CLIENT_SECRET");
+      uri = getEnvironmentVariable("OUTLOOK_REDIRECT_URI");
+    }
+    if (!id || !secret || !uri) {
+      throw new Error("Missing clientId, clientSecret or redirectUri.");
+    }
+    super(id);
+    this.clientSecret = secret;
+    this.redirectUri = uri;
+    const parsedUrl = new URL(this.redirectUri);
+    this.port = parsedUrl.port ? parseInt(parsedUrl.port, 10) : 3000;
+    this.pathname = parsedUrl.pathname || "";
+  }
+
+  // Function to construct the OAuth URL
+  private openAuthUrl(): string {
+    const loginEndpoint =
+      "https://login.microsoftonline.com/common/oauth2/v2.0/authorize";
+    const { clientId } = this; // client ID regestered in Azure
+    const response_type = "code";
+    const response_mode = "query";
+    const redirectUri = encodeURIComponent(this.redirectUri); // redirect URI regestered in Azure
+    const scope = encodeURIComponent(
+      "openid offline_access https://graph.microsoft.com/.default"
+    );
+    const state = "12345";
+
+    const url = [
+      `${loginEndpoint}?client_id=${clientId}`,
+      `&response_type=${response_type}`,
+      `&response_mode=${response_mode}`,
+      `&redirect_uri=${redirectUri}`,
+      `&scope=${scope}`,
+      `&state=${state}`,
+    ].join("");
+    openurl.open(url);
+    return url;
+  }
+
+  // Function to start the server
+  private startServer(): Promise<http.Server> {
+    return new Promise((resolve, reject) => {
+      const server = http.createServer();
+
+      server.listen(this.port, () => {
+        console.log(`Server listening at http://localhost:${this.port}`);
+        resolve(server);
+      });
+
+      server.on("error", (err) => {
+        reject(err);
+      });
+    });
+  }
+
+  // Function to listen for the authorization code
+  private async listenForCode(server: http.Server): Promise<string> {
+    return new Promise((resolve, reject) => {
+      server.on("request", (req, res) => {
+        try {
+          const reqUrl = url.parse(req.url || "", true);
+
+          if (reqUrl.pathname === this.pathname) {
+            const authCode = reqUrl.query.code as string;
+
+            res.writeHead(200, { "Content-Type": "text/html" });
+            res.end("Authorization code received. You can close this window.");
+
+            server.close();
+            console.log("Server closed");
+            resolve(authCode); // Resolve the Promise with the authorization code
+          } else {
+            res.writeHead(404);
+            res.end("404 Not Found");
+          }
+        } catch (err) {
+          res.writeHead(500);
+          res.end("Server error");
+          reject(err);
+        }
+      });
+    });
+  }
+
+  // Main function to run the auth flow
+  private async getCode(): Promise<string> {
+    // check credentials
+    if (!this.clientId || !this.redirectUri) {
+      throw new Error("Missing clientId or redirectUri.");
+    }
+
+    const server = await this.startServer();
+    this.openAuthUrl();
+    const code = await this.listenForCode(server);
+    return code;
+  }
+
+  // Function to get the token using the code and client credentials
+  public async getAccessToken(): Promise<string> {
+    // fetch auth code from user login
+    const code = await this.getCode();
+    // fetch access token using auth code
+    const req_body =
+      `client_id=${encodeURIComponent(this.clientId)}&` +
+      `client_secret=${encodeURIComponent(this.clientSecret)}&` +
+      `scope=${encodeURIComponent("https://graph.microsoft.com/.default")}&` +
+      `redirect_uri=${encodeURIComponent(this.redirectUri)}&` +
+      `grant_type=authorization_code&` +
+      `code=${encodeURIComponent(code)}`;
+
+    const response = await fetch(
+      "https://login.microsoftonline.com/common/oauth2/v2.0/token",
+      {
+        method: "POST",
+        headers: {
+          "Content-Type": "application/x-www-form-urlencoded",
+        },
+        body: req_body,
+      }
+    );
+
+    if (!response.ok) {
+      throw new Error(`fetch token error! response: ${response.status}`);
+    }
+    // save access token and refresh token
+    const json = (await response.json()) as AccessTokenResponse;
+    this.accessToken = json.access_token;
+    this.refreshToken = json.refresh_token;
+    return this.accessToken;
+  }
+
+  public async refreshAccessToken(): Promise<string> {
+    // fetch new access token using refresh token
+    const req_body =
+      `client_id=${encodeURIComponent(this.clientId)}&` +
+      `client_secret=${encodeURIComponent(this.clientSecret)}&` +
+      `scope=${encodeURIComponent("https://graph.microsoft.com/.default")}&` +
+      `redirect_uri=${encodeURIComponent(this.redirectUri)}&` +
+      `grant_type=refresh_token&` +
+      `refresh_token=${encodeURIComponent(this.refreshToken)}`;
+
+    const response = await fetch(
+      "https://login.microsoftonline.com/common/oauth2/v2.0/token",
+      {
+        method: "POST",
+        headers: {
+          "Content-Type": "application/x-www-form-urlencoded",
+        },
+        body: req_body,
+      }
+    );
+
+    if (!response.ok) {
+      throw new Error(`fetch token error! response: ${response.status}`);
+    }
+    // save new access token
+    const json = (await response.json()) as AccessTokenResponse;
+    this.accessToken = json.access_token;
+    return this.accessToken;
+  }
+}

langchain/src/tools/outlook/authFlowToken.ts

@@ -0,0 +1,103 @@
+import { AuthFlowBase } from "./authFlowBase.js";
+import { getEnvironmentVariable } from "../../util/env.js";
+
+interface AccessTokenResponse {
+  access_token: string;
+  refresh_token: string;
+}
+
+// if you have the token, and no need to refresh it, warning: token expires in 1 hour
+export class AuthFlowToken extends AuthFlowBase {
+  constructor(accessToken?: string) {
+    let token = accessToken;
+    if (!token) {
+      token = getEnvironmentVariable("OUTLOOK_ACCESS_TOKEN");
+    }
+    if (!token) {
+      throw new Error("Missing access_token.");
+    }
+    super("");
+    this.accessToken = token;
+  }
+
+  public async refreshAccessToken(): Promise<string> {
+    return this.accessToken;
+  }
+
+  public async getAccessToken(): Promise<string> {
+    return this.accessToken;
+  }
+}
+
+// if you have the refresh token and other credentials
+export class AuthFlowRefresh extends AuthFlowBase {
+  private clientSecret: string;
+
+  private redirectUri: string;
+
+  private refreshToken: string;
+
+  constructor(
+    clientId?: string,
+    clientSecret?: string,
+    redirectUri?: string,
+    refreshToken?: string
+  ) {
+    let id = clientId;
+    let secret = clientSecret;
+    let uri = redirectUri;
+    let token = refreshToken;
+    if (!id || !secret || !uri || !token) {
+      id = getEnvironmentVariable("OUTLOOK_CLIENT_ID");
+      secret = getEnvironmentVariable("OUTLOOK_CLIENT_SECRET");
+      uri = getEnvironmentVariable("OUTLOOK_REDIRECT_URI");
+      token = getEnvironmentVariable("OUTLOOK_REFRESH_TOKEN");
+    }
+    if (!id || !secret || !uri || !token) {
+      throw new Error(
+        "Missing clientId, clientSecret, redirectUri or refreshToken."
+      );
+    }
+    super(id);
+    this.clientSecret = secret;
+    this.redirectUri = uri;
+    this.refreshToken = token;
+  }
+
+  public async refreshAccessToken(): Promise<string> {
+    // fetch new access token using refresh token
+    const req_body =
+      `client_id=${encodeURIComponent(this.clientId)}&` +
+      `client_secret=${encodeURIComponent(this.clientSecret)}&` +
+      `scope=${encodeURIComponent("https://graph.microsoft.com/.default")}&` +
+      `redirect_uri=${encodeURIComponent(this.redirectUri)}&` +
+      `grant_type=refresh_token&` +
+      `refresh_token=${encodeURIComponent(this.refreshToken)}`;
+
+    const response = await fetch(
+      "https://login.microsoftonline.com/common/oauth2/v2.0/token",
+      {
+        method: "POST",
+        headers: {
+          "Content-Type": "application/x-www-form-urlencoded",
+        },
+        body: req_body,
+      }
+    );
+
+    if (!response.ok) {
+      throw new Error(`fetch token error! response: ${response.status}`);
+    }
+    // save new access token
+    const json = (await response.json()) as AccessTokenResponse;
+    this.accessToken = json.access_token;
+    return this.accessToken;
+  }
+
+  // Function to get the token using the code and client credentials
+  public async getAccessToken(): Promise<string> {
+    const accessToken = await this.refreshAccessToken();
+    this.accessToken = accessToken;
+    return accessToken;
+  }
+}