schizo

2026-05-11 10:50:12 -07:00 · 2024-03-18 10:01:49 +10:00
parent a1f7d18af9
commit f88fe7faf9
4 changed files with 100 additions and 20 deletions
@@ -14,11 +14,15 @@ class APIKey:
            self.organizations = []
            self.rpm = 0
            self.tier = ""
+            self.has_special_models = False
+            self.the_one = False

        elif provider == Provider.ANTHROPIC:
            self.pozzed = False
            self.rate_limited = False
            self.has_quota = True
+            self.tier = ""
+            self.remaining_tokens = 0

        elif provider == Provider.AI21:
            self.trial_elapsed = False
@@ -57,6 +61,12 @@ class APIKey:
            self.limit_reached = False
            self.bought_credits = False

+    def clone(self):
+        cloned_key = APIKey(self.provider, self.api_key)
+        cloned_key.__dict__ = self.__dict__.copy()
+        return cloned_key
+
+
 class Provider(Enum):
    OPENAI = 1,
    ANTHROPIC = 2