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

fixed other guys fuckup

Browse Source
This commit is contained in:
based
2024-11-27 18:28:11 +10:00
parent 3b3ea46360
commit c0c12fadce
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
AWSAsync.py
+2 -2
View File
@@ -66,7 +66,7 @@ async def test_invoke_perms(key: APIKey, session, model):
if message and ('The request signature we calculated does not match the signature you provided' in message or 'The security token included in the request is invalid' in message): if message and ('The request signature we calculated does not match the signature you provided' in message or 'The security token included in the request is invalid' in message):
return False return False
elif response.status == 400 or response.status == 404: elif response.status == 400 or response.status == 404:
if message and 'The provided body' in message: if message and 'Malformed input request' in message:
if key.region == "": if key.region == "":
key.region = region key.region = region
else: else:
@@ -190,7 +190,7 @@ async def invoke_model(key: APIKey, session, region, model):
async with session.post(url, headers=signed_headers, data=signed_data) as response: async with session.post(url, headers=signed_headers, data=signed_data) as response:
if response.status == 400: if response.status == 400:
resp = await response.json() resp = await response.json()
if resp['message'] and 'The provided body' in resp['message']: if resp['message'] and 'Malformed input request' in resp['message']:
key.models[region].append(model_name) key.models[region].append(model_name)
elif resp['message'] and 'Operation not' in resp['message']: elif resp['message'] and 'Operation not' in resp['message']:
key.useless = True key.useless = True