pi-free 2.0.2 → 2.0.4

package/index.ts

@@ -1,148 +1,175 @@
-/**
- * Pi-Free Providers Index
- *
- * Provides free model filtering for ALL providers (built-in + extension)
- * plus unique free/paid providers not covered by pi's built-in providers.
- *
- * Unique providers:
- * - Kilo: OAuth-based free models
- * - Cline: Cline bot integration
- * - NVIDIA: NVIDIA NIM hosting (free tier available)
- * - Ollama Cloud: Ollama's cloud-hosted models with usage-based free tier
- * - Qwen: OAuth-based Qwen access (deprecated)
- * - Modal: Modal Labs hosting
- */
-
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
-import { setupBuiltInProviderToggles } from "./lib/built-in-toggle.ts";
-import { createLogger } from "./lib/logger.ts";
-import {
-	applyGlobalFilter,
-	getGlobalFreeOnly,
-	getProviderRegistry,
-	isFreeModel,
-	registerWithGlobalToggle,
-} from "./lib/registry.ts";
-// Import unique provider extensions (only providers NOT built into pi)
-import cline from "./providers/cline/cline.ts";
-import cloudflare from "./providers/cloudflare/cloudflare.ts";
-import kilo from "./providers/kilo/kilo.ts";
-import modal from "./providers/modal/modal.ts";
-import nvidia from "./providers/nvidia/nvidia.ts";
-import ollama from "./providers/ollama/ollama.ts";
-import qwen from "./providers/qwen/qwen.ts";
-
-const _logger = createLogger("pi-free");
-
-// =============================================================================
-// Global Commands
-// =============================================================================
-
-function setupGlobalCommands(pi: ExtensionAPI) {
-	// /free-providers - Show free model counts by provider
-	pi.registerCommand("free-providers", {
-		description: "Show free/paid model counts for all pi-free providers",
-		handler: async (_args, ctx) => {
-			const lines = ["📊 Pi-Free Providers:", ""];
-			const registry = getProviderRegistry();
-
-			// Providers known to not expose pricing via API (all models show as "free")
-			// OpenRouter and OpenCode expose actual pricing
-			const noPricingApi = new Set([
-				"mistral",
-				"xai",
-				"huggingface",
-				"groq",
-				"cerebras",
-			]);
-			// Freemium providers - all models share a free tier quota
-			const freemiumProviders = new Set(["nvidia"]);
-
-			for (const [id, entry] of registry) {
-				const free = entry.stored.free.length;
-				const all = entry.stored.all.length || free;
-				const indicator = entry.hasKey ? "🔑" : "🆓";
-				const paid = all - free;
-
-				if (freemiumProviders.has(id)) {
-					// Freemium: all models share a free tier (e.g., 1,000 reqs/month)
-					lines.push(`${indicator} ${id}: ${all} models (freemium)`);
-				} else if (noPricingApi.has(id)) {
-					// Provider doesn't expose pricing - can't determine free vs paid
-					lines.push(
-						`${indicator} ${id}: ${all} models (pricing not exposed by API)`,
-					);
-				} else if (paid === 0 && free > 0) {
-					// All models are actually free
-					lines.push(`${indicator} ${id}: ${free} free models`);
-				} else {
-					// Mix of free and paid
-					lines.push(
-						`${indicator} ${id}: ${free} free / ${paid} paid (${all} total)`,
-					);
-				}
-			}
-
-			if (registry.size === 0) {
-				lines.push("(No providers registered yet)");
-			}
-
-			ctx.ui.notify(lines.join("\n"), "info");
-		},
-	});
-}
-
-// =============================================================================
-// Main Entry Point
-// =============================================================================
-
-export default async function (pi: ExtensionAPI) {
-	const globalFreeOnly = getGlobalFreeOnly();
-	_logger.info(`[pi-free] Initializing (global free-only: ${globalFreeOnly})`);
-
-	// Setup global commands first
-	setupGlobalCommands(pi);
-
-	// Load all unique providers
-	// Each provider will register itself with the global toggle system
-	await Promise.allSettled([
-		cloudflare(pi),
-		modal(pi),
-		nvidia(pi),
-		kilo(pi),
-		ollama(pi),
-		// Qwen is deprecated
-		qwen(pi).catch((err) => {
-			_logger.warn("[pi-free] Qwen provider failed to load (deprecated)", err);
-		}),
-		cline(pi),
-	]);
-
-	// Setup dynamic built-in providers (Mistral, Groq, Cerebras, xAI, Hugging Face, OpenRouter)
-	// These only activate if the user has configured API keys (OpenRouter works without key too)
-	const { setupDynamicBuiltInProviders } = await import(
-		"./providers/dynamic-built-in/index.ts"
-	);
-	await setupDynamicBuiltInProviders(pi);
-
-	// Setup toggles for pi's built-in providers (e.g., OpenCode)
-	setupBuiltInProviderToggles(pi);
-
-	// Apply initial global filter if free-only mode is enabled
-	if (globalFreeOnly) {
-		_logger.info("[pi-free] Applying initial free-only filter");
-		await applyGlobalFilter(pi, true);
-	}
-
-	const registry = getProviderRegistry();
-	_logger.info(`[pi-free] Loaded with ${registry.size} providers`);
-}
-
-// Re-export registry helpers so consumers don't need deep imports
-export {
-	applyGlobalFilter,
-	getGlobalFreeOnly,
-	getProviderRegistry,
-	isFreeModel,
-	registerWithGlobalToggle,
-};
+/**
+ * Pi-Free Providers Index
+ *
+ * Provides free model filtering for ALL providers (built-in + extension)
+ * plus unique free/paid providers not covered by pi's built-in providers.
+ *
+ * Unique providers:
+ * - Kilo: OAuth-based free models
+ * - Cline: Cline bot integration
+ * - NVIDIA: NVIDIA NIM hosting (free tier available)
+ * - Ollama Cloud: Ollama's cloud-hosted models with usage-based free tier
+ * - ZenMux: Unified AI API gateway with 200+ models
+ */
+
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { setupBuiltInProviderToggles } from "./lib/built-in-toggle.ts";
+import { createLogger } from "./lib/logger.ts";
+import {
+	applyGlobalFilter,
+	getGlobalFreeOnly,
+	getProviderRegistry,
+	isFreeModel,
+	registerWithGlobalToggle,
+} from "./lib/registry.ts";
+// Import unique provider extensions (only providers NOT built into pi)
+import cline from "./providers/cline/cline.ts";
+import crofai from "./providers/crofai/crofai.ts";
+import kilo from "./providers/kilo/kilo.ts";
+import nvidia from "./providers/nvidia/nvidia.ts";
+import ollama from "./providers/ollama/ollama.ts";
+import zenmux from "./providers/zenmux/zenmux.ts";
+
+const _logger = createLogger("pi-free");
+
+// =============================================================================
+// Global Commands
+// =============================================================================
+
+function setupGlobalCommands(pi: ExtensionAPI) {
+	// /toggle-free - Global free-only mode toggle
+	pi.registerCommand("toggle-free", {
+		description: "Toggle global free-only mode for all providers",
+		handler: async (_args, ctx) => {
+			const current = getGlobalFreeOnly();
+			const next = !current;
+			applyGlobalFilter(pi, next);
+
+			const registry = getProviderRegistry();
+			const providerCount = registry.size;
+
+			if (next) {
+				const totalFree = [...registry.values()].reduce(
+					(sum, e) => sum + e.stored.free.length,
+					0,
+				);
+				ctx.ui.notify(
+					`Free-only mode: ON (${totalFree} free models across ${providerCount} providers)`,
+					"info",
+				);
+			} else {
+				const totalAll = [...registry.values()].reduce(
+					(sum, e) => sum + (e.stored.all.length || e.stored.free.length),
+					0,
+				);
+				ctx.ui.notify(
+					`Free-only mode: OFF (all ${totalAll} models visible across ${providerCount} providers)`,
+					"info",
+				);
+			}
+		},
+	});
+
+	// /free-providers - Show free model counts by provider
+	pi.registerCommand("free-providers", {
+		description: "Show free/paid model counts for all pi-free providers",
+		handler: async (_args, ctx) => {
+			const lines = ["📊 Pi-Free Providers:", ""];
+			const registry = getProviderRegistry();
+
+			// Providers known to not expose pricing via API (all models show as "free")
+			// OpenRouter and OpenCode expose actual pricing
+			const noPricingApi = new Set([
+				"mistral",
+				"xai",
+				"huggingface",
+				"groq",
+				"cerebras",
+			]);
+			// Freemium providers - all models share a free tier quota
+			const freemiumProviders = new Set(["nvidia"]);
+
+			for (const [id, entry] of registry) {
+				const free = entry.stored.free.length;
+				const all = entry.stored.all.length || free;
+				const indicator = entry.hasKey ? "🔑" : "🆓";
+				const paid = all - free;
+
+				if (freemiumProviders.has(id)) {
+					// Freemium: all models share a free tier (e.g., 1,000 reqs/month)
+					lines.push(`${indicator} ${id}: ${all} models (freemium)`);
+				} else if (noPricingApi.has(id)) {
+					// Provider doesn't expose pricing - can't determine free vs paid
+					lines.push(
+						`${indicator} ${id}: ${all} models (pricing not exposed by API)`,
+					);
+				} else if (paid === 0 && free > 0) {
+					// All models are actually free
+					lines.push(`${indicator} ${id}: ${free} free models`);
+				} else {
+					// Mix of free and paid
+					lines.push(
+						`${indicator} ${id}: ${free} free / ${paid} paid (${all} total)`,
+					);
+				}
+			}
+
+			if (registry.size === 0) {
+				lines.push("(No providers registered yet)");
+			}
+
+			ctx.ui.notify(lines.join("\n"), "info");
+		},
+	});
+}
+
+// =============================================================================
+// Main Entry Point
+// =============================================================================
+
+export default async function (pi: ExtensionAPI) {
+	const globalFreeOnly = getGlobalFreeOnly();
+	_logger.info(`[pi-free] Initializing (global free-only: ${globalFreeOnly})`);
+
+	// Setup global commands first
+	setupGlobalCommands(pi);
+
+	// Load all unique providers
+	// Each provider will register itself with the global toggle system
+	await Promise.allSettled([
+		nvidia(pi),
+		kilo(pi),
+		ollama(pi),
+		cline(pi),
+		zenmux(pi),
+		crofai(pi),
+	]);
+
+	// Setup dynamic built-in providers (Mistral, Groq, Cerebras, xAI, Hugging Face)
+	// These only activate if the user has configured API keys (OpenRouter works without key too)
+	const { setupDynamicBuiltInProviders } = await import(
+		"./providers/dynamic-built-in/index.ts"
+	);
+	await setupDynamicBuiltInProviders(pi);
+
+	// Setup toggles for pi's built-in providers (e.g., OpenCode)
+	setupBuiltInProviderToggles(pi);
+
+	// Apply initial global filter if free-only mode is enabled
+	if (globalFreeOnly) {
+		_logger.info("[pi-free] Applying initial free-only filter");
+		await applyGlobalFilter(pi, true);
+	}
+
+	const registry = getProviderRegistry();
+	_logger.info(`[pi-free] Loaded with ${registry.size} providers`);
+}
+
+// Re-export registry helpers so consumers don't need deep imports
+export {
+	applyGlobalFilter,
+	getGlobalFreeOnly,
+	getProviderRegistry,
+	isFreeModel,
+	registerWithGlobalToggle,
+};

package/lib/built-in-toggle.ts

@@ -60,6 +60,7 @@ export function setupBuiltInProviderToggles(pi: ExtensionAPI): void {
 		for (const config of BUILT_IN_TOGGLE_PROVIDERS) {
 			registerToggleCommand(pi, config);
 		}
+		setupStatusBar(pi);
 		commandsRegistered = true;
 	}
 
@@ -79,7 +80,9 @@ export function setupBuiltInProviderToggles(pi: ExtensionAPI): void {
 			if (providerModels.length === 0) continue;
 
 			const allModels = providerModels.map(modelToProviderConfig);
-			const freeModels = allModels.filter(isFreeModel);
+			const freeModels = allModels.filter((m) =>
+				isFreeModel({ ...m, provider: config.id }, allModels),
+			);
 
 			const baseUrl = providerModels[0].baseUrl;
 			const api = providerModels[0].api;
@@ -178,6 +181,42 @@ function modelToProviderConfig(m: Model<Api>): ProviderModelConfig {
 	};
 }
 
+// =============================================================================
+// Status bar for provider selection
+// =============================================================================
+
+function setupStatusBar(pi: ExtensionAPI): void {
+	pi.on("model_select", (_event, ctx) => {
+		const selected = _event.model?.provider;
+
+		// Clear status for all built-in toggle providers
+		for (const config of BUILT_IN_TOGGLE_PROVIDERS) {
+			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/model-enhancer.ts

@@ -1,20 +1,20 @@
-/**
- * Model name enhancement helper
- * Adds Coding Index scores to model names for display in /model
- */
-
-import type { ProviderModelConfig } from "@mariozechner/pi-coding-agent";
-import { enhanceModelNameWithCodingIndex } from "../provider-failover/benchmark-lookup.ts";
-
-/**
- * Enhance model names with Coding Index scores
- * Use this before registering providers to show CI in /model list
- */
-export function enhanceModelsWithCodingIndex(
-	models: ProviderModelConfig[],
-): ProviderModelConfig[] {
-	return models.map((m) => ({
-		...m,
-		name: enhanceModelNameWithCodingIndex(m.name, m.id),
-	}));
-}
+/**
+ * Model name enhancement helper
+ * Adds Coding Index scores to model names for display in /model
+ */
+
+import type { ProviderModelConfig } from "@mariozechner/pi-coding-agent";
+import { enhanceModelNameWithCodingIndex } from "../provider-failover/benchmark-lookup.ts";
+
+/**
+ * Enhance model names with Coding Index scores
+ * Use this before registering providers to show CI in /model list
+ */
+export function enhanceModelsWithCodingIndex(
+	models: ProviderModelConfig[],
+): ProviderModelConfig[] {
+	return models.map((m) => ({
+		...m,
+		name: enhanceModelNameWithCodingIndex(m.name, m.id),
+	}));
+}

package/lib/open-browser.ts

@@ -25,7 +25,7 @@ export function openBrowser(url: string): void {
 					"-NoProfile",
 					"-NonInteractive",
 					"-Command",
-					`Start-Process "${url.replace(/"/g, '\\"')}"`,
+					`Start-Process "${url.replace(/[\\"]/g, "\\$&")}"`,
 				],
 				{ detached: true, shell: false, windowsHide: true },
 			).unref();

package/lib/provider-compat.ts

@@ -0,0 +1,46 @@
+import type { ProviderModelConfig } from "@mariozechner/pi-coding-agent";
+
+export interface ProviderModelIdentity {
+	id: string;
+	name?: string;
+}
+
+export const DEEPSEEK_PROXY_COMPAT: NonNullable<ProviderModelConfig["compat"]> =
+	{
+		supportsStore: false,
+		supportsDeveloperRole: false,
+		supportsReasoningEffort: true,
+		requiresReasoningContentOnAssistantMessages: true,
+		thinkingFormat: "deepseek",
+	};
+
+export function isDeepSeekModel(model: ProviderModelIdentity): boolean {
+	const haystack = `${model.id} ${model.name ?? ""}`.toLowerCase();
+	return haystack.includes("deepseek");
+}
+
+export function isLikelyReasoningModel(model: ProviderModelIdentity): boolean {
+	const haystack = `${model.id} ${model.name ?? ""}`.toLowerCase();
+	return (
+		isDeepSeekModel(model) ||
+		haystack.includes("thinking") ||
+		haystack.includes("reasoning") ||
+		haystack.includes("reasoner") ||
+		haystack.includes("r1") ||
+		haystack.includes("qwq")
+	);
+}
+
+/**
+ * For gateway/proxy providers that mask the upstream DeepSeek base URL,
+ * add explicit compat so pi-ai preserves and replays reasoning_content.
+ */
+export function getProxyModelCompat(
+	model: ProviderModelIdentity,
+): ProviderModelConfig["compat"] | undefined {
+	if (isDeepSeekModel(model)) {
+		return DEEPSEEK_PROXY_COMPAT;
+	}
+
+	return undefined;
+}