@posthog/agent 2.0.2 → 2.0.3

package/dist/server/agent-server.js

@@ -1183,7 +1183,7 @@ import { v7 as uuidv7 } from "uuid";
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.0.2",
+  version: "2.0.3",
   repository: "https://github.com/PostHog/twig",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -1229,7 +1229,7 @@ var package_default = {
     }
   },
   bin: {
-    "agent-server": "./dist/server/bin.js"
+    "agent-server": "./dist/server/bin.cjs"
   },
   type: "module",
   keywords: [
@@ -4349,9 +4349,22 @@ var SessionLogWriter = class {
     }
     try {
       const message = JSON.parse(line);
+      const timestamp = (/* @__PURE__ */ new Date()).toISOString();
+      if (this.isAgentMessageChunk(message)) {
+        const text2 = this.extractChunkText(message);
+        if (text2) {
+          if (!session.chunkBuffer) {
+            session.chunkBuffer = { text: text2, firstTimestamp: timestamp };
+          } else {
+            session.chunkBuffer.text += text2;
+          }
+        }
+        return;
+      }
+      this.emitCoalescedMessage(sessionId, session);
       const entry = {
         type: "notification",
-        timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+        timestamp,
         notification: message
       };
       if (session.otelWriter) {
@@ -4373,6 +4386,7 @@ var SessionLogWriter = class {
   async flush(sessionId) {
     const session = this.sessions.get(sessionId);
     if (!session) return;
+    this.emitCoalescedMessage(sessionId, session);
     if (session.otelWriter) {
       await session.otelWriter.flush();
     }
@@ -4394,6 +4408,49 @@ var SessionLogWriter = class {
       this.logger.error("Failed to persist session logs:", error);
     }
   }
+  isAgentMessageChunk(message) {
+    if (message.method !== "session/update") return false;
+    const params = message.params;
+    const update = params?.update;
+    return update?.sessionUpdate === "agent_message_chunk";
+  }
+  extractChunkText(message) {
+    const params = message.params;
+    const update = params?.update;
+    const content = update?.content;
+    if (content?.type === "text" && content.text) {
+      return content.text;
+    }
+    return "";
+  }
+  emitCoalescedMessage(sessionId, session) {
+    if (!session.chunkBuffer) return;
+    const { text: text2, firstTimestamp } = session.chunkBuffer;
+    session.chunkBuffer = void 0;
+    const entry = {
+      type: "notification",
+      timestamp: firstTimestamp,
+      notification: {
+        jsonrpc: "2.0",
+        method: "session/update",
+        params: {
+          update: {
+            sessionUpdate: "agent_message",
+            content: { type: "text", text: text2 }
+          }
+        }
+      }
+    };
+    if (session.otelWriter) {
+      session.otelWriter.emit({ notification: entry });
+    }
+    if (this.posthogAPI) {
+      const pending = this.pendingEntries.get(sessionId) ?? [];
+      pending.push(entry);
+      this.pendingEntries.set(sessionId, pending);
+      this.scheduleFlush(sessionId);
+    }
+  }
   scheduleFlush(sessionId) {
     const existing = this.flushTimeouts.get(sessionId);
     if (existing) clearTimeout(existing);