Merge pull request #5 from hourianto/master

Fix Mistral key checking
This commit is contained in:
jia tan
2024-07-29 21:35:10 +00:00
committed by GitHub
+4 -4
View File
@@ -13,15 +13,15 @@ async def check_sub_status(key: APIKey, session):
data = {
'model': 'open-mistral-7b',
'messages': [{'role': 'user', 'content': ''}],
'max_tokens': 1
'max_tokens': -1
}
async with session.post(f'https://api.mistral.ai/v1/chat/completions', headers={'Authorization': f'Bearer {key.api_key}'}, json=data) as response:
if response.status == 401 or response.status == 429:
return False
return True
# Since we do an invalid request, if the key is active and has quota, the API returns 422 Unprocessable Entity
return response.status == 422
def pretty_print_mistral_keys(keys):
keys = sorted(keys, key=lambda x: x.subbed, reverse=True)
print('-' * 90)
subbed = 0
print(f'Validated {len(keys)} Mistral keys:')