astrocode-workflow 0.1.39 → 0.1.41

package/dist/hooks/continuation-enforcer.js

@@ -16,7 +16,7 @@ export function createContinuationEnforcer(opts) {
         const cur = sessions.get(sessionId);
         if (cur)
             return cur;
-        const state = { lastHash: null, lastAtMs: 0, repeats: 0, autoSteps: 0, idleTimer: null };
+        const state = { lastHash: null, lastAtMs: 0, repeats: 0, autoSteps: 0, idleTimer: null, periodicTimer: null };
         sessions.set(sessionId, state);
         return state;
     }
@@ -27,6 +27,13 @@ export function createContinuationEnforcer(opts) {
             s.idleTimer = null;
         }
     }
+    function clearPeriodicTimer(sessionId) {
+        const s = getState(sessionId);
+        if (s.periodicTimer) {
+            clearInterval(s.periodicTimer);
+            s.periodicTimer = null;
+        }
+    }
     function scheduleIdleInjection(sessionId) {
         clearIdleTimer(sessionId);
         if (!config.continuation.enabled)
@@ -40,6 +47,18 @@ export function createContinuationEnforcer(opts) {
             void maybeInjectContinue(sessionId, "idle_timer");
         }, delay);
     }
+    function schedulePeriodicInjection(sessionId) {
+        clearPeriodicTimer(sessionId);
+        if (!config.continuation.enabled)
+            return;
+        // Inject every 5 minutes (300,000 ms)
+        const interval = 5 * 60 * 1000;
+        const s = getState(sessionId);
+        s.periodicTimer = setInterval(() => {
+            // Fire and forget
+            void maybeInjectContinue(sessionId, "periodic_timer");
+        }, interval);
+    }
     function shouldDedupe(sessionId, directive) {
         const s = getState(sessionId);
         const now = Date.now();
@@ -156,9 +175,11 @@ export function createContinuationEnforcer(opts) {
             if (type === "session.created") {
                 // When a session is created and there is an active run, nudge.
                 scheduleIdleInjection(sessionId);
+                schedulePeriodicInjection(sessionId);
             }
             if (type === "session.deleted") {
                 clearIdleTimer(sessionId);
+                clearPeriodicTimer(sessionId);
                 sessions.delete(sessionId);
             }
         },

package/dist/tools/stage.js

@@ -137,9 +137,8 @@ Expected format:
 
 Ensure JSON has required fields (stage_key, status) and valid syntax.`;
             }
-            if (parsed.astro_json.stage_key !== sk) {
-                return `❌ Stage Key Mismatch: ASTRO JSON has stage_key="${parsed.astro_json.stage_key}" but expected "${sk}". Update the JSON to match the current stage.`;
-            }
+            // Override stage_key to match the expected stage (agents sometimes get this wrong)
+            parsed.astro_json.stage_key = sk;
             // Context validation (warnings, not errors)
             if (parsed.astro_json.run_id && parsed.astro_json.run_id !== rid) {
                 console.warn(`[Astrocode] ⚠️ Run ID mismatch in baton: expected "${rid}", got "${parsed.astro_json.run_id}". Proceeding anyway.`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "astrocode-workflow",
-  "version": "0.1.39",
+  "version": "0.1.41",
   "type": "module",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/hooks/continuation-enforcer.ts

@@ -13,6 +13,7 @@ type SessionState = {
   repeats: number;
   autoSteps: number;
   idleTimer: NodeJS.Timeout | null;
+  periodicTimer: NodeJS.Timeout | null;
 };
 
 type ToolExecuteAfterInput = {
@@ -48,7 +49,7 @@ export function createContinuationEnforcer(opts: {
   function getState(sessionId: string): SessionState {
     const cur = sessions.get(sessionId);
     if (cur) return cur;
-    const state: SessionState = { lastHash: null, lastAtMs: 0, repeats: 0, autoSteps: 0, idleTimer: null };
+    const state: SessionState = { lastHash: null, lastAtMs: 0, repeats: 0, autoSteps: 0, idleTimer: null, periodicTimer: null };
     sessions.set(sessionId, state);
     return state;
   }
@@ -61,6 +62,14 @@ export function createContinuationEnforcer(opts: {
     }
   }
 
+  function clearPeriodicTimer(sessionId: string) {
+    const s = getState(sessionId);
+    if (s.periodicTimer) {
+      clearInterval(s.periodicTimer);
+      s.periodicTimer = null;
+    }
+  }
+
   function scheduleIdleInjection(sessionId: string) {
     clearIdleTimer(sessionId);
     if (!config.continuation.enabled) return;
@@ -75,6 +84,20 @@ export function createContinuationEnforcer(opts: {
     }, delay);
   }
 
+  function schedulePeriodicInjection(sessionId: string) {
+    clearPeriodicTimer(sessionId);
+    if (!config.continuation.enabled) return;
+
+    // Inject every 5 minutes (300,000 ms)
+    const interval = 5 * 60 * 1000;
+
+    const s = getState(sessionId);
+    s.periodicTimer = setInterval(() => {
+      // Fire and forget
+      void maybeInjectContinue(sessionId, "periodic_timer");
+    }, interval);
+  }
+
   function shouldDedupe(sessionId: string, directive: BuiltDirective): boolean {
     const s = getState(sessionId);
     const now = Date.now();
@@ -206,10 +229,12 @@ export function createContinuationEnforcer(opts: {
       if (type === "session.created") {
         // When a session is created and there is an active run, nudge.
         scheduleIdleInjection(sessionId);
+        schedulePeriodicInjection(sessionId);
       }
 
       if (type === "session.deleted") {
         clearIdleTimer(sessionId);
+        clearPeriodicTimer(sessionId);
         sessions.delete(sessionId);
       }
     },

package/src/tools/stage.ts

@@ -174,9 +174,8 @@ Expected format:
 Ensure JSON has required fields (stage_key, status) and valid syntax.`;
        }
 
-        if (parsed.astro_json.stage_key !== sk) {
-          return `❌ Stage Key Mismatch: ASTRO JSON has stage_key="${parsed.astro_json.stage_key}" but expected "${sk}". Update the JSON to match the current stage.`;
-        }
+        // Override stage_key to match the expected stage (agents sometimes get this wrong)
+        parsed.astro_json.stage_key = sk;
 
         // Context validation (warnings, not errors)
         if (parsed.astro_json.run_id && parsed.astro_json.run_id !== rid) {