feat: persist model selection
This commit is contained in:
@@ -20,12 +20,22 @@ export function Chat({ projectPath, onCloseProject }: ChatProps) {
|
||||
|
||||
useEffect(() => {
|
||||
invoke<string[]>("get_ollama_models")
|
||||
.then((models) => {
|
||||
.then(async (models) => {
|
||||
if (models.length > 0) {
|
||||
setAvailableModels(models);
|
||||
// If we have models and current one isn't valid, switch to first
|
||||
if (!models.includes(model)) {
|
||||
setModel(models[0]);
|
||||
|
||||
// Check backend store for saved model
|
||||
try {
|
||||
const savedModel = await invoke<string | null>(
|
||||
"get_model_preference",
|
||||
);
|
||||
if (savedModel && models.includes(savedModel)) {
|
||||
setModel(savedModel);
|
||||
} else if (!models.includes(model)) {
|
||||
setModel(models[0]);
|
||||
}
|
||||
} catch (e) {
|
||||
console.error(e);
|
||||
}
|
||||
}
|
||||
})
|
||||
@@ -161,7 +171,13 @@ export function Chat({ projectPath, onCloseProject }: ChatProps) {
|
||||
{availableModels.length > 0 ? (
|
||||
<select
|
||||
value={model}
|
||||
onChange={(e) => setModel(e.target.value)}
|
||||
onChange={(e) => {
|
||||
const newModel = e.target.value;
|
||||
setModel(newModel);
|
||||
invoke("set_model_preference", { model: newModel }).catch(
|
||||
console.error,
|
||||
);
|
||||
}}
|
||||
style={{
|
||||
padding: "6px 32px 6px 16px",
|
||||
borderRadius: "99px",
|
||||
@@ -188,7 +204,13 @@ export function Chat({ projectPath, onCloseProject }: ChatProps) {
|
||||
) : (
|
||||
<input
|
||||
value={model}
|
||||
onChange={(e) => setModel(e.target.value)}
|
||||
onChange={(e) => {
|
||||
const newModel = e.target.value;
|
||||
setModel(newModel);
|
||||
invoke("set_model_preference", { model: newModel }).catch(
|
||||
console.error,
|
||||
);
|
||||
}}
|
||||
placeholder="Model"
|
||||
style={{
|
||||
padding: "6px 12px",
|
||||
|
||||
Reference in New Issue
Block a user