klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-04 05:39:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Prevent infinite loops in model selection logic by configuring an initial model state

Browse Source
This commit is contained in:
sabaimran
2025-02-03 18:11:01 -08:00
parent 489fa71143
commit 51874c25d5
2 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/interface/web/app/common/modelSelector.tsx
View File

@@ -31,6 +31,7 @@ interface ModelSelectorProps extends PopoverProps {
onSelect: (model: ModelOptions, userModification: boolean) => void;
selectedModel?: string;
disabled?: boolean;
initialModel?: string;
}
export function ModelSelector({ ...props }: ModelSelectorProps) {
@@ -46,25 +47,31 @@ export function ModelSelector({ ...props }: ModelSelectorProps) {
if (userConfig) {
setModels(userConfig.chat_model_options);
const selectedChatModelOption = userConfig.chat_model_options.find(model => model.id === userConfig.selected_chat_model_config);
setSelectedModel(selectedChatModelOption);
return;
if (!props.initialModel) {
const selectedChatModelOption = userConfig.chat_model_options.find(model => model.id === userConfig.selected_chat_model_config);
setSelectedModel(selectedChatModelOption);
} else {
const model = userConfig.chat_model_options.find(model => model.name === props.initialModel);
setSelectedModel(model);
}
} else {
setSelectedModel(models[0]);
}
}, [models, props.selectedModel, userConfig]);
}, [userConfig, props.initialModel, isLoadingUserConfig]);
useEffect(() => {
if (props.selectedModel) {
if (props.selectedModel && selectedModel && props.selectedModel !== selectedModel.name) {
const model = models.find(model => model.name === props.selectedModel);
setSelectedModel(model);
}
}, [props.selectedModel]);
}, [props.selectedModel, models]);
useEffect(() => {
if (selectedModel) {
const userModification = selectedModel.id !== userConfig?.selected_chat_model_config;
props.onSelect(selectedModel, userModification);
if (props.selectedModel !== selectedModel.name) {
props.onSelect(selectedModel, userModification);
}
}
}, [selectedModel]);