feat: detect capacity limit

transitive-bullshit · Dec 12, 2022 · 1dc12af · 1dc12af
1 parent 658be86
commit 1dc12af
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/src/openai-auth-2captcha.ts b/src/openai-auth-2captcha.ts
@@ -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()

diff --git a/src/openai-auth.ts b/src/openai-auth.ts
@@ -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 })