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

Update for the new AWS error

Browse Source
This commit is contained in:
Raka Hourianto
2024-11-17 23:15:48 +03:00
committed by GitHub
parent 3f42982290
commit 4899a29aea
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):
return False
elif response.status == 400 or response.status == 404:
if message and 'Malformed input request' in message:
if message and 'The provided body' in message:
if key.region == "":
key.region = region
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:
if response.status == 400:
resp = await response.json()
if resp['message'] and 'Malformed input request' in resp['message']:
if resp['message'] and 'The provided body' in resp['message']:
key.models[region].append(model_name)
elif resp['message'] and 'Operation not' in resp['message']:
key.useless = True