@posthog/agent 2.3.202 → 2.3.209

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@posthog/agent",
-  "version": "2.3.202",
+  "version": "2.3.209",
   "repository": "https://github.com/PostHog/code",
   "description": "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   "exports": {

package/src/acp-extensions.ts

@@ -5,10 +5,6 @@
  * - Custom notification methods are prefixed with `_posthog/`
  * - Custom data can be attached via `_meta` fields
  *
- * Note: When using `extNotification()` from the ACP SDK, it automatically
- * adds an extra underscore prefix (e.g., `_posthog/tree_snapshot` becomes
- * `__posthog/tree_snapshot` in the log). Code that reads logs should handle both.
- *
  * See: https://agentclientprotocol.com/docs/extensibility
  */
 
@@ -68,3 +64,18 @@ export const POSTHOG_NOTIFICATIONS = {
   /** Token usage update for a session turn */
   USAGE_UPDATE: "_posthog/usage_update",
 } as const;
+
+type NotificationMethod =
+  (typeof POSTHOG_NOTIFICATIONS)[keyof typeof POSTHOG_NOTIFICATIONS];
+
+/**
+ * Check if an ACP method matches a PostHog notification, handling the
+ * possible `__posthog/` double-prefix from extNotification().
+ */
+export function isNotification(
+  method: string | undefined,
+  notification: NotificationMethod,
+): boolean {
+  if (!method) return false;
+  return method === notification || method === `_${notification}`;
+}

package/src/adapters/claude/claude-agent.ts

@@ -371,8 +371,18 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
         switch (message.type) {
           case "system":
             if (message.subtype === "compact_boundary") {
+              // Send used:0 immediately so the client doesn't keep showing
+              // the stale pre-compaction context size until the next turn.
               lastAssistantTotalUsage = 0;
               promptReplayed = true;
+              await this.client.sessionUpdate({
+                sessionId: params.sessionId,
+                update: {
+                  sessionUpdate: "usage_update",
+                  used: 0,
+                  size: lastContextWindowSize,
+                },
+              });
             }
             if (message.subtype === "local_command_output") {
               promptReplayed = true;
@@ -530,6 +540,11 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
             }
 
             // Store latest assistant usage (excluding subagents)
+            // Sum all token types as a proxy for post-turn context occupancy:
+            // current turn's output will become next turn's input.
+            // Note: per the Anthropic API, input_tokens excludes cache tokens —
+            // cache_read and cache_creation are reported separately, so summing
+            // all four fields is not double-counting.
             if (
               "usage" in message.message &&
               message.parent_tool_use_id === null
@@ -544,6 +559,7 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
               };
               lastAssistantTotalUsage =
                 usage.input_tokens +
+                usage.output_tokens +
                 usage.cache_read_input_tokens +
                 usage.cache_creation_input_tokens;
 
@@ -931,7 +947,7 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
       ),
       ...(meta?.taskRunId
         ? [
-            this.client.extNotification("_posthog/sdk_session", {
+            this.client.extNotification(POSTHOG_NOTIFICATIONS.SDK_SESSION, {
               taskRunId: meta.taskRunId,
               sessionId,
               adapter: "claude",

package/src/adapters/claude/conversion/sdk-to-acp.ts

@@ -17,6 +17,7 @@ import type {
   BetaContentBlock,
   BetaRawContentBlockDelta,
 } from "@anthropic-ai/sdk/resources/beta.mjs";
+import { POSTHOG_NOTIFICATIONS } from "@/acp-extensions";
 import { image, text } from "../../../utils/acp-content";
 import { unreachable } from "../../../utils/common";
 import type { Logger } from "../../../utils/logger";
@@ -550,7 +551,7 @@ export async function handleSystemMessage(
     case "init":
       break;
     case "compact_boundary":
-      await client.extNotification("_posthog/compact_boundary", {
+      await client.extNotification(POSTHOG_NOTIFICATIONS.COMPACT_BOUNDARY, {
         sessionId,
         trigger: message.compact_metadata.trigger,
         preTokens: message.compact_metadata.pre_tokens,
@@ -566,7 +567,7 @@ export async function handleSystemMessage(
     case "status":
       if (message.status === "compacting") {
         logger.info("Session compacting started", { sessionId });
-        await client.extNotification("_posthog/status", {
+        await client.extNotification(POSTHOG_NOTIFICATIONS.STATUS, {
           sessionId,
           status: "compacting",
         });
@@ -579,7 +580,7 @@ export async function handleSystemMessage(
         status: message.status,
         summary: message.summary,
       });
-      await client.extNotification("_posthog/task_notification", {
+      await client.extNotification(POSTHOG_NOTIFICATIONS.TASK_NOTIFICATION, {
         sessionId,
         taskId: message.task_id,
         status: message.status,

package/src/index.ts

@@ -1,3 +1,4 @@
+export { isNotification, POSTHOG_NOTIFICATIONS } from "./acp-extensions";
 export {
   getMcpToolMetadata,
   isMcpToolReadOnly,

package/src/sagas/resume-saga.test.ts

@@ -441,35 +441,6 @@ describe("ResumeSaga", () => {
       expect(result.data.latestSnapshot?.treeHash).toBe("hash-2");
     });
 
-    it("finds snapshot with SDK double-underscore prefix", async () => {
-      (mockApiClient.getTaskRun as ReturnType<typeof vi.fn>).mockResolvedValue(
-        createTaskRun(),
-      );
-      (
-        mockApiClient.fetchTaskRunLogs as ReturnType<typeof vi.fn>
-      ).mockResolvedValue([
-        createNotification(`_${POSTHOG_NOTIFICATIONS.TREE_SNAPSHOT}`, {
-          treeHash: "sdk-prefixed-hash",
-          baseCommit: "abc",
-          changes: [],
-          timestamp: new Date().toISOString(),
-        }),
-      ]);
-
-      const saga = new ResumeSaga(mockLogger);
-      const result = await saga.run({
-        taskId: "task-1",
-        runId: "run-1",
-        repositoryPath: repo.path,
-        apiClient: mockApiClient,
-      });
-
-      expect(result.success).toBe(true);
-      if (!result.success) return;
-
-      expect(result.data.latestSnapshot?.treeHash).toBe("sdk-prefixed-hash");
-    });
-
     it("returns interrupted flag from snapshot", async () => {
       (mockApiClient.getTaskRun as ReturnType<typeof vi.fn>).mockResolvedValue(
         createTaskRun(),

package/src/sagas/resume-saga.ts

@@ -1,6 +1,6 @@
 import type { ContentBlock } from "@agentclientprotocol/sdk";
 import { Saga } from "@posthog/shared";
-import { POSTHOG_NOTIFICATIONS } from "../acp-extensions";
+import { isNotification, POSTHOG_NOTIFICATIONS } from "../acp-extensions";
 import type { PostHogAPIClient } from "../posthog-api";
 import { TreeTracker } from "../tree-tracker";
 import type {
@@ -178,14 +178,13 @@ export class ResumeSaga extends Saga<ResumeInput, ResumeOutput> {
   private findLatestTreeSnapshot(
     entries: StoredNotification[],
   ): TreeSnapshotEvent | null {
-    const sdkPrefixedMethod = `_${POSTHOG_NOTIFICATIONS.TREE_SNAPSHOT}`;
-
     for (let i = entries.length - 1; i >= 0; i--) {
       const entry = entries[i];
-      const method = entry.notification?.method;
       if (
-        method === sdkPrefixedMethod ||
-        method === POSTHOG_NOTIFICATIONS.TREE_SNAPSHOT
+        isNotification(
+          entry.notification?.method,
+          POSTHOG_NOTIFICATIONS.TREE_SNAPSHOT,
+        )
       ) {
         const params = entry.notification.params as
           | TreeSnapshotEvent