feat: Retain model selection upon starting a new chat session

mlc-ai · May 19, 2024 · 98264f6 · 98264f6
1 parent 434da32
commit 98264f6
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 298 deletions.
diff --git a/app/components/new-chat.module.scss b/app/components/new-chat.module.scss
diff --git a/app/components/new-chat.tsx b/app/components/new-chat.tsx
diff --git a/app/store/chat.ts b/app/store/chat.ts
@@ -191,6 +191,9 @@ export const useChatStore = createPersistStore(
             },
           };
           session.topic = mask.name;
+        } else {
+          // inherit current model config
+          session.mask.modelConfig = this.currentSession().mask.modelConfig;
         }
 
         set((state) => ({

diff --git a/app/store/mask.ts b/app/store/mask.ts
@@ -1,6 +1,6 @@
 import { BUILTIN_MASKS } from "../masks";
 import { getLang, Lang } from "../locales";
-import { DEFAULT_TOPIC, ChatMessage } from "./chat";
+import { DEFAULT_TOPIC, ChatMessage, useChatStore } from "./chat";
 import { ModelConfig, useAppConfig } from "./config";
 import { StoreKey } from "../constant";
 import { nanoid } from "nanoid";
@@ -33,7 +33,9 @@ export const createEmptyMask = () =>
     name: DEFAULT_TOPIC,
     context: [],
     syncGlobalConfig: true, // use global config as default
-    modelConfig: { ...useAppConfig.getState().modelConfig },
+    modelConfig: {
+      ...useAppConfig.getState().modelConfig,
+    },
     lang: getLang(),
     builtin: false,
     createdAt: Date.now(),