From e5cad7a55e6d8f3ed7613117b0909b635b787735 Mon Sep 17 00:00:00 2001
From: Alex Shand <alex.shand@cartacapitalmarkets.com>
Date: Sat, 17 Dec 2022 00:02:33 +1000
Subject: [PATCH] feature: Add browserPath option

---
 src/chatgpt-api-browser.ts | 19 ++++++++++++++++---
 src/openai-auth.ts         |  8 ++++++--
 2 files changed, 22 insertions(+), 5 deletions(-)

diff --git a/src/chatgpt-api-browser.ts b/src/chatgpt-api-browser.ts
index 14bd5f9..0db4a3f 100644
--- a/src/chatgpt-api-browser.ts
+++ b/src/chatgpt-api-browser.ts
@@ -3,7 +3,11 @@ import type { Browser, HTTPRequest, HTTPResponse, Page } from 'puppeteer'
 import { v4 as uuidv4 } from 'uuid'
 
 import * as types from './types'
-import { getBrowser, getOpenAIAuth } from './openai-auth'
+import {
+  defaultChromeExecutablePath,
+  getBrowser,
+  getOpenAIAuth
+} from './openai-auth'
 import {
   browserPostEventStream,
   isRelevantRequest,
@@ -22,6 +26,7 @@ export class ChatGPTAPIBrowser {
   protected _email: string
   protected _password: string
 
+  protected _browserPath: string
   protected _browser: Browser
   protected _page: Page
 
@@ -46,6 +51,9 @@ export class ChatGPTAPIBrowser {
 
     /** @defaultValue `undefined` **/
     captchaToken?: string
+
+    /** @defaultValue `undefined` **/
+    browserPath?: string
   }) {
     const {
       email,
@@ -54,7 +62,8 @@ export class ChatGPTAPIBrowser {
       debug = false,
       isGoogleLogin = false,
       minimize = true,
-      captchaToken
+      captchaToken,
+      browserPath = defaultChromeExecutablePath()
     } = opts
 
     this._email = email
@@ -65,6 +74,7 @@ export class ChatGPTAPIBrowser {
     this._isGoogleLogin = !!isGoogleLogin
     this._minimize = !!minimize
     this._captchaToken = captchaToken
+    this._browserPath = browserPath
   }
 
   async init() {
@@ -75,7 +85,10 @@ export class ChatGPTAPIBrowser {
     }
 
     try {
-      this._browser = await getBrowser({ captchaToken: this._captchaToken })
+      this._browser = await getBrowser({
+        captchaToken: this._captchaToken,
+        executablePath: this._browserPath
+      })
       this._page =
         (await this._browser.pages())[0] || (await this._browser.newPage())
 
diff --git a/src/openai-auth.ts b/src/openai-auth.ts
index 588750b..66abf9e 100644
--- a/src/openai-auth.ts
+++ b/src/openai-auth.ts
@@ -186,7 +186,11 @@ export async function getBrowser(
     captchaToken?: string
   } = {}
 ) {
-  const { captchaToken = process.env.CAPTCHA_TOKEN, ...launchOptions } = opts
+  const {
+    captchaToken = process.env.CAPTCHA_TOKEN,
+    executablePath = defaultChromeExecutablePath(),
+    ...launchOptions
+  } = opts
 
   if (captchaToken && !hasRecaptchaPlugin) {
     hasRecaptchaPlugin = true
@@ -207,7 +211,7 @@ export async function getBrowser(
     headless: false,
     args: ['--no-sandbox', '--exclude-switches', 'enable-automation'],
     ignoreHTTPSErrors: true,
-    executablePath: defaultChromeExecutablePath(),
+    executablePath,
     ...launchOptions
   })
 }