@posthog/agent 2.3.370 → 2.3.386

package/dist/server/bin.cjs

@@ -8729,7 +8729,7 @@ var import_zod3 = require("zod");
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.3.370",
+  version: "2.3.386",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -8785,6 +8785,14 @@ var package_default = {
       types: "./dist/execution-mode.d.ts",
       import: "./dist/execution-mode.js"
     },
+    "./resume": {
+      types: "./dist/resume.d.ts",
+      import: "./dist/resume.js"
+    },
+    "./tree-tracker": {
+      types: "./dist/tree-tracker.d.ts",
+      import: "./dist/tree-tracker.js"
+    },
     "./server": {
       types: "./dist/server/agent-server.d.ts",
       import: "./dist/server/agent-server.js"
@@ -16452,14 +16460,6 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
     return this.listSessions(params);
   }
   async prompt(params) {
-    this.session.cancelled = false;
-    this.session.interruptReason = void 0;
-    this.session.accumulatedUsage = {
-      inputTokens: 0,
-      outputTokens: 0,
-      cachedReadTokens: 0,
-      cachedWriteTokens: 0
-    };
     const userMessage = promptToClaude(params);
     const promptUuid = (0, import_node_crypto.randomUUID)();
     userMessage.uuid = promptUuid;
@@ -16495,6 +16495,14 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
     } else {
       this.session.input.push(userMessage);
     }
+    this.session.cancelled = false;
+    this.session.interruptReason = void 0;
+    this.session.accumulatedUsage = {
+      inputTokens: 0,
+      outputTokens: 0,
+      cachedReadTokens: 0,
+      cachedWriteTokens: 0
+    };
     await this.broadcastUserMessage(params);
     this.session.promptRunning = true;
     let handedOff = false;
@@ -16700,9 +16708,6 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
           }
           case "user":
           case "assistant": {
-            if (this.session.cancelled) {
-              break;
-            }
             if (message.type === "user" && "uuid" in message && message.uuid) {
               if (message.uuid === promptUuid) {
                 promptReplayed = true;
@@ -16715,9 +16720,14 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
                 pending.resolve(false);
                 this.session.pendingMessages.delete(message.uuid);
                 handedOff = true;
-                return { stopReason: "end_turn" };
+                return {
+                  stopReason: this.session.cancelled ? "cancelled" : "end_turn"
+                };
               }
             }
+            if (this.session.cancelled) {
+              break;
+            }
             if ("isReplay" in message && message.isReplay) {
               break;
             }
@@ -18321,45 +18331,6 @@ function createCodexConnection(config) {
   };
 }
 
-// src/adapters/claude/session/jsonl-hydration.ts
-var import_node_crypto2 = require("crypto");
-var fs10 = __toESM(require("fs/promises"), 1);
-var os6 = __toESM(require("os"), 1);
-var path12 = __toESM(require("path"), 1);
-var CHARS_PER_TOKEN = 4;
-var DEFAULT_MAX_TOKENS = 15e4;
-function estimateTurnTokens(turn) {
-  let chars = 0;
-  for (const block of turn.content) {
-    if ("text" in block && typeof block.text === "string") {
-      chars += block.text.length;
-    }
-  }
-  if (turn.toolCalls) {
-    for (const tc of turn.toolCalls) {
-      chars += JSON.stringify(tc.input ?? "").length;
-      if (tc.result !== void 0) {
-        chars += typeof tc.result === "string" ? tc.result.length : JSON.stringify(tc.result).length;
-      }
-    }
-  }
-  return Math.ceil(chars / CHARS_PER_TOKEN);
-}
-function selectRecentTurns(turns, maxTokens = DEFAULT_MAX_TOKENS) {
-  let budget = maxTokens;
-  let startIndex = turns.length;
-  for (let i2 = turns.length - 1; i2 >= 0; i2--) {
-    const cost = estimateTurnTokens(turns[i2]);
-    if (cost > budget) break;
-    budget -= cost;
-    startIndex = i2;
-  }
-  while (startIndex < turns.length && turns[startIndex].role !== "user") {
-    startIndex++;
-  }
-  return turns.slice(startIndex);
-}
-
 // src/utils/gateway.ts
 function getGatewayBaseUrl(posthogHost) {
   const url = new URL(posthogHost);
@@ -18561,6 +18532,45 @@ var PostHogAPIClient = class {
   }
 };
 
+// src/adapters/claude/session/jsonl-hydration.ts
+var import_node_crypto2 = require("crypto");
+var fs10 = __toESM(require("fs/promises"), 1);
+var os6 = __toESM(require("os"), 1);
+var path12 = __toESM(require("path"), 1);
+var CHARS_PER_TOKEN = 4;
+var DEFAULT_MAX_TOKENS = 15e4;
+function estimateTurnTokens(turn) {
+  let chars = 0;
+  for (const block of turn.content) {
+    if ("text" in block && typeof block.text === "string") {
+      chars += block.text.length;
+    }
+  }
+  if (turn.toolCalls) {
+    for (const tc of turn.toolCalls) {
+      chars += JSON.stringify(tc.input ?? "").length;
+      if (tc.result !== void 0) {
+        chars += typeof tc.result === "string" ? tc.result.length : JSON.stringify(tc.result).length;
+      }
+    }
+  }
+  return Math.ceil(chars / CHARS_PER_TOKEN);
+}
+function selectRecentTurns(turns, maxTokens = DEFAULT_MAX_TOKENS) {
+  let budget = maxTokens;
+  let startIndex = turns.length;
+  for (let i2 = turns.length - 1; i2 >= 0; i2--) {
+    const cost = estimateTurnTokens(turns[i2]);
+    if (cost > budget) break;
+    budget -= cost;
+    startIndex = i2;
+  }
+  while (startIndex < turns.length && turns[startIndex].role !== "user") {
+    startIndex++;
+  }
+  return turns.slice(startIndex);
+}
+
 // ../shared/dist/index.js
 var CLOUD_PROMPT_PREFIX = "__twig_cloud_prompt_v1__:";
 function deserializeCloudPrompt(value) {
@@ -19530,6 +19540,37 @@ async function resumeFromLog(config) {
     logEntryCount: result.data.logEntryCount
   };
 }
+var RESUME_HISTORY_TOKEN_BUDGET = 5e4;
+var TOOL_RESULT_MAX_CHARS = 2e3;
+function formatConversationForResume(conversation) {
+  const selected = selectRecentTurns(conversation, RESUME_HISTORY_TOKEN_BUDGET);
+  const parts2 = [];
+  if (selected.length < conversation.length) {
+    parts2.push(
+      `*(${conversation.length - selected.length} earlier turns omitted)*`
+    );
+  }
+  for (const turn of selected) {
+    const role = turn.role === "user" ? "User" : "Assistant";
+    const textParts = turn.content.filter((block) => block.type === "text").map((block) => block.text);
+    if (textParts.length > 0) {
+      parts2.push(`**${role}**: ${textParts.join("\n")}`);
+    }
+    if (turn.toolCalls?.length) {
+      const toolSummary = turn.toolCalls.map((tc) => {
+        let resultStr = "";
+        if (tc.result !== void 0) {
+          const raw = typeof tc.result === "string" ? tc.result : JSON.stringify(tc.result);
+          resultStr = raw.length > TOOL_RESULT_MAX_CHARS ? ` \u2192 ${raw.substring(0, TOOL_RESULT_MAX_CHARS)}...(truncated)` : ` \u2192 ${raw}`;
+        }
+        return `  - ${tc.toolName}${resultStr}`;
+      }).join("\n");
+      parts2.push(`**${role} (tools)**:
+${toolSummary}`);
+    }
+  }
+  return parts2.join("\n\n");
+}
 
 // src/session-log-writer.ts
 var import_node_fs5 = __toESM(require("fs"), 1);
@@ -20115,7 +20156,7 @@ function getTaskRunStateString(taskRun, key) {
   const value = state[key];
   return typeof value === "string" ? value : null;
 }
-var AgentServer = class _AgentServer {
+var AgentServer = class {
   config;
   logger;
   server = null;
@@ -20840,7 +20881,7 @@ var AgentServer = class _AgentServer {
   async sendResumeMessage(payload, taskRun) {
     if (!this.session || !this.resumeState) return;
     try {
-      const conversationSummary = this.formatConversationForResume(
+      const conversationSummary = formatConversationForResume(
         this.resumeState.conversation
       );
       const pendingUserPrompt = await this.getPendingUserPrompt(taskRun);
@@ -20911,40 +20952,6 @@ Continue from where you left off. The user is waiting for your response.`
       await this.classifyAndSignalFailure(payload, "resume", error);
     }
   }
-  static RESUME_HISTORY_TOKEN_BUDGET = 5e4;
-  static TOOL_RESULT_MAX_CHARS = 2e3;
-  formatConversationForResume(conversation) {
-    const selected = selectRecentTurns(
-      conversation,
-      _AgentServer.RESUME_HISTORY_TOKEN_BUDGET
-    );
-    const parts2 = [];
-    if (selected.length < conversation.length) {
-      parts2.push(
-        `*(${conversation.length - selected.length} earlier turns omitted)*`
-      );
-    }
-    for (const turn of selected) {
-      const role = turn.role === "user" ? "User" : "Assistant";
-      const textParts = turn.content.filter((block) => block.type === "text").map((block) => block.text);
-      if (textParts.length > 0) {
-        parts2.push(`**${role}**: ${textParts.join("\n")}`);
-      }
-      if (turn.toolCalls?.length) {
-        const toolSummary = turn.toolCalls.map((tc) => {
-          let resultStr = "";
-          if (tc.result !== void 0) {
-            const raw = typeof tc.result === "string" ? tc.result : JSON.stringify(tc.result);
-            resultStr = raw.length > _AgentServer.TOOL_RESULT_MAX_CHARS ? ` \u2192 ${raw.substring(0, _AgentServer.TOOL_RESULT_MAX_CHARS)}...(truncated)` : ` \u2192 ${raw}`;
-          }
-          return `  - ${tc.toolName}${resultStr}`;
-        }).join("\n");
-        parts2.push(`**${role} (tools)**:
-${toolSummary}`);
-      }
-    }
-    return parts2.join("\n\n");
-  }
   getInitialPromptOverride(taskRun) {
     const state = taskRun.state;
     const override = state?.initial_prompt_override;