fixes OAI trial keys bricking the dall-e queue

This commit is contained in:
nai-degen 2024-01-25 01:47:51 -06:00
parent 02a54bf4e3
commit 6f626e623e
1 changed files with 6 additions and 0 deletions

View File

@ -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.