zack3d/keychecker
1
0
Fork 0
mirror of https://github.com/cunnymessiah/keychecker.git synced 2026-05-10 18:39:04 -07:00
Code Issues Packages Projects Releases Wiki Activity

implement a concurrent connection limit so we don't overflow the windows network buffer

Browse Source 
(oops)
This commit is contained in:
based
2024-01-18 12:14:21 +10:00
parent 873a8f51f7
commit 3ad16793a3
7 changed files with 152 additions and 162 deletions
Download Patch File Download Diff File Expand all files Collapse all files
AI21.py
+7 -9
View File
@@ -1,8 +1,7 @@
import aiohttp
import APIKey
async def check_ai21(key: APIKey):
async def check_ai21(key: APIKey, session):
url = "https://api.ai21.com/studio/v1/j2-light/complete"
payload = {
@@ -15,15 +14,14 @@ async def check_ai21(key: APIKey):
"Authorization": f"Bearer {key.api_key}"
}
async with aiohttp.ClientSession() as session:
async with session.post(url, json=payload, headers=headers) as response:
if response.status not in [200, 402]:
return
async with session.post(url, json=payload, headers=headers) as response:
if response.status not in [200, 402]:
return
if response.status == 402: # unsure if this error code also applies to empty keys
key.trial_elapsed = True
if response.status == 402: # unsure if this error code also applies to empty keys
key.trial_elapsed = True
return True
return True
def pretty_print_ai21_keys(keys):