mirror of
https://github.com/cunnymessiah/keychecker.git
synced 2026-05-10 18:39:04 -07:00
based
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
import APIKey
|
|
|
|
|
|
async def check_mistral(key: APIKey, session):
|
|
async with session.get(f'https://api.mistral.ai/v1/models', headers={'Authorization': f'Bearer {key.api_key}'}) as response:
|
|
if response.status != 200:
|
|
return
|
|
key.subbed = await check_sub_status(key, session)
|
|
return True
|
|
|
|
|
|
async def check_sub_status(key: APIKey, session):
|
|
data = {
|
|
'model': 'open-mistral-7b',
|
|
'messages': [{'role': 'user', 'content': ''}],
|
|
'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
|
|
|
|
|
|
def pretty_print_mistral_keys(keys):
|
|
print('-' * 90)
|
|
subbed = 0
|
|
print(f'Validated {len(keys)} Mistral keys:')
|
|
for key in keys:
|
|
if key.subbed:
|
|
subbed += 1
|
|
print(f'{key.api_key}' + (' | has sub active' if key.subbed else ''))
|
|
print(f'\n--- Total Valid Mistral Keys: {len(keys)} ({subbed} with an active subscription) ---\n') |