mirror of
https://github.com/cunnymessiah/keychecker.git
synced 2026-05-11 02:40:13 -07:00
based
2bce800767
this will help differentiate fake t4/t5 keys reporting as 10k rpm but in reality have just had their rpm limit increased by asking openai
57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
from enum import Enum
|
|
|
|
|
|
class APIKey:
|
|
def __init__(self, provider, api_key):
|
|
self.provider = provider
|
|
self.api_key = api_key
|
|
|
|
if provider == Provider.OPENAI:
|
|
self.model = ""
|
|
self.trial = False
|
|
self.has_quota = False
|
|
self.default_org = ""
|
|
self.organizations = []
|
|
self.rpm = 0
|
|
self.tier = ""
|
|
|
|
elif provider == Provider.ANTHROPIC:
|
|
self.pozzed = False
|
|
self.rate_limited = False
|
|
|
|
elif provider == Provider.AI21:
|
|
self.trial_elapsed = False
|
|
|
|
elif provider == Provider.AWS:
|
|
self.username = ""
|
|
self.useless = True
|
|
self.admin_priv = False
|
|
self.bedrock_enabled = False
|
|
self.region = ""
|
|
self.useless_reasons = []
|
|
self.logged = False
|
|
|
|
elif provider == Provider.AZURE:
|
|
self.endpoint = ""
|
|
self.best_deployment = ""
|
|
self.model = ""
|
|
self.deployments = []
|
|
self.unfiltered = False
|
|
|
|
elif provider == Provider.VERTEXAI:
|
|
self.project_id = ""
|
|
|
|
elif provider == Provider.MISTRAL:
|
|
self.subbed = False
|
|
|
|
|
|
class Provider(Enum):
|
|
OPENAI = 1,
|
|
ANTHROPIC = 2
|
|
AI21 = 3
|
|
MAKERSUITE = 4
|
|
AWS = 5
|
|
AZURE = 6
|
|
VERTEXAI = 7
|
|
MISTRAL = 8
|