diff --git a/enjoy/src/renderer/components/preferences/default-engine-settings.tsx b/enjoy/src/renderer/components/preferences/default-engine-settings.tsx index ffc19b33..1a41d97c 100644 --- a/enjoy/src/renderer/components/preferences/default-engine-settings.tsx +++ b/enjoy/src/renderer/components/preferences/default-engine-settings.tsx @@ -52,6 +52,16 @@ export const DefaultEngineSettings = () => { }, }); + const modelOptions = () => { + if (form.watch("name") === "openai") { + const customModels = openai?.models?.split(",")?.filter(Boolean); + + return customModels.length ? customModels : providers.openai.models; + } else { + return providers.enjoyai.models; + } + }; + const onSubmit = async (data: z.infer) => { const { name, models } = data; @@ -155,11 +165,7 @@ export const DefaultEngineSettings = () => { > - {(form.watch("name") === "openai" - ? openai?.models?.split(",") || - providers.openai.models - : providers.enjoyai.models - ).map((model: string) => ( + {modelOptions().map((model: string) => ( {model} @@ -192,11 +198,7 @@ export const DefaultEngineSettings = () => { > - {(form.watch("name") === "openai" - ? openai?.models?.split(",") || - providers.openai.models - : providers.enjoyai.models - ).map((model: string) => ( + {modelOptions().map((model: string) => ( {model} @@ -227,11 +229,7 @@ export const DefaultEngineSettings = () => { > - {(form.watch("name") === "openai" - ? openai?.models?.split(",") || - providers.openai.models - : providers.enjoyai.models - ).map((model: string) => ( + {modelOptions().map((model: string) => ( {model} @@ -262,11 +260,7 @@ export const DefaultEngineSettings = () => { > - {(form.watch("name") === "openai" - ? openai?.models?.split(",") || - providers.openai.models - : providers.enjoyai.models - ).map((model: string) => ( + {modelOptions().map((model: string) => ( {model} @@ -297,11 +291,7 @@ export const DefaultEngineSettings = () => { > - {(form.watch("name") === "openai" - ? openai?.models?.split(",") || - providers.openai.models - : providers.enjoyai.models - ).map((model: string) => ( + {modelOptions().map((model: string) => ( {model}