Fix dreamer key-files lease loss during long LLM call

The key-files post-task phase ran outside the dreamer task wrapper that
provides periodic lease renewal, so when the LLM call took longer than
the 2-minute lease TTL (e.g. 197s on this project's corpus), the
commit-time peekLeaseHolderAndExpiry check failed with "lease lost"
even though no other holder had taken over.

Three consecutive dream runs failed with this exact pattern after
v6 key-files shipped. The fix mirrors the proven setInterval pattern
used by the smart-notes post-task phase at runner.ts:1068.

Wraps the LLM call + commit in a setInterval that calls renewLease
every 60s, cleared in finally.

Author: ualtinok

packages/plugin/src/features/magic-context/key-files/identify-key-files.ts

@@ -513,30 +513,47 @@ export async function runKeyFilesTask(args: {
 
     const prompt = buildV6KeyFilesPrompt({ candidates, currentRows, config: args.config });
     let validated: ValidatedKeyFilesOutput;
+    // Renew the dream lease every 60s while the LLM call is in flight so the
+    // commit-time lease check (peekLeaseHolderAndExpiry) sees a live expiry.
+    // Without this, key-files runs longer than the 2-minute lease TTL fail at
+    // commit time with "lease lost" even though no other holder took over.
+    const leaseInterval = setInterval(() => {
+        try {
+            if (!renewLease(args.db, args.holderId)) {
+                log("[key-files] lease renewal failed during LLM call");
+            }
+        } catch (error) {
+            log(`[key-files] lease renewal threw: ${getErrorMessage(error)}`);
+        }
+    }, 60_000);
     try {
-        const raw = await runKeyFilesLlm({
-            client: args.client,
-            parentSessionId: args.parentSessionId,
+        try {
+            const raw = await runKeyFilesLlm({
+                client: args.client,
+                parentSessionId: args.parentSessionId,
+                projectPath,
+                prompt,
+                deadline: args.deadline,
+                fallbackModels: args.fallbackModels,
+            });
+            validated = validateLlmOutput(raw, args.config, projectPath);
+        } catch (error) {
+            log(`[key-files] LLM validation failed: ${getErrorMessage(error)}`);
+            throw error;
+        }
+        if (validated.no_change)
+            return { committedVersion: null, candidates: candidates.length, noChange: true };
+        const committedVersion = commitKeyFiles({
+            db: args.db,
             projectPath,
-            prompt,
-            deadline: args.deadline,
-            fallbackModels: args.fallbackModels,
+            validated,
+            configHash,
+            modelId: args.fallbackModels?.[0] ?? "dreamer",
+            leaseHolderId: args.holderId,
         });
-        validated = validateLlmOutput(raw, args.config, projectPath);
-    } catch (error) {
-        log(`[key-files] LLM validation failed: ${getErrorMessage(error)}`);
-        throw error;
+        renewLease(args.db, args.holderId);
+        return { committedVersion, candidates: candidates.length, noChange: false };
+    } finally {
+        clearInterval(leaseInterval);
     }
-    if (validated.no_change)
-        return { committedVersion: null, candidates: candidates.length, noChange: true };
-    const committedVersion = commitKeyFiles({
-        db: args.db,
-        projectPath,
-        validated,
-        configHash,
-        modelId: args.fallbackModels?.[0] ?? "dreamer",
-        leaseHolderId: args.holderId,
-    });
-    renewLease(args.db, args.holderId);
-    return { committedVersion, candidates: candidates.length, noChange: false };
 }