pi-free 2.0.4 → 2.0.5

package/banner.svg

@@ -0,0 +1,132 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 1280 320" width="1280" height="320">
+  <defs>
+    <linearGradient id="bg" x1="0" y1="0" x2="1" y2="1">
+      <stop offset="0%" stop-color="#0f0b1a"/>
+      <stop offset="50%" stop-color="#1a1130"/>
+      <stop offset="100%" stop-color="#0d0a1a"/>
+    </linearGradient>
+    <linearGradient id="accent" x1="0" y1="0" x2="1" y2="0">
+      <stop offset="0%" stop-color="#7c3aed"/>
+      <stop offset="50%" stop-color="#a78bfa"/>
+      <stop offset="100%" stop-color="#7c3aed"/>
+    </linearGradient>
+    <linearGradient id="accent2" x1="0" y1="0" x2="1" y2="0">
+      <stop offset="0%" stop-color="#06b6d4"/>
+      <stop offset="50%" stop-color="#22d3ee"/>
+      <stop offset="100%" stop-color="#06b6d4"/>
+    </linearGradient>
+    <linearGradient id="card1" x1="0" y1="0" x2="0" y2="1">
+      <stop offset="0%" stop-color="#7c3aed" stop-opacity="0.15"/>
+      <stop offset="100%" stop-color="#7c3aed" stop-opacity="0.05"/>
+    </linearGradient>
+    <linearGradient id="card2" x1="0" y1="0" x2="0" y2="1">
+      <stop offset="0%" stop-color="#06b6d4" stop-opacity="0.15"/>
+      <stop offset="100%" stop-color="#06b6d4" stop-opacity="0.05"/>
+    </linearGradient>
+    <linearGradient id="card3" x1="0" y1="0" x2="0" y2="1">
+      <stop offset="0%" stop-color="#10b981" stop-opacity="0.15"/>
+      <stop offset="100%" stop-color="#10b981" stop-opacity="0.05"/>
+    </linearGradient>
+    <linearGradient id="card4" x1="0" y1="0" x2="0" y2="1">
+      <stop offset="0%" stop-color="#f59e0b" stop-opacity="0.15"/>
+      <stop offset="100%" stop-color="#f59e0b" stop-opacity="0.05"/>
+    </linearGradient>
+    <filter id="glow">
+      <feGaussianBlur stdDeviation="3" result="blur"/>
+      <feMerge>
+        <feMergeNode in="blur"/>
+        <feMergeNode in="SourceGraphic"/>
+      </feMerge>
+    </filter>
+    <filter id="shadow">
+      <feDropShadow dx="0" dy="2" stdDeviation="4" flood-color="#000" flood-opacity="0.3"/>
+    </filter>
+    <pattern id="grid" width="40" height="40" patternUnits="userSpaceOnUse">
+      <path d="M 40 0 L 0 0 0 40" fill="none" stroke="#7c3aed" stroke-opacity="0.04" stroke-width="1"/>
+    </pattern>
+  </defs>
+
+  <!-- Background -->
+  <rect width="1280" height="320" fill="url(#bg)"/>
+  <rect width="1280" height="320" fill="url(#grid)"/>
+
+  <!-- Decorative circles -->
+  <circle cx="1200" cy="60" r="180" fill="#7c3aed" opacity="0.06"/>
+  <circle cx="80" cy="280" r="140" fill="#06b6d4" opacity="0.05"/>
+  <circle cx="640" cy="320" r="200" fill="#a78bfa" opacity="0.04"/>
+
+  <!-- Decorative lines -->
+  <line x1="0" y1="280" x2="1280" y2="280" stroke="url(#accent)" stroke-opacity="0.15" stroke-width="1"/>
+  <line x1="0" y1="282" x2="1280" y2="282" stroke="url(#accent2)" stroke-opacity="0.08" stroke-width="0.5"/>
+
+  <!-- Pi icon (stylized) -->
+  <g transform="translate(80, 80)">
+    <rect x="0" y="0" width="56" height="56" rx="14" fill="url(#accent)" opacity="0.15"/>
+    <rect x="2" y="2" width="52" height="52" rx="12" fill="none" stroke="url(#accent)" stroke-width="1.5" opacity="0.3"/>
+    <text x="28" y="40" font-family="monospace" font-size="36" font-weight="bold" fill="url(#accent)" text-anchor="middle" filter="url(#glow)">π</text>
+  </g>
+
+  <!-- Title -->
+  <text x="160" y="115" font-family="system-ui, -apple-system, sans-serif" font-size="44" font-weight="800" fill="#f0ecfc" letter-spacing="-0.02em">
+    pi-free
+  </text>
+  <text x="160" y="148" font-family="system-ui, -apple-system, sans-serif" font-size="18" fill="#a78bfa" letter-spacing="0.01em" opacity="0.9">
+    Free &amp; Paid AI Model Providers for Pi
+  </text>
+
+  <!-- Subtitle -->
+  <text x="160" y="178" font-family="system-ui, -apple-system, sans-serif" font-size="13" fill="#7c8a9a" letter-spacing="0.02em">
+    </text>
+
+  <!-- Provider badges -->
+  <g transform="translate(160, 200)">
+    <!-- Free -->
+    <rect x="0" y="0" width="90" height="28" rx="6" fill="#10b981" opacity="0.15" stroke="#10b981" stroke-width="0.5" stroke-opacity="0.3"/>
+    <text x="45" y="18" font-family="system-ui, sans-serif" font-size="11" fill="#6ee7b7" text-anchor="middle" font-weight="600">✅ Free</text>
+  </g>
+
+  <g transform="translate(260, 200)">
+    <rect x="0" y="0" width="100" height="28" rx="6" fill="#f59e0b" opacity="0.15" stroke="#f59e0b" stroke-width="0.5" stroke-opacity="0.3"/>
+    <text x="50" y="18" font-family="system-ui, sans-serif" font-size="11" fill="#fcd34d" text-anchor="middle" font-weight="600">🔄 Freemium</text>
+  </g>
+
+  <g transform="translate(370, 200)">
+    <rect x="0" y="0" width="90" height="28" rx="6" fill="#7c3aed" opacity="0.15" stroke="#7c3aed" stroke-width="0.5" stroke-opacity="0.3"/>
+    <text x="45" y="18" font-family="system-ui, sans-serif" font-size="11" fill="#a78bfa" text-anchor="middle" font-weight="600">🔧 Dynamic</text>
+  </g>
+
+  <g transform="translate(470, 200)">
+    <rect x="0" y="0" width="90" height="28" rx="6" fill="#ef4444" opacity="0.15" stroke="#ef4444" stroke-width="0.5" stroke-opacity="0.3"/>
+    <text x="45" y="18" font-family="system-ui, sans-serif" font-size="11" fill="#fca5a5" text-anchor="middle" font-weight="600">💳 Paid</text>
+  </g>
+
+  <!-- Provider cards (right side) -->
+  <g transform="translate(700, 55)">
+    <!-- Card 1: Custom Providers -->
+    <rect x="0" y="0" width="250" height="170" rx="12" fill="url(#card1)" stroke="#7c3aed" stroke-width="0.5" stroke-opacity="0.2" filter="url(#shadow)"/>
+    <rect x="0" y="0" width="250" height="170" rx="12" fill="none" stroke="url(#accent)" stroke-width="0.5" opacity="0.1"/>
+    <text x="20" y="28" font-family="system-ui, sans-serif" font-size="13" font-weight="700" fill="#c4b5fd">Custom Providers</text>
+    <text x="20" y="55" font-family="system-ui, sans-serif" font-size="12" fill="#8b9aaa">Kilo · Cline</text>
+    <text x="20" y="78" font-family="system-ui, sans-serif" font-size="12" fill="#8b9aaa">NVIDIA</text>
+    <text x="20" y="101" font-family="system-ui, sans-serif" font-size="12" fill="#8b9aaa">Ollama Cloud · ZenMux</text>
+    <text x="20" y="124" font-family="system-ui, sans-serif" font-size="12" fill="#8b9aaa">CrofAI</text>
+  </g>
+
+  <g transform="translate(970, 55)">
+    <!-- Card 2: Features -->
+    <rect x="0" y="0" width="250" height="170" rx="12" fill="url(#card2)" stroke="#06b6d4" stroke-width="0.5" stroke-opacity="0.2" filter="url(#shadow)"/>
+    <rect x="0" y="0" width="250" height="170" rx="12" fill="none" stroke="url(#accent2)" stroke-width="0.5" opacity="0.1"/>
+    <text x="20" y="28" font-family="system-ui, sans-serif" font-size="13" font-weight="700" fill="#67e8f9">Features</text>
+    <text x="20" y="52" font-family="system-ui, sans-serif" font-size="12" fill="#8b9aaa">✦ Free model auto-detection</text>
+    <text x="20" y="72" font-family="system-ui, sans-serif" font-size="12" fill="#8b9aaa">✦ Per-provider toggles</text>
+    <text x="20" y="92" font-family="system-ui, sans-serif" font-size="12" fill="#8b9aaa">✦ OAuth flows (Kilo, Cline)</text>
+    <text x="20" y="112" font-family="system-ui, sans-serif" font-size="12" fill="#8b9aaa">✦ Coding Index (CI) scores</text>
+    <text x="20" y="132" font-family="system-ui, sans-serif" font-size="12" fill="#8b9aaa">✦ Model health probes</text>
+    <text x="20" y="152" font-family="system-ui, sans-serif" font-size="12" fill="#8b9aaa">✦ 404/403 auto-hide on probes</text>
+  </g>
+
+  <!-- Bottom tagline -->
+  <text x="640" y="305" font-family="system-ui, -apple-system, sans-serif" font-size="12" fill="#5a6a7a" text-anchor="middle" letter-spacing="0.04em">
+    npx pi install git:github.com/apmantza/pi-free
+  </text>
+</svg>

package/index.ts

@@ -158,7 +158,7 @@ export default async function (pi: ExtensionAPI) {
 	// 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);
+		applyGlobalFilter(pi, true);
 	}
 
 	const registry = getProviderRegistry();

package/lib/model-detection.ts

@@ -60,6 +60,83 @@ export function toProviderModelInfo(model: ProviderModelConfig): ModelInfo {
 	};
 }
 
+// =============================================================================
+// Shared helpers for model family detection
+// =============================================================================
+
+const VERSION_RE = /^v?\d+(\.\d+)?$/;
+const ROUTER_RE = /\b(?:router|auto)\b/;
+const SKIP_PARTS = new Set([
+	"latest",
+	"preview",
+	"rc",
+	"beta",
+	"alpha",
+	"dev",
+	"free",
+]);
+
+interface BrandMapping {
+	keywords: string[];
+	familyId: string;
+	familyName: string;
+}
+
+const BRAND_MAPPINGS: BrandMapping[] = [
+	{ keywords: ["claude"], familyId: "claude", familyName: "Claude" },
+	{ keywords: ["deepseek"], familyId: "deepseek", familyName: "DeepSeek" },
+	{ keywords: ["gemini"], familyId: "gemini", familyName: "Gemini" },
+	{ keywords: ["gpt"], familyId: "gpt", familyName: "GPT" },
+	{ keywords: ["llama"], familyId: "llama", familyName: "Llama" },
+	{ keywords: ["minimax"], familyId: "minimax", familyName: "MiniMax" },
+	{ keywords: ["qwen"], familyId: "qwen", familyName: "Qwen" },
+	{ keywords: ["nemotron"], familyId: "nemotron", familyName: "Nemotron" },
+	{ keywords: ["kimi", "moonshot"], familyId: "kimi", familyName: "Kimi" },
+	{ keywords: ["glm", "chatglm"], familyId: "glm", familyName: "GLM" },
+	{ keywords: ["mistral"], familyId: "mistral", familyName: "Mistral" },
+	{ keywords: ["arcee", "trinity"], familyId: "arcee", familyName: "Arcee" },
+	{ keywords: ["o1", "o3"], familyId: "openai-o", familyName: "OpenAI o" },
+];
+
+const PROVIDER_MAPPINGS: Record<
+	string,
+	{ familyId: string; familyName: string }
+> = {
+	minimax: { familyId: "minimax", familyName: "MiniMax" },
+	minimaxai: { familyId: "minimax", familyName: "MiniMax" },
+	deepseek: { familyId: "deepseek", familyName: "DeepSeek" },
+	nvidia: { familyId: "nemotron", familyName: "Nemotron" },
+	moonshot: { familyId: "kimi", familyName: "Kimi" },
+	zhipu: { familyId: "glm", familyName: "GLM" },
+};
+
+function capitalize(s: string): string {
+	return s.charAt(0).toUpperCase() + s.slice(1);
+}
+
+function findBrandInText(
+	text: string,
+): { familyId: string; familyName: string } | null {
+	for (const mapping of BRAND_MAPPINGS) {
+		for (const keyword of mapping.keywords) {
+			if (text.includes(keyword)) {
+				return { familyId: mapping.familyId, familyName: mapping.familyName };
+			}
+		}
+	}
+	return null;
+}
+
+function findBrandInParts(
+	parts: string[],
+): { familyId: string; familyName: string } | null {
+	for (const part of parts) {
+		const result = findBrandInText(part);
+		if (result) return result;
+	}
+	return null;
+}
+
 /**
  * Detect the model family from a model's ID or name.
  * Returns the family ID and display name.
@@ -72,116 +149,41 @@ export function detectModelFamily(
 	const fullText = `${id} ${name}`;
 
 	// Router models (gateways to free models) - group into "other"
-	if (/\brouter\b/.test(fullText) || /\bauto\b/.test(fullText) || id === "kilo-auto/free") {
+	if (ROUTER_RE.test(fullText) || id === "kilo-auto/free") {
 		return { familyId: "other", familyName: "Other" };
 	}
 
-	// Known brand keywords - order matters: more specific/longer matches first
-	const brandMappings: {
-		keywords: string[];
-		familyId: string;
-		familyName: string;
-	}[] = [
-		{ keywords: ["claude"], familyId: "claude", familyName: "Claude" },
-		{ keywords: ["deepseek"], familyId: "deepseek", familyName: "DeepSeek" },
-		{ keywords: ["gemini"], familyId: "gemini", familyName: "Gemini" },
-		{ keywords: ["gpt"], familyId: "gpt", familyName: "GPT" },
-		{ keywords: ["llama"], familyId: "llama", familyName: "Llama" },
-		{ keywords: ["minimax"], familyId: "minimax", familyName: "MiniMax" },
-		{ keywords: ["qwen"], familyId: "qwen", familyName: "Qwen" },
-		{ keywords: ["nemotron"], familyId: "nemotron", familyName: "Nemotron" },
-		{ keywords: ["kimi", "moonshot"], familyId: "kimi", familyName: "Kimi" },
-		{ keywords: ["glm", "chatglm"], familyId: "glm", familyName: "GLM" },
-		{ keywords: ["mistral"], familyId: "mistral", familyName: "Mistral" },
-		{ keywords: ["arcee", "trinity"], familyId: "arcee", familyName: "Arcee" },
-		{ keywords: ["o1", "o3"], familyId: "openai-o", familyName: "OpenAI o" },
-	];
-
-	// Check for known brands in ID or name
-	for (const mapping of brandMappings) {
-		for (const keyword of mapping.keywords) {
-			if (fullText.includes(keyword)) {
-				return { familyId: mapping.familyId, familyName: mapping.familyName };
-			}
-		}
-	}
+	// Known brand keywords in full text
+	const brandFromText = findBrandInText(fullText);
+	if (brandFromText) return brandFromText;
 
 	// Provider-specific fallbacks for models without brand in ID/name
-	const providerMappings: Record<string, { familyId: string; familyName: string }> = {
-		minimax: { familyId: "minimax", familyName: "MiniMax" },
-		minimaxai: { familyId: "minimax", familyName: "MiniMax" },
-		deepseek: { familyId: "deepseek", familyName: "DeepSeek" },
-		nvidia: { familyId: "nemotron", familyName: "Nemotron" },
-		moonshot: { familyId: "kimi", familyName: "Kimi" },
-		zhipu: { familyId: "glm", familyName: "GLM" },
-	};
-
-	if (providerMappings[model.provider]) {
-		return providerMappings[model.provider];
-	}
-
-	// Helper to find brand in ID parts
-	function findBrandInParts(parts: string[]): { familyId: string; familyName: string } | null {
-		for (const part of parts) {
-			for (const mapping of brandMappings) {
-				for (const keyword of mapping.keywords) {
-					if (part.includes(keyword)) {
-						return { familyId: mapping.familyId, familyName: mapping.familyName };
-					}
-				}
-			}
-		}
-		return null;
-	}
+	const providerResult = PROVIDER_MAPPINGS[model.provider];
+	if (providerResult) return providerResult;
 
-	// Smart fallback: try to identify brand from model ID structure
+	// Fallback: try to identify brand from model ID structure
 	const parts = id.split(/[-_:.@]/);
 	const firstPart = parts[0];
 
-	// If ID starts with a version number, check remaining parts for brand
-	if (firstPart && /^v?\d+(\.\d+)?$/.test(firstPart)) {
-		const brandFromParts = findBrandInParts(parts.slice(1));
-		if (brandFromParts) {
-			return brandFromParts;
-		}
-	}
-
-	// If ID has multiple parts, check ALL parts for brand keywords
-	if (parts.length > 1) {
-		const brandFromParts = findBrandInParts(parts);
-		if (brandFromParts) {
-			return brandFromParts;
-		}
+	const brandFromParts = findBrandInParts(parts);
+	if (brandFromParts) return brandFromParts;
 
-		// Use first part as brand if it looks brand-like
-		if (firstPart && !/^v?\d+(\.\d+)?$/.test(firstPart)) {
-			return {
-				familyId: firstPart,
-				familyName: firstPart.charAt(0).toUpperCase() + firstPart.slice(1),
-			};
-		}
+	// Use first part as brand if it looks brand-like
+	if (firstPart && !VERSION_RE.test(firstPart)) {
+		return { familyId: firstPart, familyName: capitalize(firstPart) };
 	}
 
-	// Last resort: use first non-version part
-	if (firstPart && /^v?\d+(\.\d+)?$/.test(firstPart) && parts.length > 1) {
-		for (let i = 1; i < parts.length; i++) {
-			const part = parts[i];
-			if (
-				part &&
-				!/^v?\d+(\.\d+)?$/.test(part) &&
-				!["latest", "preview", "rc", "beta", "alpha", "dev", "free"].includes(part)
-			) {
-				return {
-					familyId: part,
-					familyName: part.charAt(0).toUpperCase() + part.slice(1),
-				};
-			}
-		}
+	// First non-version, non-skip part
+	const nonVersion = parts.find(
+		(p) => p && !VERSION_RE.test(p) && !SKIP_PARTS.has(p),
+	);
+	if (nonVersion) {
+		return { familyId: nonVersion, familyName: capitalize(nonVersion) };
 	}
 
 	return {
 		familyId: firstPart || id,
-		familyName: (firstPart || id).charAt(0).toUpperCase() + (firstPart || id).slice(1),
+		familyName: capitalize(firstPart || id),
 	};
 }
 
@@ -189,21 +191,83 @@ export function detectModelFamily(
  * Normalize a model name for comparison by removing provider-specific suffixes
  * and common qualifiers. This helps detect when the same model is offered by
  * multiple providers with slightly different naming.
+ *
+ * Uses string operations instead of regex backtracking to avoid ReDoS warnings.
  */
 export function normalizeModelName(name: string): string {
-	return (
-		name
-			.toLowerCase()
-			// Remove common suffixes added by providers
-			.replace(/\s*\(free\)\s*$/i, "")
-			.replace(/\s*\(cline\)\s*$/i, "")
-			.replace(/\s*\(ci:\s*[\d.]+\)\s*$/i, "")
-			.replace(/\s*\[ci:\s*[\d.]+\]\s*$/i, "")
-			.replace(/\s*\([^)]*\)\s*$/g, "") // Remove any trailing parenthetical
-			.replace(/\s*-\s*free\s*$/i, "") // e.g., "minimax-m2.5-free"
-			.replace(/\s*free\s*$/i, "") // trailing "free"
-			.trim()
-	);
+	const suffixes = ["(free)", "(cline)", "-free", "free"];
+	let normalized = name.toLowerCase().trimEnd();
+
+	// Remove common literal suffixes — simple string ops, no regex backtracking
+	for (const suffix of suffixes) {
+		while (normalized.endsWith(suffix)) {
+			normalized = normalized.slice(0, -suffix.length).trimEnd();
+		}
+	}
+
+	// CI score suffix — regex with disjoint char classes (linear)
+	normalized = normalized.replace(/\(ci:\s*[\d.]+\)$/, "").trimEnd();
+	normalized = normalized.replace(/\[ci:\s*[\d.]+\]$/, "").trimEnd();
+
+	// Remove any trailing parenthetical — non-regex loop
+	while (normalized.endsWith(")")) {
+		const idx = normalized.lastIndexOf("(", normalized.length - 1);
+		if (idx === -1) break;
+		normalized = normalized.slice(0, idx).trimEnd();
+	}
+
+	return normalized.trim();
+}
+
+/**
+ * Try to merge a model into another family if its normalized name
+ * matches a model in a different family.
+ */
+function tryMergeFamily(
+	byFamily: Map<string, ModelInfo[]>,
+	nameToFamilyId: Map<string, string>,
+	familyId: string,
+	model: ModelInfo,
+): boolean {
+	const normalizedName = normalizeModelName(model.name || model.id);
+	if (!normalizedName) return false;
+
+	const existingFamilyForName = nameToFamilyId.get(normalizedName);
+	if (!existingFamilyForName || existingFamilyForName === familyId) {
+		nameToFamilyId.set(normalizedName, familyId);
+		return false;
+	}
+
+	// Same model name found in different family - merge them
+	const targetFamily = byFamily.get(existingFamilyForName);
+	const sourceFamily = byFamily.get(familyId);
+	if (!targetFamily || !sourceFamily) return false;
+
+	targetFamily.push(...sourceFamily);
+	byFamily.delete(familyId);
+	return true;
+}
+
+/**
+ * Build a sorted list of ModelFamily from a by-family grouping map.
+ */
+function buildFamiliesList(byFamily: Map<string, ModelInfo[]>): ModelFamily[] {
+	const families: ModelFamily[] = [];
+	for (const [id, familyModels] of byFamily) {
+		const firstModel = familyModels[0]!;
+		const familyInfo = detectModelFamily(firstModel)!;
+
+		families.push({
+			id,
+			displayName: familyInfo.familyName,
+			models: familyModels.sort(
+				(a, b) =>
+					a.provider.localeCompare(b.provider) || b.id.localeCompare(a.id),
+			),
+		});
+	}
+
+	return families.sort((a, b) => a.displayName.localeCompare(b.displayName));
 }
 
 /**
@@ -214,6 +278,7 @@ export function getModelFamilies(models: ModelInfo[]): ModelFamily[] {
 	const byFamily = new Map<string, ModelInfo[]>();
 	const nameToFamilyId = new Map<string, string>();
 
+	// First pass: group models by detected family
 	for (const model of models) {
 		const family = detectModelFamily(model);
 		if (!family) continue;
@@ -223,46 +288,18 @@ export function getModelFamilies(models: ModelInfo[]): ModelFamily[] {
 		byFamily.set(family.familyId, existing);
 	}
 
-	// Second pass: merge families with models that have the same normalized name
+	// Second pass: merge families whose models have the same normalized name
 	const familyIds = [...byFamily.keys()];
 	for (const familyId of familyIds) {
 		const familyModels = byFamily.get(familyId);
 		if (!familyModels) continue;
 
 		for (const model of familyModels) {
-			const normalizedName = normalizeModelName(model.name || model.id);
-			if (!normalizedName) continue;
-
-			const existingFamilyForName = nameToFamilyId.get(normalizedName);
-			if (existingFamilyForName && existingFamilyForName !== familyId) {
-				// Same model name found in different family - merge them
-				const targetFamily = byFamily.get(existingFamilyForName);
-				const sourceFamily = byFamily.get(familyId);
-				if (targetFamily && sourceFamily) {
-					targetFamily.push(...sourceFamily);
-					byFamily.delete(familyId);
-					break;
-				}
-			} else {
-				nameToFamilyId.set(normalizedName, familyId);
+			if (tryMergeFamily(byFamily, nameToFamilyId, familyId, model)) {
+				break;
 			}
 		}
 	}
 
-	const families: ModelFamily[] = [];
-	for (const [id, familyModels] of byFamily) {
-		const firstModel = familyModels[0]!;
-		const familyInfo = detectModelFamily(firstModel)!;
-
-		families.push({
-			id,
-			displayName: familyInfo.familyName,
-			models: familyModels.sort(
-				(a, b) =>
-					a.provider.localeCompare(b.provider) || b.id.localeCompare(a.id),
-			),
-		});
-	}
-
-	return families.sort((a, b) => a.displayName.localeCompare(b.displayName));
+	return buildFamiliesList(byFamily);
 }

package/lib/registry.ts

@@ -156,33 +156,40 @@ export function getProviderRegistry(): ReadonlyMap<string, ProviderEntry> {
 // Global filter application
 // =============================================================================
 
+function applyFilterToProvider(
+	providerId: string,
+	entry: ProviderEntry,
+	freeOnly: boolean,
+): void {
+	if (freeOnly) {
+		if (entry.stored.free.length > 0) {
+			entry.reRegister(entry.stored.free);
+			_logger.info(
+				`[pi-free] ${providerId}: filtered to ${entry.stored.free.length} free models`,
+			);
+		} else {
+			_logger.warn(`[pi-free] ${providerId}: no free models available`);
+		}
+	} else {
+		// Show all models (paid + free)
+		const allModels =
+			entry.stored.all.length > 0 ? entry.stored.all : entry.stored.free;
+		if (allModels.length > 0) {
+			entry.reRegister(allModels);
+			_logger.info(
+				`[pi-free] ${providerId}: showing all ${allModels.length} models`,
+			);
+		}
+	}
+}
+
 export function applyGlobalFilter(_pi: ExtensionAPI, freeOnly: boolean): void {
 	globalFreeOnly = freeOnly;
 	saveConfig({ free_only: freeOnly });
 
 	for (const [providerId, entry] of providerRegistry) {
 		try {
-			if (freeOnly) {
-				// Show only free models
-				if (entry.stored.free.length > 0) {
-					entry.reRegister(entry.stored.free);
-					_logger.info(
-						`[pi-free] ${providerId}: filtered to ${entry.stored.free.length} free models`,
-					);
-				} else {
-					_logger.warn(`[pi-free] ${providerId}: no free models available`);
-				}
-			} else {
-				// Show all models (paid + free)
-				const allModels =
-					entry.stored.all.length > 0 ? entry.stored.all : entry.stored.free;
-				if (allModels.length > 0) {
-					entry.reRegister(allModels);
-					_logger.info(
-						`[pi-free] ${providerId}: showing all ${allModels.length} models`,
-					);
-				}
-			}
+			applyFilterToProvider(providerId, entry, freeOnly);
 		} catch (err) {
 			_logger.error(
 				`[pi-free] Failed to apply filter to ${providerId}`,

package/lib/util.ts

@@ -205,10 +205,16 @@ export function isUsableModel(modelId: string, minSizeB?: number): boolean {
  */
 export function cleanModelName(name: string): string {
 	// Handle patterns like "Provider : Model Name" or "Provider / Model Name"
-	// Match colon or slash separator with optional surrounding whitespace
-	const separatorMatch = name.match(/^[^:]+\s*[:/]\s*(.+)$/);
-	if (separatorMatch) {
-		return separatorMatch[1].trim();
+	const colonIdx = name.indexOf(":");
+	const slashIdx = name.indexOf("/");
+	const idx =
+		colonIdx === -1
+			? slashIdx
+			: slashIdx === -1
+				? colonIdx
+				: Math.min(colonIdx, slashIdx);
+	if (idx > 0) {
+		return name.slice(idx + 1).trim();
 	}
 	return name.trim();
 }

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "pi-free",
-	"version": "2.0.4",
+	"version": "2.0.5",
 	"type": "module",
 	"description": "AI model providers for Pi with free model filtering. Shows only $0 cost models by default. Supports Kilo (free OAuth), Cline (free), NVIDIA (freemium), ZenMux, CrofAI, Ollama Cloud, and more.",
 	"keywords": [
@@ -40,6 +40,7 @@
 		"README.md",
 		"LICENSE",
 		"CHANGELOG.md",
+		"banner.svg",
 		"scripts/check-extensions.mjs"
 	],
 	"scripts": {