diff --git a/ui/desktop/src/components/settings/models/subcomponents/SwitchModelModal.tsx b/ui/desktop/src/components/settings/models/subcomponents/SwitchModelModal.tsx index a4cfa00ca8df..f6d2f485e0e1 100644 --- a/ui/desktop/src/components/settings/models/subcomponents/SwitchModelModal.tsx +++ b/ui/desktop/src/components/settings/models/subcomponents/SwitchModelModal.tsx @@ -76,12 +76,14 @@ export const SwitchModelModal = ({ titleOverride, }: SwitchModelModalProps) => { const { getProviders, getProviderModels, read } = useConfig(); - const { changeModel } = useModelAndProvider(); + const { changeModel, currentModel, currentProvider } = useModelAndProvider(); const [providerOptions, setProviderOptions] = useState<{ value: string; label: string }[]>([]); type ModelOption = { value: string; label: string; provider: string; isDisabled?: boolean }; const [modelOptions, setModelOptions] = useState<{ options: ModelOption[] }[]>([]); - const [provider, setProvider] = useState(initialProvider || null); - const [model, setModel] = useState(''); + const [provider, setProvider] = useState( + initialProvider || currentProvider || null + ); + const [model, setModel] = useState(currentModel || ''); const [isCustomModel, setIsCustomModel] = useState(false); const [validationErrors, setValidationErrors] = useState({ provider: '',