@posthog/agent 2.1.53 → 2.1.60

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.1.53",
+  version: "2.1.60",
   repository: "https://github.com/PostHog/twig",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -1248,7 +1248,8 @@ var package_default = {
     test: "vitest run",
     "test:watch": "vitest",
     typecheck: "pnpm exec tsc --noEmit",
-    prepublishOnly: "pnpm run build"
+    prepublishOnly: "pnpm run build",
+    clean: "rm -rf dist .turbo"
   },
   engines: {
     node: ">=20.0.0"
@@ -2059,14 +2060,25 @@ ${text2}${text2.endsWith("\n") ? "" : "\n"}${escapedText}`;
 function messageUpdateType(role) {
   return role === "assistant" ? "agent_message_chunk" : "user_message_chunk";
 }
-function toolMeta(toolName, toolResponse) {
-  return toolResponse ? { claudeCode: { toolName, toolResponse } } : { claudeCode: { toolName } };
+function toolMeta(toolName, toolResponse, parentToolCallId) {
+  const meta = { toolName };
+  if (toolResponse !== void 0) meta.toolResponse = toolResponse;
+  if (parentToolCallId) meta.parentToolCallId = parentToolCallId;
+  return { claudeCode: meta };
 }
-function handleTextChunk(chunk, role) {
-  return {
+function handleTextChunk(chunk, role, parentToolCallId) {
+  const update = {
     sessionUpdate: messageUpdateType(role),
     content: text(chunk.text)
   };
+  if (parentToolCallId) {
+    update._meta = toolMeta(
+      "__text__",
+      void 0,
+      parentToolCallId
+    );
+  }
+  return update;
 }
 function handleImageChunk(chunk, role) {
   return {
@@ -2078,11 +2090,19 @@ function handleImageChunk(chunk, role) {
     )
   };
 }
-function handleThinkingChunk(chunk) {
-  return {
+function handleThinkingChunk(chunk, parentToolCallId) {
+  const update = {
     sessionUpdate: "agent_thought_chunk",
     content: text(chunk.thinking)
   };
+  if (parentToolCallId) {
+    update._meta = toolMeta(
+      "__thinking__",
+      void 0,
+      parentToolCallId
+    );
+  }
+  return update;
 }
 function handleToolUseChunk(chunk, ctx) {
   ctx.toolUseCache[chunk.id] = chunk;
@@ -2103,7 +2123,7 @@ function handleToolUseChunk(chunk, ctx) {
         await ctx.client.sessionUpdate({
           sessionId: ctx.sessionId,
           update: {
-            _meta: toolMeta(toolUse.name, toolResponse),
+            _meta: toolMeta(toolUse.name, toolResponse, ctx.parentToolCallId),
             toolCallId: toolUseId,
             sessionUpdate: "tool_call_update"
           }
@@ -2121,7 +2141,7 @@ function handleToolUseChunk(chunk, ctx) {
   } catch {
   }
   return {
-    _meta: toolMeta(chunk.name),
+    _meta: toolMeta(chunk.name, void 0, ctx.parentToolCallId),
     toolCallId: chunk.id,
     sessionUpdate: "tool_call",
     rawInput,
@@ -2141,7 +2161,7 @@ function handleToolResultChunk(chunk, ctx) {
     return null;
   }
   return {
-    _meta: toolMeta(toolUse.name),
+    _meta: toolMeta(toolUse.name, void 0, ctx.parentToolCallId),
     toolCallId: chunk.tool_use_id,
     sessionUpdate: "tool_call_update",
     status: chunk.is_error ? "failed" : "completed",
@@ -2155,12 +2175,12 @@ function processContentChunk(chunk, role, ctx) {
   switch (chunk.type) {
     case "text":
     case "text_delta":
-      return handleTextChunk(chunk, role);
+      return handleTextChunk(chunk, role, ctx.parentToolCallId);
     case "image":
       return handleImageChunk(chunk, role);
     case "thinking":
     case "thinking_delta":
-      return handleThinkingChunk(chunk);
+      return handleThinkingChunk(chunk, ctx.parentToolCallId);
     case "tool_use":
     case "server_tool_use":
     case "mcp_tool_use":
@@ -2190,24 +2210,28 @@ function processContentChunk(chunk, role, ctx) {
       return null;
   }
 }
-function toAcpNotifications(content, role, sessionId, toolUseCache, fileContentCache, client, logger) {
+function toAcpNotifications(content, role, sessionId, toolUseCache, fileContentCache, client, logger, parentToolCallId) {
   if (typeof content === "string") {
-    return [
-      {
-        sessionId,
-        update: {
-          sessionUpdate: messageUpdateType(role),
-          content: text(content)
-        }
-      }
-    ];
+    const update = {
+      sessionUpdate: messageUpdateType(role),
+      content: text(content)
+    };
+    if (parentToolCallId) {
+      update._meta = toolMeta(
+        "__text__",
+        void 0,
+        parentToolCallId
+      );
+    }
+    return [{ sessionId, update }];
   }
   const ctx = {
     sessionId,
     toolUseCache,
     fileContentCache,
     client,
-    logger
+    logger,
+    parentToolCallId
   };
   const output = [];
   for (const chunk of content) {
@@ -2218,7 +2242,7 @@ function toAcpNotifications(content, role, sessionId, toolUseCache, fileContentC
   }
   return output;
 }
-function streamEventToAcpNotifications(message, sessionId, toolUseCache, fileContentCache, client, logger) {
+function streamEventToAcpNotifications(message, sessionId, toolUseCache, fileContentCache, client, logger, parentToolCallId) {
   const event = message.event;
   switch (event.type) {
     case "content_block_start":
@@ -2229,7 +2253,8 @@ function streamEventToAcpNotifications(message, sessionId, toolUseCache, fileCon
         toolUseCache,
         fileContentCache,
         client,
-        logger
+        logger,
+        parentToolCallId
       );
     case "content_block_delta":
       return toAcpNotifications(
@@ -2239,7 +2264,8 @@ function streamEventToAcpNotifications(message, sessionId, toolUseCache, fileCon
         toolUseCache,
         fileContentCache,
         client,
-        logger
+        logger,
+        parentToolCallId
       );
     case "message_start":
     case "message_delta":
@@ -2352,13 +2378,15 @@ function handleResultMessage(message, context) {
 }
 async function handleStreamEvent(message, context) {
   const { sessionId, client, toolUseCache, fileContentCache, logger } = context;
+  const parentToolCallId = message.parent_tool_use_id ?? void 0;
   for (const notification of streamEventToAcpNotifications(
     message,
     sessionId,
     toolUseCache,
     fileContentCache,
     client,
-    logger
+    logger,
+    parentToolCallId
   )) {
     await client.sessionUpdate(notification);
     context.session.notificationHistory.push(notification);
@@ -2410,6 +2438,7 @@ async function handleUserAssistantMessage(message, context) {
   }
   const content = message.message.content;
   const contentToProcess = filterMessageContent(content);
+  const parentToolCallId = "parent_tool_use_id" in message ? message.parent_tool_use_id ?? void 0 : void 0;
   for (const notification of toAcpNotifications(
     contentToProcess,
     message.message.role,
@@ -2417,7 +2446,8 @@ async function handleUserAssistantMessage(message, context) {
     toolUseCache,
     fileContentCache,
     client,
-    logger
+    logger,
+    parentToolCallId
   )) {
     await client.sessionUpdate(notification);
     session.notificationHistory.push(notification);