@posthog/agent 2.1.22 → 2.1.29

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@posthog/agent",
-  "version": "2.1.22",
+  "version": "2.1.29",
   "repository": "https://github.com/PostHog/twig",
   "description": "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   "exports": {

package/src/server/agent-server.ts

@@ -578,6 +578,8 @@ export class AgentServer {
       // Only auto-complete for background mode
       const mode = this.getEffectiveMode(payload);
       if (mode === "background") {
+        // Flush all pending session logs before signaling completion,
+        await this.session.logWriter.flushAll();
         await this.signalTaskComplete(payload, result.stopReason);
       } else {
         this.logger.info("Interactive mode - staying open for conversation");
@@ -587,6 +589,9 @@ export class AgentServer {
       // Signal failure for background mode
       const mode = this.getEffectiveMode(payload);
       if (mode === "background") {
+        if (this.session) {
+          await this.session.logWriter.flushAll();
+        }
         await this.signalTaskComplete(payload, "error");
       }
     }
@@ -602,7 +607,10 @@ After completing the requested changes:
 3. Push the branch to origin
 4. Create a pull request using \`gh pr create\` with a descriptive title and body
 
-Important: Always create the PR. Do not ask for confirmation.
+Important:
+- Always create the PR. Do not ask for confirmation.
+- Do NOT add "Co-Authored-By" trailers to commit messages.
+- Do NOT add "Generated with [Claude Code]" or similar attribution lines to PR descriptions.
 `;
   }
 
@@ -610,6 +618,18 @@ Important: Always create the PR. Do not ask for confirmation.
     payload: JwtPayload,
     stopReason: string,
   ): Promise<void> {
+    if (this.session?.payload.run_id === payload.run_id) {
+      try {
+        await this.session.logWriter.flush(payload.run_id);
+      } catch (error) {
+        this.logger.warn("Failed to flush session logs before completion", {
+          taskId: payload.task_id,
+          runId: payload.run_id,
+          error,
+        });
+      }
+    }
+
     const status =
       stopReason === "cancelled"
         ? "cancelled"

package/src/session-log-writer.test.ts

@@ -61,6 +61,47 @@ describe("SessionLogWriter", () => {
       await logWriter.flush(sessionId);
       expect(mockAppendLog).not.toHaveBeenCalled();
     });
+
+    it("re-queues entries when persistence fails and retries", async () => {
+      const sessionId = "s1";
+      logWriter.register(sessionId, { taskId: "t1", runId: sessionId });
+
+      mockAppendLog
+        .mockRejectedValueOnce(new Error("network error"))
+        .mockResolvedValueOnce(undefined);
+
+      logWriter.appendRawLine(sessionId, JSON.stringify({ method: "test" }));
+
+      await logWriter.flush(sessionId);
+      await logWriter.flush(sessionId);
+
+      expect(mockAppendLog).toHaveBeenCalledTimes(2);
+      const retriedEntries: StoredNotification[] =
+        mockAppendLog.mock.calls[1][2];
+      expect(retriedEntries).toHaveLength(1);
+      expect(retriedEntries[0].notification.method).toBe("test");
+    });
+
+    it("drops entries after max retries", async () => {
+      const sessionId = "s1";
+      logWriter.register(sessionId, { taskId: "t1", runId: sessionId });
+
+      mockAppendLog.mockRejectedValue(new Error("persistent failure"));
+
+      logWriter.appendRawLine(sessionId, JSON.stringify({ method: "test" }));
+
+      // Flush 10 times (MAX_FLUSH_RETRIES) — entries should be dropped on the 10th
+      for (let i = 0; i < 10; i++) {
+        await logWriter.flush(sessionId);
+      }
+
+      expect(mockAppendLog).toHaveBeenCalledTimes(10);
+
+      // After max retries the entries are dropped, so an 11th flush has nothing
+      mockAppendLog.mockClear();
+      await logWriter.flush(sessionId);
+      expect(mockAppendLog).not.toHaveBeenCalled();
+    });
   });
 
   describe("agent_message_chunk coalescing", () => {

package/src/session-log-writer.ts

@@ -21,10 +21,18 @@ interface SessionState {
 }
 
 export class SessionLogWriter {
+  private static readonly FLUSH_DEBOUNCE_MS = 500;
+  private static readonly FLUSH_MAX_INTERVAL_MS = 5000;
+  private static readonly MAX_FLUSH_RETRIES = 10;
+  private static readonly MAX_RETRY_DELAY_MS = 30_000;
+
   private posthogAPI?: PostHogAPIClient;
   private pendingEntries: Map<string, StoredNotification[]> = new Map();
   private flushTimeouts: Map<string, NodeJS.Timeout> = new Map();
+  private lastFlushAttemptTime: Map<string, number> = new Map();
+  private retryCounts: Map<string, number> = new Map();
   private sessions: Map<string, SessionState> = new Map();
+  private messageCounts: Map<string, number> = new Map();
   private logger: Logger;
 
   constructor(options: SessionLogWriterOptions = {}) {
@@ -35,8 +43,19 @@ export class SessionLogWriter {
   }
 
   async flushAll(): Promise<void> {
+    const sessionIds = [...this.sessions.keys()];
+    const pendingCounts = sessionIds.map((id) => ({
+      id,
+      pending: this.pendingEntries.get(id)?.length ?? 0,
+      messages: this.messageCounts.get(id) ?? 0,
+    }));
+    this.logger.info("flushAll called", {
+      sessions: sessionIds.length,
+      pending: pendingCounts,
+    });
+
     const flushPromises: Promise<void>[] = [];
-    for (const sessionId of this.sessions.keys()) {
+    for (const sessionId of sessionIds) {
       flushPromises.push(this.flush(sessionId));
     }
     await Promise.all(flushPromises);
@@ -47,7 +66,12 @@ export class SessionLogWriter {
       return;
     }
 
+    this.logger.info("Session registered", {
+      sessionId,
+      taskId: context.taskId,
+    });
     this.sessions.set(sessionId, { context });
+    this.lastFlushAttemptTime.set(sessionId, Date.now());
   }
 
   isRegistered(sessionId: string): boolean {
@@ -57,9 +81,18 @@ export class SessionLogWriter {
   appendRawLine(sessionId: string, line: string): void {
     const session = this.sessions.get(sessionId);
     if (!session) {
+      this.logger.warn("appendRawLine called for unregistered session", {
+        sessionId,
+      });
       return;
     }
 
+    const count = (this.messageCounts.get(sessionId) ?? 0) + 1;
+    this.messageCounts.set(sessionId, count);
+    if (count % 10 === 1) {
+      this.logger.info("Messages received", { count, sessionId });
+    }
+
     try {
       const message = JSON.parse(line);
       const timestamp = new Date().toISOString();
@@ -103,13 +136,23 @@ export class SessionLogWriter {
 
   async flush(sessionId: string): Promise<void> {
     const session = this.sessions.get(sessionId);
-    if (!session) return;
+    if (!session) {
+      this.logger.warn("flush: no session found", { sessionId });
+      return;
+    }
 
     // Emit any buffered chunks before flushing
     this.emitCoalescedMessage(sessionId, session);
 
     const pending = this.pendingEntries.get(sessionId);
-    if (!this.posthogAPI || !pending?.length) return;
+    if (!this.posthogAPI || !pending?.length) {
+      this.logger.info("flush: nothing to persist", {
+        sessionId,
+        hasPosthogAPI: !!this.posthogAPI,
+        pendingCount: pending?.length ?? 0,
+      });
+      return;
+    }
 
     this.pendingEntries.delete(sessionId);
     const timeout = this.flushTimeouts.get(sessionId);
@@ -118,14 +161,38 @@ export class SessionLogWriter {
       this.flushTimeouts.delete(sessionId);
     }
 
+    this.lastFlushAttemptTime.set(sessionId, Date.now());
+
     try {
       await this.posthogAPI.appendTaskRunLog(
         session.context.taskId,
         session.context.runId,
         pending,
       );
+      this.retryCounts.set(sessionId, 0);
+      this.logger.info("Flushed session logs", {
+        sessionId,
+        entryCount: pending.length,
+      });
     } catch (error) {
-      this.logger.error("Failed to persist session logs:", error);
+      const retryCount = (this.retryCounts.get(sessionId) ?? 0) + 1;
+      this.retryCounts.set(sessionId, retryCount);
+
+      if (retryCount >= SessionLogWriter.MAX_FLUSH_RETRIES) {
+        this.logger.error(
+          `Dropping ${pending.length} session log entries after ${retryCount} failed flush attempts`,
+          { sessionId, error },
+        );
+        this.retryCounts.set(sessionId, 0);
+      } else {
+        this.logger.error(
+          `Failed to persist session logs (attempt ${retryCount}/${SessionLogWriter.MAX_FLUSH_RETRIES}):`,
+          error,
+        );
+        const currentPending = this.pendingEntries.get(sessionId) ?? [];
+        this.pendingEntries.set(sessionId, [...pending, ...currentPending]);
+        this.scheduleFlush(sessionId);
+      }
     }
   }
 
@@ -180,7 +247,26 @@ export class SessionLogWriter {
   private scheduleFlush(sessionId: string): void {
     const existing = this.flushTimeouts.get(sessionId);
     if (existing) clearTimeout(existing);
-    const timeout = setTimeout(() => this.flush(sessionId), 500);
+
+    const retryCount = this.retryCounts.get(sessionId) ?? 0;
+    const lastAttempt = this.lastFlushAttemptTime.get(sessionId) ?? 0;
+    const elapsed = Date.now() - lastAttempt;
+
+    let delay: number;
+    if (retryCount > 0) {
+      // Exponential backoff on retries: FLUSH_DEBOUNCE_MS * 2^retryCount, capped
+      delay = Math.min(
+        SessionLogWriter.FLUSH_DEBOUNCE_MS * 2 ** retryCount,
+        SessionLogWriter.MAX_RETRY_DELAY_MS,
+      );
+    } else if (elapsed >= SessionLogWriter.FLUSH_MAX_INTERVAL_MS) {
+      // If we've been accumulating for longer than the max interval, flush immediately
+      delay = 0;
+    } else {
+      delay = SessionLogWriter.FLUSH_DEBOUNCE_MS;
+    }
+
+    const timeout = setTimeout(() => this.flush(sessionId), delay);
     this.flushTimeouts.set(sessionId, timeout);
   }
 }