diff --git a/enjoy/src/renderer/components/preferences/default-engine-settings.tsx b/enjoy/src/renderer/components/preferences/default-engine-settings.tsx index c4a27629..ffc19b33 100644 --- a/enjoy/src/renderer/components/preferences/default-engine-settings.tsx +++ b/enjoy/src/renderer/components/preferences/default-engine-settings.tsx @@ -55,11 +55,16 @@ export const DefaultEngineSettings = () => { const onSubmit = async (data: z.infer) => { const { name, models } = data; - models.default ||= providers[name].models[0]; + let options = [...providers[name].models]; + if (name === "openai" && openai?.models) { + options = openai.models.split(","); + } + + models.default ||= options[0]; Object.keys(models).forEach((key: keyof typeof models) => { - if (!providers[name].models.includes(models[key])) { + if (!options.includes(models[key])) { if (key === "default") { - models[key] = providers[name].models[0]; + models[key] = options[0]; } else { delete models[key]; } @@ -150,13 +155,15 @@ export const DefaultEngineSettings = () => { > - {providers[form.watch("name")].models.map( - (model: string) => ( - - {model} - - ) - )} + {(form.watch("name") === "openai" + ? openai?.models?.split(",") || + providers.openai.models + : providers.enjoyai.models + ).map((model: string) => ( + + {model} + + ))} @@ -185,13 +192,15 @@ export const DefaultEngineSettings = () => { > - {providers[form.watch("name")].models.map( - (model: string) => ( - - {model} - - ) - )} + {(form.watch("name") === "openai" + ? openai?.models?.split(",") || + providers.openai.models + : providers.enjoyai.models + ).map((model: string) => ( + + {model} + + ))} @@ -218,13 +227,15 @@ export const DefaultEngineSettings = () => { > - {providers[form.watch("name")].models.map( - (model: string) => ( - - {model} - - ) - )} + {(form.watch("name") === "openai" + ? openai?.models?.split(",") || + providers.openai.models + : providers.enjoyai.models + ).map((model: string) => ( + + {model} + + ))} @@ -251,13 +262,15 @@ export const DefaultEngineSettings = () => { > - {providers[form.watch("name")].models.map( - (model: string) => ( - - {model} - - ) - )} + {(form.watch("name") === "openai" + ? openai?.models?.split(",") || + providers.openai.models + : providers.enjoyai.models + ).map((model: string) => ( + + {model} + + ))} @@ -284,13 +297,15 @@ export const DefaultEngineSettings = () => { > - {providers[form.watch("name")].models.map( - (model: string) => ( - - {model} - - ) - )} + {(form.watch("name") === "openai" + ? openai?.models?.split(",") || + providers.openai.models + : providers.enjoyai.models + ).map((model: string) => ( + + {model} + + ))} diff --git a/enjoy/src/renderer/context/ai-settings-provider.tsx b/enjoy/src/renderer/context/ai-settings-provider.tsx index 97889d31..a7702100 100644 --- a/enjoy/src/renderer/context/ai-settings-provider.tsx +++ b/enjoy/src/renderer/context/ai-settings-provider.tsx @@ -136,7 +136,6 @@ export const AISettingsProvider = ({ ? Object.assign(gptEngine, { key: openai.key, baseUrl: openai.baseUrl, - models: openai.models, }) : Object.assign(gptEngine, { key: user?.accessToken, diff --git a/enjoy/src/renderer/hooks/use-conversation.tsx b/enjoy/src/renderer/hooks/use-conversation.tsx index b396a99e..0ed296a4 100644 --- a/enjoy/src/renderer/hooks/use-conversation.tsx +++ b/enjoy/src/renderer/hooks/use-conversation.tsx @@ -21,9 +21,7 @@ export const useConversation = () => { const { EnjoyApp, webApi, user, apiUrl, learningLanguage } = useContext( AppSettingsProviderContext ); - const { openai, currentEngine } = useContext( - AISettingsProviderContext - ); + const { openai, currentEngine } = useContext(AISettingsProviderContext); const pickLlm = (conversation: ConversationType) => { const { @@ -152,7 +150,7 @@ export const useConversation = () => { const chain = new ConversationChain({ llm: llm as any, memory, - prompt, + prompt: prompt as any, verbose: true, }); let response: LLMResult["generations"][0] = [];