@posthog/agent 2.3.72 → 2.3.73

package/package.json

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

package/src/adapters/base-acp-agent.ts

@@ -20,6 +20,7 @@ import {
   DEFAULT_GATEWAY_MODEL,
   fetchGatewayModels,
   formatGatewayModelName,
+  type GatewayModel,
   isAnthropicModel,
 } from "../gateway-models";
 import { Logger } from "../utils/logger";
@@ -33,6 +34,8 @@ export interface BaseSession {
   settingsManager: SettingsManager;
 }
 
+const DEFAULT_CONTEXT_WINDOW = 200_000;
+
 export abstract class BaseAcpAgent implements Agent {
   abstract readonly adapterName: string;
   protected session!: BaseSession;
@@ -40,6 +43,7 @@ export abstract class BaseAcpAgent implements Agent {
   client: AgentSideConnection;
   logger: Logger;
   fileContentCache: { [key: string]: string } = {};
+  protected gatewayModels: GatewayModel[] = [];
 
   constructor(client: AgentSideConnection) {
     this.client = client;
@@ -119,9 +123,9 @@ export abstract class BaseAcpAgent implements Agent {
     currentModelId: string;
     options: SessionConfigSelectOption[];
   }> {
-    const gatewayModels = await fetchGatewayModels();
+    this.gatewayModels = await fetchGatewayModels();
 
-    const options = gatewayModels
+    const options = this.gatewayModels
       .filter((model) => isAnthropicModel(model))
       .map((model) => ({
         value: model.id,
@@ -150,4 +154,9 @@ export abstract class BaseAcpAgent implements Agent {
 
     return { currentModelId, options };
   }
+
+  getContextWindowForModel(modelId: string): number {
+    const match = this.gatewayModels.find((m) => m.id === modelId);
+    return match?.context_window ?? DEFAULT_CONTEXT_WINDOW;
+  }
 }

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

@@ -64,7 +64,6 @@ import { getAvailableSlashCommands } from "./session/commands";
 import { parseMcpServers } from "./session/mcp-config";
 import {
   DEFAULT_MODEL,
-  getDefaultContextWindow,
   getEffortOptions,
   resolveModelPreference,
   toSdkModelId,
@@ -323,6 +322,16 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
     this.session.promptRunning = true;
     let handedOff = false;
     let lastAssistantTotalUsage: number | null = null;
+    if (this.session.lastContextWindowSize == null) {
+      this.session.lastContextWindowSize = this.getContextWindowForModel(
+        this.session.modelId ?? "",
+      );
+      this.logger.debug("Initial context window size from gateway", {
+        modelId: this.session.modelId,
+        contextWindowSize: this.session.lastContextWindowSize,
+      });
+    }
+    let lastContextWindowSize = this.session.lastContextWindowSize;
 
     const supportsTerminalOutput =
       (
@@ -393,16 +402,25 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
             this.session.accumulatedUsage.cachedWriteTokens +=
               message.usage.cache_creation_input_tokens;
 
-            // Calculate context window size from modelUsage (minimum across all models used)
+            // SDK can underreport context window (e.g. 200k for 1M models).
+            // Use SDK value only if it's larger than what gateway reported.
             const contextWindows = Object.values(message.modelUsage).map(
               (m) => m.contextWindow,
             );
-            const contextWindowSize =
-              contextWindows.length > 0
-                ? Math.min(...contextWindows)
-                : getDefaultContextWindow(this.session.modelId ?? "");
+            if (contextWindows.length > 0) {
+              const sdkContextWindow = Math.min(...contextWindows);
+              if (sdkContextWindow > lastContextWindowSize) {
+                lastContextWindowSize = sdkContextWindow;
+              }
+            }
+            this.session.lastContextWindowSize = lastContextWindowSize;
+            this.logger.debug("Context window size from result", {
+              sdkReported: contextWindows,
+              resolved: lastContextWindowSize,
+              modelId: this.session.modelId,
+            });
 
-            this.session.contextSize = contextWindowSize;
+            this.session.contextSize = lastContextWindowSize;
             if (lastAssistantTotalUsage !== null) {
               this.session.contextUsed = lastAssistantTotalUsage;
             }
@@ -414,7 +432,7 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
                 update: {
                   sessionUpdate: "usage_update",
                   used: lastAssistantTotalUsage,
-                  size: contextWindowSize,
+                  size: lastContextWindowSize,
                   cost: {
                     amount: message.total_cost_usd,
                     currency: "USD",
@@ -521,9 +539,18 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
               };
               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: {
+                  sessionUpdate: "usage_update",
+                  used: lastAssistantTotalUsage,
+                  size: lastContextWindowSize,
+                  cost: null,
+                },
+              });
             }
 
             const result = await handleUserAssistantMessage(message, context);
@@ -595,9 +622,11 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
   async unstable_setSessionModel(
     params: SetSessionModelRequest,
   ): Promise<SetSessionModelResponse | undefined> {
-    const sdkModelId = toSdkModelId(params.modelId);
-    await this.session.query.setModel(sdkModelId);
+    await this.session.query.setModel(toSdkModelId(params.modelId));
     this.session.modelId = params.modelId;
+    this.session.lastContextWindowSize = this.getContextWindowForModel(
+      params.modelId,
+    );
     this.rebuildEffortConfigOption(params.modelId);
     await this.updateConfigOption("model", params.modelId);
     return {};
@@ -674,6 +703,8 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
       const sdkModelId = toSdkModelId(resolvedValue);
       await this.session.query.setModel(sdkModelId);
       this.session.modelId = resolvedValue;
+      this.session.lastContextWindowSize =
+        this.getContextWindowForModel(resolvedValue);
       this.rebuildEffortConfigOption(resolvedValue);
     } else if (params.configId === "effort") {
       const newEffort = resolvedValue as EffortLevel;
@@ -893,12 +924,12 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
     const modelOptions = await this.getModelConfigOptions();
     const resolvedModelId = settingsModel || modelOptions.currentModelId;
     session.modelId = resolvedModelId;
+    session.lastContextWindowSize =
+      this.getContextWindowForModel(resolvedModelId);
 
-    if (!isResume) {
-      const resolvedSdkModel = toSdkModelId(resolvedModelId);
-      if (resolvedSdkModel !== DEFAULT_MODEL) {
-        await this.session.query.setModel(resolvedSdkModel);
-      }
+    const resolvedSdkModel = toSdkModelId(resolvedModelId);
+    if (!isResume && resolvedSdkModel !== DEFAULT_MODEL) {
+      await this.session.query.setModel(resolvedSdkModel);
     }
 
     const availableModes = getAvailableModes();

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

@@ -544,7 +544,7 @@ export async function handleSystemMessage(
   message: Extract<SDKMessage, { type: "system" }>,
   context: MessageHandlerContext,
 ): Promise<void> {
-  const { sessionId, client, logger } = context;
+  const { session, sessionId, client, logger } = context;
 
   switch (message.subtype) {
     case "init":
@@ -554,6 +554,7 @@ export async function handleSystemMessage(
         sessionId,
         trigger: message.compact_metadata.trigger,
         preTokens: message.compact_metadata.pre_tokens,
+        contextSize: session.contextSize,
       });
       break;
     case "hook_response":

package/src/adapters/claude/session/models.ts

@@ -21,10 +21,6 @@ export function supports1MContext(modelId: string): boolean {
   return MODELS_WITH_1M_CONTEXT.has(modelId);
 }
 
-export function getDefaultContextWindow(modelId: string): number {
-  return supports1MContext(modelId) ? 1_000_000 : 200_000;
-}
-
 const MODELS_WITH_EFFORT = new Set([
   "claude-opus-4-5",
   "claude-opus-4-6",

package/src/adapters/claude/types.ts

@@ -57,6 +57,8 @@ export type Session = BaseSession & {
   contextUsed?: number;
   /** Context window size in tokens */
   contextSize?: number;
+  /** Persists across prompt() calls so SDK-reported values survive turn boundaries */
+  lastContextWindowSize?: number;
   promptRunning: boolean;
   pendingMessages: Map<string, PendingMessage>;
   nextPendingOrder: number;