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

        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 = ""

        elif provider == Provider.AZURE:
            self.endpoint = ""
            self.deployment = ""
            self.unfiltered = False


class Provider(Enum):
    OPENAI = 1,
    ANTHROPIC = 2
    AI21 = 3
    PALM = 4
    AWS = 5
    AZURE = 6