@posthog/agent 2.3.513 → 2.3.517

package/dist/server/bin.cjs

@@ -8755,7 +8755,7 @@ var import_zod3 = require("zod");
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.3.513",
+  version: "2.3.517",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -21058,6 +21058,7 @@ var agentErrorClassificationSchema = import_zod3.z.enum([
 var errorWithClassificationSchema = import_zod3.z.object({
   data: import_zod3.z.object({ classification: agentErrorClassificationSchema })
 });
+var SSE_KEEPALIVE_INTERVAL_MS = 25e3;
 var NdJsonTap = class {
   constructor(onMessage) {
     this.onMessage = onMessage;
@@ -21236,41 +21237,67 @@ var AgentServer = class {
           401
         );
       }
+      let keepaliveInterval = null;
+      const clearKeepalive = () => {
+        if (keepaliveInterval) {
+          clearInterval(keepaliveInterval);
+          keepaliveInterval = null;
+        }
+      };
       const stream = new ReadableStream({
         start: async (controller) => {
-          const sseController = {
+          let sseController = null;
+          const encoder = new TextEncoder();
+          const detachCurrentSseController = () => {
+            if (sseController) {
+              this.detachSseController(sseController);
+            }
+          };
+          const enqueueSseFrame = (frame) => {
+            try {
+              controller.enqueue(encoder.encode(frame));
+            } catch {
+              clearKeepalive();
+              detachCurrentSseController();
+            }
+          };
+          sseController = {
             send: (data) => {
-              try {
-                controller.enqueue(
-                  new TextEncoder().encode(`data: ${JSON.stringify(data)}
+              enqueueSseFrame(`data: ${JSON.stringify(data)}
 
-`)
-                );
-              } catch {
-                this.detachSseController(sseController);
-              }
+`);
             },
             close: () => {
               try {
+                clearKeepalive();
                 controller.close();
               } catch {
-                this.detachSseController(sseController);
+                detachCurrentSseController();
               }
             }
           };
-          if (!this.session || this.session.payload.run_id !== payload.run_id) {
-            await this.initializeSession(payload, sseController);
-          } else {
-            this.session.sseController = sseController;
-            this.session.hasDesktopConnected = true;
-            this.replayPendingEvents();
+          keepaliveInterval = setInterval(() => {
+            enqueueSseFrame(": keepalive\n\n");
+          }, SSE_KEEPALIVE_INTERVAL_MS);
+          try {
+            if (!this.session || this.session.payload.run_id !== payload.run_id) {
+              await this.initializeSession(payload, sseController);
+            } else {
+              this.session.sseController = sseController;
+              this.session.hasDesktopConnected = true;
+              this.replayPendingEvents();
+            }
+            this.sendSseEvent(sseController, {
+              type: "connected",
+              run_id: payload.run_id
+            });
+          } catch (error) {
+            clearKeepalive();
+            throw error;
           }
-          this.sendSseEvent(sseController, {
-            type: "connected",
-            run_id: payload.run_id
-          });
         },
         cancel: () => {
+          clearKeepalive();
           this.logger.debug("SSE connection closed");
           if (this.session?.sseController) {
             this.session.sseController = null;