@episoda/cli 0.2.182 → 0.2.184

package/dist/daemon/daemon-process.js

@@ -3046,7 +3046,7 @@ var require_package = __commonJS({
   "package.json"(exports2, module2) {
     module2.exports = {
       name: "@episoda/cli",
-      version: "0.2.182",
+      version: "0.2.184",
       description: "CLI tool for Episoda local development workflow orchestration",
       main: "dist/index.js",
       types: "dist/index.d.ts",
@@ -15311,10 +15311,7 @@ var Daemon = class _Daemon {
     });
   }
   static {
-    this.AGENT_HEARTBEAT_INTERVAL_MS = 15e3;
-  }
-  static {
-    this.AGENT_HEARTBEAT_REQUEST_TIMEOUT_MS = 12e3;
+    this.AGENT_HEARTBEAT_INTERVAL_MS = 6e4;
   }
   static {
     this.HEALTH_PORT = 9999;
@@ -15341,59 +15338,58 @@ var Daemon = class _Daemon {
     loop.stop();
     this.agentHeartbeatLoops.delete(sessionId);
   }
-  async handleServerRequestedAgentStop(sessionId) {
+  async shouldContinueAgentSession(sessionId) {
+    try {
+      const config = await (0, import_core22.loadConfig)();
+      const apiUrl = config?.api_url || "https://episoda.dev";
+      const response = await fetchWithAuth(`${apiUrl}/api/agent-sessions/${encodeURIComponent(sessionId)}/heartbeat`, {
+        method: "POST"
+      });
+      if (!response.ok) {
+        console.warn(`[Daemon] EP1429: Heartbeat API returned ${response.status} for session ${sessionId}; continuing session`);
+        return true;
+      }
+      const payload = await response.json().catch(() => null);
+      return payload?.continue !== false;
+    } catch (error) {
+      console.warn(`[Daemon] EP1429: Heartbeat API check failed for session ${sessionId}; continuing session`, error);
+      return true;
+    }
+  }
+  async stopAgentSessionFromServerRequest(sessionId) {
     try {
       const agentManager = getAgentControlPlane();
+      await agentManager.initialize();
       await agentManager.stopSession(sessionId);
-      console.log(`[Daemon] EP1430: Stopped session ${sessionId} due to server continue=false`);
+      this.agentEventSeq.delete(sessionId);
+      console.log(`[Daemon] EP1429: Stopped agent session ${sessionId} on server heartbeat continue=false`);
     } catch (error) {
-      console.warn(
-        `[Daemon] EP1430: Failed to stop session ${sessionId} after server continue=false:`,
-        error
-      );
+      console.warn(`[Daemon] EP1429: Failed to stop session ${sessionId} after continue=false`, error);
     }
   }
-  startAgentHeartbeatLoop(sessionId) {
+  startAgentHeartbeatLoop(sessionId, client) {
     if (this.agentHeartbeatLoops.has(sessionId)) return;
     let stopped = false;
-    let inFlight = false;
     let timer = null;
     const runHeartbeat = async () => {
-      if (stopped || inFlight) return;
-      inFlight = true;
-      const abortController = new AbortController();
-      const requestTimeout = setTimeout(() => {
-        abortController.abort();
-      }, _Daemon.AGENT_HEARTBEAT_REQUEST_TIMEOUT_MS);
+      if (stopped) return;
       try {
-        const config = await (0, import_core22.loadConfig)();
-        if (!config?.access_token) return;
-        const apiUrl = config.api_url || "https://episoda.dev";
-        const response = await fetchWithAuth(
-          `${apiUrl}/api/agent-sessions/${sessionId}/heartbeat`,
-          {
-            method: "POST",
-            signal: abortController.signal
-          }
-        );
-        if (!response.ok && response.status !== 409) {
-          return;
-        }
-        const payload = await response.json().catch(() => null);
-        const shouldContinue = response.status === 409 ? false : Boolean(payload?.data?.continue ?? payload?.continue ?? true);
+        await client.send({
+          type: "agent_heartbeat",
+          sessionId,
+          pid: process.pid,
+          status: "running"
+        });
+        const shouldContinue = await this.shouldContinueAgentSession(sessionId);
         if (!shouldContinue) {
-          await this.handleServerRequestedAgentStop(sessionId);
           this.stopAgentHeartbeatLoop(sessionId);
+          await this.stopAgentSessionFromServerRequest(sessionId);
         }
       } catch (error) {
-        if (error instanceof Error && error.name === "AbortError") {
-          console.warn(
-            `[Daemon] EP1430: Heartbeat request timed out for session ${sessionId} after ${_Daemon.AGENT_HEARTBEAT_REQUEST_TIMEOUT_MS}ms`
-          );
-        }
-      } finally {
-        clearTimeout(requestTimeout);
-        inFlight = false;
+        console.warn(
+          `[Daemon] EP1429: Failed to send agent_heartbeat for session ${sessionId}:`,
+          error
+        );
       }
     };
     timer = setInterval(() => {
@@ -15848,7 +15844,7 @@ var Daemon = class _Daemon {
           await agentManager.initialize();
           let result;
           if (cmd.action === "start") {
-            this.startAgentHeartbeatLoop(cmd.sessionId);
+            this.startAgentHeartbeatLoop(cmd.sessionId, client);
             const callbacks = createStreamingCallbacks(cmd.sessionId, message.id);
             const agentWorkingDir = await resolveAgentWorkingDirectory({
               command: cmd,
@@ -15889,7 +15885,7 @@ var Daemon = class _Daemon {
               error: startResult.error
             };
           } else if (cmd.action === "message") {
-            this.startAgentHeartbeatLoop(cmd.sessionId);
+            this.startAgentHeartbeatLoop(cmd.sessionId, client);
             const callbacks = createStreamingCallbacks(cmd.sessionId, message.id);
             const sendResult = await agentManager.sendMessage({
               sessionId: cmd.sessionId,