Move context state restoration until after checkpoint file synchronization is complete

Author: robertpiosik

packages/vscode/package.json

@@ -2,7 +2,7 @@
   "name": "gemini-coder",
   "displayName": "Code Web Chat",
   "description": "CWC is a free AI coding assistant without agents",
-  "version": "1.572.0",
+  "version": "1.573.0",
   "scripts": {
     "build": "npx vsce package --no-dependencies",
     "vscode:prepublish": "rimraf out && npm run compile",

packages/vscode/src/commands/checkpoints-command/actions/restore-checkpoint.ts

@@ -130,21 +130,6 @@ export const restore_checkpoint = async (params: {
       CHECKPOINT_OPERATION_IN_PROGRESS_STATE_KEY,
       Date.now()
     )
-
-    if (params.checkpoint.checked_files) {
-      await params.context.workspaceState.update(
-        CONTEXT_CHECKED_PATHS_STATE_KEY,
-        params.checkpoint.checked_files
-      )
-      params.workspace_provider.load_checked_files_state()
-    }
-    if (params.checkpoint.checked_websites) {
-      await params.context.workspaceState.update(
-        CONTEXT_CHECKED_URLS_STATE_KEY,
-        params.checkpoint.checked_websites
-      )
-      params.websites_provider.load_checked_websites_state()
-    }
     await vscode.window.withProgress(
       {
         location: vscode.ProgressLocation.Notification,
@@ -353,6 +338,21 @@ export const restore_checkpoint = async (params: {
       } catch {}
     }
 
+    if (params.checkpoint.checked_files) {
+      await params.context.workspaceState.update(
+        CONTEXT_CHECKED_PATHS_STATE_KEY,
+        params.checkpoint.checked_files
+      )
+      params.workspace_provider.load_checked_files_state()
+    }
+    if (params.checkpoint.checked_websites) {
+      await params.context.workspaceState.update(
+        CONTEXT_CHECKED_URLS_STATE_KEY,
+        params.checkpoint.checked_websites
+      )
+      params.websites_provider.load_checked_websites_state()
+    }
+
     if (
       params.checkpoint.title == 'Accepted response preview' &&
       params.checkpoint.response_preview_item_created_at &&