pi-free 2.0.1 → 2.0.4

package/config.ts

@@ -24,54 +24,45 @@ const _logger = createLogger("config");
 
 interface PiFreeConfig {
 	nvidia_api_key?: string;
-	cloudflare_api_token?: string;
-	cloudflare_account_id?: string;
 	ollama_api_key?: string;
-	modal_api_key?: string;
-	opencode_api_key?: string;
+	zenmux_api_key?: string;
+	crofai_api_key?: string;
 	mistral_api_key?: string;
 	groq_api_key?: string;
 	cerebras_api_key?: string;
 	xai_api_key?: string;
 	hf_token?: string;
-	openrouter_api_key?: string;
 	kilo_free_only?: boolean;
 	hidden_models?: string[];
 	free_only?: boolean;
 	kilo_show_paid?: boolean;
-	nvidia_show_paid?: boolean;
-	cloudflare_show_paid?: boolean;
 	ollama_show_paid?: boolean;
 	cline_show_paid?: boolean;
-	qwen_show_paid?: boolean;
-	modal_show_paid?: boolean;
+	zenmux_show_paid?: boolean;
+	crofai_show_paid?: boolean;
 	openrouter_show_paid?: boolean;
 	opencode_show_paid?: boolean;
 }
 
 const CONFIG_TEMPLATE: PiFreeConfig = {
 	nvidia_api_key: "",
-	cloudflare_api_token: "",
-	cloudflare_account_id: "",
 	ollama_api_key: "",
-	modal_api_key: "",
-	opencode_api_key: "",
+	zenmux_api_key: "",
+	crofai_api_key: "",
 	mistral_api_key: "",
 	groq_api_key: "",
 	cerebras_api_key: "",
 	xai_api_key: "",
 	hf_token: "",
-	openrouter_api_key: "",
+
 	kilo_free_only: false,
 	hidden_models: [],
 	free_only: true,
 	kilo_show_paid: false,
-	nvidia_show_paid: false,
-	cloudflare_show_paid: false,
 	ollama_show_paid: false,
 	cline_show_paid: false,
-	qwen_show_paid: false,
-	modal_show_paid: false,
+	zenmux_show_paid: false,
+	crofai_show_paid: false,
 	openrouter_show_paid: false,
 	opencode_show_paid: false,
 };
@@ -144,34 +135,22 @@ export function getKiloShowPaid(): boolean {
 	return resolveBool("KILO_SHOW_PAID", loadConfigFile().kilo_show_paid);
 }
 
-export function getNvidiaShowPaid(): boolean {
-	return resolveBool("NVIDIA_SHOW_PAID", loadConfigFile().nvidia_show_paid);
-}
-
 export function getClineShowPaid(): boolean {
 	return resolveBool("CLINE_SHOW_PAID", loadConfigFile().cline_show_paid);
 }
 
-/** @deprecated Qwen provider is deprecated. */
-export function getQwenShowPaid(): boolean {
-	return resolveBool("QWEN_SHOW_PAID", loadConfigFile().qwen_show_paid);
+export function getZenmuxShowPaid(): boolean {
+	return resolveBool("ZENMUX_SHOW_PAID", loadConfigFile().zenmux_show_paid);
 }
 
-export function getModalShowPaid(): boolean {
-	return resolveBool("MODAL_SHOW_PAID", loadConfigFile().modal_show_paid);
+export function getCrofaiShowPaid(): boolean {
+	return resolveBool("CROFAI_SHOW_PAID", loadConfigFile().crofai_show_paid);
 }
 
 export function getOllamaShowPaid(): boolean {
 	return resolveBool("OLLAMA_SHOW_PAID", loadConfigFile().ollama_show_paid);
 }
 
-export function getCloudflareShowPaid(): boolean {
-	return resolveBool(
-		"CLOUDFLARE_SHOW_PAID",
-		loadConfigFile().cloudflare_show_paid,
-	);
-}
-
 export function getOpenrouterShowPaid(): boolean {
 	return resolveBool(
 		"OPENROUTER_SHOW_PAID",
@@ -203,27 +182,16 @@ export function getNvidiaApiKey(): string | undefined {
 	return resolve("NVIDIA_API_KEY", loadConfigFile().nvidia_api_key);
 }
 
-export function getModalApiKey(): string | undefined {
-	return resolve("MODAL_API_KEY", loadConfigFile().modal_api_key);
-}
-
-export function getOllamaApiKey(): string | undefined {
-	return resolve("OLLAMA_API_KEY", loadConfigFile().ollama_api_key);
+export function getZenmuxApiKey(): string | undefined {
+	return resolve("ZENMUX_API_KEY", loadConfigFile().zenmux_api_key);
 }
 
-export function getCloudflareApiToken(): string | undefined {
-	return resolve("CLOUDFLARE_API_TOKEN", loadConfigFile().cloudflare_api_token);
+export function getCrofaiApiKey(): string | undefined {
+	return resolve("CROFAI_API_KEY", loadConfigFile().crofai_api_key);
 }
 
-export function getCloudflareAccountId(): string | undefined {
-	return resolve(
-		"CLOUDFLARE_ACCOUNT_ID",
-		loadConfigFile().cloudflare_account_id,
-	);
-}
-
-export function getOpencodeApiKey(): string | undefined {
-	return resolve("OPENCODE_API_KEY", loadConfigFile().opencode_api_key);
+export function getOllamaApiKey(): string | undefined {
+	return resolve("OLLAMA_API_KEY", loadConfigFile().ollama_api_key);
 }
 
 export function getMistralApiKey(): string | undefined {
@@ -246,18 +214,42 @@ export function getHfToken(): string | undefined {
 	return resolve("HF_TOKEN", loadConfigFile().hf_token);
 }
 
+/**
+ * OpenRouter key — pi's built-in provider reads from ~/.pi/agent/auth.json.
+ * pi-free only checks the env var to avoid stale keys from free.json.
+ */
 export function getOpenrouterApiKey(): string | undefined {
-	return resolve("OPENROUTER_API_KEY", loadConfigFile().openrouter_api_key);
+	return process.env.OPENROUTER_API_KEY;
 }
 
 // =============================================================================
 // Hidden models (re-reads config on every call)
 // =============================================================================
 
-export function applyHidden<T extends { id: string }>(models: T[]): T[] {
+/**
+ * Apply hidden models filter with provider scoping.
+ * Hidden models can be specified as:
+ *   - "model-id" (global, applies to all providers - deprecated)
+ *   - "provider/model-id" (provider-specific, preferred)
+ */
+export function applyHidden<T extends { id: string }>(
+	models: T[],
+	providerId?: string,
+): T[] {
 	const hidden = new Set(loadConfigFile().hidden_models ?? []);
 	if (hidden.size === 0) return models;
-	return models.filter((m) => !hidden.has(m.id));
+
+	return models.filter((m) => {
+		// Check provider-scoped ID (preferred format: "provider/model-id")
+		if (providerId && hidden.has(`${providerId}/${m.id}`)) {
+			return false;
+		}
+		// Check global ID (legacy format, still supported for backward compat)
+		if (hidden.has(m.id)) {
+			return false;
+		}
+		return true;
+	});
 }
 
 // =============================================================================

package/constants.ts

@@ -15,6 +15,8 @@ export const PROVIDER_OLLAMA = "ollama-cloud";
 /** @deprecated Qwen provider is deprecated. The 1,000 req/day free tier is no longer available. */
 export const PROVIDER_QWEN = "qwen";
 export const PROVIDER_MODAL = "modal";
+export const PROVIDER_ZENMUX = "zenmux";
+export const PROVIDER_CROFAI = "crofai";
 
 export const ALL_UNIQUE_PROVIDERS = [
 	PROVIDER_KILO,
@@ -24,6 +26,8 @@ export const ALL_UNIQUE_PROVIDERS = [
 	PROVIDER_QWEN,
 	PROVIDER_MODAL,
 	PROVIDER_OLLAMA,
+	PROVIDER_ZENMUX,
+	PROVIDER_CROFAI,
 ] as const;
 
 // =============================================================================
@@ -38,6 +42,8 @@ export const BASE_URL_CLINE = "https://api.cline.bot/api/v1";
 export const BASE_URL_MODAL = "https://api.us-west-2.modal.direct/v1";
 export const BASE_URL_QWEN =
 	"https://dashscope.aliyuncs.com/compatible-mode/v1";
+export const BASE_URL_ZENMUX = "https://zenmux.ai/api/v1";
+export const BASE_URL_CROFAI = "https://crof.ai/v1";
 
 /** Cline fetches free models from OpenRouter */
 export const BASE_URL_OPENROUTER = "https://openrouter.ai/api/v1";

package/index.ts

@@ -9,8 +9,7 @@
  * - 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
+ * - ZenMux: Unified AI API gateway with 200+ models
  */
 
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
@@ -25,12 +24,11 @@ import {
 } 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 crofai from "./providers/crofai/crofai.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";
+import zenmux from "./providers/zenmux/zenmux.ts";
 
 const _logger = createLogger("pi-free");
 
@@ -39,6 +37,39 @@ const _logger = createLogger("pi-free");
 // =============================================================================
 
 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",
@@ -106,19 +137,15 @@ export default async function (pi: ExtensionAPI) {
 	// 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),
+		zenmux(pi),
+		crofai(pi),
 	]);
 
-	// Setup dynamic built-in providers (Mistral, Groq, Cerebras, xAI, Hugging Face, OpenRouter)
+	// 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"

package/lib/built-in-toggle.ts

@@ -16,17 +16,10 @@ import type {
 	ExtensionAPI,
 	ProviderModelConfig,
 } from "@mariozechner/pi-coding-agent";
-import {
-	getOpencodeShowPaid,
-	getOpenrouterShowPaid,
-	saveConfig,
-} from "../config.ts";
+import { getOpencodeShowPaid, getOpenrouterShowPaid } from "../config.ts";
 import { createLogger } from "./logger.ts";
-import {
-	getGlobalFreeOnly,
-	isFreeModel,
-	registerWithGlobalToggle,
-} from "./registry.ts";
+import { isFreeModel, registerWithGlobalToggle } from "./registry.ts";
+import { createToggleState } from "./toggle-state.ts";
 
 const _logger = createLogger("built-in-toggle");
 
@@ -49,10 +42,9 @@ const BUILT_IN_TOGGLE_PROVIDERS: BuiltInToggleConfig[] = [
 // =============================================================================
 
 interface BuiltInProviderState {
-	free: ProviderModelConfig[];
-	all: ProviderModelConfig[];
+	stored: { free: ProviderModelConfig[]; all: ProviderModelConfig[] };
 	reRegister: (models: ProviderModelConfig[]) => void;
-	showPaid: boolean;
+	toggleState: ReturnType<typeof createToggleState<ProviderModelConfig>>;
 }
 
 const providerStates = new Map<string, BuiltInProviderState>();
@@ -68,6 +60,7 @@ export function setupBuiltInProviderToggles(pi: ExtensionAPI): void {
 		for (const config of BUILT_IN_TOGGLE_PROVIDERS) {
 			registerToggleCommand(pi, config);
 		}
+		setupStatusBar(pi);
 		commandsRegistered = true;
 	}
 
@@ -87,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;
@@ -102,35 +97,29 @@ export function setupBuiltInProviderToggles(pi: ExtensionAPI): void {
 				});
 			};
 
+			const stored = { free: freeModels, all: allModels };
+			const toggleState = createToggleState<ProviderModelConfig>({
+				providerId: config.id,
+				initialShowPaid: config.getShowPaid(),
+				initialModels: stored,
+			});
+
 			providerStates.set(config.id, {
-				free: freeModels,
-				all: allModels,
+				stored,
 				reRegister,
-				showPaid: config.getShowPaid(),
+				toggleState,
 			});
 
-			// Register with global free-only filter
-			registerWithGlobalToggle(
-				config.id,
-				{ free: freeModels, all: allModels },
-				reRegister,
-				true,
-			);
+			registerWithGlobalToggle(config.id, stored, reRegister, true);
 
 			_logger.info(
 				`[built-in-toggle] ${config.id}: captured ${allModels.length} models (${freeModels.length} free)`,
 			);
 
-			// Respect global free-only setting at capture time
-			if (!getGlobalFreeOnly() && !config.getShowPaid()) {
-				// Default: show free only (same as other pi-free providers)
-				if (freeModels.length > 0) {
-					reRegister(freeModels);
-					_logger.info(
-						`[built-in-toggle] ${config.id}: applied free-only filter`,
-					);
-				}
-			}
+			const applied = toggleState.applyCurrent(reRegister);
+			_logger.info(
+				`[built-in-toggle] ${config.id}: applied ${applied.mode} mode with ${applied.models.length} models`,
+			);
 		}
 	});
 }
@@ -156,21 +145,16 @@ function registerToggleCommand(
 				return;
 			}
 
-			state.showPaid = !state.showPaid;
-
-			// Persist preference
-			saveConfig({ [`${config.id}_show_paid`]: state.showPaid });
+			const applied = state.toggleState.toggle(state.reRegister);
 
-			if (state.showPaid) {
-				state.reRegister(state.all);
+			if (applied.mode === "all") {
 				ctx.ui.notify(
-					`${config.id}: showing all ${state.all.length} models`,
+					`${config.id}: showing all ${state.stored.all.length} models`,
 					"info",
 				);
 			} else {
-				state.reRegister(state.free);
 				ctx.ui.notify(
-					`${config.id}: showing ${state.free.length} free models`,
+					`${config.id}: showing ${state.stored.free.length} free models`,
 					"info",
 				);
 			}
@@ -197,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;
+}