pi-free 2.1.0 → 2.2.0

package/providers/cline/cline.ts

@@ -163,29 +163,6 @@ export default async function clineProvider(pi: ExtensionAPI) {
 		},
 	});
 
-	// ── Status bar for provider selection ─────────────────────────
-
-	pi.on("model_select", (_event, ctx) => {
-		if (_event.model?.provider !== PROVIDER_CLINE) {
-			ctx.ui.setStatus(`${PROVIDER_CLINE}-status`, undefined);
-			return;
-		}
-
-		const free = stored.free.length;
-		const total = stored.all.length;
-		const paid = total - free;
-		const mode = toggleState.getCurrentMode();
-		let status: string;
-		if (paid === 0) {
-			status = `cline: ${free} free models`;
-		} else if (mode === "all") {
-			status = `cline: ${total} models (free + paid)`;
-		} else {
-			status = `cline: ${free} free \u00b7 ${paid} paid`;
-		}
-		ctx.ui.setStatus(`${PROVIDER_CLINE}-status`, status);
-	});
-
 	pi.on("before_agent_start", (_event, ctx) => {
 		if (ctx.model?.provider !== PROVIDER_CLINE) return;
 		_currentTaskId = generateUlid();

package/providers/codestral/codestral.ts

@@ -125,15 +125,4 @@ export default async function codestralProvider(pi: ExtensionAPI) {
 
 	_logger.info(`[codestral] Registered codestral-latest via ${keySource}`);
 
-	// Status bar
-	pi.on("model_select", (_event, ctx) => {
-		if (_event.model?.provider !== PROVIDER_CODESTRAL) {
-			ctx.ui.setStatus(`${PROVIDER_CODESTRAL}-status`, undefined);
-			return;
-		}
-		ctx.ui.setStatus(
-			`${PROVIDER_CODESTRAL}-status`,
-			`codestral: 1 model (free tier) 🔑`,
-		);
-	});
 }

package/providers/dynamic-built-in/index.ts

@@ -45,6 +45,7 @@ import { createLogger } from "../../lib/logger.ts";
 import { safeEnrichModelsWithModelsDev } from "../../lib/model-metadata.ts";
 import { getProxyModelCompat } from "../../lib/provider-compat.ts";
 import {
+	areAllModelsFresh,
 	getModelsDueForProbe,
 	recordModelProbeResults,
 } from "../../lib/probe-cache.ts";
@@ -563,26 +564,6 @@ async function registerProvider(
 		true,
 	);
 
-	// Status bar
-	const pid = config.providerId;
-	pi.on("model_select", (_event, ctx) => {
-		if (_event.model?.provider !== pid) {
-			ctx.ui.setStatus(`${pid}-status`, undefined);
-			return;
-		}
-		const f = freeModels.length;
-		const t = allModels.length;
-		const p = t - f;
-		const mode = toggleState.getCurrentMode();
-		const status =
-			p === 0
-				? `${pid}: ${f} free models`
-				: mode === "all"
-					? `${pid}: ${t} models (free + paid)`
-					: `${pid}: ${f} free \u00b7 ${p} paid`;
-		ctx.ui.setStatus(`${pid}-status`, `${status} 🔑`);
-	});
-
 	// Register models (this swaps in our discovered models over Pi's defaults)
 	toggleState.applyCurrent(reRegister);
 
@@ -625,6 +606,17 @@ async function registerProvider(
 			wrapSessionStartHandler(`${config.providerId}-auto-probe`, async () => {
 				if (_autoProbeDone) return;
 				_autoProbeDone = true;
+				if (
+					areAllModelsFresh(
+						config.providerId,
+						stored.free.map((m) => m.id),
+					)
+				) {
+					_logger.info(
+						`[probe] ${config.providerId}: auto-probe cache is fresh`,
+					);
+					return;
+				}
 				_logger.info(
 					`Starting lazy auto-probe of ${config.providerId} free models...`,
 				);

package/providers/kilo/kilo.ts

@@ -326,27 +326,10 @@ export default async function kiloProvider(pi: ExtensionAPI) {
 		},
 	});
 
-	// Status bar + ToS notice on provider selection
+	// ToS notice on provider selection
 	let tosShown = false;
 	pi.on("model_select", async (_event, ctx) => {
-		if (ctx.model?.provider !== PROVIDER_KILO) {
-			ctx.ui.setStatus(`${PROVIDER_KILO}-status`, undefined);
-			return;
-		}
-
-		// Build status line
-		const free = freeModels.length;
-		const total = allModels.length;
-		const paid = total - free;
-		let status: string;
-		if (paid === 0) {
-			status = `kilo: ${free} free models`;
-		} else if (showPaidModels) {
-			status = `kilo: ${total} models (free + paid)`;
-		} else {
-			status = `kilo: ${free} free \u00b7 ${paid} paid`;
-		}
-		ctx.ui.setStatus(`${PROVIDER_KILO}-status`, status);
+		if (ctx.model?.provider !== PROVIDER_KILO) return;
 
 		// ToS notice (once)
 		if (tosShown) return;

package/providers/ollama/ollama.ts

@@ -43,6 +43,7 @@ import {
 } from "../../lib/provider-cache.ts";
 import { wrapSessionStartHandler } from "../../lib/session-start-metrics.ts";
 import {
+	areAllModelsFresh,
 	getModelsDueForProbe,
 	recordModelProbeResults,
 } from "../../lib/probe-cache.ts";
@@ -591,20 +592,19 @@ export default async function ollamaProvider(pi: ExtensionAPI) {
 		},
 	});
 
-	// ── Status bar for provider selection ───────────────────────────
-
-	// eslint-disable-next-line @typescript-eslint/no-explicit-any
-	pi.on("model_select" as any, (_event: any, ctx: any) => {
-		if (_event.model?.provider !== PROVIDER_OLLAMA) {
-			ctx.ui.setStatus(`${PROVIDER_OLLAMA}-status`, undefined);
+	const runProbeInBackground = (models: ProviderModelConfig[]) => {
+		// Skip scheduling entirely if every model was probed recently.
+		// Without this check the probe runs on every session_start and
+		// only then discovers the cache is fresh inside runOllamaProbe.
+		if (
+			areAllModelsFresh(
+				PROVIDER_OLLAMA,
+				models.map((m) => m.id),
+			)
+		) {
+			_logger.info("Auto-probe: Ollama probe cache is fresh");
 			return;
 		}
-
-		const count = allModels.length;
-		ctx.ui.setStatus(`${PROVIDER_OLLAMA}-status`, `ollama: ${count} models`);
-	});
-
-	const runProbeInBackground = (models: ProviderModelConfig[]) => {
 		runOllamaProbe(apiKey, models, applyModelList, { useCache: true }).catch(
 			(error) => {
 				_logger.warn("Auto-probe failed", {

package/providers/routeway/routeway.ts

@@ -37,6 +37,7 @@ import {
 	isLikelyReasoningModel,
 } from "../../lib/provider-compat.ts";
 import {
+	areAllModelsFresh,
 	getModelsDueForProbe,
 	recordModelProbeResults,
 } from "../../lib/probe-cache.ts";
@@ -350,6 +351,15 @@ export default async function routewayProvider(pi: ExtensionAPI) {
 		wrapSessionStartHandler("routeway", async () => {
 			if (_autoProbeDone || !apiKey) return;
 			_autoProbeDone = true;
+			if (
+				areAllModelsFresh(
+					PROVIDER_ROUTEWAY,
+					allModels.map((m) => m.id),
+				)
+			) {
+				_logger.info("Auto-probe: Routeway probe cache is fresh");
+				return;
+			}
 			_logger.info("Starting lazy auto-probe of Routeway models...");
 			runRoutewayProbe(apiKey, allModels, stored, reRegister, {
 				useCache: true,