@posthog/agent 2.0.2 → 2.0.3

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-export { A as AcpConnection, a as AcpConnectionConfig, b as Agent, c as AgentAdapter, C as CodexProcessOptions, I as InProcessAcpConnection, O as OtelLogConfig, d as OtelLogWriter, S as SessionContext, e as SessionLogWriter, f as SessionLogWriterOptions, g as createAcpConnection } from './agent-DBQY1BfC.js';
+export { A as AcpConnection, a as AcpConnectionConfig, b as Agent, c as AgentAdapter, C as CodexProcessOptions, I as InProcessAcpConnection, O as OtelLogConfig, d as OtelLogWriter, S as SessionContext, e as SessionLogWriter, f as SessionLogWriterOptions, g as createAcpConnection } from './agent-kRbaLUfe.js';
 import { McpServerConfig } from '@anthropic-ai/claude-agent-sdk';
 import { L as Logger } from './logger-DDBiMOOD.js';
 export { a as LoggerConfig } from './logger-DDBiMOOD.js';

package/dist/index.js

@@ -1174,7 +1174,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: {
@@ -1220,7 +1220,7 @@ var package_default = {
     }
   },
   bin: {
-    "agent-server": "./dist/server/bin.js"
+    "agent-server": "./dist/server/bin.cjs"
   },
   type: "module",
   keywords: [
@@ -4375,9 +4375,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) {
@@ -4399,6 +4412,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();
     }
@@ -4420,6 +4434,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);
@@ -10160,6 +10217,18 @@ var ResumeSaga = class extends Saga {
             });
             break;
           }
+          case "agent_message": {
+            const content = update.content;
+            if (content) {
+              if (content.type === "text" && currentAssistantContent.length > 0 && currentAssistantContent[currentAssistantContent.length - 1].type === "text") {
+                const lastBlock = currentAssistantContent[currentAssistantContent.length - 1];
+                lastBlock.text += content.text;
+              } else {
+                currentAssistantContent.push(content);
+              }
+            }
+            break;
+          }
           case "agent_message_chunk": {
             const content = update.content;
             if (content) {