@posthog/agent 2.1.22 → 2.1.29

package/dist/{agent-LrKyX9KN.d.ts → agent-DcBmoTR4.d.ts}

@@ -46,10 +46,17 @@ interface SessionLogWriterOptions {
     logger?: Logger;
 }
 declare class SessionLogWriter {
+    private static readonly FLUSH_DEBOUNCE_MS;
+    private static readonly FLUSH_MAX_INTERVAL_MS;
+    private static readonly MAX_FLUSH_RETRIES;
+    private static readonly MAX_RETRY_DELAY_MS;
     private posthogAPI?;
     private pendingEntries;
     private flushTimeouts;
+    private lastFlushAttemptTime;
+    private retryCounts;
     private sessions;
+    private messageCounts;
     private logger;
     constructor(options?: SessionLogWriterOptions);
     flushAll(): Promise<void>;

package/dist/agent.d.ts

@@ -1,4 +1,4 @@
-export { b as Agent } from './agent-LrKyX9KN.js';
+export { b as Agent } from './agent-DcBmoTR4.js';
 import './types.js';
 import '@agentclientprotocol/sdk';
 import './logger-DDBiMOOD.js';

package/dist/agent.js

@@ -276,7 +276,7 @@ import { v7 as uuidv7 } from "uuid";
 // package.json
 var package_default = {
   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: {
@@ -3386,19 +3386,36 @@ var PostHogAPIClient = class {
 };
 
 // src/session-log-writer.ts
-var SessionLogWriter = class {
+var SessionLogWriter = class _SessionLogWriter {
+  static FLUSH_DEBOUNCE_MS = 500;
+  static FLUSH_MAX_INTERVAL_MS = 5e3;
+  static MAX_FLUSH_RETRIES = 10;
+  static MAX_RETRY_DELAY_MS = 3e4;
   posthogAPI;
   pendingEntries = /* @__PURE__ */ new Map();
   flushTimeouts = /* @__PURE__ */ new Map();
+  lastFlushAttemptTime = /* @__PURE__ */ new Map();
+  retryCounts = /* @__PURE__ */ new Map();
   sessions = /* @__PURE__ */ new Map();
+  messageCounts = /* @__PURE__ */ new Map();
   logger;
   constructor(options = {}) {
     this.posthogAPI = options.posthogAPI;
     this.logger = options.logger ?? new Logger({ debug: false, prefix: "[SessionLogWriter]" });
   }
   async flushAll() {
+    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 = [];
-    for (const sessionId of this.sessions.keys()) {
+    for (const sessionId of sessionIds) {
       flushPromises.push(this.flush(sessionId));
     }
     await Promise.all(flushPromises);
@@ -3407,7 +3424,12 @@ var SessionLogWriter = class {
     if (this.sessions.has(sessionId)) {
       return;
     }
+    this.logger.info("Session registered", {
+      sessionId,
+      taskId: context.taskId
+    });
     this.sessions.set(sessionId, { context });
+    this.lastFlushAttemptTime.set(sessionId, Date.now());
   }
   isRegistered(sessionId) {
     return this.sessions.has(sessionId);
@@ -3415,8 +3437,16 @@ var SessionLogWriter = class {
   appendRawLine(sessionId, line) {
     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 = (/* @__PURE__ */ new Date()).toISOString();
@@ -3452,24 +3482,56 @@ var SessionLogWriter = class {
   }
   async flush(sessionId) {
     const session = this.sessions.get(sessionId);
-    if (!session) return;
+    if (!session) {
+      this.logger.warn("flush: no session found", { sessionId });
+      return;
+    }
     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);
     if (timeout) {
       clearTimeout(timeout);
       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);
+      }
     }
   }
   isAgentMessageChunk(message) {
@@ -3515,7 +3577,21 @@ var SessionLogWriter = class {
   scheduleFlush(sessionId) {
     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;
+    if (retryCount > 0) {
+      delay = Math.min(
+        _SessionLogWriter.FLUSH_DEBOUNCE_MS * 2 ** retryCount,
+        _SessionLogWriter.MAX_RETRY_DELAY_MS
+      );
+    } else if (elapsed >= _SessionLogWriter.FLUSH_MAX_INTERVAL_MS) {
+      delay = 0;
+    } else {
+      delay = _SessionLogWriter.FLUSH_DEBOUNCE_MS;
+    }
+    const timeout = setTimeout(() => this.flush(sessionId), delay);
     this.flushTimeouts.set(sessionId, timeout);
   }
 };