Merge pull request #9 from bytemate/feat/add-user-data

Feat/add user data
fff455 · Jan 21, 2023 · 347b6e1 · 347b6e1
2 parents b444dcd + bcab6b0
commit 347b6e1
Show file tree

Hide file tree

Showing 6 changed files with 139 additions and 11 deletions.
diff --git a/.env.example b/.env.example
@@ -21,3 +21,4 @@ CAPTCHA_TOKEN=
 NOPECHA_KEY=
 EXECUTABLE_PATH=
 PROXY_SERVER=
+USER_DATA_DIR=
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -54,7 +54,7 @@
     "async-retry": "^1.3.3",
     "bcrypt": "5.0.1",
     "cache-manager": "^5.1.4",
-    "chatgpt": "^3.4.2",
+    "chatgpt": "^3.5.0",
     "class-transformer": "0.5.1",
     "class-validator": "0.13.2",
     "graphql": "16.5.0",

diff --git a/src/chatgpt/chatgpt-pool/chatgpt-pool.service.ts b/src/chatgpt/chatgpt-pool/chatgpt-pool.service.ts
@@ -29,6 +29,9 @@ export class ChatgptPoolService {
     const chatgpt = new ChatGPTAPIBrowser({
       ...opts,
       ...this.chatgptConfig,
+      userDataDir: this.chatgptConfig.userDataDir
+        ? `${this.chatgptConfig.userDataDir}/${opts.email}`
+        : undefined,
     });
     await retry(
       async (_: any, num: number) => {

diff --git a/src/configs/config.interface.ts b/src/configs/config.interface.ts
@@ -26,4 +26,5 @@ export interface ChatgptConfig {
   nopechaKey?: string;
   executablePath?: string;
   proxyServer?: string;
+  userDataDir?: string;
 }
diff --git a/src/configs/config.ts b/src/configs/config.ts
@@ -20,6 +20,7 @@ const config: Config = {
     nopechaKey: process.env.NOPECHA_KEY,
     executablePath: process.env.EXECUTABLE_PATH || undefined,
     proxyServer: process.env.PROXY_SERVER || undefined,
+    userDataDir: process.env.USER_DATA_DIR || undefined,
   },
 };