Mirrors
/
oai-reverse-proxy
mirror of https://gitgud.io/khanon/oai-reverse-proxy.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

adds temporary keychecker var to treat dall-e-2 the same as dall-e-3

This commit is contained in:
nai-degen 2023-11-17 20:24:36 -06:00
parent 6c02e9b265
commit dc61291933
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/shared/key-management/openai/checker.ts
View File

@ -98,10 +98,10 @@ export class OpenAIKeyChecker extends KeyCheckerBase<OpenAIKey> {
const families = new Set<OpenAIModelFamily>();
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