feat: implement todo status tracking and nudge reminders

- Added functionality to track todo statuses and set a nudge reminder when a todo is marked as completed.
- Introduced `todoStatusById` and `todoCompletionNudgePending` properties in the session state.
- Updated the `insertPruneToolContext` function to conditionally insert nudge messages based on the new todo completion state.
- Refactored prompt loading to use the correct directory path.

Author: mcheviron

lib/messages/inject.ts

@@ -117,12 +117,18 @@ export const insertPruneToolContext = (
         logger.debug("prunable-tools: \n" + prunableToolsList)
 
         let nudgeString = ""
-        if (
+        const shouldNudge =
             config.tools.settings.nudgeEnabled &&
-            state.nudgeCounter >= config.tools.settings.nudgeFrequency
-        ) {
+            (state.nudgeCounter >= config.tools.settings.nudgeFrequency ||
+                state.todoCompletionNudgePending)
+
+        if (shouldNudge) {
             logger.info("Inserting prune nudge message")
             nudgeString = "\n" + getNudgeString(config)
+
+            if (state.todoCompletionNudgePending) {
+                state.todoCompletionNudgePending = false
+            }
         }
 
         prunableToolsContent = prunableToolsList + nudgeString

lib/prompts/index.ts

@@ -1,8 +1,11 @@
 import { readFileSync } from "fs"
-import { join } from "path"
+import { dirname, join } from "path"
+import { fileURLToPath } from "url"
+
+const PROMPTS_DIR = dirname(fileURLToPath(import.meta.url))
 
 export function loadPrompt(name: string, vars?: Record<string, string>): string {
-    const filePath = join(__dirname, `${name}.txt`)
+    const filePath = join(PROMPTS_DIR, `${name}.txt`)
     let content = readFileSync(filePath, "utf8").trim()
     if (vars) {
         for (const [key, value] of Object.entries(vars)) {

lib/state/state.ts

@@ -31,6 +31,8 @@ export const checkSession = async (
         state.lastCompaction = lastCompactionTimestamp
         state.toolParameters.clear()
         state.prune.toolIds = []
+        state.todoStatusById.clear()
+        state.todoCompletionNudgePending = false
         logger.info("Detected compaction from messages - cleared tool cache", {
             timestamp: lastCompactionTimestamp,
         })
@@ -56,6 +58,9 @@ export function createSessionState(): SessionState {
         lastCompaction: 0,
         currentTurn: 0,
         variant: undefined,
+
+        todoStatusById: new Map<string, string>(),
+        todoCompletionNudgePending: false,
     }
 }
 
@@ -75,6 +80,9 @@ export function resetSessionState(state: SessionState): void {
     state.lastCompaction = 0
     state.currentTurn = 0
     state.variant = undefined
+
+    state.todoStatusById.clear()
+    state.todoCompletionNudgePending = false
 }
 
 export async function ensureSessionInitialized(
@@ -101,6 +109,8 @@ export async function ensureSessionInitialized(
     state.lastCompaction = findLastCompactionTimestamp(messages)
     state.currentTurn = countTurns(state, messages)
 
+    seedTodoStatusFromMessages(state, messages)
+
     const persisted = await loadSessionState(sessionId, logger)
     if (persisted === null) {
         return
@@ -115,6 +125,28 @@ export async function ensureSessionInitialized(
     }
 }
 
+function seedTodoStatusFromMessages(state: SessionState, messages: WithParts[]): void {
+    for (const msg of messages) {
+        if (isMessageCompacted(state, msg)) {
+            continue
+        }
+        for (const part of msg.parts) {
+            if (part.type !== "tool" || part.tool !== "todowrite") {
+                continue
+            }
+            const todos = (part.state?.input as any)?.todos
+            if (!Array.isArray(todos)) {
+                continue
+            }
+            for (const todo of todos) {
+                if (todo && typeof todo.id === "string" && typeof todo.status === "string") {
+                    state.todoStatusById.set(todo.id, todo.status)
+                }
+            }
+        }
+    }
+}
+
 function findLastCompactionTimestamp(messages: WithParts[]): number {
     for (let i = messages.length - 1; i >= 0; i--) {
         const msg = messages[i]

lib/state/tool-cache.ts

@@ -60,6 +60,30 @@ export async function syncToolCache(
                     continue
                 }
 
+                if (part.tool === "todowrite") {
+                    const todos = (part.state?.input as any)?.todos
+                    if (Array.isArray(todos)) {
+                        for (const todo of todos) {
+                            if (
+                                !todo ||
+                                typeof todo.id !== "string" ||
+                                typeof todo.status !== "string"
+                            ) {
+                                continue
+                            }
+                            const previousStatus = state.todoStatusById.get(todo.id)
+                            if (
+                                previousStatus !== undefined &&
+                                previousStatus !== "completed" &&
+                                todo.status === "completed"
+                            ) {
+                                state.todoCompletionNudgePending = true
+                            }
+                            state.todoStatusById.set(todo.id, todo.status)
+                        }
+                    }
+                }
+
                 state.toolParameters.set(part.callID, {
                     tool: part.tool,
                     parameters: part.state?.input ?? {},

lib/state/types.ts

@@ -35,4 +35,7 @@ export interface SessionState {
     lastCompaction: number
     currentTurn: number
     variant: string | undefined
+
+    todoStatusById: Map<string, string>
+    todoCompletionNudgePending: boolean
 }