feat: detect capacity limit

pull/110/head
Mrlolpro 2022-12-12 20:49:57 +01:00
rodzic 658be86e24
commit 1dc12afe7d
2 zmienionych plików z 9 dodań i 4 usunięć

Wyświetl plik

@ -67,6 +67,11 @@ export async function getOpenAIAuth2Captcha({
// NOTE: this is where you may encounter a CAPTCHA
await page.solveRecaptchas()
var capacityLimit = await page.$('[role="alert"]')
if (capacityLimit) {
throw `ChatGPT is at capacity right now`
}
await page.waitForSelector('#__next .btn-primary', { timeout: timeoutMs })
// once we get to this point, the Cloudflare cookies are available
@ -91,10 +96,6 @@ export async function getOpenAIAuth2Captcha({
waitUntil: 'networkidle0'
})
])
/*var capacityLimit = await page.$('')
if (capacityLimit) {
throw `ChatGPT is at capacity right now`
}*/
}
const pageCookies = await page.cookies()

Wyświetl plik

@ -62,6 +62,10 @@ export async function getOpenAIAuth({
await page.goto('https://chat.openai.com/auth/login')
// NOTE: this is where you may encounter a CAPTCHA
var capacityLimit = await page.$('[role="alert"]')
if (capacityLimit) {
throw `ChatGPT is at capacity right now`
}
await page.waitForSelector('#__next .btn-primary', { timeout: timeoutMs })