feat: persist task dialog model selections

Author: safe1ine

frontend/src/components/console/project/start-develop-task-dialog.tsx

@@ -16,6 +16,7 @@ import { useState, useEffect, useMemo, useRef } from "react"
 import { useNavigate } from "react-router-dom"
 import { toast } from "sonner"
 import { TaskConcurrentLimitDialog } from "@/components/console/task/task-concurrent-limit-dialog"
+import { readStoredTaskDialogParams, writeStoredTaskDialogParams } from "@/components/console/task/task-dialog-params-storage"
 
 interface StartDevelopTaskDialogProps {
   open: boolean
@@ -111,8 +112,15 @@ export default function StartDevelopTaskDialog({
       const justOpened = !prevOpenRef.current
       prevOpenRef.current = true
       if (justOpened) {
+        const storedParams = readStoredTaskDialogParams()
         setUserMessage('')
-        setSelectedModelId(selectPreferredTaskModel(models, subscription))
+        setSelectedModelId(
+          storedParams.modelId
+            && models.some((model) => model.id === storedParams.modelId)
+            && canUseModelBySubscription(models.find((model) => model.id === storedParams.modelId), subscription)
+            ? storedParams.modelId
+            : selectPreferredTaskModel(models, subscription)
+        )
         setSelectedBranch('main')
       }
       fetchBranches()
@@ -155,6 +163,12 @@ export default function StartDevelopTaskDialog({
       return
     }
 
+    const storedParams = readStoredTaskDialogParams()
+    writeStoredTaskDialogParams({
+      ...storedParams,
+      modelId: selectedModelId,
+    })
+
     setSubmitting(true)
 
     // 创建任务

frontend/src/components/console/task/task-dialog-params-storage.ts

@@ -0,0 +1,33 @@
+const TASK_PARAM_STORAGE_KEY = "task_input_dialog_params";
+
+export interface StoredTaskDialogParams {
+  modelId?: string;
+  hostId?: string;
+  imageId?: string;
+}
+
+export function readStoredTaskDialogParams(): StoredTaskDialogParams {
+  try {
+    const raw = window.localStorage.getItem(TASK_PARAM_STORAGE_KEY);
+    if (!raw) {
+      return {};
+    }
+
+    const parsed = JSON.parse(raw) as StoredTaskDialogParams;
+    return {
+      modelId: typeof parsed.modelId === "string" ? parsed.modelId : undefined,
+      hostId: typeof parsed.hostId === "string" ? parsed.hostId : undefined,
+      imageId: typeof parsed.imageId === "string" ? parsed.imageId : undefined,
+    };
+  } catch {
+    return {};
+  }
+}
+
+export function writeStoredTaskDialogParams(params: StoredTaskDialogParams) {
+  try {
+    window.localStorage.setItem(TASK_PARAM_STORAGE_KEY, JSON.stringify(params));
+  } catch {
+    // ignore storage write failures
+  }
+}

frontend/src/components/console/task/task-input.tsx

@@ -30,6 +30,7 @@ import { Tooltip, TooltipContent, TooltipTrigger } from "@/components/ui/tooltip
 import { Tabs, TabsContent, TabsList, TabsTrigger } from "@/components/ui/tabs";
 import { Empty, EmptyDescription, EmptyHeader, EmptyMedia } from "@/components/ui/empty";
 import { defaultSkills } from "@/utils/config";
+import { readStoredTaskDialogParams, writeStoredTaskDialogParams } from "./task-dialog-params-storage";
 
 
 interface SkillItemProps {
@@ -210,16 +211,39 @@ export function TaskInput({ repos, onTaskCreated }: TaskInputProps) {
   };
 
   const setDefaultConfig = () => {
-    setSelectedModelId(selectPreferredTaskModel(models, subscription))
-    setSelectedHostId(selectHost(hosts, true))
+    const storedParams = readStoredTaskDialogParams();
+    const defaultModelId = selectPreferredTaskModel(models, subscription);
+    const nextModelId = (
+      storedParams.modelId
+      && models.some((model) => model.id === storedParams.modelId)
+      && canUseModelBySubscription(models.find((model) => model.id === storedParams.modelId), subscription)
+    )
+      ? storedParams.modelId
+      : defaultModelId;
+
+    setSelectedModelId(nextModelId);
 
     if (user.role === ConstsUserRole.UserRoleSubAccount) {
-      setSelectedHostId(selectHost(hosts, true))
-      setSelectedImageId(selectImage(images, true));
-    } else {
-      setSelectedHostId(selectHost(hosts, false))
-      setSelectedImageId(selectImage(images, false));
+      const nextHostId = (
+        storedParams.hostId === "public_host"
+        || hosts.some((host) => host.id === storedParams.hostId && host.status === ConstsHostStatus.HostStatusOnline)
+      )
+        ? (storedParams.hostId || "public_host")
+        : selectHost(hosts, true);
+      const nextImageId = (
+        storedParams.imageId
+        && images.some((image) => image.id === storedParams.imageId)
+      )
+        ? storedParams.imageId
+        : selectImage(images, true);
+
+      setSelectedHostId(nextHostId);
+      setSelectedImageId(nextImageId);
+      return;
     }
+
+    setSelectedHostId(selectHost(hosts, false));
+    setSelectedImageId(selectImage(images, false));
   };
 
   const adaptedModelForTool = (): boolean => {
@@ -265,6 +289,13 @@ export function TaskInput({ repos, onTaskCreated }: TaskInputProps) {
       return;
     }
 
+    const storedParams = readStoredTaskDialogParams();
+    writeStoredTaskDialogParams({
+      modelId: selectedModelId,
+      hostId: user.role === ConstsUserRole.UserRoleSubAccount ? selectedHostId : storedParams.hostId,
+      imageId: user.role === ConstsUserRole.UserRoleSubAccount ? selectedImageId : storedParams.imageId,
+    });
+
     executeTask();
   };