@posthog/agent 2.3.202 → 2.3.209

package/dist/server/bin.cjs

@@ -904,7 +904,7 @@ var import_hono = require("hono");
 // package.json
 var package_default = {
   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: {
@@ -1066,6 +1066,10 @@ var POSTHOG_NOTIFICATIONS = {
   /** Token usage update for a session turn */
   USAGE_UPDATE: "_posthog/usage_update"
 };
+function isNotification(method, notification) {
+  if (!method) return false;
+  return method === notification || method === `_${notification}`;
+}
 
 // src/adapters/acp-connection.ts
 var import_sdk4 = require("@agentclientprotocol/sdk");
@@ -2689,7 +2693,7 @@ async function handleSystemMessage(message, context) {
     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,
@@ -2705,7 +2709,7 @@ async function handleSystemMessage(message, context) {
     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"
         });
@@ -2718,7 +2722,7 @@ async function handleSystemMessage(message, context) {
         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,
@@ -4350,6 +4354,14 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
             if (message.subtype === "compact_boundary") {
               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;
@@ -4465,7 +4477,7 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
             }
             if ("usage" in message.message && message.parent_tool_use_id === null) {
               const usage = message.message.usage;
-              lastAssistantTotalUsage = usage.input_tokens + usage.cache_read_input_tokens + usage.cache_creation_input_tokens;
+              lastAssistantTotalUsage = usage.input_tokens + usage.output_tokens + usage.cache_read_input_tokens + usage.cache_creation_input_tokens;
               await this.client.sessionUpdate({
                 sessionId: params.sessionId,
                 update: {
@@ -4746,7 +4758,7 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
         settingsManager.getSettings().model || meta?.model || void 0
       ),
       ...meta?.taskRunId ? [
-        this.client.extNotification("_posthog/sdk_session", {
+        this.client.extNotification(POSTHOG_NOTIFICATIONS.SDK_SESSION, {
           taskRunId: meta.taskRunId,
           sessionId,
           adapter: "claude"
@@ -11521,11 +11533,12 @@ var ResumeSaga = class extends Saga {
     };
   }
   findLatestTreeSnapshot(entries) {
-    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) {
+      if (isNotification(
+        entry.notification?.method,
+        POSTHOG_NOTIFICATIONS.TREE_SNAPSHOT
+      )) {
         const params = entry.notification.params;
         if (params?.treeHash) {
           return params;