@trevonistrevon/pi-loop 0.1.9 → 0.2.1

package/dist/index.js

@@ -30,10 +30,8 @@ function debug(...args) {
 function textResult(msg) {
     return { content: [{ type: "text", text: msg }], details: undefined };
 }
-const LOOP_TOOL_NAMES = new Set(["LoopCreate", "LoopList", "LoopDelete", "MonitorCreate", "MonitorList", "MonitorStop"]);
-const REMINDER_INTERVAL = 3;
 const SYSTEM_REMINDER_TEMPLATE = `<system-reminder>
-Scheduled loop "%propmpt%" fired. Trigger: %trigger_info%.
+Scheduled loop "%prompt%" fired. Trigger: %trigger_info%.
 [loop:%loop_id%]
 </system-reminder>`;
 export default function (pi) {
@@ -150,10 +148,11 @@ export default function (pi) {
         if (persistedShown)
             return;
         persistedShown = true;
+        const sessionStartedAt = Date.now();
         const loops = store.list();
         if (loops.length > 0) {
             store.clearExpired();
-            store.expireEventLoops();
+            store.expireEventLoops(sessionStartedAt);
             triggerSystem.start();
             widget.update();
         }
@@ -183,11 +182,9 @@ export default function (pi) {
         showPersistedLoops(isResume);
     });
     // ── System-reminder injection for loop fires ──
-    let currentTurn = 0;
-    let lastLoopToolUseTurn = 0;
-    let reminderInjectedThisCycle = false;
+    let canInjectReminder = true;
     const pendingReminders = [];
-    pi.on("loop:fire", (data) => {
+    pi.events.on("loop:fire", (data) => {
         const triggerInfo = typeof data.trigger === "string"
             ? data.trigger
             : data.trigger?.type === "cron"
@@ -201,22 +198,15 @@ export default function (pi) {
             .replace("%loop_id%", data.loopId || "unknown");
         pendingReminders.push(reminder);
     });
+    // Allow one reminder injection per agent turn (at the first tool call),
+    // so the reminder arrives between batches of work, not mid-execution.
     pi.on("turn_start", async () => {
-        currentTurn++;
+        canInjectReminder = true;
     });
     pi.on("tool_result", async (event) => {
-        if (LOOP_TOOL_NAMES.has(event.toolName)) {
-            lastLoopToolUseTurn = currentTurn;
-            reminderInjectedThisCycle = false;
+        if (!canInjectReminder || pendingReminders.length === 0)
             return {};
-        }
-        if (currentTurn - lastLoopToolUseTurn < REMINDER_INTERVAL || reminderInjectedThisCycle) {
-            return {};
-        }
-        if (pendingReminders.length === 0)
-            return {};
-        reminderInjectedThisCycle = true;
-        lastLoopToolUseTurn = currentTurn;
+        canInjectReminder = false;
         const reminder = pendingReminders.shift();
         return {
             content: [...event.content, { type: "text", text: reminder }],

package/dist/store.d.ts

@@ -25,7 +25,7 @@ export declare class LoopStore {
     };
     delete(id: string): boolean;
     clearExpired(): number;
-    expireEventLoops(): number;
+    expireEventLoops(sessionStartedAt: number): number;
     clearAll(): number;
     deleteFileIfEmpty(): boolean;
 }

package/dist/store.js

@@ -176,15 +176,17 @@ export class LoopStore {
             return count;
         });
     }
-    expireEventLoops() {
+    expireEventLoops(sessionStartedAt) {
         return this.withLock(() => {
             let count = 0;
             for (const [_id, entry] of this.loops) {
                 if (entry.status !== "active")
                     continue;
                 if (entry.trigger.type === "event" || entry.trigger.type === "hybrid") {
-                    entry.status = "expired";
-                    count++;
+                    if (entry.createdAt < sessionStartedAt) {
+                        entry.status = "expired";
+                        count++;
+                    }
                 }
             }
             return count;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@trevonistrevon/pi-loop",
-  "version": "0.1.9",
+  "version": "0.2.1",
   "description": "A pi extension for cron/event-based agent re-wake loops and background process monitoring.",
   "author": "trevonistrevon",
   "license": "MIT",

package/src/index.ts

@@ -35,11 +35,8 @@ function textResult(msg: string) {
   return { content: [{ type: "text" as const, text: msg }], details: undefined as any };
 }
 
-const LOOP_TOOL_NAMES = new Set(["LoopCreate", "LoopList", "LoopDelete", "MonitorCreate", "MonitorList", "MonitorStop"]);
-const REMINDER_INTERVAL = 3;
-
 const SYSTEM_REMINDER_TEMPLATE = `<system-reminder>
-Scheduled loop "%propmpt%" fired. Trigger: %trigger_info%.
+Scheduled loop "%prompt%" fired. Trigger: %trigger_info%.
 [loop:%loop_id%]
 </system-reminder>`;
 
@@ -156,10 +153,11 @@ export default function (pi: ExtensionAPI) {
   function showPersistedLoops(_isResume = false) {
     if (persistedShown) return;
     persistedShown = true;
+    const sessionStartedAt = Date.now();
     const loops = store.list();
     if (loops.length > 0) {
       store.clearExpired();
-      store.expireEventLoops();
+      store.expireEventLoops(sessionStartedAt);
       triggerSystem.start();
       widget.update();
     }
@@ -197,12 +195,10 @@ export default function (pi: ExtensionAPI) {
 
   // ── System-reminder injection for loop fires ──
 
-  let currentTurn = 0;
-  let lastLoopToolUseTurn = 0;
-  let reminderInjectedThisCycle = false;
+  let canInjectReminder = true;
   const pendingReminders: string[] = [];
 
-  pi.on("loop:fire" as any, (data: any) => {
+  pi.events.on("loop:fire", (data: any) => {
     const triggerInfo = typeof data.trigger === "string"
       ? data.trigger
       : data.trigger?.type === "cron"
@@ -219,25 +215,16 @@ export default function (pi: ExtensionAPI) {
     pendingReminders.push(reminder);
   });
 
+  // Allow one reminder injection per agent turn (at the first tool call),
+  // so the reminder arrives between batches of work, not mid-execution.
   pi.on("turn_start", async () => {
-    currentTurn++;
+    canInjectReminder = true;
   });
 
   pi.on("tool_result", async (event) => {
-    if (LOOP_TOOL_NAMES.has(event.toolName)) {
-      lastLoopToolUseTurn = currentTurn;
-      reminderInjectedThisCycle = false;
-      return {};
-    }
-
-    if (currentTurn - lastLoopToolUseTurn < REMINDER_INTERVAL || reminderInjectedThisCycle) {
-      return {};
-    }
-
-    if (pendingReminders.length === 0) return {};
+    if (!canInjectReminder || pendingReminders.length === 0) return {};
 
-    reminderInjectedThisCycle = true;
-    lastLoopToolUseTurn = currentTurn;
+    canInjectReminder = false;
     const reminder = pendingReminders.shift()!;
     return {
       content: [...event.content, { type: "text" as const, text: reminder }],

package/src/store.ts

@@ -173,14 +173,16 @@ export class LoopStore {
     });
   }
 
-  expireEventLoops(): number {
+  expireEventLoops(sessionStartedAt: number): number {
     return this.withLock(() => {
       let count = 0;
       for (const [_id, entry] of this.loops) {
         if (entry.status !== "active") continue;
         if (entry.trigger.type === "event" || entry.trigger.type === "hybrid") {
-          entry.status = "expired";
-          count++;
+          if (entry.createdAt < sessionStartedAt) {
+            entry.status = "expired";
+            count++;
+          }
         }
       }
       return count;