mirror of
https://github.com/cunnymessiah/keychecker.git
synced 2026-05-10 18:39:04 -07:00
based
25 lines
549 B
Python
25 lines
549 B
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
|
|
|
|
elif provider == Provider.ANTHROPIC:
|
|
self.pozzed = False
|
|
self.rate_limited = False
|
|
|
|
|
|
class Provider(Enum):
|
|
OPENAI = 1,
|
|
ANTHROPIC = 2
|