pi-free 2.0.9 → 2.0.10

package/providers/ollama/ollama.ts

@@ -1,295 +1,610 @@
-/**
- * Ollama Cloud Provider Extension
- *
- * Provides access to Ollama's cloud-hosted models via ollama.com API.
- * All models use Ollama's usage-based pricing system:
- *   - Free tier: Unlimited public models (session limits reset every 5 hours,
- *     weekly limits reset every 7 days)
- *   - Pro tier: 50x more cloud usage than Free
- *   - Max tier: 5x more usage than Pro
- *
- * Requires OLLAMA_API_KEY with cloud access.
- * Get a free key at: https://ollama.com/settings/keys
- *
- * Responds to global free-only filter (shows models but warns they're freemium).
- *
- * Usage:
- *   pi install git:github.com/apmantza/pi-free
- *   # Set OLLAMA_API_KEY env var
- *   # Models appear in /model selector
- *   # Use /toggle-ollama to show all vs limited set
- */
-
-import type {
-	ExtensionAPI,
-	ProviderModelConfig,
-} from "@earendil-works/pi-coding-agent";
-import {
-	applyHidden,
-	getOllamaApiKey,
-	getOllamaShowPaid,
-	loadConfigFile,
-	saveConfig,
-} from "../../config.ts";
-import {
-	BASE_URL_OLLAMA,
-	DEFAULT_FETCH_TIMEOUT_MS,
-	PROVIDER_OLLAMA,
-} from "../../constants.ts";
-import { createLogger } from "../../lib/logger.ts";
-import { registerWithGlobalToggle } from "../../lib/registry.ts";
-import { fetchWithRetry, fetchWithTimeout } from "../../lib/util.ts";
-import { createReRegister, enhanceWithCI } from "../../provider-helper.ts";
-
-const _logger = createLogger("ollama-cloud");
-
-// =============================================================================
-// Known 403 models (listed but return "access denied" on /v1/chat/completions)
-// These are models that appear in /v1/models but aren't provisioned for chat.
-// Add new IDs here as they surface via /probe-ollama command.
-// =============================================================================
-const OLLAMA_KNOWN_403_MODELS: ReadonlySet<string> = new Set([
-	// Example entries - populate via probe-ollama.mjs
-	// "model-id-that-403s",
-]);
-
-// =============================================================================
-// Fetch + map
-// =============================================================================
-
-async function fetchOllamaModels(
-	apiKey: string,
-): Promise<ProviderModelConfig[]> {
-	// Use OpenAI-compatible /v1/models endpoint for consistency
-	// The native /api/tags returns :cloud suffixes that may not work with /v1/chat/completions
-	const response = await fetchWithRetry(
-		`${BASE_URL_OLLAMA}/models`,
-		{
-			headers: {
-				Authorization: `Bearer ${apiKey}`,
-				"Content-Type": "application/json",
-			},
-		},
-		3,
-		1000,
-		DEFAULT_FETCH_TIMEOUT_MS,
-	);
-
-	if (!response.ok) {
-		throw new Error(
-			`Failed to fetch Ollama models: ${response.status} ${response.statusText}`,
-		);
-	}
-
-	const json = (await response.json()) as {
-		data?: Array<{ id: string; owned_by?: string }>;
-	};
-	const models = json.data ?? [];
-
-	_logger.info(
-		`[ollama-cloud] Fetched ${models.length} models from Ollama Cloud`,
-	);
-
-	// Filter to chat/text generation models only
-	const chatModels = models
-		.filter((m) => {
-			// Skip embedding-only models (typically have "embed" in name)
-			const name = m.id.toLowerCase();
-			if (name.includes("embed")) return false;
-			return true;
-		})
-		// Filter out known 403 models (listed but not provisioned for chat)
-		.filter((m) => {
-			if (OLLAMA_KNOWN_403_MODELS.has(m.id)) {
-				return false;
-			}
-			return true;
-		});
-
-	const result = applyHidden(
-		chatModels.map(
-			(m): ProviderModelConfig => ({
-				id: m.id,
-				name: m.id,
-				// Try to infer reasoning from model name
-				reasoning:
-					m.id.toLowerCase().includes("reasoning") ||
-					m.id.toLowerCase().includes("r1") ||
-					m.id.toLowerCase().includes("thinking"),
-				input: ["text"],
-				// Ollama Cloud uses usage-based pricing (GPU time), not per-token
-				// Free tier has limits but no direct cost per token
-				cost: {
-					input: 0, // Freemium: usage-based, not per-token
-					output: 0,
-					cacheRead: 0,
-					cacheWrite: 0,
-				},
-				// Default context window - Ollama doesn't expose this via /v1/models
-				contextWindow: 32768,
-				maxTokens: 4096, // Default, varies by model
-			}),
-		),
-		PROVIDER_OLLAMA,
-	);
-
-	return result;
-}
-
-// =============================================================================
-// Extension Entry Point
-// =============================================================================
-
-export default async function ollamaProvider(pi: ExtensionAPI) {
-	const apiKey = getOllamaApiKey();
-
-	if (!apiKey) {
-		_logger.info(
-			"[ollama-cloud] Skipping - OLLAMA_API_KEY not set (env var or ~/.pi/free.json)",
-		);
-		return;
-	}
-
-	// Fetch models
-	let allModels: ProviderModelConfig[] = [];
-
-	try {
-		allModels = await fetchOllamaModels(apiKey);
-	} catch (error) {
-		_logger.error("[ollama-cloud] Failed to fetch models at startup", {
-			error: error instanceof Error ? error.message : String(error),
-		});
-		return;
-	}
-
-	// For Ollama, all models share the same free tier
-	// So "free" and "all" are the same set
-	const freeModels = allModels;
-	const stored = { free: freeModels, all: allModels };
-	const hasKey = true;
-
-	// Create re-register function
-	const reRegister = createReRegister(pi, {
-		providerId: PROVIDER_OLLAMA,
-		baseUrl: BASE_URL_OLLAMA,
-		apiKey,
-	});
-
-	// Register with global toggle system
-	registerWithGlobalToggle(PROVIDER_OLLAMA, stored, reRegister, hasKey);
-
-	// Register initial models
-	const initialModels = getOllamaShowPaid() ? allModels : freeModels;
-	pi.registerProvider(PROVIDER_OLLAMA, {
-		baseUrl: BASE_URL_OLLAMA,
-		apiKey,
-		api: "openai-completions" as const,
-		models: enhanceWithCI(initialModels),
-	});
-
-	_logger.info(
-		`[ollama-cloud] Registered ${initialModels.length} models (usage-based free tier)`,
-	);
-
-	// ── Probe command: test all registered models for 403s ─────────────
-	pi.registerCommand("probe-ollama", {
-		description: "Test all Ollama Cloud models for 403 'access denied' errors",
-		handler: async (_args, ctx) => {
-			if (!apiKey) {
-				ctx.ui.notify("OLLAMA_API_KEY not set", "error");
-				return;
-			}
-
-			const modelsToTest = allModels;
-			ctx.ui.notify(`Probing ${modelsToTest.length} Ollama models…`, "info");
-
-			const notFound: string[] = [];
-			const batchSize = 5;
-
-			for (let i = 0; i < modelsToTest.length; i += batchSize) {
-				const batch = modelsToTest.slice(i, i + batchSize);
-				const results = await Promise.all(
-					batch.map(async (m) => {
-						const ok = await probeOllamaModel(apiKey, m.id);
-						return { id: m.id, ok };
-					}),
-				);
-				for (const r of results) {
-					if (!r.ok) notFound.push(r.id);
-				}
-			}
-
-			if (notFound.length === 0) {
-				ctx.ui.notify("All Ollama models are accessible ✅", "info");
-				return;
-			}
-
-			// Auto-hide 403 models in config (provider-scoped)
-			const config = loadConfigFile();
-			const existingHidden = new Set(config.hidden_models ?? []);
-			for (const id of notFound) existingHidden.add(`${PROVIDER_OLLAMA}/${id}`);
-			saveConfig({ hidden_models: Array.from(existingHidden) });
-
-			// Re-register so hidden models disappear immediately
-			const filtered = await fetchOllamaModels(apiKey);
-			stored.free = filtered;
-			stored.all = filtered;
-			reRegister(filtered);
-
-			ctx.ui.notify(
-				`Found ${notFound.length} broken models (auto-hidden):\n${notFound.join("\n")}`,
-				"warning",
-			);
-		},
-	});
-
-	// ── Status bar for provider selection ─────────────────────────
-
-	pi.on("model_select", (_event, ctx) => {
-		if (_event.model?.provider !== PROVIDER_OLLAMA) {
-			ctx.ui.setStatus(`${PROVIDER_OLLAMA}-status`, undefined);
-			return;
-		}
-
-		const count = allModels.length;
-		ctx.ui.setStatus(
-			`${PROVIDER_OLLAMA}-status`,
-			`ollama: ${count} models (usage-based)`,
-		);
-	});
-}
-
-/**
- * Probe a single Ollama model with a minimal chat request.
- * Returns true if the model is accessible (not 403), false if it 403s.
- */
-async function probeOllamaModel(
-	apiKey: string,
-	modelId: string,
-): Promise<boolean> {
-	try {
-		const response = await fetchWithTimeout(
-			`${BASE_URL_OLLAMA}/chat/completions`,
-			{
-				method: "POST",
-				headers: {
-					Authorization: `Bearer ${apiKey}`,
-					"Content-Type": "application/json",
-					"User-Agent": "pi-free-providers",
-				},
-				body: JSON.stringify({
-					model: modelId,
-					messages: [{ role: "user", content: "hi" }],
-					max_tokens: 1,
-				}),
-			},
-			10000, // 10 second timeout
-		);
-		// 403 = access denied (model not provisioned)
-		// 200/400/401/etc = at least accessible
-		return response.status !== 403;
-	} catch {
-		// Network errors / timeouts are not "access denied"
-		return true;
-	}
-}
+/**
+ * Ollama Cloud Provider Extension
+ *
+ * Provides access to Ollama's cloud-hosted models via ollama.com API.
+ * Fetches per-model capabilities via /api/show for accurate reasoning,
+ * vision, and context window detection.
+ *
+ * Requires OLLAMA_API_KEY with cloud access.
+ * Get a free key at: https://ollama.com/settings/keys
+ *
+ * Usage:
+ *   pi install git:github.com/apmantza/pi-free
+ *   # Set OLLAMA_API_KEY env var
+ *   # Models appear in /model selector
+ *   # Use /toggle-ollama to show all vs limited set
+ *   # Use /probe-ollama to detect and hide 403 models
+ *   # Use /ollama-cloud-refresh to re-fetch models live
+ */
+
+import type {
+	ExtensionAPI,
+	ExtensionCommandContext,
+	ProviderModelConfig,
+} from "@earendil-works/pi-coding-agent";
+import {
+	applyHidden,
+	getOllamaApiKey,
+	getOllamaShowPaid,
+	loadConfigFile,
+	saveConfig,
+} from "../../config.ts";
+import {
+	BASE_URL_OLLAMA,
+	DEFAULT_FETCH_TIMEOUT_MS,
+	PROVIDER_OLLAMA,
+} from "../../constants.ts";
+import { createLogger } from "../../lib/logger.ts";
+import {
+	loadProviderCache,
+	saveProviderCache,
+} from "../../lib/provider-cache.ts";
+import { registerWithGlobalToggle } from "../../lib/registry.ts";
+import { fetchWithRetry, fetchWithTimeout } from "../../lib/util.ts";
+import { createReRegister, enhanceWithCI } from "../../provider-helper.ts";
+import { resolveThinkingMap } from "./thinking-levels.ts";
+
+const _logger = createLogger("ollama-cloud");
+
+// =============================================================================
+// Constants
+// =============================================================================
+
+/** Base URL for non-OpenAI-compatible endpoints (e.g. /api/show). */
+const OLLAMA_API_BASE = BASE_URL_OLLAMA.replace(/\/v1\/?$/, "");
+const DETAIL_FETCH_TIMEOUT_MS = 10000;
+const DETAIL_CONCURRENCY = 8;
+
+// =============================================================================
+// Known 403 models (listed but return "access denied" on /v1/chat/completions)
+// These are models that appear in /v1/models but aren't provisioned for chat.
+// Add new IDs here as they surface via /probe-ollama command.
+// =============================================================================
+const OLLAMA_KNOWN_403_MODELS: ReadonlySet<string> = new Set([
+	// Example entries - populate via probe-ollama.mjs
+	// "model-id-that-403s",
+]);
+
+// =============================================================================
+// Fallback models (used when API is unreachable and no cache exists)
+// =============================================================================
+const FALLBACK_MODELS: ProviderModelConfig[] = [
+	{
+		id: "glm-5.1",
+		name: "GLM 5.1",
+		reasoning: false,
+		input: ["text"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 202752,
+		maxTokens: 32768,
+		compat: { supportsDeveloperRole: false },
+	},
+	{
+		id: "gemma4:31b",
+		name: "Gemma 4 31B",
+		reasoning: false,
+		input: ["text"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 262144,
+		maxTokens: 32768,
+		compat: { supportsDeveloperRole: false },
+	},
+	{
+		id: "deepseek-v4-pro",
+		name: "DeepSeek V4 Pro",
+		reasoning: true,
+		thinkingLevelMap: resolveThinkingMap("deepseek-v4-pro", [
+			"thinking",
+			"tools",
+		]),
+		input: ["text"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 1000000,
+		maxTokens: 32768,
+		compat: { supportsDeveloperRole: false },
+	},
+	{
+		id: "qwen3.5",
+		name: "Qwen 3.5",
+		reasoning: true,
+		thinkingLevelMap: resolveThinkingMap("qwen3.5", ["thinking", "tools"]),
+		input: ["text"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 131072,
+		maxTokens: 32768,
+		compat: { supportsDeveloperRole: false },
+	},
+	{
+		id: "kimi-k2.6",
+		name: "Kimi K2.6",
+		reasoning: true,
+		thinkingLevelMap: resolveThinkingMap("kimi-k2.6", ["thinking", "tools"]),
+		input: ["text"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 131072,
+		maxTokens: 32768,
+		compat: { supportsDeveloperRole: false },
+	},
+];
+
+// =============================================================================
+// Types
+// =============================================================================
+
+/** Response from POST /api/show */
+interface OllamaShowResponse {
+	details: {
+		parent_model: string;
+		format: string;
+		family: string;
+		families: string[] | null;
+		parameter_size: string;
+		quantization_level: string;
+	};
+	model_info: Record<string, unknown>;
+	capabilities: string[];
+	modified_at: string;
+}
+
+// =============================================================================
+// Utility: concurrent map with bounded parallelism
+// =============================================================================
+
+async function concurrentMap<T, R>(
+	items: T[],
+	workers: number,
+	fn: (item: T) => Promise<R>,
+): Promise<PromiseSettledResult<R>[]> {
+	const results: PromiseSettledResult<R>[] = new Array(items.length);
+	let next = 0;
+	await Promise.all(
+		Array.from({ length: Math.max(1, workers) }, async () => {
+			while (next < items.length) {
+				const index = next++;
+				try {
+					results[index] = {
+						status: "fulfilled",
+						value: await fn(items[index]),
+					};
+				} catch (reason) {
+					results[index] = { status: "rejected", reason };
+				}
+			}
+		}),
+	);
+	return results;
+}
+
+// =============================================================================
+// Fetch: /v1/models → list of model IDs
+// =============================================================================
+
+async function fetchModelIds(apiKey: string): Promise<string[]> {
+	const response = await fetchWithRetry(
+		`${BASE_URL_OLLAMA}/models`,
+		{
+			headers: {
+				Authorization: `Bearer ${apiKey}`,
+				"Content-Type": "application/json",
+			},
+		},
+		3,
+		1000,
+		DEFAULT_FETCH_TIMEOUT_MS,
+	);
+
+	if (!response.ok) {
+		throw new Error(
+			`Failed to fetch Ollama model list: ${response.status} ${response.statusText}`,
+		);
+	}
+
+	const json = (await response.json()) as {
+		data?: Array<{ id: string; owned_by?: string }>;
+	};
+	return (json.data ?? []).map((m) => m.id);
+}
+
+// =============================================================================
+// Fetch: /api/show → per-model capabilities
+// =============================================================================
+
+async function fetchModelDetails(
+	apiKey: string,
+	modelId: string,
+): Promise<OllamaShowResponse> {
+	const response = await fetchWithTimeout(
+		`${OLLAMA_API_BASE}/api/show`,
+		{
+			method: "POST",
+			headers: {
+				Authorization: `Bearer ${apiKey}`,
+				"Content-Type": "application/json",
+			},
+			body: JSON.stringify({ model: modelId }),
+		},
+		DETAIL_FETCH_TIMEOUT_MS,
+	);
+
+	if (!response.ok) {
+		throw new Error(
+			`/api/show failed for ${modelId}: ${response.status} ${response.statusText}`,
+		);
+	}
+
+	return (await response.json()) as OllamaShowResponse;
+}
+
+// =============================================================================
+// Assembly: raw /api/show data → ProviderModelConfig[]
+// =============================================================================
+
+function getContextLength(modelInfo: Record<string, unknown>): number {
+	for (const [key, value] of Object.entries(modelInfo)) {
+		if (key.endsWith(".context_length") && typeof value === "number") {
+			return value;
+		}
+	}
+	return 128000; // fallback
+}
+
+/**
+ * Build a human-readable display name from model ID and details.
+ * Enriches with parameter size and quantization when available.
+ */
+function buildModelName(
+	id: string,
+	details: OllamaShowResponse["details"],
+): string {
+	// Convert dashes/colons to spaces for readability
+	const base = id.replace(/[:-]/g, " ");
+	const parts: string[] = [base];
+
+	const params = details?.parameter_size;
+	const quant = details?.quantization_level;
+
+	if (params && quant) {
+		parts.push(`(${params}, ${quant})`);
+	} else if (params) {
+		parts.push(`(${params})`);
+	}
+
+	return parts.join(" ");
+}
+
+function assembleModels(
+	raw: Record<string, OllamaShowResponse>,
+): ProviderModelConfig[] {
+	return Object.entries(raw)
+		.filter(([, data]) => data.capabilities?.includes("tools"))
+		.map(([id, data]) => {
+			const reasoning = data.capabilities?.includes("thinking") ?? false;
+			const thinkingMap = resolveThinkingMap(id, data.capabilities ?? []);
+
+			return {
+				id,
+				name: buildModelName(id, data.details),
+				reasoning,
+				thinkingLevelMap: thinkingMap,
+				input: (data.capabilities?.includes("vision")
+					? ["text", "image"]
+					: ["text"]) as ("text" | "image")[],
+				cost: {
+					input: 0,
+					output: 0,
+					cacheRead: 0,
+					cacheWrite: 0,
+				},
+				contextWindow: getContextLength(data.model_info ?? {}),
+				maxTokens: 32768,
+				compat: {
+					supportsDeveloperRole: false,
+					// When we provide a thinkingLevelMap, tell Pi not to use its own
+					// reasoning_effort logic — we handle it ourselves.
+					supportsReasoningEffort: thinkingMap != null,
+				},
+			};
+		});
+}
+
+// =============================================================================
+// Fetch all models (orchestrates /v1/models + /api/show)
+// =============================================================================
+
+async function fetchAllModels(apiKey: string): Promise<ProviderModelConfig[]> {
+	// Step 1: Get model IDs
+	const modelIds = await fetchModelIds(apiKey);
+	_logger.info(
+		`[ollama-cloud] Found ${modelIds.length} model IDs, fetching details...`,
+	);
+
+	// Step 2: Filter out known-broken and embedding models early
+	const candidateIds = modelIds.filter((id) => {
+		if (OLLAMA_KNOWN_403_MODELS.has(id)) return false;
+		const name = id.toLowerCase();
+		if (name.includes("embed")) return false;
+		return true;
+	});
+
+	// Step 3: Fetch per-model details concurrently
+	let succeeded = 0;
+	let failed = 0;
+
+	const detailResults = await concurrentMap(
+		candidateIds,
+		DETAIL_CONCURRENCY,
+		async (id) => {
+			try {
+				const result = await fetchModelDetails(apiKey, id);
+				succeeded++;
+				return [id, result] as const;
+			} catch {
+				failed++;
+				throw new Error(`detail fetch failed for ${id}`);
+			} finally {
+				if (
+					(succeeded + failed) % 10 === 0 ||
+					succeeded + failed === candidateIds.length
+				) {
+					_logger.debug(
+						`[ollama-cloud] Detail progress: ${succeeded + failed}/${candidateIds.length} (${failed} failed)`,
+					);
+				}
+			}
+		},
+	);
+
+	// Step 4: Collect successful results
+	const raw: Record<string, OllamaShowResponse> = {};
+	for (const result of detailResults) {
+		if (result.status === "fulfilled") {
+			const [id, data] = result.value;
+			raw[id] = data;
+		}
+	}
+
+	_logger.info(
+		`[ollama-cloud] Fetched ${Object.keys(raw).length} model details` +
+			(failed ? ` (${failed} failed)` : ""),
+	);
+
+	if (Object.keys(raw).length === 0) {
+		throw new Error("Failed to fetch any model details");
+	}
+
+	// Step 5: Assemble into Pi model configs
+	const models = assembleModels(raw);
+
+	// Step 6: Apply user-configured hidden models
+	return applyHidden(models, PROVIDER_OLLAMA);
+}
+
+// =============================================================================
+// Extension Entry Point
+// =============================================================================
+
+export default async function ollamaProvider(pi: ExtensionAPI) {
+	const apiKey = getOllamaApiKey();
+
+	if (!apiKey) {
+		_logger.info(
+			"[ollama-cloud] Skipping - OLLAMA_API_KEY not set (env var or ~/.pi/free.json)",
+		);
+		return;
+	}
+
+	// ── Try cache first for fast startup ────────────────────────────
+	let allModels: ProviderModelConfig[];
+	let fromCache = false;
+
+	const cachedModels = loadProviderCache(PROVIDER_OLLAMA);
+	if (cachedModels && cachedModels.length > 0) {
+		allModels = cachedModels;
+		fromCache = true;
+		_logger.info(
+			`[ollama-cloud] Using ${cachedModels.length} cached models for fast startup`,
+		);
+	} else {
+		allModels = FALLBACK_MODELS;
+		_logger.info("[ollama-cloud] No cache available, using fallback models");
+	}
+
+	// ── Register immediately with cached/fallback models ────────────
+	const freeModels = allModels;
+	let stored = { free: freeModels, all: allModels };
+	const hasKey = true;
+
+	const reRegister = createReRegister(pi, {
+		providerId: PROVIDER_OLLAMA,
+		baseUrl: BASE_URL_OLLAMA,
+		apiKey,
+	});
+
+	registerWithGlobalToggle(PROVIDER_OLLAMA, stored, reRegister, hasKey);
+
+	const initialModels = getOllamaShowPaid() ? allModels : freeModels;
+	pi.registerProvider(PROVIDER_OLLAMA, {
+		baseUrl: BASE_URL_OLLAMA,
+		apiKey,
+		api: "openai-completions" as const,
+		models: enhanceWithCI(initialModels),
+	});
+
+	_logger.info(
+		`[ollama-cloud] Registered ${initialModels.length} models` +
+			(fromCache ? " (from cache)" : " (fallback)") +
+			", fetching fresh in background...",
+	);
+
+	// ── Background refresh ─────────────────────────────────────────
+	async function refreshModels(): Promise<ProviderModelConfig[]> {
+		try {
+			const freshModels = await fetchAllModels(apiKey!);
+			saveProviderCache(PROVIDER_OLLAMA, freshModels);
+			return freshModels;
+		} catch (error) {
+			_logger.error("[ollama-cloud] Background refresh failed", {
+				error: error instanceof Error ? error.message : String(error),
+			});
+			// Return current models so we don't lose what we have
+			return allModels;
+		}
+	}
+
+	// ── /ollama-cloud-refresh command ───────────────────────────────
+	pi.registerCommand("ollama-cloud-refresh", {
+		description:
+			"Re-fetch Ollama Cloud models from the API and update the provider live",
+		handler: async (_args: string, ctx: ExtensionCommandContext) => {
+			ctx.ui.notify("Refreshing Ollama Cloud models…", "info");
+			try {
+				const fresh = await fetchAllModels(apiKey!);
+				saveProviderCache(PROVIDER_OLLAMA, fresh);
+				allModels = fresh;
+				stored = { free: fresh, all: fresh };
+				reRegister(fresh);
+				ctx.ui.notify(
+					`Registered ${fresh.length} Ollama Cloud models (refresh complete)`,
+					"info",
+				);
+			} catch (error) {
+				ctx.ui.notify(
+					`Refresh failed: ${error instanceof Error ? error.message : String(error)}`,
+					"error",
+				);
+			}
+		},
+	});
+
+	// ── /probe-ollama command ───────────────────────────────────────
+	pi.registerCommand("probe-ollama", {
+		description: "Test all Ollama Cloud models for 403 'access denied' errors",
+		handler: async (_args: string, ctx: ExtensionCommandContext) => {
+			if (!apiKey) {
+				ctx.ui.notify("OLLAMA_API_KEY not set", "error");
+				return;
+			}
+
+			const modelsToTest = allModels;
+			ctx.ui.notify(`Probing ${modelsToTest.length} Ollama models…`, "info");
+
+			const notFound: string[] = [];
+			const batchSize = 5;
+
+			for (let i = 0; i < modelsToTest.length; i += batchSize) {
+				const batch = modelsToTest.slice(i, i + batchSize);
+				const results = await Promise.all(
+					batch.map(async (m) => {
+						const ok = await probeOllamaModel(apiKey, m.id);
+						return { id: m.id, ok };
+					}),
+				);
+				for (const r of results) {
+					if (!r.ok) notFound.push(r.id);
+				}
+			}
+
+			if (notFound.length === 0) {
+				ctx.ui.notify("All Ollama models are accessible ✅", "info");
+				return;
+			}
+
+			// Auto-hide 403 models in config (provider-scoped)
+			const config = loadConfigFile();
+			const existingHidden = new Set(config.hidden_models ?? []);
+			for (const id of notFound) existingHidden.add(`${PROVIDER_OLLAMA}/${id}`);
+			saveConfig({
+				hidden_models: Array.from(existingHidden),
+			});
+
+			// Re-fetch and re-register so hidden models disappear immediately
+			try {
+				const fresh = await fetchAllModels(apiKey!);
+				saveProviderCache(PROVIDER_OLLAMA, fresh);
+				allModels = fresh;
+				stored = { free: fresh, all: fresh };
+				reRegister(fresh);
+			} catch {
+				// If refresh fails, just re-register current models
+				reRegister(allModels);
+			}
+
+			ctx.ui.notify(
+				`Found ${notFound.length} broken models (auto-hidden):\n${notFound.join("\n")}`,
+				"warning",
+			);
+		},
+	});
+
+	// ── 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);
+			return;
+		}
+
+		const count = allModels.length;
+		ctx.ui.setStatus(`${PROVIDER_OLLAMA}-status`, `ollama: ${count} models`);
+	});
+
+	// ── Background refresh on session_start ─────────────────────────
+	let bgRefreshed = false;
+	// eslint-disable-next-line @typescript-eslint/no-explicit-any
+	pi.on("session_start" as any, async (_event: any, ctx: any) => {
+		if (bgRefreshed) {
+			return;
+		}
+		bgRefreshed = true;
+
+		try {
+			const fresh = await refreshModels();
+			allModels = fresh;
+			stored = { free: fresh, all: fresh };
+			reRegister(fresh);
+			ctx.ui.notify(`Ollama Cloud: ${fresh.length} models ready`, "info");
+		} catch {
+			// Already logged in refreshModels()
+		}
+	});
+}
+
+// =============================================================================
+// Probe helper
+// =============================================================================
+
+/**
+ * Probe a single Ollama model with a minimal chat request.
+ * Returns true if the model is accessible (not 403), false if it 403s.
+ */
+async function probeOllamaModel(
+	apiKey: string,
+	modelId: string,
+): Promise<boolean> {
+	try {
+		const response = await fetchWithTimeout(
+			`${BASE_URL_OLLAMA}/chat/completions`,
+			{
+				method: "POST",
+				headers: {
+					Authorization: `Bearer ${apiKey}`,
+					"Content-Type": "application/json",
+					"User-Agent": "pi-free-providers",
+				},
+				body: JSON.stringify({
+					model: modelId,
+					messages: [{ role: "user", content: "hi" }],
+					max_tokens: 1,
+				}),
+			},
+			10000,
+		);
+		// 403 = access denied (model not provisioned)
+		// 200/400/401/etc = at least accessible
+		return response.status !== 403;
+	} catch {
+		// Network errors / timeouts are not "access denied"
+		return true;
+	}
+}