agent-state-machine 2.1.3 → 2.1.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-state-machine",
-  "version": "2.1.3",
+  "version": "2.1.5",
   "type": "module",
   "description": "A workflow orchestrator for running agents and scripts in sequence with state management",
   "main": "lib/index.js",

package/templates/project-builder/config.js

@@ -8,6 +8,5 @@ export const config = {
     gemini: process.env.GEMINI_API_KEY,
     anthropic: process.env.ANTHROPIC_API_KEY,
     openai: process.env.OPENAI_API_KEY,
-  },
-  remotePath: "TczrLmUecnqZPpPhBTrvU374CGlfzDfINrr0eN0nMgQ",
+  }
 };

package/vercel-server/api/events/[token].js

@@ -57,13 +57,13 @@ export default async function handler(req, res) {
 
     // Track current position for polling new events
     let lastEventIndex = await getEventsLength(token);
+    let pollCount = 0;
 
     const pollInterval = setInterval(async () => {
       try {
-        // Refresh session TTL
-        await refreshSession(token);
+        pollCount++;
 
-        // Check for new events
+        // Check for new events (most important, do this every poll)
         const newLength = await getEventsLength(token);
 
         if (newLength > lastEventIndex) {
@@ -82,18 +82,23 @@ export default async function handler(req, res) {
           lastEventIndex = newLength;
         }
 
-        // Check CLI status
-        const updatedSession = await getSession(token);
-        if (updatedSession && updatedSession.cliConnected !== session.cliConnected) {
-          session.cliConnected = updatedSession.cliConnected;
-          res.write(`data: ${JSON.stringify({
-            type: updatedSession.cliConnected ? 'cli_reconnected' : 'cli_disconnected',
-          })}\n\n`);
+        // Only check CLI status and refresh session every 5th poll (~15 seconds)
+        // This reduces Redis calls significantly
+        if (pollCount % 5 === 0) {
+          await refreshSession(token);
+
+          const updatedSession = await getSession(token);
+          if (updatedSession && updatedSession.cliConnected !== session.cliConnected) {
+            session.cliConnected = updatedSession.cliConnected;
+            res.write(`data: ${JSON.stringify({
+              type: updatedSession.cliConnected ? 'cli_reconnected' : 'cli_disconnected',
+            })}\n\n`);
+          }
         }
       } catch (err) {
         console.error('Error polling events:', err);
       }
-    }, 1000); // Poll every 1 second for faster updates
+    }, 3000); // Poll every 3 seconds (was 1 second) - 3x reduction
 
     // Clean up on client disconnect
     req.on('close', () => {

package/vercel-server/api/ws/cli.js

@@ -44,14 +44,16 @@ export default async function handler(req, res) {
  */
 async function handlePost(req, res) {
   const body = typeof req.body === 'string' ? JSON.parse(req.body) : req.body;
-  const { type, sessionToken } = body;
+  const { sessionToken } = body;
+  // Support both _action (new) and type (legacy) for message routing
+  const action = body._action || body.type;
 
   if (!sessionToken) {
     return res.status(400).json({ error: 'Missing sessionToken' });
   }
 
   try {
-    switch (type) {
+    switch (action) {
       case 'session_init': {
         const { workflowName, history } = body;
 
@@ -89,9 +91,9 @@ async function handlePost(req, res) {
           ...eventData,
         };
 
-        // Remove sessionToken and type from event data
+        // Remove routing fields, preserve type (interaction type like 'choice')
         delete historyEvent.sessionToken;
-        delete historyEvent.type;
+        delete historyEvent._action;
 
         // Add to events list (single source of truth)
         await addEvent(sessionToken, historyEvent);
@@ -125,7 +127,7 @@ async function handlePost(req, res) {
       }
 
       default:
-        return res.status(400).json({ error: `Unknown message type: ${type}` });
+        return res.status(400).json({ error: `Unknown action: ${action}` });
     }
   } catch (err) {
     console.error('Error handling CLI message:', err);
@@ -135,6 +137,7 @@ async function handlePost(req, res) {
 
 /**
  * Handle GET requests - long-poll for interaction responses
+ * Uses efficient polling with 5-second intervals (Upstash doesn't support BLPOP)
  */
 async function handleGet(req, res) {
   const { token, timeout = '30000' } = req.query;
@@ -148,16 +151,15 @@ async function handleGet(req, res) {
     return res.status(404).json({ error: 'Session not found' });
   }
 
-  const timeoutMs = Math.min(parseInt(timeout, 10), 55000); // Max 55s for Vercel
+  // Max 50s for Vercel (leave buffer for response)
+  const timeoutMs = Math.min(parseInt(timeout, 10), 50000);
   const channel = KEYS.interactions(token);
-
-  // Check for pending interactions using a list
   const pendingKey = `${channel}:pending`;
 
   try {
-    // Try to get a pending interaction
     const startTime = Date.now();
 
+    // Poll every 5 seconds (10 calls per 50s timeout vs 50 calls before)
     while (Date.now() - startTime < timeoutMs) {
       const pending = await redis.lpop(pendingKey);
 
@@ -169,8 +171,8 @@ async function handleGet(req, res) {
         });
       }
 
-      // Wait before checking again
-      await new Promise((resolve) => setTimeout(resolve, 1000));
+      // Wait 5 seconds before checking again (was 1 second)
+      await new Promise((resolve) => setTimeout(resolve, 5000));
     }
 
     // Timeout - no interaction received