@posthog/agent 2.3.385 → 2.3.386

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@posthog/agent",
-  "version": "2.3.385",
+  "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": {
@@ -56,6 +56,14 @@
       "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"

package/src/resume.ts

@@ -16,6 +16,7 @@
  */
 
 import type { ContentBlock } from "@agentclientprotocol/sdk";
+import { selectRecentTurns } from "./adapters/claude/session/jsonl-hydration";
 import type { PostHogAPIClient } from "./posthog-api";
 import { ResumeSaga } from "./sagas/resume-saga";
 import type { DeviceInfo, TreeSnapshotEvent } from "./types";
@@ -113,3 +114,53 @@ export function conversationToPromptHistory(
     content: turn.content,
   }));
 }
+
+const RESUME_HISTORY_TOKEN_BUDGET = 50_000;
+const TOOL_RESULT_MAX_CHARS = 2000;
+
+export function formatConversationForResume(
+  conversation: ConversationTurn[],
+): string {
+  const selected = selectRecentTurns(conversation, RESUME_HISTORY_TOKEN_BUDGET);
+  const parts: string[] = [];
+
+  if (selected.length < conversation.length) {
+    parts.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 as { type: "text"; text: string }).text);
+
+    if (textParts.length > 0) {
+      parts.push(`**${role}**: ${textParts.join("\n")}`);
+    }
+
+    if (turn.toolCalls?.length) {
+      const toolSummary = turn.toolCalls
+        .map((tc) => {
+          let resultStr = "";
+          if (tc.result !== undefined) {
+            const raw =
+              typeof tc.result === "string"
+                ? tc.result
+                : JSON.stringify(tc.result);
+            resultStr =
+              raw.length > TOOL_RESULT_MAX_CHARS
+                ? ` → ${raw.substring(0, TOOL_RESULT_MAX_CHARS)}...(truncated)`
+                : ` → ${raw}`;
+          }
+          return `  - ${tc.toolName}${resultStr}`;
+        })
+        .join("\n");
+      parts.push(`**${role} (tools)**:\n${toolSummary}`);
+    }
+  }
+
+  return parts.join("\n\n");
+}

package/src/server/agent-server.ts

@@ -25,12 +25,11 @@ import {
   type AgentErrorClassification,
   classifyAgentError,
 } from "../adapters/claude/conversion/sdk-to-acp";
-import { selectRecentTurns } from "../adapters/claude/session/jsonl-hydration";
 import type { PermissionMode } from "../execution-mode";
 import { DEFAULT_CODEX_MODEL } from "../gateway-models";
 import { PostHogAPIClient } from "../posthog-api";
 import {
-  type ConversationTurn,
+  formatConversationForResume,
   type ResumeState,
   resumeFromLog,
 } from "../resume";
@@ -1147,7 +1146,7 @@ export class AgentServer {
     if (!this.session || !this.resumeState) return;
 
     try {
-      const conversationSummary = this.formatConversationForResume(
+      const conversationSummary = formatConversationForResume(
         this.resumeState.conversation,
       );
 
@@ -1229,59 +1228,6 @@ export class AgentServer {
     }
   }
 
-  private static RESUME_HISTORY_TOKEN_BUDGET = 50_000;
-  private static TOOL_RESULT_MAX_CHARS = 2000;
-
-  private formatConversationForResume(
-    conversation: ConversationTurn[],
-  ): string {
-    const selected = selectRecentTurns(
-      conversation,
-      AgentServer.RESUME_HISTORY_TOKEN_BUDGET,
-    );
-    const parts: string[] = [];
-
-    if (selected.length < conversation.length) {
-      parts.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 as { type: "text"; text: string }).text);
-
-      if (textParts.length > 0) {
-        parts.push(`**${role}**: ${textParts.join("\n")}`);
-      }
-
-      if (turn.toolCalls?.length) {
-        const toolSummary = turn.toolCalls
-          .map((tc) => {
-            let resultStr = "";
-            if (tc.result !== undefined) {
-              const raw =
-                typeof tc.result === "string"
-                  ? tc.result
-                  : JSON.stringify(tc.result);
-              resultStr =
-                raw.length > AgentServer.TOOL_RESULT_MAX_CHARS
-                  ? ` → ${raw.substring(0, AgentServer.TOOL_RESULT_MAX_CHARS)}...(truncated)`
-                  : ` → ${raw}`;
-            }
-            return `  - ${tc.toolName}${resultStr}`;
-          })
-          .join("\n");
-        parts.push(`**${role} (tools)**:\n${toolSummary}`);
-      }
-    }
-
-    return parts.join("\n\n");
-  }
-
   private getInitialPromptOverride(taskRun: TaskRun): string | null {
     const state = taskRun.state as Record<string, unknown> | undefined;
     const override = state?.initial_prompt_override;