pi-free 2.1.0 → 2.2.0

package/lib/built-in-toggle.ts

@@ -95,7 +95,6 @@ export function setupBuiltInProviderToggles(pi: ExtensionAPI): void {
 		for (const config of activeConfigs) {
 			registerToggleCommand(pi, config);
 		}
-		setupStatusBar(pi, activeConfigs);
 		commandsRegistered = true;
 	}
 
@@ -250,45 +249,6 @@ function modelToProviderConfig(
 	return base;
 }
 
-// =============================================================================
-// Status bar for provider selection
-// =============================================================================
-
-function setupStatusBar(
-	pi: ExtensionAPI,
-	configs: BuiltInToggleConfig[],
-): void {
-	pi.on("model_select", (_event, ctx) => {
-		const selected = _event.model?.provider;
-
-		// Clear status for all fallback-captured built-in providers
-		for (const config of configs) {
-			if (selected !== config.id) {
-				ctx.ui.setStatus(`${config.id}-status`, undefined);
-			}
-		}
-
-		if (!selected) return;
-
-		const state = providerStates.get(selected);
-		if (!state) return;
-
-		const free = state.stored.free.length;
-		const total = state.stored.all.length;
-		const paid = total - free;
-		const mode = state.toggleState.getCurrentMode();
-		let status: string;
-		if (paid === 0) {
-			status = `${selected}: ${free} free models`;
-		} else if (mode === "all") {
-			status = `${selected}: ${total} models (free + paid)`;
-		} else {
-			status = `${selected}: ${free} free \u00b7 ${paid} paid`;
-		}
-		ctx.ui.setStatus(`${selected}-status`, status);
-	});
-}
-
 function getApiKeyEnvForProvider(providerId: string): string {
 	const envMap: Record<string, string> = {
 		opencode: "$OPENCODE_API_KEY",

package/lib/probe-cache.ts

@@ -60,6 +60,14 @@ export function getModelsDueForProbe(
 	});
 }
 
+export function areAllModelsFresh(
+	providerId: string,
+	modelIds: string[],
+	ttlMs = DEFAULT_PROBE_TTL_MS,
+): boolean {
+	return getModelsDueForProbe(providerId, modelIds, ttlMs).length === 0;
+}
+
 export async function recordModelProbeResults(
 	providerId: string,
 	results: ModelProbeResult[],

package/lib/provider-probe.ts

@@ -24,6 +24,7 @@ import type { ProviderModelConfig } from "@earendil-works/pi-coding-agent";
 import { updateConfig } from "../config.ts";
 import { createLogger } from "./logger.ts";
 import {
+	areAllModelsFresh,
 	getModelsDueForProbe,
 	recordModelProbeResults,
 	type ModelProbeResult,
@@ -175,6 +176,20 @@ export function createProviderProbe(
 		return () => {
 			if (done) return;
 			done = true;
+
+			// Skip scheduling entirely if every model was probed recently.
+			// Without this check the handler fires on every session_start and
+			// only then discovers the cache is fresh inside run().
+			if (
+				areAllModelsFresh(
+					providerId,
+					models.map((m) => m.id),
+				)
+			) {
+				_logger.info(`[probe] ${providerId}: auto-probe cache is fresh`);
+				return;
+			}
+
 			_logger.info(`[probe] Starting lazy auto-probe for ${providerId}...`);
 			run(apiKey, models, { useCache: true }).catch((err) => {
 				_logger.warn(`[probe] ${providerId}: auto-probe failed`, {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "pi-free",
-	"version": "2.1.0",
+	"version": "2.2.0",
 	"type": "module",
 	"description": "AI model providers for Pi with free model filtering and dynamic model fetching",
 	"keywords": [
@@ -55,14 +55,14 @@
 		"smoke:cline": "tsx scripts/smoke-cline-xml-bridge.ts"
 	},
 	"peerDependencies": {
-		"@earendil-works/pi-ai": "*",
-		"@earendil-works/pi-coding-agent": "*",
-		"@earendil-works/pi-tui": "*"
+		"@earendil-works/pi-ai": "^0.79.8",
+		"@earendil-works/pi-coding-agent": "^0.79.8",
+		"@earendil-works/pi-tui": "^0.79.8"
 	},
 	"devDependencies": {
 		"@vitest/ui": "^4.1.5",
 		"tsx": "^4.0.0",
-		"typescript": "^6.0.2",
+		"typescript": "^6.0.3",
 		"vitest": "^4.1.5"
 	},
 	"pi": {

package/provider-helper.ts

@@ -222,31 +222,7 @@ export function setupProvider(
 		});
 	}
 
-	// ── Status bar for selected provider ───────────────────────────
-
-	pi.on("model_select", (_event, ctx) => {
-		if (_event.model?.provider !== providerId) {
-			ctx.ui.setStatus(`${providerId}-status`, undefined);
-			return;
-		}
-
-		// Build status line for this provider
-		const free = stored.free.length;
-		const total = stored.all.length || free;
-		const paid = total - free;
-		let status: string;
-
-		if (paid === 0) {
-			status = `${providerId}: ${free} free models`;
-		} else if (currentShowPaid) {
-			status = `${providerId}: ${total} models (free + paid)`;
-		} else {
-			status = `${providerId}: ${free} free · ${paid} paid`;
-		}
-
-		if (config.hasKey) status += " 🔑";
-		ctx.ui.setStatus(`${providerId}-status`, status);
-	});
+	// ── Model count status bar removed (keeps footer clean) ───────
 
 	// ── Error handling / usage tracking are temporarily deprecated ─────────
 

package/providers/bai/bai.ts

@@ -0,0 +1,232 @@
+/**
+ * B.AI Provider Extension
+ *
+ * B.AI (https://b.ai) is an OpenAI-compatible LLM gateway providing access
+ * to many models (OpenAI, Anthropic, Google, DeepSeek, Qwen, GLM, Kimi).
+ *
+ * API: https://api.b.ai/v1
+ * Models: /v1/models
+ * Chat: /v1/chat/completions
+ *
+ * Pricing is not exposed via the /v1/models endpoint, so all models
+ * default to cost=0. The `isFreeModel` Route B detection (name contains
+ * "free") is therefore used. As a result, with `free_only: true` no b.ai
+ * models will be visible until you run `/toggle-bai` to enable paid models.
+ *
+ * A small set of known-promotional models are hardcoded as known-free so
+ * they remain visible even when free-only mode is on (mirrors the
+ * TokenRouter approach for `MiniMax-M3`).
+ *
+ * Setup:
+ *   BAI_API_KEY=sk-...
+ *   # or add bai_api_key to ~/.pi/free.json
+ */
+
+import type {
+	ExtensionAPI,
+	ProviderModelConfig,
+} from "@earendil-works/pi-coding-agent";
+import { getBaiApiKey, getBaiShowPaid, applyHidden } from "../../config.ts";
+import {
+	BASE_URL_BAI,
+	DEFAULT_FETCH_TIMEOUT_MS,
+	PROVIDER_BAI,
+} from "../../constants.ts";
+import { createLogger } from "../../lib/logger.ts";
+import { safeEnrichModelsWithModelsDev } from "../../lib/model-metadata.ts";
+import {
+	getProxyModelCompat,
+	isLikelyReasoningModel,
+} from "../../lib/provider-compat.ts";
+import { isFreeModel, registerWithGlobalToggle } from "../../lib/registry.ts";
+import { cleanModelName, fetchWithRetry } from "../../lib/util.ts";
+import { createReRegister, setupProvider } from "../../provider-helper.ts";
+
+const _logger = createLogger("bai");
+
+// =============================================================================
+// Known Free Models
+// B.AI doesn't expose pricing via /v1/models, so known-free models are
+// hardcoded. The site currently advertises `MiniMax-M3` as a limited-time
+// free promotional model; we hardcode that alias and any future `:free`
+// suffixed IDs (catches dynamic promotional additions).
+// =============================================================================
+
+const BAI_KNOWN_FREE_MODELS = new Set(["minimax-m3", "MiniMax-M3"]);
+
+function isBaiKnownFree(modelId: string): boolean {
+	if (BAI_KNOWN_FREE_MODELS.has(modelId)) return true;
+	// Catch any future `:free` suffixed model the gateway advertises
+	return modelId.toLowerCase().endsWith(":free");
+}
+
+// =============================================================================
+// Types
+// =============================================================================
+
+interface BaiModel {
+	id: string;
+	object?: string;
+	created?: number;
+	owned_by?: string;
+	supported_endpoint_types?: string[];
+}
+
+// =============================================================================
+// Helpers
+// =============================================================================
+
+/** Text-capable chat endpoints (excludes image/video/audio-only types) */
+const CHAT_ENDPOINT_TYPES = new Set([
+	"openai",
+	"openai-response",
+	"anthropic",
+	"anthropic-compatible",
+	"gemini",
+]);
+
+function isTextChatModel(model: BaiModel): boolean {
+	const endpoints = model.supported_endpoint_types ?? [];
+	if (endpoints.length === 0) {
+		// No endpoint info — assume text chat (matches TokenRouter fallback)
+		return true;
+	}
+	return endpoints.some((t) => CHAT_ENDPOINT_TYPES.has(t));
+}
+
+function mapBaiModel(model: BaiModel): ProviderModelConfig & {
+	_pricingKnown?: boolean;
+	_freeKnown?: boolean;
+	_isFree?: boolean;
+} {
+	const name = cleanModelName(model.id);
+	const reasoning = isLikelyReasoningModel({ id: model.id, name });
+	const isKnownFree = isBaiKnownFree(model.id);
+
+	return {
+		id: model.id,
+		name,
+		reasoning,
+		input: ["text"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 128_000,
+		maxTokens: 16_384,
+		compat: getProxyModelCompat({ id: model.id, name }),
+		// Known-free models bypass name-based detection entirely
+		_freeKnown: isKnownFree,
+		_isFree: isKnownFree,
+		// Non-free models signal no pricing data (name-based detection only)
+		_pricingKnown: false,
+	} as ProviderModelConfig & {
+		_pricingKnown?: boolean;
+		_freeKnown?: boolean;
+		_isFree?: boolean;
+	};
+}
+
+// =============================================================================
+// Fetch Models
+// =============================================================================
+
+async function fetchBaiModels(apiKey: string): Promise<ProviderModelConfig[]> {
+	_logger.info("[bai] Fetching models from B.AI API...");
+
+	try {
+		const response = await fetchWithRetry(
+			`${BASE_URL_BAI}/models`,
+			{
+				headers: {
+					Authorization: `Bearer ${apiKey}`,
+					Accept: "application/json",
+					"Content-Type": "application/json",
+				},
+			},
+			3,
+			1000,
+			DEFAULT_FETCH_TIMEOUT_MS,
+		);
+
+		if (!response.ok) {
+			throw new Error(`B.AI API error: ${response.status}`);
+		}
+
+		const json = (await response.json()) as { data?: BaiModel[] };
+		const models = (json.data ?? []).filter(isTextChatModel);
+
+		_logger.info(`[bai] Fetched ${models.length} text chat models`);
+		const enriched = await safeEnrichModelsWithModelsDev(
+			models.map(mapBaiModel),
+			{ providerId: PROVIDER_BAI },
+		);
+		return applyHidden(enriched, PROVIDER_BAI);
+	} catch (error) {
+		_logger.error("[bai] Failed to fetch models", {
+			error: error instanceof Error ? error.message : String(error),
+		});
+		return [];
+	}
+}
+
+// =============================================================================
+// Extension Entry Point
+// =============================================================================
+
+export default async function baiProvider(pi: ExtensionAPI) {
+	const apiKey = getBaiApiKey();
+
+	if (!apiKey) {
+		_logger.info(
+			"[bai] Skipping — BAI_API_KEY not set. Sign up at https://b.ai/",
+		);
+		return;
+	}
+
+	const allModels = await fetchBaiModels(apiKey);
+
+	if (allModels.length === 0) {
+		_logger.warn("[bai] No text chat models available");
+		return;
+	}
+
+	// Use isFreeModel with allModels for proper detection
+	// B.AI doesn't expose pricing, so Route B (name-based) applies:
+	// FREE if name contains "free" OR _isFree is true (known-free hardcoded).
+	const freeModels = allModels.filter((m) =>
+		isFreeModel({ ...m, provider: PROVIDER_BAI }, allModels),
+	);
+	const stored = { free: freeModels, all: allModels };
+
+	_logger.info(
+		`[bai] Registered ${allModels.length} models (${freeModels.length} free)`,
+	);
+
+	const reRegister = createReRegister(pi, {
+		providerId: PROVIDER_BAI,
+		baseUrl: BASE_URL_BAI,
+		apiKey,
+	});
+
+	registerWithGlobalToggle(PROVIDER_BAI, stored, reRegister, true);
+
+	setupProvider(
+		pi,
+		{
+			providerId: PROVIDER_BAI,
+			initialShowPaid: getBaiShowPaid(),
+			tosUrl: "https://b.ai/",
+			reRegister: (models, _stored) => {
+				if (_stored) {
+					stored.free = _stored.free;
+					stored.all = _stored.all;
+				}
+				reRegister(models);
+			},
+		},
+		stored,
+	);
+
+	const showPaid = getBaiShowPaid();
+	const initialModels =
+		showPaid && stored.all.length > 0 ? stored.all : freeModels;
+	reRegister(initialModels);
+}