added ai21 support

2026-05-10 18:39:04 -07:00 · 2023-11-10 07:57:50 +10:00
parent 03bed91409
commit 16536c9bde
4 changed files with 68 additions and 10 deletions
@@ -18,7 +18,11 @@ class APIKey:
            self.pozzed = False
            self.rate_limited = False

+        elif provider == Provider.AI21:
+            self.trial_elapsed = False
+

 class Provider(Enum):
    OPENAI = 1,
    ANTHROPIC = 2
+    AI21 = 3