@posthog/agent 2.3.169 → 2.3.171

package/dist/agent.js

@@ -245,7 +245,7 @@ import { v7 as uuidv7 } from "uuid";
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.3.169",
+  version: "2.3.171",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -4161,6 +4161,22 @@ function createCodexClient(upstreamClient, logger, sessionState, callbacks) {
         const size = update.size;
         if (used !== void 0) sessionState.contextUsed = used;
         if (size !== void 0) sessionState.contextSize = size;
+        const inputTokens = update.inputTokens;
+        const outputTokens = update.outputTokens;
+        if (inputTokens !== void 0) {
+          sessionState.accumulatedUsage.inputTokens += inputTokens;
+        }
+        if (outputTokens !== void 0) {
+          sessionState.accumulatedUsage.outputTokens += outputTokens;
+        }
+        const cachedRead = update.cachedReadTokens;
+        const cachedWrite = update.cachedWriteTokens;
+        if (cachedRead !== void 0) {
+          sessionState.accumulatedUsage.cachedReadTokens += cachedRead;
+        }
+        if (cachedWrite !== void 0) {
+          sessionState.accumulatedUsage.cachedWriteTokens += cachedWrite;
+        }
         callbacks?.onUsageUpdate?.(update);
       }
       await upstreamClient.sessionUpdate(params);
@@ -4340,6 +4356,10 @@ function buildConfigArgs(options) {
   if (options.model) {
     args.push("-c", `model="${options.model}"`);
   }
+  if (options.instructions) {
+    const escaped = options.instructions.replace(/\\/g, "\\\\").replace(/"/g, '\\"');
+    args.push("-c", `instructions="${escaped}"`);
+  }
   return args;
 }
 function findCodexBinary(options) {
@@ -4566,17 +4586,37 @@ var CodexAcpAgent = class extends BaseAcpAgent {
       resetUsage(this.sessionState);
     }
     const response = await this.codexConnection.prompt(params);
-    if (this.sessionState?.taskRunId && response.usage) {
-      await this.client.extNotification("_posthog/usage_update", {
+    if (this.sessionState && response.usage) {
+      this.sessionState.accumulatedUsage.inputTokens += response.usage.inputTokens ?? 0;
+      this.sessionState.accumulatedUsage.outputTokens += response.usage.outputTokens ?? 0;
+      this.sessionState.accumulatedUsage.cachedReadTokens += response.usage.cachedReadTokens ?? 0;
+      this.sessionState.accumulatedUsage.cachedWriteTokens += response.usage.cachedWriteTokens ?? 0;
+    }
+    if (this.sessionState?.taskRunId) {
+      const { accumulatedUsage } = this.sessionState;
+      await this.client.extNotification(POSTHOG_NOTIFICATIONS.TURN_COMPLETE, {
         sessionId: params.sessionId,
-        used: {
-          inputTokens: response.usage.inputTokens ?? 0,
-          outputTokens: response.usage.outputTokens ?? 0,
-          cachedReadTokens: response.usage.cachedReadTokens ?? 0,
-          cachedWriteTokens: response.usage.cachedWriteTokens ?? 0
-        },
-        cost: null
+        stopReason: response.stopReason ?? "end_turn",
+        usage: {
+          inputTokens: accumulatedUsage.inputTokens,
+          outputTokens: accumulatedUsage.outputTokens,
+          cachedReadTokens: accumulatedUsage.cachedReadTokens,
+          cachedWriteTokens: accumulatedUsage.cachedWriteTokens,
+          totalTokens: accumulatedUsage.inputTokens + accumulatedUsage.outputTokens + accumulatedUsage.cachedReadTokens + accumulatedUsage.cachedWriteTokens
+        }
       });
+      if (response.usage) {
+        await this.client.extNotification("_posthog/usage_update", {
+          sessionId: params.sessionId,
+          used: {
+            inputTokens: response.usage.inputTokens ?? 0,
+            outputTokens: response.usage.outputTokens ?? 0,
+            cachedReadTokens: response.usage.cachedReadTokens ?? 0,
+            cachedWriteTokens: response.usage.cachedWriteTokens ?? 0
+          },
+          cost: null
+        });
+      }
     }
     return response;
   }
@@ -5393,7 +5433,8 @@ var Agent = class {
         apiBaseUrl: `${gatewayConfig.gatewayUrl}/v1`,
         apiKey: gatewayConfig.apiKey,
         binaryPath: options.codexBinaryPath,
-        model: sanitizedModel
+        model: sanitizedModel,
+        instructions: options.instructions
       } : void 0
     });
     return this.acpConnection;