From b8565b7bd2da07c057beee44c222500b356ca9e5 Mon Sep 17 00:00:00 2001
From: Travis Fischer <fisch0920@gmail.com>
Date: Sun, 11 Dec 2022 19:40:06 -0600
Subject: [PATCH] feat: allow for adding extra headers

---
 legacy/src/chatgpt-api.ts | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/legacy/src/chatgpt-api.ts b/legacy/src/chatgpt-api.ts
index 53528306..6921556f 100644
--- a/legacy/src/chatgpt-api.ts
+++ b/legacy/src/chatgpt-api.ts
@@ -55,6 +55,8 @@ export class ChatGPTAPI {
     accessTokenTTL?: number
 
     accessToken?: string
+
+    headers?: Record<string, string>
   }) {
     const {
       sessionToken,
@@ -64,7 +66,8 @@ export class ChatGPTAPI {
       backendApiBaseUrl = 'https://chat.openai.com/backend-api',
       userAgent = USER_AGENT,
       accessTokenTTL = 60 * 60000, // 1 hour
-      accessToken
+      accessToken,
+      headers
     } = opts
 
     this._sessionToken = sessionToken
@@ -74,7 +77,7 @@ export class ChatGPTAPI {
     this._backendApiBaseUrl = backendApiBaseUrl
     this._userAgent = userAgent
     this._headers = {
-      'User-Agent': this._userAgent,
+      'user-agent': this._userAgent,
       'x-openai-assistant-app-id': '',
       'accept-language': 'en-US,en;q=0.9',
       origin: 'https://chat.openai.com',
@@ -84,7 +87,8 @@ export class ChatGPTAPI {
       'sec-ch-ua-platform': '"macOS"',
       'sec-fetch-dest': 'empty',
       'sec-fetch-mode': 'cors',
-      'sec-fetch-site': 'same-origin'
+      'sec-fetch-site': 'same-origin',
+      ...headers
     }
 
     this._accessTokenCache = new ExpiryMap<string, string>(accessTokenTTL)