diff --git a/service/src/chatgpt/index.ts b/service/src/chatgpt/index.ts index 8f18600..80f8ecc 100644 --- a/service/src/chatgpt/index.ts +++ b/service/src/chatgpt/index.ts @@ -27,8 +27,17 @@ let api: ChatGPTAPI | ChatGPTUnofficialProxyAPI apiKey: process.env.OPENAI_API_KEY, debug: false, } - - api = new ChatGPTAPI({ ...options }) + let fetchFn + if (process.env.SOCKS_PROXY_HOST && process.env.SOCKS_PROXY_PORT) { + const agent = new SocksProxyAgent({ + hostname: process.env.SOCKS_PROXY_HOST, + port: process.env.SOCKS_PROXY_PORT, + }) + fetchFn = (url, options) => { + return fetch(url, { agent, ...options }) + } + } + api = new ChatGPTAPI({ ...options, fetch: fetchFn }) apiModel = 'ChatGPTAPI' } else {