@posthog/agent 2.3.72 → 2.3.73

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.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: {
@@ -1372,12 +1372,14 @@ function formatGatewayModelName(model) {
 }
 
 // src/adapters/base-acp-agent.ts
+var DEFAULT_CONTEXT_WINDOW = 2e5;
 var BaseAcpAgent = class {
   session;
   sessionId;
   client;
   logger;
   fileContentCache = {};
+  gatewayModels = [];
   constructor(client) {
     this.client = client;
     this.logger = new Logger({ debug: true, prefix: "[BaseAcpAgent]" });
@@ -1430,8 +1432,8 @@ var BaseAcpAgent = class {
     throw new Error("Method not implemented.");
   }
   async getModelConfigOptions(currentModelOverride) {
-    const gatewayModels = await fetchGatewayModels();
-    const options = gatewayModels.filter((model) => isAnthropicModel(model)).map((model) => ({
+    this.gatewayModels = await fetchGatewayModels();
+    const options = this.gatewayModels.filter((model) => isAnthropicModel(model)).map((model) => ({
       value: model.id,
       name: formatGatewayModelName(model),
       description: `Context: ${model.context_window.toLocaleString()} tokens`
@@ -1452,6 +1454,10 @@ var BaseAcpAgent = class {
     }
     return { currentModelId, options };
   }
+  getContextWindowForModel(modelId) {
+    const match = this.gatewayModels.find((m) => m.id === modelId);
+    return match?.context_window ?? DEFAULT_CONTEXT_WINDOW;
+  }
 };
 
 // src/adapters/claude/conversion/acp-to-sdk.ts
@@ -2627,7 +2633,7 @@ function streamEventToAcpNotifications(message, sessionId, toolUseCache, fileCon
   }
 }
 async function handleSystemMessage(message, context) {
-  const { sessionId, client, logger } = context;
+  const { session, sessionId, client, logger } = context;
   switch (message.subtype) {
     case "init":
       break;
@@ -2635,7 +2641,8 @@ async function handleSystemMessage(message, context) {
       await client.extNotification("_posthog/compact_boundary", {
         sessionId,
         trigger: message.compact_metadata.trigger,
-        preTokens: message.compact_metadata.pre_tokens
+        preTokens: message.compact_metadata.pre_tokens,
+        contextSize: session.contextSize
       });
       break;
     case "hook_response":
@@ -3460,16 +3467,6 @@ var GATEWAY_TO_SDK_MODEL = {
 function toSdkModelId(modelId) {
   return GATEWAY_TO_SDK_MODEL[modelId] ?? modelId;
 }
-var MODELS_WITH_1M_CONTEXT = /* @__PURE__ */ new Set([
-  "claude-opus-4-6",
-  "claude-sonnet-4-6"
-]);
-function supports1MContext(modelId) {
-  return MODELS_WITH_1M_CONTEXT.has(modelId);
-}
-function getDefaultContextWindow(modelId) {
-  return supports1MContext(modelId) ? 1e6 : 2e5;
-}
 var MODELS_WITH_EFFORT = /* @__PURE__ */ new Set([
   "claude-opus-4-5",
   "claude-opus-4-6",
@@ -4202,6 +4199,16 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
     this.session.promptRunning = true;
     let handedOff = false;
     let lastAssistantTotalUsage = 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 = this.clientCapabilities?._meta?.terminal_output === true;
     const context = {
       session: this.session,
@@ -4253,8 +4260,19 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
             const contextWindows = Object.values(message.modelUsage).map(
               (m) => m.contextWindow
             );
-            const contextWindowSize = contextWindows.length > 0 ? Math.min(...contextWindows) : getDefaultContextWindow(this.session.modelId ?? "");
-            this.session.contextSize = contextWindowSize;
+            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 = lastContextWindowSize;
             if (lastAssistantTotalUsage !== null) {
               this.session.contextUsed = lastAssistantTotalUsage;
             }
@@ -4264,7 +4282,7 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
                 update: {
                   sessionUpdate: "usage_update",
                   used: lastAssistantTotalUsage,
-                  size: contextWindowSize,
+                  size: lastContextWindowSize,
                   cost: {
                     amount: message.total_cost_usd,
                     currency: "USD"
@@ -4330,7 +4348,16 @@ 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.output_tokens + usage.cache_read_input_tokens + usage.cache_creation_input_tokens;
+              lastAssistantTotalUsage = usage.input_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);
             if (result.error) throw result.error;
@@ -4388,9 +4415,11 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
     await this.session.query.interrupt();
   }
   async unstable_setSessionModel(params) {
-    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 {};
@@ -4441,6 +4470,7 @@ var ClaudeAcpAgent = class 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;
@@ -4601,11 +4631,10 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
     const modelOptions = await this.getModelConfigOptions();
     const resolvedModelId = settingsModel || modelOptions.currentModelId;
     session.modelId = resolvedModelId;
-    if (!isResume) {
-      const resolvedSdkModel = toSdkModelId(resolvedModelId);
-      if (resolvedSdkModel !== DEFAULT_MODEL) {
-        await this.session.query.setModel(resolvedSdkModel);
-      }
+    session.lastContextWindowSize = this.getContextWindowForModel(resolvedModelId);
+    const resolvedSdkModel = toSdkModelId(resolvedModelId);
+    if (!isResume && resolvedSdkModel !== DEFAULT_MODEL) {
+      await this.session.query.setModel(resolvedSdkModel);
     }
     const availableModes2 = getAvailableModes();
     const modes = {