fix vscode bugs (#558)

AbanteAI · Apr 5, 2024 · 93926e4 · 93926e4
1 parent 0a984b9
commit 93926e4
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 13 deletions.
diff --git a/mentat-vscode/package.json b/mentat-vscode/package.json
@@ -2,7 +2,7 @@
     "name": "mentat",
     "displayName": "Mentat",
     "description": "Mentat - The AI Coding Assistant",
-    "version": "0.0.4",
+    "version": "0.0.6",
     "private": true,
     "publisher": "AbanteAI",
     "license": "Apache-2.0",

diff --git a/mentat-vscode/src/webviews/views/Chat.tsx b/mentat-vscode/src/webviews/views/Chat.tsx
@@ -199,13 +199,14 @@ export default function Chat() {
                 const subchannel = message.channel.split(":").at(1);
                 switch (subchannel) {
                     case "newSession": {
+                        setLoaded(true);
                         const state: any = vscode.getState();
                         const statePreviousMessages: Message[] =
                             state.previousMessages ?? [];
                         const stateMessages: Message[] = state.messages ?? [];
                         if (
                             stateMessages.some(
-                                (message) => message.source == "user"
+                                (message) => message.source === "user"
                             )
                         ) {
                             setPreviousMessages([
@@ -228,23 +229,22 @@ export default function Chat() {
                             },
                         ]);
                         setWorkspaceRoot(message.extra.workspaceRoot);
-                        setLoaded(true);
                         break;
                     }
                     case "continuingSession": {
+                        setLoaded(true);
                         const state: any = vscode.getState();
                         if (state) {
-                            setPreviousMessages(state.previousMessages);
-                            setMessages(state.messages);
-                            setInputRequestId(state.inputRequestId);
-                            setSessionActive(state.sessionActive);
-                            setTextAreaValue(state.textAreaValue);
-                            setInterruptable(state.interruptable);
-                            setActiveEdits(state.activeEdits);
-                            setWorkspaceRoot(state.workspaceRoot);
-                            setContextUpdateData(state.contextUpdataData);
+                            setPreviousMessages(state.previousMessages ?? []);
+                            setMessages(state.messages ?? []);
+                            setInputRequestId(state.inputRequestId ?? null);
+                            setSessionActive(state.sessionActive ?? true);
+                            setTextAreaValue(state.textAreaValue ?? "");
+                            setInterruptable(state.interruptable ?? false);
+                            setActiveEdits(state.activeEdits ?? []);
+                            setWorkspaceRoot(state.workspaceRoot ?? "");
+                            setContextUpdateData(state.contextUpdataData ?? {});
                         }
-                        setLoaded(true);
                         break;
                     }
                     case "clearChatbox": {