pi-ui-extend 1.0.6 → 1.0.7

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/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.7",
   "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",