@posthog/agent 2.3.110 → 2.3.116

package/dist/agent.js

@@ -371,7 +371,7 @@ import { v7 as uuidv7 } from "uuid";
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.3.110",
+  version: "2.3.116",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -4730,9 +4730,9 @@ var SessionLogWriter = class _SessionLogWriter {
     this.logger = options.logger ?? new Logger({ debug: false, prefix: "[SessionLogWriter]" });
   }
   async flushAll() {
-    const sessionIds = [...this.sessions.keys()];
     const flushPromises = [];
-    for (const sessionId of sessionIds) {
+    for (const [sessionId, session] of this.sessions) {
+      this.emitCoalescedMessage(sessionId, session);
       flushPromises.push(this.flush(sessionId));
     }
     await Promise.all(flushPromises);
@@ -4788,7 +4788,11 @@ var SessionLogWriter = class _SessionLogWriter {
         }
         return;
       }
-      this.emitCoalescedMessage(sessionId, session);
+      if (this.isDirectAgentMessage(message) && session.chunkBuffer) {
+        session.chunkBuffer = void 0;
+      } else {
+        this.emitCoalescedMessage(sessionId, session);
+      }
       const nonChunkAgentText = this.extractAgentMessageText(message);
       if (nonChunkAgentText) {
         session.lastAgentMessage = nonChunkAgentText;
@@ -4814,7 +4818,13 @@ var SessionLogWriter = class _SessionLogWriter {
       });
     }
   }
-  async flush(sessionId) {
+  async flush(sessionId, { coalesce = false } = {}) {
+    if (coalesce) {
+      const session = this.sessions.get(sessionId);
+      if (session) {
+        this.emitCoalescedMessage(sessionId, session);
+      }
+    }
     const prev = this.flushQueues.get(sessionId) ?? Promise.resolve();
     const next = prev.catch(() => {
     }).then(() => this._doFlush(sessionId));
@@ -4832,7 +4842,6 @@ var SessionLogWriter = class _SessionLogWriter {
       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;
@@ -4881,11 +4890,17 @@ var SessionLogWriter = class _SessionLogWriter {
       }
     }
   }
-  isAgentMessageChunk(message) {
-    if (message.method !== "session/update") return false;
+  getSessionUpdateType(message) {
+    if (message.method !== "session/update") return void 0;
     const params = message.params;
     const update = params?.update;
-    return update?.sessionUpdate === "agent_message_chunk";
+    return update?.sessionUpdate;
+  }
+  isDirectAgentMessage(message) {
+    return this.getSessionUpdateType(message) === "agent_message";
+  }
+  isAgentMessageChunk(message) {
+    return this.getSessionUpdateType(message) === "agent_message_chunk";
   }
   extractChunkText(message) {
     const params = message.params;
@@ -4930,6 +4945,15 @@ var SessionLogWriter = class _SessionLogWriter {
   getFullAgentResponse(sessionId) {
     const session = this.sessions.get(sessionId);
     if (!session || session.currentTurnMessages.length === 0) return void 0;
+    if (session.chunkBuffer) {
+      this.logger.warn(
+        "getFullAgentResponse called with non-empty chunk buffer",
+        {
+          sessionId,
+          bufferedLength: session.chunkBuffer.text.length
+        }
+      );
+    }
     return session.currentTurnMessages.join("\n\n");
   }
   resetTurnMessages(sessionId) {
@@ -5148,7 +5172,7 @@ var Agent = class {
   }
   async cleanup() {
     if (this.sessionLogWriter && this.taskRunId) {
-      await this.sessionLogWriter.flush(this.taskRunId);
+      await this.sessionLogWriter.flush(this.taskRunId, { coalesce: true });
     }
     await this.acpConnection?.cleanup();
   }