pi-openai-toolkit 0.17.1 → 0.17.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-openai-toolkit",
-  "version": "0.17.1",
+  "version": "0.17.3",
   "private": false,
   "type": "module",
   "engines": {

package/src/codex-astra/extension.ts

@@ -1,6 +1,5 @@
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import { ASTRA_MODEL_ID } from "../types";
-import { CODEX_CLIENT_VERSION } from "../responses-headers";
 import {
 	getEffortControlState,
 	planStableEffort,
@@ -101,20 +100,6 @@ export function registerCodexAstraExtension(pi: ExtensionAPI): void {
 			return undefined;
 		}
 	});
-
-	pi.on("before_provider_headers", (event, ctx) => {
-		try {
-			const model = ctx.model;
-			if (!model || model.api !== "openai-codex-responses") return;
-			// Live requests need the same backend version gate as our synthetic
-			// ones: an old or missing `version` never reaches gated SKUs.
-			if (!event.headers.version) {
-				event.headers.version = CODEX_CLIENT_VERSION;
-			}
-		} catch {
-			// Header injection is best-effort.
-		}
-	});
 }
 
 export default function codexAstraExtension(pi: ExtensionAPI): void {

package/src/context-management/codex-provider.ts

@@ -146,7 +146,6 @@ export function codexContextProviderHeaders(
 		headers.set("ChatGPT-Account-ID", provider.accountId);
 		headers.set("originator", CODEX_TOOL_ORIGINATOR);
 		headers.set("User-Agent", userAgent(CODEX_TOOL_VERSION));
-		headers.set("version", CODEX_TOOL_VERSION);
 	} else {
 		headers.set("Authorization", `Bearer ${provider.apiKey}`);
 		headers.delete("ChatGPT-Account-ID");

package/src/context-management/window-manager.ts

@@ -1,12 +1,14 @@
 import { randomUUID } from "node:crypto";
 import type { AgentMessage } from "@earendil-works/pi-agent-core";
 import type { ProviderHeaders } from "@earendil-works/pi-ai";
-import type {
-	CompactionResult,
-	ExtensionAPI,
-	ExtensionContext,
-	SessionBeforeCompactEvent,
-	SessionEntry,
+import {
+	calculateContextTokens,
+	estimateTokens,
+	type CompactionResult,
+	type ExtensionAPI,
+	type ExtensionContext,
+	type SessionBeforeCompactEvent,
+	type SessionEntry,
 } from "@earendil-works/pi-coding-agent";
 import { ContextWindowBudget, type ContextRemaining } from "./window-budget";
 import type { WindowBulkReport } from "./window-bulk";
@@ -86,6 +88,12 @@ export class CodexContextWindowManager {
 	private lastKnownSystemHead: AgentMessage | undefined;
 	private bulkReport: WindowBulkReport | undefined;
 	private bulkSurfacedKey: string | undefined;
+	/**
+	 * Pi's getContextUsage() measures the durable transcript. Remote windows are
+	 * projected only by this extension, so remember the last provider-visible
+	 * estimate for budget decisions and get_context_remaining.
+	 */
+	private projectedContextTokens: number | undefined;
 	private readonly loadThreadHint: ThreadHintLoader;
 
 	constructor(loadThreadHint?: ThreadHintLoader) {
@@ -149,6 +157,7 @@ export class CodexContextWindowManager {
 		this.branchStateInvalidated = false;
 		this.budget.reset();
 		this.trimPendingWindowId = undefined;
+		this.projectedContextTokens = undefined;
 	}
 
 	restore(entries: readonly SessionEntry[], sessionId?: string): void {
@@ -240,6 +249,7 @@ export class CodexContextWindowManager {
 		mode: "off" | "remote",
 	): AgentMessage[] {
 		if (mode === "off") {
+			this.projectedContextTokens = undefined;
 			return messages.filter(
 				(message) => message.role !== "custom" || message.customType !== CODEX_CONTEXT_WINDOW_MESSAGE_TYPE,
 			);
@@ -266,11 +276,16 @@ export class CodexContextWindowManager {
 			this.restoredMarkerId = undefined;
 			this.budget.reset();
 			this.trimPendingWindowId = undefined;
+			this.projectedContextTokens = undefined;
 		}
 		// Projection observes the request view, never durable session state. A
 		// queued rollover marker must not clear the duplicate guard here.
 		const trimmed = boundaryIndex < 0 ? [...messages] : this.trimToWindow(messages, boundaryIndex);
-		return mode === "remote" ? projectEncryptedToolResults(trimmed) : trimmed;
+		const projected = mode === "remote" ? projectEncryptedToolResults(trimmed) : trimmed;
+		this.projectedContextTokens = mode === "remote" && this.identity
+			? estimateProjectedWindowTokens(projected)
+			: undefined;
+		return projected;
 	}
 
 	/**
@@ -363,7 +378,12 @@ export class CodexContextWindowManager {
 		// once-per-window reminder on a false alarm seconds after a successful
 		// rollover, leaving the window silent for the rest of its life.
 		if (!hasAssistantUsageSinceWindowBoundary(ctx.sessionManager.getBranch(), this.sessionId)) return;
-		const reminder = this.budget.record(ctx, this.identity, contextTokens, contextReminderThresholdPercent);
+		const reminder = this.budget.record(
+			ctx,
+			this.identity,
+			contextTokens ?? this.projectedContextTokens,
+			contextReminderThresholdPercent,
+		);
 		if (!reminder) return;
 		sendContextWindowMessage(
 			pi,
@@ -375,7 +395,7 @@ export class CodexContextWindowManager {
 	}
 
 	remaining(ctx: ExtensionContext, contextTokens?: number): ContextRemaining {
-		return this.budget.remaining(ctx, this.identity, contextTokens);
+		return this.budget.remaining(ctx, this.identity, contextTokens ?? this.projectedContextTokens);
 	}
 
 	/**
@@ -703,6 +723,35 @@ function matchesSession(markerSessionId: string | undefined, sessionId: string |
 	return sessionId === undefined || markerSessionId === sessionId;
 }
 
+function estimateProjectedWindowTokens(messages: readonly AgentMessage[]): number {
+	let lastUsageIndex = -1;
+	let usageTokens = 0;
+	for (let index = messages.length - 1; index >= 0; index -= 1) {
+		const message = messages[index]!;
+		if (message.role !== "assistant" || message.stopReason === "aborted" || message.stopReason === "error" || !message.usage) continue;
+		const tokens = calculateContextTokens(message.usage);
+		if (tokens <= 0) continue;
+		lastUsageIndex = index;
+		usageTokens = tokens;
+		break;
+	}
+	if (lastUsageIndex < 0) return messages.reduce((total, message) => total + estimateProjectedMessageTokens(message), 0);
+	return usageTokens + messages
+		.slice(lastUsageIndex + 1)
+		.reduce((total, message) => total + estimateProjectedMessageTokens(message), 0);
+}
+
+function estimateProjectedMessageTokens(message: AgentMessage): number {
+	// A synthetic custom marker may be represented without content in a test or
+	// by an older host. It contributes no measurable text; it must not make the
+	// context hook fail closed merely because the diagnostic estimate is missing.
+	if (
+		!("content" in message) ||
+		(typeof message.content !== "string" && !Array.isArray(message.content))
+	) return 0;
+	return estimateTokens(message);
+}
+
 function projectEncryptedToolResults(messages: readonly AgentMessage[]): AgentMessage[] {
 	let changed = false;
 	const projected = messages.map((message) => {

package/src/extension-runtime.ts

@@ -876,13 +876,16 @@ async function handleContextInternal(
 		const remoteActive = await remoteContextActive(ctx, config);
 		if (remoteActive) {
 			try {
+				// Measure the provider-visible current window, not Pi's durable transcript.
+				// The latter still contains retired remote windows and can be inflated by
+				// an upstream error even though this request only carries the live window.
+				const projected = contextWindows.project(event.messages, "remote");
 				contextWindows.recordBudget(
 					pi,
 					ctx,
 					true,
 					config.contextReminderThresholdPercent,
 				);
-				const projected = contextWindows.project(event.messages, "remote");
 				reportProjectionDiagnostics(contextWindows, config, ctx);
 				return projected.length === event.messages.length && projected.every((message, index) => message === event.messages[index])
 					? undefined

package/src/responses-headers.ts

@@ -90,11 +90,9 @@ export function buildCodexCliUserAgent(version: string): string {
 }
 
 /**
- * The Codex backend version-gates model availability on the `version` header
- * (the pinned @openai/codex client version). `gpt-6-astra` requires >= 0.153.0;
- * an older or absent pin makes the SKU invisible to discovery and rejects
- * requests that name it. Keep this in sync with the newest gate Oh My Pi pins
- * in @oh-my-pi/pi-wire/codex (`CODEX_CLIENT_VERSION`).
+ * Codex client identity used by synthetic gateway requests in their User-Agent.
+ * Native live and context-management requests do not generate a `version`
+ * header; explicitly configured provider values remain caller-owned.
  */
 export const CODEX_CLIENT_VERSION = "0.153.0";
 
@@ -148,8 +146,6 @@ export function buildResponsesRequestHeaders(
 			headers.set("originator", "pi");
 			headers.set("user-agent", buildCodexUserAgent());
 			headers.set("openai-beta", "responses=experimental");
-			// Preserve the existing native Codex compatibility pin.
-			headers.set("version", CODEX_CLIENT_VERSION);
 		}
 		// Gateways own their version policy. Explicit versions still pass through
 		// the header merge and allowlist, but Toolkit does not generate one.