From dc61291933cecaee26234ee184567f7ce21b4db9 Mon Sep 17 00:00:00 2001 From: nai-degen Date: Fri, 17 Nov 2023 20:24:36 -0600 Subject: [PATCH] adds temporary keychecker var to treat dall-e-2 the same as dall-e-3 --- src/shared/key-management/openai/checker.ts | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/shared/key-management/openai/checker.ts b/src/shared/key-management/openai/checker.ts index 88d7b59..645a02b 100644 --- a/src/shared/key-management/openai/checker.ts +++ b/src/shared/key-management/openai/checker.ts @@ -98,10 +98,10 @@ export class OpenAIKeyChecker extends KeyCheckerBase { const families = new Set(); models.forEach(({ id }) => families.add(getOpenAIModelFamily(id, "turbo"))); - // For now we remove dall-e from the list of provisioned models if only - // dall-e-2 is available. if (families.has("dall-e") && !models.find(({ id }) => id === "dall-e-3")) { - families.delete("dall-e"); + if (process.env.DALLE_2_OVERRIDE !== "true") { + families.delete("dall-e"); + } } // We want to update the key's model families here, but we don't want to