pi-ui-extend 1.0.6 → 1.0.8

package/dist/bundled-extensions/session-title/title-generation-compat.d.ts

@@ -1,17 +1,6 @@
-import type { Api, Model } from "@earendil-works/pi-ai";
+import type { ModelRegistry } from "@earendil-works/pi-coding-agent";
 import type { SessionTitleConfig } from "./config.js";
-type TitleModelRegistry = {
-    find(provider: string, modelId: string): Model<Api> | undefined;
-    getApiKeyAndHeaders(model: Model<Api>): Promise<{
-        ok: true;
-        apiKey?: string;
-        headers?: Record<string, string>;
-        env?: Record<string, string>;
-    } | {
-        ok: false;
-        error: string;
-    }>;
-};
+type TitleModelRegistry = Pick<ModelRegistry, "complete" | "find">;
 /** Extension-side title generation through Pi's public ModelRegistry facade. */
 export declare function generateSessionTitle(input: string, modelRegistry: TitleModelRegistry, config: SessionTitleConfig, modelRef: string, signal: AbortSignal, onWarning?: (message: string) => void): Promise<string | undefined>;
 export {};

package/dist/bundled-extensions/session-title/title-generation-compat.js

@@ -1,4 +1,3 @@
-import { complete } from "@earendil-works/pi-ai/compat";
 import { buildTitlePrompt, parseTitleModelRef, sanitizeSessionTitle, TITLE_SYSTEM_PROMPT, titleResponseText, } from "./title-generation.js";
 /** Extension-side title generation through Pi's public ModelRegistry facade. */
 export async function generateSessionTitle(input, modelRegistry, config, modelRef, signal, onWarning) {
@@ -12,12 +11,7 @@ export async function generateSessionTitle(input, modelRegistry, config, modelRe
         onWarning?.(`Session-title model not found: ${modelRef}`);
         return undefined;
     }
-    const auth = await modelRegistry.getApiKeyAndHeaders(model);
-    if (auth.ok === false) {
-        onWarning?.(auth.error);
-        return undefined;
-    }
-    const response = await complete(model, {
+    const response = await modelRegistry.complete(model, {
         systemPrompt: TITLE_SYSTEM_PROMPT,
         messages: [
             {
@@ -27,9 +21,6 @@ export async function generateSessionTitle(input, modelRegistry, config, modelRe
             },
         ],
     }, {
-        ...(auth.apiKey === undefined ? {} : { apiKey: auth.apiKey }),
-        ...(auth.headers === undefined ? {} : { headers: auth.headers }),
-        ...(auth.env === undefined ? {} : { env: auth.env }),
         cacheRetention: "none",
         maxRetries: config.maxRetries,
         maxTokens: config.maxTokens,

package/external/pi-tools-suite/package.json

@@ -43,9 +43,9 @@
     "vscode-languageserver-protocol": "^3.17.5"
   },
   "peerDependencies": {
-    "@earendil-works/pi-ai": "0.83.0",
-    "@earendil-works/pi-coding-agent": "0.83.0",
-    "@earendil-works/pi-tui": "0.83.0",
+    "@earendil-works/pi-ai": "0.84.1",
+    "@earendil-works/pi-coding-agent": "0.84.1",
+    "@earendil-works/pi-tui": "0.84.1",
     "typebox": "*"
   },
   "devDependencies": {

package/external/pi-tools-suite/src/antigravity-auth/oauth.ts

@@ -221,7 +221,11 @@ export async function addAntigravityAccount(
 	};
 }
 
-async function refreshAccountToken(account: OpencodeAntigravityAccount, oauthClient?: GoogleOAuthClientCredentials): Promise<RefreshedAntigravityAccount> {
+async function refreshAccountToken(
+	account: OpencodeAntigravityAccount,
+	oauthClient?: GoogleOAuthClientCredentials,
+	signal?: AbortSignal,
+): Promise<RefreshedAntigravityAccount> {
 	const refreshToken = getAccountRefreshToken(account);
 	if (!refreshToken) throw new Error(`Missing refresh token for Antigravity account ${account.email ?? "<unknown>"}`);
 	const clientCredentials = getGoogleOAuthClientCredentials(account, oauthClient);
@@ -230,6 +234,7 @@ async function refreshAccountToken(account: OpencodeAntigravityAccount, oauthCli
 	const start = Date.now();
 	const response = await fetch("https://oauth2.googleapis.com/token", {
 		method: "POST",
+		signal,
 		headers: {
 			"Content-Type": "application/x-www-form-urlencoded;charset=UTF-8",
 			Accept: "*/*",
@@ -256,7 +261,7 @@ async function refreshAccountToken(account: OpencodeAntigravityAccount, oauthCli
 	};
 }
 
-export async function refreshAntigravityToken(credentials: OAuthCredentials): Promise<OAuthCredentials> {
+export async function refreshAntigravityToken(credentials: OAuthCredentials, signal: AbortSignal): Promise<OAuthCredentials> {
 	const credentialDetails = credentials as OAuthCredentials & PiAuthCredential;
 	const oauthClient = getGoogleOAuthClientCredentials(credentialDetails);
 	const storedAccounts = getStoredAccounts(credentialDetails);
@@ -272,6 +277,7 @@ export async function refreshAntigravityToken(credentials: OAuthCredentials): Pr
 			email: credentialDetails.email,
 		},
 		oauthClient,
+		signal,
 	);
 	return {
 		...refreshed.credentials,

package/external/pi-tools-suite/src/antigravity-auth/stream.ts

@@ -68,17 +68,21 @@ async function sendAntigravityRequest(
 	const headerStyle = getModelHeaderStyle(model);
 	const endpoints = headerStyle === "gemini-cli" ? [ENDPOINT_PROD] : STREAM_ENDPOINTS;
 	for (const endpoint of endpoints) {
+		const headers = new Headers({
+			Authorization: `Bearer ${apiKey}`,
+			"Content-Type": "application/json",
+			Accept: "text/event-stream",
+			...getAntigravityHeaders(headerStyle),
+			...requestHeaders,
+		});
+		for (const [name, value] of Object.entries(options?.headers ?? {})) {
+			if (value === null) headers.delete(name);
+			else headers.set(name, value);
+		}
 		response = await fetch(`${endpoint}/v1internal:streamGenerateContent?alt=sse`, {
 			method: "POST",
 			signal: options?.signal,
-			headers: {
-				Authorization: `Bearer ${apiKey}`,
-				"Content-Type": "application/json",
-				Accept: "text/event-stream",
-				...getAntigravityHeaders(headerStyle),
-				...requestHeaders,
-				...(options?.headers ?? {}),
-			},
+			headers,
 			body: JSON.stringify(payload),
 		});
 		await options?.onResponse?.({ status: response.status, headers: Object.fromEntries(response.headers.entries()) }, model);

package/external/pi-tools-suite/src/async-subagents/core/routing.ts

@@ -1,4 +1,4 @@
-import type { Api, Model } from "@earendil-works/pi-ai";
+import type { Api, Model, ProviderHeaders } from "@earendil-works/pi-ai";
 import { completeWithModelRegistry, type ModelCompletionRegistry } from "../../model-completion.js";
 import type { AgentTask } from "./types.js";
 import {
@@ -14,7 +14,7 @@ export interface SubagentRoutingContext {
 	modelRegistry?: ModelCompletionRegistry & {
 		find(provider: string, modelId: string): Model<Api> | undefined;
 		getApiKeyAndHeaders(model: Model<Api>): Promise<
-			| { ok?: true; apiKey?: string; headers?: Record<string, string>; env?: Record<string, string> }
+			| { ok?: true; apiKey?: string; headers?: ProviderHeaders; baseUrl?: string; env?: Record<string, string> }
 			| { ok: false; error: string }
 		>;
 	};
@@ -180,7 +180,7 @@ async function resolveRoutingModels(
 interface RoutingCandidate {
 	model: Model<Api>;
 	apiKey?: string;
-	headers?: Record<string, string>;
+	headers?: ProviderHeaders;
 	env?: Record<string, string>;
 }
 
@@ -189,7 +189,7 @@ type RoutingResponse = Awaited<ReturnType<typeof completeWithModelRegistry>>;
 async function resolveModelRef(ctx: SubagentRoutingContext, modelRef: string): Promise<{
 	model: Model<Api>;
 	apiKey?: string;
-	headers?: Record<string, string>;
+	headers?: ProviderHeaders;
 	env?: Record<string, string>;
 } | undefined> {
 	const parsed = parseModelRef(modelRef);

package/external/pi-tools-suite/src/async-subagents/core/ultrawork-auto.ts

@@ -1,4 +1,4 @@
-import type { Api, Model } from "@earendil-works/pi-ai";
+import type { Api, Model, ProviderHeaders } from "@earendil-works/pi-ai";
 import { completeWithModelRegistry, type ModelCompletionRegistry } from "../../model-completion.js";
 import { currentModelRef, resolveSubagentRoutingConfig, type SubagentConfig } from "./config.js";
 
@@ -9,7 +9,7 @@ export interface UltraworkAutoContext {
 	modelRegistry?: ModelCompletionRegistry & {
 		find(provider: string, modelId: string): Model<Api> | undefined;
 		getApiKeyAndHeaders(model: Model<Api>): Promise<
-			| { ok?: true; apiKey?: string; headers?: Record<string, string>; env?: Record<string, string> }
+			| { ok?: true; apiKey?: string; headers?: ProviderHeaders; baseUrl?: string; env?: Record<string, string> }
 			| { ok: false; error: string }
 		>;
 	};
@@ -131,7 +131,7 @@ function buildClassifierPrompt(userText: string): string {
 async function resolveClassifierModel(ctx: UltraworkAutoContext, modelRef: string): Promise<{
 	model: Model<Api>;
 	apiKey?: string;
-	headers?: Record<string, string>;
+	headers?: ProviderHeaders;
 	env?: Record<string, string>;
 } | undefined> {
 	const configured = await resolveModelRef(ctx, modelRef);
@@ -143,7 +143,7 @@ async function resolveClassifierModel(ctx: UltraworkAutoContext, modelRef: strin
 async function resolveModelRef(ctx: UltraworkAutoContext, modelRef: string): Promise<{
 	model: Model<Api>;
 	apiKey?: string;
-	headers?: Record<string, string>;
+	headers?: ProviderHeaders;
 	env?: Record<string, string>;
 } | undefined> {
 	const parsed = parseModelRef(modelRef);

package/external/pi-tools-suite/src/coding-discipline/index.ts

@@ -1,6 +1,6 @@
 import * as fs from "node:fs";
 import * as path from "node:path";
-import type { Api, AssistantMessage, ImageContent, Model, TextContent } from "@earendil-works/pi-ai";
+import type { Api, AssistantMessage, ImageContent, Model, ProviderHeaders, TextContent } from "@earendil-works/pi-ai";
 import { Type } from "typebox";
 
 import { loadPiToolsSuiteConfig, DEFAULT_CODING_DISCIPLINE_STRICTNESS, type CodingDisciplineStrictness } from "../config.js";
@@ -28,7 +28,7 @@ type ResolvedLookupModel = {
 	modelRegistry: ModelCompletionRegistry;
 	model: Model<Api>;
 	apiKey?: string;
-	headers?: Record<string, string>;
+	headers?: ProviderHeaders;
 	env?: Record<string, string>;
 };
 
@@ -825,7 +825,8 @@ async function resolveLookupModel(ctx: unknown, modelRef: string): Promise<Resol
 	const auth = await registry.getApiKeyAndHeaders(model) as {
 		ok?: true;
 		apiKey?: string;
-		headers?: Record<string, string>;
+		headers?: ProviderHeaders;
+		baseUrl?: string;
 		env?: Record<string, string>;
 	} | { ok: false; error: string };
 	if (auth.ok === false) return undefined;

package/external/pi-tools-suite/src/dcp/auto-compress.ts

@@ -13,7 +13,7 @@
 // automatic fallback to the programmatic digest on any failure/timeout.
 // ---------------------------------------------------------------------------
 
-import type { Model, Api } from "@earendil-works/pi-ai"
+import type { Model, Api, ProviderHeaders } from "@earendil-works/pi-ai"
 import { completeWithModelRegistry, type ModelCompletionRegistry } from "../model-completion.js"
 import type { DcpState } from "./state.js"
 import type { DcpConfig } from "./config.js"
@@ -137,7 +137,7 @@ export interface ModelSummaryResult {
 type ModelSummaryRegistry = ModelCompletionRegistry & {
 	find(provider: string, modelId: string): Model<Api> | undefined
 	getApiKeyAndHeaders(model: Model<Api>): Promise<
-		| { ok: true; apiKey?: string; headers?: Record<string, string>; env?: Record<string, string> }
+		| { ok: true; apiKey?: string; headers?: ProviderHeaders; baseUrl?: string; env?: Record<string, string> }
 		| { ok: false; error: string }
 	>
 }

package/external/pi-tools-suite/src/dcp/index.ts

@@ -809,6 +809,7 @@ export default async function dcpModule(pi: ExtensionAPI): Promise<void> {
 			await saveDcpState(ctx, state)
 		}
 
+		const anchorsBeforeFinalization = state.nudgeAnchors.length
 		if (state.manualMode) {
 			state.nudgeAnchors = state.nudgeAnchors.filter((anchor) =>
 				anchor.type === "context-strong" || anchor.type === "context-soft",
@@ -817,6 +818,9 @@ export default async function dcpModule(pi: ExtensionAPI): Promise<void> {
 		applyAnchoredNudges(prunedMessages, state, (anchor) =>
 			appendConcreteNudgeGuidance(baseNudgeText(anchor.type), candidate, messageCandidates, state),
 		)
+		if (state.nudgeAnchors.length !== anchorsBeforeFinalization) {
+			await saveDcpState(ctx, state)
+		}
 
 		return finishContext("complete", prunedMessages, {
 			candidate,

package/external/pi-tools-suite/src/dcp/pruner-nudge.ts

@@ -51,7 +51,6 @@ function isRealAnchorCandidate(msg: any): boolean {
   const role = msg?.role ?? "";
   if (role !== "user" && role !== "assistant") return false;
   const text = messageText(msg);
-  if (text.includes("<dcp-system-reminder>")) return false;
   if (extractBlockId(text) !== undefined) return false;
   return true;
 }
@@ -94,6 +93,12 @@ function anchorMatchesMessage(anchor: DcpNudgeAnchor, msg: any, index: number):
   return msg?.timestamp === anchor.anchorTimestamp;
 }
 
+function isNewerAnchor(candidate: DcpNudgeAnchor, current: DcpNudgeAnchor): boolean {
+  if (candidate.updatedAt !== current.updatedAt) return candidate.updatedAt > current.updatedAt;
+  if (candidate.createdAt !== current.createdAt) return candidate.createdAt > current.createdAt;
+  return candidate.id > current.id;
+}
+
 function appendTextToMessage(msg: any, text: string): void {
   const suffix = `\n\n${text}`;
   if (typeof msg.content === "string") {
@@ -205,15 +210,27 @@ export function upsertNudgeAnchor(
   options: { contextPercent?: number } = {},
 ): { anchor: DcpNudgeAnchor | null; created: boolean; updated: boolean } {
   const target = findAnchorMessage(messages);
-  if (!target) return { anchor: null, created: false, updated: false };
+  if (!target) {
+    // The caller will append one synthetic tail reminder. Drop persisted
+    // anchors first so applyAnchoredNudges cannot render a second reminder.
+    state.nudgeAnchors = [];
+    state.lastNudge = undefined;
+    return { anchor: null, created: false, updated: false };
+  }
 
   const key = `${target.stableId}|${target.timestamp}`;
-  const existing = state.nudgeAnchors.find(
-    (anchor) => `${anchor.anchorStableId ?? ""}|${anchor.anchorTimestamp}` === key,
-  );
+  let existing: DcpNudgeAnchor | null = null;
+  for (const anchor of state.nudgeAnchors) {
+    const anchorKey = `${anchor.anchorStableId ?? ""}|${anchor.anchorTimestamp}`;
+    if (anchorKey !== key) continue;
+    if (!existing || isNewerAnchor(anchor, existing)) existing = anchor;
+  }
 
   const now = Date.now();
   if (existing) {
+    // Older sidecars may contain several anchors. Keep only the anchor for the
+    // current target so every subsequent context pass has singleton state.
+    state.nudgeAnchors = [existing];
     const shouldUpgrade = typePriority(type) > typePriority(existing.type);
     if (shouldUpgrade) existing.type = type;
     existing.updatedAt = now;
@@ -240,7 +257,9 @@ export function upsertNudgeAnchor(
     createdAt: now,
     updatedAt: now,
   };
-  state.nudgeAnchors.push(anchor);
+  // A nudge follows the latest useful message. Replacing the previous anchor
+  // prevents one full reminder from accumulating per turn/assistant response.
+  state.nudgeAnchors = [anchor];
   state.lastNudge = {
     type,
     anchorId: anchor.id,
@@ -259,15 +278,19 @@ export function applyAnchoredNudges(
 ): void {
   if (state.nudgeAnchors.length === 0) return;
 
-  const activeAnchors: DcpNudgeAnchor[] = [];
+  let selected: { anchor: DcpNudgeAnchor; index: number } | null = null;
   for (const anchor of state.nudgeAnchors) {
     const index = messages.findIndex((msg, messageIndex) => anchorMatchesMessage(anchor, msg, messageIndex));
     if (index === -1) continue;
-    activeAnchors.push(anchor);
-    appendTextToMessage(messages[index], render(anchor));
+    if (!selected || isNewerAnchor(anchor, selected.anchor)) {
+      selected = { anchor, index };
+    }
   }
 
-  state.nudgeAnchors = activeAnchors;
+  // Defensive migration for persisted pre-singleton state: render only the
+  // newest valid anchor and discard every stale predecessor.
+  state.nudgeAnchors = selected ? [selected.anchor] : [];
+  if (selected) appendTextToMessage(messages[selected.index], render(selected.anchor));
 }
 
 export function clearDcpNudgeAnchors(state: DcpState): number {

package/external/pi-tools-suite/src/model-completion.ts

@@ -1,37 +1,31 @@
 import type {
 	Api,
 	AssistantMessage,
-	AssistantMessageEventStream,
 	Context,
 	Model,
+	ModelsApiStreamOptions,
 	SimpleStreamOptions,
 } from "@earendil-works/pi-ai";
 import { completeSimple } from "@earendil-works/pi-ai/compat";
+import type { ModelRegistry } from "@earendil-works/pi-coding-agent";
 
-type RegisteredProviderConfig = {
-	streamSimple?: (
-		model: Model<Api>,
-		context: Context,
-		options?: SimpleStreamOptions,
-	) => AssistantMessageEventStream;
-};
-
-export type ModelCompletionRegistry = {
-	getRegisteredProviderConfig?(providerId: string): RegisteredProviderConfig | undefined;
-};
+export type ModelCompletionRegistry = Partial<Pick<ModelRegistry, "complete" | "getRegisteredProviderConfig">>;
 
 /**
- * Complete through an extension provider's registered stream when available.
- * Pi 0.80.8+ no longer copies extension streams into pi-ai's global compat
- * registry, so falling back to compat is valid only for built-in APIs.
+ * Complete through Pi's model runtime when available so custom providers and
+ * resolved auth endpoints are preserved. The older stream/compat branches are
+ * retained for narrow test doubles that do not expose ModelRegistry.complete().
  */
 export async function completeWithModelRegistry(
 	modelRegistry: ModelCompletionRegistry | undefined,
 	model: Model<Api>,
 	context: Context,
-	options?: SimpleStreamOptions,
+	options?: ModelsApiStreamOptions<Api>,
 ): Promise<AssistantMessage> {
+	if (modelRegistry?.complete) return modelRegistry.complete(model, context, options);
+
 	const providerConfig = modelRegistry?.getRegisteredProviderConfig?.(model.provider);
-	if (providerConfig?.streamSimple) return providerConfig.streamSimple(model, context, options).result();
-	return completeSimple(model, context, options);
+	const simpleOptions = options as SimpleStreamOptions | undefined;
+	if (providerConfig?.streamSimple) return providerConfig.streamSimple(model, context, simpleOptions).result();
+	return completeSimple(model, context, simpleOptions);
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-ui-extend",
-  "version": "1.0.6",
+  "version": "1.0.8",
   "description": "Pix: a workspace-first terminal UI for Pi with tabs, readable tool activity, voice input, and bundled agent tools.",
   "private": false,
   "repository": {
@@ -75,9 +75,9 @@
     "prepublishOnly": "npm run check && npm run build:pix && npm run generate-schemas"
   },
   "dependencies": {
-    "@earendil-works/pi-ai": "0.83.0",
-    "@earendil-works/pi-coding-agent": "0.83.0",
-    "@earendil-works/pi-tui": "0.83.0",
+    "@earendil-works/pi-ai": "0.84.1",
+    "@earendil-works/pi-coding-agent": "0.84.1",
+    "@earendil-works/pi-tui": "0.84.1",
     "@mariozechner/clipboard": "^0.3.9",
     "jsonc-parser": "3.3.1",
     "typebox": "1.1.38",