From 37bf84bd2a22db07e8454070954c7ad156afb1a9 Mon Sep 17 00:00:00 2001
From: liyu <liyu@iftech.io>
Date: Mon, 12 Dec 2022 00:46:03 +0800
Subject: [PATCH] feat: set accessToken while initialize ChatGPTAPI

---
 src/chatgpt-api.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/chatgpt-api.ts b/src/chatgpt-api.ts
index 0ffcf98..31d4e8b 100644
--- a/src/chatgpt-api.ts
+++ b/src/chatgpt-api.ts
@@ -52,6 +52,8 @@ export class ChatGPTAPI {
 
     /** @defaultValue 60000 (60 seconds) */
     accessTokenTTL?: number
+
+    accessToken?: string
   }) {
     const {
       sessionToken,
@@ -59,7 +61,8 @@ export class ChatGPTAPI {
       apiBaseUrl = 'https://chat.openai.com/api',
       backendApiBaseUrl = 'https://chat.openai.com/backend-api',
       userAgent = USER_AGENT,
-      accessTokenTTL = 60000 // 60 seconds
+      accessTokenTTL = 60000, // 60 seconds
+      accessToken
     } = opts
 
     this._sessionToken = sessionToken
@@ -76,6 +79,7 @@ export class ChatGPTAPI {
     }
 
     this._accessTokenCache = new ExpiryMap<string, string>(accessTokenTTL)
+    this._accessTokenCache.set(KEY_ACCESS_TOKEN, accessToken ?? '')
 
     if (!this._sessionToken) {
       throw new types.ChatGPTError('ChatGPT invalid session token')