✨ Add OpenAi Proxy config

Author: fuergaosi233

config.yaml.example

@@ -2,4 +2,5 @@ chatGPTAccountPool:
   - email: email
     password: password
     session_token: session_token
-chatPrivateTiggerKeyword: ""
+chatPrivateTiggerKeyword: ""
+openAIProxy: ""

src/chatgpt.ts

@@ -58,7 +58,12 @@ export class ChatGPTPoole {
     const platform = process.platform;
     const { stdout, stderr, exitCode } = await execa(
       platform === "win32" ? "powershell" : "sh",
-      [platform === "win32" ? "/c" : "-c", cmd]
+      [platform === "win32" ? "/c" : "-c", cmd],
+      {
+        env: {
+          https_proxy: config.openAIProxy || process.env.https_proxy,
+        },
+      }
     );
     if (exitCode !== 0) {
       console.error(`${email} login failed: ${stderr}`);

src/config.ts

@@ -19,6 +19,7 @@ if (fs.existsSync("./config.yaml")) {
     ],
     chatGptRetryTimes: Number(process.env.CHAT_GPT_RETRY_TIMES),
     chatPrivateTiggerKeyword: process.env.CHAT_PRIVATE_TRIGGER_KEYWORD,
+    openAIProxy: process.env.OPENAI_PROXY,
   };
 }
 dotenv.config();
@@ -35,4 +36,6 @@ export const config: IConfig = {
       ""
     ) ||
     "",
+  // Support openai-js use this proxy
+  openAIProxy: configFile.openAIProxy,
 };

src/generate_session.py

@@ -20,7 +20,7 @@ def login(email: str, password: str) -> str:
         config={
             "email": email,
             "password": password,
-            "proxy": os.environ.get("http_proxy", None),
+            "proxy": os.environ.get("http_proxy",None) or os.environ.get("openAIProxy", None) ,
         },
         conversation_id=None,
     )

src/interface.ts

@@ -36,4 +36,5 @@ export interface IConfig {
   chatGPTAccountPool: IAccount[];
   chatGptRetryTimes: number;
   chatPrivateTiggerKeyword: string;
+  openAIProxy?: string;
 }