From 6f626e623ebea6e5bf665895eba3148ed497f5f6 Mon Sep 17 00:00:00 2001
From: nai-degen <khoners@protonmail.com>
Date: Thu, 25 Jan 2024 01:47:51 -0600
Subject: [PATCH] fixes OAI trial keys bricking the dall-e queue

---
 src/shared/key-management/openai/checker.ts | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/shared/key-management/openai/checker.ts b/src/shared/key-management/openai/checker.ts
index 6afdb3c..52d933a 100644
--- a/src/shared/key-management/openai/checker.ts
+++ b/src/shared/key-management/openai/checker.ts
@@ -73,6 +73,12 @@ export class OpenAIKeyChecker extends KeyCheckerBase<OpenAIKey> {
 
     const families = new Set<OpenAIModelFamily>();
     models.forEach(({ id }) => families.add(getOpenAIModelFamily(id, "turbo")));
+    
+    // disable dall-e for trial keys due to very low per-day quota that tends to
+    // render the key unusable.
+    if (key.isTrial) {
+      families.delete("dall-e");
+    }
 
     // as of 2023-11-18, many keys no longer return the dalle3 model but still
     // have access to it via the api for whatever reason.