@posthog/agent 2.3.508 → 2.3.510

package/dist/agent.js

@@ -4030,7 +4030,7 @@ import { v7 as uuidv7 } from "uuid";
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.3.508",
+  version: "2.3.510",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -8681,6 +8681,54 @@ function toolContent() {
   return new ToolContentBuilder();
 }
 
+// src/utils/partial-json.ts
+function tryParsePartialJson(s) {
+  const trimmed2 = s.trim();
+  if (!trimmed2) return null;
+  try {
+    return JSON.parse(trimmed2);
+  } catch {
+  }
+  const closers = [];
+  let inString = false;
+  let escaped = false;
+  for (let i2 = 0; i2 < trimmed2.length; i2++) {
+    const ch = trimmed2[i2];
+    if (inString) {
+      if (escaped) {
+        escaped = false;
+      } else if (ch === "\\") {
+        escaped = true;
+      } else if (ch === '"') {
+        inString = false;
+      }
+      continue;
+    }
+    if (ch === '"') inString = true;
+    else if (ch === "{") closers.push("}");
+    else if (ch === "[") closers.push("]");
+    else if (ch === "}" || ch === "]") closers.pop();
+  }
+  const closeBrackets = (str) => {
+    let out2 = str;
+    for (let i2 = closers.length - 1; i2 >= 0; i2--) out2 += closers[i2];
+    return out2;
+  };
+  const candidates = [];
+  const closedString = inString ? `${trimmed2}"` : trimmed2;
+  candidates.push(closeBrackets(closedString));
+  let stripped = closedString.replace(/[,:]\s*$/, "");
+  stripped = stripped.replace(/,?\s*"[^"]*"\s*:?\s*$/, "");
+  candidates.push(closeBrackets(stripped));
+  for (const candidate of candidates) {
+    try {
+      return JSON.parse(candidate);
+    } catch {
+    }
+  }
+  return null;
+}
+
 // src/adapters/claude/permissions/posthog-exec-gate.ts
 var POSTHOG_EXEC_TOOL_RE = /^mcp__posthog(?:_[^_]+)*__exec$/;
 var POSTHOG_CALL_COMMAND_RE = /^\s*call\s+(?:--json\s+)?([a-zA-Z0-9_-]+)/;
@@ -9847,12 +9895,19 @@ function toAcpNotifications(content, role, sessionId, toolUseCache, fileContentC
   }
   return output;
 }
-function streamEventToAcpNotifications(message, sessionId, toolUseCache, fileContentCache, client, logger, parentToolCallId, registerHooks, supportsTerminalOutput, cwd, enrichedReadCache) {
+function streamEventToAcpNotifications(message, sessionId, toolUseCache, toolUseStreamCache, fileContentCache, client, logger, parentToolCallId, registerHooks, supportsTerminalOutput, cwd, enrichedReadCache) {
   const event = message.event;
   switch (event.type) {
-    case "content_block_start":
+    case "content_block_start": {
+      const block = event.content_block;
+      if (block.type === "tool_use" || block.type === "mcp_tool_use") {
+        toolUseStreamCache.set(event.index, {
+          toolUseId: block.id,
+          partialJson: ""
+        });
+      }
       return toAcpNotifications(
-        [event.content_block],
+        [block],
         "assistant",
         sessionId,
         toolUseCache,
@@ -9866,7 +9921,16 @@ function streamEventToAcpNotifications(message, sessionId, toolUseCache, fileCon
         void 0,
         enrichedReadCache
       );
-    case "content_block_delta":
+    }
+    case "content_block_delta": {
+      if (event.delta.type === "input_json_delta") {
+        return inputJsonDeltaToAcpNotifications(
+          event.index,
+          event.delta.partial_json,
+          sessionId,
+          toolUseStreamCache
+        );
+      }
       return toAcpNotifications(
         [event.delta],
         "assistant",
@@ -9882,16 +9946,36 @@ function streamEventToAcpNotifications(message, sessionId, toolUseCache, fileCon
         void 0,
         enrichedReadCache
       );
+    }
+    case "content_block_stop":
+      toolUseStreamCache.delete(event.index);
+      return [];
     case "message_start":
     case "message_delta":
     case "message_stop":
-    case "content_block_stop":
       return [];
     default:
       unreachable(event, logger);
       return [];
   }
 }
+function inputJsonDeltaToAcpNotifications(index, partialJson, sessionId, toolUseStreamCache) {
+  const entry = toolUseStreamCache.get(index);
+  if (!entry) return [];
+  entry.partialJson += partialJson;
+  const parsed = tryParsePartialJson(entry.partialJson);
+  if (!parsed || typeof parsed !== "object") return [];
+  return [
+    {
+      sessionId,
+      update: {
+        sessionUpdate: "tool_call_update",
+        toolCallId: entry.toolUseId,
+        rawInput: parsed
+      }
+    }
+  ];
+}
 async function handleSystemMessage(message, context) {
   const { session, sessionId, client, logger } = context;
   switch (message.subtype) {
@@ -10035,12 +10119,20 @@ function extractUsageFromResult(message) {
   };
 }
 async function handleStreamEvent(message, context) {
-  const { sessionId, client, toolUseCache, fileContentCache, logger } = context;
+  const {
+    sessionId,
+    client,
+    toolUseCache,
+    toolUseStreamCache,
+    fileContentCache,
+    logger
+  } = context;
   const parentToolCallId = message.parent_tool_use_id ?? void 0;
   for (const notification of streamEventToAcpNotifications(
     message,
     sessionId,
     toolUseCache,
+    toolUseStreamCache,
     fileContentCache,
     client,
     logger,
@@ -16594,6 +16686,7 @@ function shouldEmitRawMessage(config, message) {
 var ClaudeAcpAgent = class extends BaseAcpAgent {
   adapterName = "claude";
   toolUseCache;
+  toolUseStreamCache;
   backgroundTerminals = {};
   clientCapabilities;
   options;
@@ -16603,6 +16696,7 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
     super(client);
     this.options = options;
     this.toolUseCache = {};
+    this.toolUseStreamCache = /* @__PURE__ */ new Map();
     this.logger = new Logger({ debug: true, prefix: "[ClaudeAcpAgent]" });
     this.enrichment = createEnrichment(options?.posthogApiConfig, this.logger);
   }
@@ -16797,6 +16891,7 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
       sessionId: params.sessionId,
       client: this.client,
       toolUseCache: this.toolUseCache,
+      toolUseStreamCache: this.toolUseStreamCache,
       fileContentCache: this.fileContentCache,
       enrichedReadCache: this.enrichedReadCache,
       logger: this.logger,
@@ -17049,6 +17144,7 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
       }
       throw error;
     } finally {
+      this.toolUseStreamCache.clear();
       if (!handedOff) {
         this.session.promptRunning = false;
         for (const [key, pending] of this.session.pendingMessages) {
@@ -17623,6 +17719,7 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
         sessionId,
         client: this.client,
         toolUseCache: this.toolUseCache,
+        toolUseStreamCache: this.toolUseStreamCache,
         fileContentCache: this.fileContentCache,
         enrichedReadCache: this.enrichedReadCache,
         logger: this.logger,