@tekmidian/pai 0.44.0 → 0.46.0

package/dist/hooks/load-project-context.mjs

@@ -135,12 +135,13 @@ function findSessionNotes(notesDir) {
 function normalizeLine(line) {
   return line.replace(/^(- \[[ x]\] |- |\* |\d+\.\s)/, "").trim().replace(/\s+/g, " ");
 }
-function extractKeyLines(content, maxChars, seen = /* @__PURE__ */ new Set()) {
+function extractKeyLines(content, maxChars, seen = /* @__PURE__ */ new Set(), seenLabels = /* @__PURE__ */ new Set()) {
   const lines = content.split("\n");
   const selected = [];
   let inTargetSection = false;
   let currentSection = "";
   let charCount = 0;
+  let pendingLabel = null;
   for (const line of lines) {
     const h2Match = line.match(/^## (.+)$/);
     const h3Match = line.match(/^### (.+)$/);
@@ -149,15 +150,12 @@ function extractKeyLines(content, maxChars, seen = /* @__PURE__ */ new Set()) {
       inTargetSection = EXTRACT_SECTIONS.some(
         (s) => currentSection.toLowerCase().includes(s.toLowerCase())
       );
+      pendingLabel = null;
       continue;
     }
     if (h3Match) {
       if (inTargetSection) {
-        const label = `[${h3Match[1]}]`;
-        if (charCount + label.length < maxChars) {
-          selected.push(label);
-          charCount += label.length + 1;
-        }
+        pendingLabel = h3Match[1];
       }
       continue;
     }
@@ -167,7 +165,21 @@ function extractKeyLines(content, maxChars, seen = /* @__PURE__ */ new Set()) {
     if (trimmed.startsWith("- ") || trimmed.startsWith("* ") || trimmed.match(/^\d+\./) || trimmed.startsWith("**")) {
       const normalized = normalizeLine(trimmed);
       if (seen.has(normalized)) continue;
-      if (charCount + trimmed.length + 1 > maxChars) break;
+      let labelText = "";
+      let labelKey = "";
+      if (pendingLabel !== null) {
+        labelKey = pendingLabel.toLowerCase().trim();
+        if (!seenLabels.has(labelKey)) {
+          labelText = `[${pendingLabel}]`;
+        }
+      }
+      if (charCount + labelText.length + trimmed.length + 2 > maxChars) break;
+      if (labelText) {
+        selected.push(labelText);
+        seenLabels.add(labelKey);
+        charCount += labelText.length + 1;
+      }
+      pendingLabel = null;
       seen.add(normalized);
       selected.push(trimmed);
       charCount += trimmed.length + 1;
@@ -183,6 +195,7 @@ function buildL1EssentialStory(rootPath, tokenBudget = L1_TOKEN_BUDGET) {
   if (noteFiles.length === 0) return "";
   const sections = [];
   const seen = /* @__PURE__ */ new Set();
+  const seenLabels = /* @__PURE__ */ new Set();
   let remaining = charBudget;
   for (const noteFile of noteFiles) {
     if (remaining <= 50) break;
@@ -197,8 +210,9 @@ function buildL1EssentialStory(rootPath, tokenBudget = L1_TOKEN_BUDGET) {
     const dateLabel = titleMatch ? titleMatch[1] : "";
     const titleLabel = titleMatch ? titleMatch[2] : name.replace(/^\d+ - /, "").replace(/\.md$/, "");
     const perNoteChars = Math.min(remaining, Math.floor(charBudget / noteFiles.length) + 200);
-    const extracted = extractKeyLines(content, perNoteChars, seen);
-    if (!extracted) continue;
+    const extracted = extractKeyLines(content, perNoteChars, seen, seenLabels);
+    const hasContentLine = extracted.split("\n").some((l) => !/^\[.*\]$/.test(l));
+    if (!extracted || !hasContentLine) continue;
     const noteBlock = `[${dateLabel} - ${titleLabel}]
 ${extracted}`;
     sections.push(noteBlock);
@@ -207,9 +221,9 @@ ${extracted}`;
   if (sections.length === 0) return "";
   return sections.join("\n\n");
 }
-function buildWakeupContext(rootPath, tokenBudget = L1_TOKEN_BUDGET) {
+function buildWakeupContext(rootPath, tokenBudget = L1_TOKEN_BUDGET, opts = {}) {
   const identity = loadL0Identity();
-  const essentialStory = rootPath ? buildL1EssentialStory(rootPath, tokenBudget) : "";
+  const essentialStory = rootPath && !opts.skipStory ? buildL1EssentialStory(rootPath, tokenBudget) : "";
   if (!identity && !essentialStory) return "";
   const parts = [];
   if (identity) {
@@ -1011,10 +1025,12 @@ Session Commands:
 </system-reminder>
 `;
   console.log(reminder);
+  let handoverInjected = false;
   if (existsSync7(todoPath)) {
     try {
       const checkpoint = readContinueCheckpoint(readFileSync6(todoPath, "utf-8"));
       if (checkpoint) {
+        handoverInjected = true;
         const from = checkpoint.meta?.session ? `
 From session: ${checkpoint.meta.session}` : "";
         const when = checkpoint.meta?.ts ? `
@@ -1073,7 +1089,7 @@ THE ABOVE INSTRUCTIONS ARE MANDATORY. Follow them exactly.
     } catch {
       wakeupRootPath = cwd;
     }
-    const wakeupBlock = buildWakeupContext(wakeupRootPath);
+    const wakeupBlock = buildWakeupContext(wakeupRootPath, void 0, { skipStory: handoverInjected });
     if (wakeupBlock) {
       const wakeupReminder = `
 <system-reminder>