pi-free 2.0.7 → 2.0.9

package/config.ts

@@ -11,7 +11,7 @@
 
 import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
 import { join } from "node:path";
-import {
+export {
 	PROVIDER_CLINE,
 	PROVIDER_KILO,
 	PROVIDER_MODAL,
@@ -27,7 +27,11 @@ interface PiFreeConfig {
 	ollama_api_key?: string;
 	zenmux_api_key?: string;
 	crofai_api_key?: string;
+	codestral_api_key?: string;
 	mistral_api_key?: string;
+	llm7_api_key?: string;
+	deepinfra_api_key?: string;
+	sambanova_api_key?: string;
 	groq_api_key?: string;
 	cerebras_api_key?: string;
 	xai_api_key?: string;
@@ -40,6 +44,10 @@ interface PiFreeConfig {
 	cline_show_paid?: boolean;
 	zenmux_show_paid?: boolean;
 	crofai_show_paid?: boolean;
+	codestral_show_paid?: boolean;
+	llm7_show_paid?: boolean;
+	deepinfra_show_paid?: boolean;
+	sambanova_show_paid?: boolean;
 	openrouter_show_paid?: boolean;
 	opencode_show_paid?: boolean;
 }
@@ -49,7 +57,11 @@ const CONFIG_TEMPLATE: PiFreeConfig = {
 	ollama_api_key: "",
 	zenmux_api_key: "",
 	crofai_api_key: "",
+	codestral_api_key: "",
 	mistral_api_key: "",
+	llm7_api_key: "",
+	deepinfra_api_key: "",
+	sambanova_api_key: "",
 	groq_api_key: "",
 	cerebras_api_key: "",
 	xai_api_key: "",
@@ -63,6 +75,10 @@ const CONFIG_TEMPLATE: PiFreeConfig = {
 	cline_show_paid: false,
 	zenmux_show_paid: false,
 	crofai_show_paid: false,
+	codestral_show_paid: false,
+	llm7_show_paid: false,
+	deepinfra_show_paid: false,
+	sambanova_show_paid: false,
 	openrouter_show_paid: false,
 	opencode_show_paid: false,
 };
@@ -147,6 +163,31 @@ export function getCrofaiShowPaid(): boolean {
 	return resolveBool("CROFAI_SHOW_PAID", loadConfigFile().crofai_show_paid);
 }
 
+export function getCodestralShowPaid(): boolean {
+	return resolveBool(
+		"CODESTRAL_SHOW_PAID",
+		loadConfigFile().codestral_show_paid,
+	);
+}
+
+export function getLlm7ShowPaid(): boolean {
+	return resolveBool("LLM7_SHOW_PAID", loadConfigFile().llm7_show_paid);
+}
+
+export function getDeepinfraShowPaid(): boolean {
+	return resolveBool(
+		"DEEPINFRA_SHOW_PAID",
+		loadConfigFile().deepinfra_show_paid,
+	);
+}
+
+export function getSambanovaShowPaid(): boolean {
+	return resolveBool(
+		"SAMBANOVA_SHOW_PAID",
+		loadConfigFile().sambanova_show_paid,
+	);
+}
+
 export function getOllamaShowPaid(): boolean {
 	return resolveBool("OLLAMA_SHOW_PAID", loadConfigFile().ollama_show_paid);
 }
@@ -190,6 +231,22 @@ export function getCrofaiApiKey(): string | undefined {
 	return resolve("CROFAI_API_KEY", loadConfigFile().crofai_api_key);
 }
 
+export function getCodestralApiKey(): string | undefined {
+	return resolve("CODESTRAL_API_KEY", loadConfigFile().codestral_api_key);
+}
+
+export function getLlm7ApiKey(): string | undefined {
+	return resolve("LLM7_API_KEY", loadConfigFile().llm7_api_key);
+}
+
+export function getDeepinfraApiKey(): string | undefined {
+	return resolve("DEEPINFRA_TOKEN", loadConfigFile().deepinfra_api_key);
+}
+
+export function getSambanovaApiKey(): string | undefined {
+	return resolve("SAMBANOVA_API_KEY", loadConfigFile().sambanova_api_key);
+}
+
 export function getOllamaApiKey(): string | undefined {
 	return resolve("OLLAMA_API_KEY", loadConfigFile().ollama_api_key);
 }
@@ -278,13 +335,3 @@ export function getConfig(): PiFreeConfig {
 }
 
 // =============================================================================
-// Re-export provider names for consistency
-// =============================================================================
-
-export {
-	PROVIDER_CLINE,
-	PROVIDER_KILO,
-	PROVIDER_MODAL,
-	PROVIDER_NVIDIA,
-	PROVIDER_QWEN,
-};

package/constants.ts

@@ -17,6 +17,10 @@ export const PROVIDER_QWEN = "qwen";
 export const PROVIDER_MODAL = "modal";
 export const PROVIDER_ZENMUX = "zenmux";
 export const PROVIDER_CROFAI = "crofai";
+export const PROVIDER_CODESTRAL = "codestral";
+export const PROVIDER_LLM7 = "llm7";
+export const PROVIDER_DEEPINFRA = "deepinfra";
+export const PROVIDER_SAMBANOVA = "sambanova";
 
 export const ALL_UNIQUE_PROVIDERS = [
 	PROVIDER_KILO,
@@ -28,6 +32,10 @@ export const ALL_UNIQUE_PROVIDERS = [
 	PROVIDER_OLLAMA,
 	PROVIDER_ZENMUX,
 	PROVIDER_CROFAI,
+	PROVIDER_CODESTRAL,
+	PROVIDER_LLM7,
+	PROVIDER_DEEPINFRA,
+	PROVIDER_SAMBANOVA,
 ] as const;
 
 // =============================================================================
@@ -44,6 +52,10 @@ 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";
+export const BASE_URL_CODESTRAL = "https://codestral.mistral.ai/v1";
+export const BASE_URL_LLM7 = "https://api.llm7.io/v1";
+export const BASE_URL_DEEPINFRA = "https://api.deepinfra.com/v1/openai";
+export const BASE_URL_SAMBANOVA = "https://api.sambanova.ai/v1";
 
 /** Cline fetches free models from OpenRouter */
 export const BASE_URL_OPENROUTER = "https://openrouter.ai/api/v1";

package/index.ts

@@ -10,11 +10,19 @@
  * - 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
+ * - Codestral: Mistral's code-focused model via codestral.mistral.ai (free tier)
+ * - DeepInfra: AI inference cloud ($5 trial credit)
+ * - SambaNova: Fast inference on RDU hardware (free tier, no credit card)
+ * - LLM7: AI gateway (free default/fast selectors)
  */
 
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import { setupBuiltInProviderToggles } from "./lib/built-in-toggle.ts";
 import { createLogger } from "./lib/logger.ts";
+import {
+	processQuotaResponse,
+	formatQuotaStatus,
+} from "./lib/quota-monitor.ts";
 import {
 	applyGlobalFilter,
 	getGlobalFreeOnly,
@@ -24,8 +32,12 @@ import {
 } from "./lib/registry.ts";
 // Import unique provider extensions (only providers NOT built into pi)
 import cline from "./providers/cline/cline.ts";
+import codestral from "./providers/codestral/codestral.ts";
 import crofai from "./providers/crofai/crofai.ts";
 import kilo from "./providers/kilo/kilo.ts";
+import llm7 from "./providers/llm7/llm7.ts";
+import deepinfra from "./providers/deepinfra/deepinfra.ts";
+import sambanova from "./providers/sambanova/sambanova.ts";
 import nvidia from "./providers/nvidia/nvidia.ts";
 import ollama from "./providers/ollama/ollama.ts";
 import zenmux from "./providers/zenmux/zenmux.ts";
@@ -87,7 +99,13 @@ function setupGlobalCommands(pi: ExtensionAPI) {
 				"cerebras",
 			]);
 			// Freemium providers - all models share a free tier quota
-			const freemiumProviders = new Set(["nvidia"]);
+			const freemiumProviders = new Set([
+				"nvidia",
+				"sambanova",
+				"ollama-cloud",
+			]);
+			// Trial credit providers - one-time credits, otherwise paid
+			const trialCreditProviders = new Set(["deepinfra"]);
 
 			for (const [id, entry] of registry) {
 				const free = entry.stored.free.length;
@@ -98,6 +116,9 @@ function setupGlobalCommands(pi: ExtensionAPI) {
 				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 (trialCreditProviders.has(id)) {
+					// Trial credit: one-time credits, otherwise paid
+					lines.push(`${indicator} ${id}: ${all} models ($5 trial credit)`);
 				} else if (noPricingApi.has(id)) {
 					// Provider doesn't expose pricing - can't determine free vs paid
 					lines.push(
@@ -123,17 +144,56 @@ function setupGlobalCommands(pi: ExtensionAPI) {
 	});
 }
 
+// =============================================================================
+// Quota Monitoring
+// =============================================================================
+
+function setupQuotaMonitoring(pi: ExtensionAPI) {
+	// Capture rate-limit headers from every provider response
+	// eslint-disable-next-line @typescript-eslint/no-explicit-any
+	(pi as any).on(
+		"after_provider_response",
+		(event: { status: number; headers: Record<string, string> }, ctx: any) => {
+			const providerId = ctx.model?.provider;
+			if (!providerId) return;
+
+			processQuotaResponse(providerId, event.headers);
+
+			// Update status bar with quota for the active provider
+			const status = formatQuotaStatus(providerId);
+			if (status) {
+				ctx.ui.setStatus("quota", status);
+			}
+		},
+	);
+
+	// Clear quota status when switching away from a provider
+	pi.on("model_select", (_event, ctx) => {
+		const providerId = ctx.model?.provider;
+		if (!providerId) {
+			ctx.ui.setStatus("quota", undefined);
+			return;
+		}
+		// Show cached quota on provider switch (if still fresh)
+		const status = formatQuotaStatus(providerId);
+		ctx.ui.setStatus("quota", status);
+	});
+}
+
 // =============================================================================
 // Main Entry Point
 // =============================================================================
 
-export default async function (pi: ExtensionAPI) {
+export default async function piFreeEntry(pi: ExtensionAPI) {
 	const globalFreeOnly = getGlobalFreeOnly();
 	_logger.info(`[pi-free] Initializing (global free-only: ${globalFreeOnly})`);
 
 	// Setup global commands first
 	setupGlobalCommands(pi);
 
+	// Setup quota monitoring (passive, no extra API calls)
+	setupQuotaMonitoring(pi);
+
 	// Load all unique providers
 	// Each provider will register itself with the global toggle system
 	await Promise.allSettled([
@@ -143,6 +203,10 @@ export default async function (pi: ExtensionAPI) {
 		cline(pi),
 		zenmux(pi),
 		crofai(pi),
+		codestral(pi),
+		llm7(pi),
+		deepinfra(pi),
+		sambanova(pi),
 	]);
 
 	// Setup dynamic built-in providers (Mistral, Groq, Cerebras, xAI, Hugging Face)

package/lib/built-in-toggle.ts

@@ -11,11 +11,11 @@
  * Usage: /toggle-opencode
  */
 
-import type { Api, Model } from "@mariozechner/pi-ai";
+import type { Api, Model } from "@earendil-works/pi-ai";
 import type {
 	ExtensionAPI,
 	ProviderModelConfig,
-} from "@mariozechner/pi-coding-agent";
+} from "@earendil-works/pi-coding-agent";
 import { getOpencodeShowPaid, getOpenrouterShowPaid } from "../config.ts";
 import { createLogger } from "./logger.ts";
 import { isFreeModel, registerWithGlobalToggle } from "./registry.ts";

package/lib/model-detection.ts

@@ -4,7 +4,7 @@
  * Used for failover when providers hit rate limits.
  */
 
-import type { Model } from "@mariozechner/pi-ai";
+import type { Model } from "@earendil-works/pi-ai";
 import type { ProviderModelConfig } from "./types.ts";
 
 export interface ModelInfo {
@@ -206,6 +206,7 @@ export function normalizeModelName(name: string): string {
 	}
 
 	// CI score suffix — regex with disjoint char classes (linear)
+	// Anchored with $, matches at most once → .replace() is correct (S4144 N/A)
 	normalized = normalized.replace(/\(ci:\s*[\d.]+\)$/, "").trimEnd();
 	normalized = normalized.replace(/\[ci:\s*[\d.]+\]$/, "").trimEnd();
 

package/lib/model-enhancer.ts

@@ -3,7 +3,7 @@
  * Adds Coding Index scores to model names for display in /model
  */
 
-import type { ProviderModelConfig } from "@mariozechner/pi-coding-agent";
+import type { ProviderModelConfig } from "@earendil-works/pi-coding-agent";
 import { enhanceModelNameWithCodingIndex } from "../provider-failover/benchmark-lookup.ts";
 
 /**

package/lib/open-browser.ts

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

package/lib/provider-compat.ts

@@ -1,4 +1,4 @@
-import type { ProviderModelConfig } from "@mariozechner/pi-coding-agent";
+import type { ProviderModelConfig } from "@earendil-works/pi-coding-agent";
 
 export interface ProviderModelIdentity {
 	id: string;

package/lib/quota-monitor.ts

@@ -0,0 +1,123 @@
+/**
+ * Quota Monitoring for pi-free providers.
+ *
+ * Subscribes to pi's `after_provider_response` event to extract rate-limit
+ * headers from provider responses and track remaining quota per provider.
+ *
+ * Inspired by free-coding-models' extractQuotaPercent and provider-quota-fetchers.
+ *
+ * Supported header formats (tried in order):
+ *   1. x-ratelimit-remaining-requests / x-ratelimit-limit-requests (SambaNova)
+ *   2. x-ratelimit-remaining / x-ratelimit-limit (Mistral, others)
+ *   3. ratelimit-remaining-requests / ratelimit-limit-requests
+ *   4. ratelimit-remaining / ratelimit-limit
+ *   5. x-ratelimit-remaining-requests-day / x-ratelimit-limit-requests-day (SambaNova daily)
+ */
+
+const _quotaState = new Map<string, QuotaSnapshot>();
+
+/** Snapshot of quota state for a single provider. */
+export interface QuotaSnapshot {
+	/** Requests remaining in the current window. */
+	remaining: number;
+	/** Total requests allowed in the current window. */
+	limit: number;
+	/** Remaining as percentage 0–100. */
+	percent: number;
+	/** Timestamp (Date.now()) when this snapshot was captured. */
+	lastUpdated: number;
+	/** Which header variant was matched (for debugging). */
+	source: string;
+}
+
+// Header key pairs to try, in priority order.
+// Each pair is [remaining, limit].
+const HEADER_PAIRS: [string, string][] = [
+	// Per-minute (most common)
+	["x-ratelimit-remaining-requests", "x-ratelimit-limit-requests"],
+	["x-ratelimit-remaining", "x-ratelimit-limit"],
+	["ratelimit-remaining-requests", "ratelimit-limit-requests"],
+	["ratelimit-remaining", "ratelimit-limit"],
+	// Per-day
+	["x-ratelimit-remaining-requests-day", "x-ratelimit-limit-requests-day"],
+	["x-ratelimit-remaining-day", "x-ratelimit-limit-day"],
+];
+
+/**
+ * Attempt to extract quota from response headers.
+ * Returns { remaining, limit, source } or null if no quota headers found.
+ */
+export function extractQuota(
+	headers: Record<string, string>,
+): { remaining: number; limit: number; source: string } | null {
+	// Normalize keys to lowercase for case-insensitive matching.
+	// Some proxies/servers vary header casing.
+	const normalized: Record<string, string> = {};
+	for (const [key, value] of Object.entries(headers)) {
+		normalized[key.toLowerCase()] = value;
+	}
+
+	for (const [remainingKey, limitKey] of HEADER_PAIRS) {
+		const remaining = Number.parseFloat(normalized[remainingKey]);
+		const limit = Number.parseFloat(normalized[limitKey]);
+		if (Number.isFinite(remaining) && Number.isFinite(limit) && limit > 0) {
+			return { remaining, limit, source: remainingKey };
+		}
+	}
+
+	return null;
+}
+
+/**
+ * Process an after_provider_response event, updating quota state.
+ * Call from the event handler in index.ts.
+ */
+export function processQuotaResponse(
+	providerId: string,
+	headers: Record<string, string>,
+): void {
+	const extracted = extractQuota(headers);
+	if (!extracted) return;
+
+	const percent = Math.round((extracted.remaining / extracted.limit) * 100);
+
+	_quotaState.set(providerId, {
+		remaining: extracted.remaining,
+		limit: extracted.limit,
+		percent: Math.max(0, Math.min(100, percent)),
+		lastUpdated: Date.now(),
+		source: extracted.source,
+	});
+}
+
+/**
+ * Get the latest quota snapshot for a provider, or null if unknown.
+ */
+export function getQuota(providerId: string): QuotaSnapshot | null {
+	return _quotaState.get(providerId) ?? null;
+}
+
+/**
+ * Get all tracked quotas.
+ */
+export function getAllQuotas(): ReadonlyMap<string, QuotaSnapshot> {
+	return _quotaState;
+}
+
+/**
+ * Build a human-readable status bar line for a provider's quota.
+ * Returns undefined if no quota data is available.
+ */
+export function formatQuotaStatus(providerId: string): string | undefined {
+	const q = _quotaState.get(providerId);
+	if (!q) return undefined;
+
+	// Stale after 5 minutes
+	if (Date.now() - q.lastUpdated > 5 * 60 * 1000) return undefined;
+
+	if (q.percent <= 10)
+		return `⚠️ ${providerId}: ${q.remaining}/${q.limit} (${q.percent}%)`;
+	if (q.percent <= 25)
+		return `⚡ ${providerId}: ${q.remaining}/${q.limit} (${q.percent}%)`;
+	return `${providerId}: ${q.remaining}/${q.limit} (${q.percent}%)`;
+}

package/lib/registry.ts

@@ -8,7 +8,7 @@
 import type {
 	ExtensionAPI,
 	ProviderModelConfig,
-} from "@mariozechner/pi-coding-agent";
+} from "@earendil-works/pi-coding-agent";
 import { getFreeOnly, saveConfig } from "../config.ts";
 import { createLogger } from "./logger.ts";
 

package/lib/types.ts

@@ -1,101 +1,101 @@
-/**
- * Shared types for pi-free-providers.
- * Interfaces duplicated across providers consolidated here.
- */
-
-// =============================================================================
-// Provider model configuration (matches Pi's ProviderModelConfig)
-// =============================================================================
-
-export interface CostConfig {
-	input: number;
-	output: number;
-	cacheRead: number;
-	cacheWrite: number;
-}
-
-export interface ProviderModelConfig {
-	id: string;
-	name: string;
-	reasoning: boolean;
-	input: ("text" | "image")[];
-	cost: CostConfig;
-	contextWindow: number;
-	maxTokens: number;
-}
-
-// =============================================================================
-// models.dev schema types
-// =============================================================================
-
-export interface ModelsDevCost {
-	input: number;
-	output: number;
-	cache_read?: number;
-	cache_write?: number;
-}
-
-export interface ModelsDevLimit {
-	context: number;
-	output: number;
-}
-
-export interface ModelsDevModalities {
-	input?: string[];
-	output?: string[];
-}
-
-export interface ModelsDevModel {
-	id: string;
-	name: string;
-	reasoning: boolean;
-	cost?: ModelsDevCost;
-	limit: ModelsDevLimit;
-	modalities?: ModelsDevModalities;
-}
-
-export interface ModelsDevProvider {
-	id: string;
-	api: string;
-	models: Record<string, ModelsDevModel>;
-}
-
-// =============================================================================
-// OpenRouter API response types
-// =============================================================================
-
-export interface OpenRouterPricing {
-	prompt?: string | null;
-	completion?: string | null;
-	input_cache_write?: string | null;
-	input_cache_read?: string | null;
-}
-
-export interface OpenRouterArchitecture {
-	input_modalities?: string[] | null;
-	output_modalities?: string[] | null;
-}
-
-export interface OpenRouterTopProvider {
-	max_completion_tokens?: number | null;
-}
-
-export interface OpenRouterModel {
-	id: string;
-	name: string;
-	context_length: number;
-	max_completion_tokens?: number | null;
-	pricing?: OpenRouterPricing;
-	architecture?: OpenRouterArchitecture;
-	top_provider?: OpenRouterTopProvider;
-	supported_parameters?: string[];
-}
-
-// =============================================================================
-// Zen gateway types
-// =============================================================================
-
-export interface ZenGatewayModel {
-	id: string;
-	object?: string;
-}
+/**
+ * Shared types for pi-free-providers.
+ * Interfaces duplicated across providers consolidated here.
+ */
+
+// =============================================================================
+// Provider model configuration (matches Pi's ProviderModelConfig)
+// =============================================================================
+
+export interface CostConfig {
+	input: number;
+	output: number;
+	cacheRead: number;
+	cacheWrite: number;
+}
+
+export interface ProviderModelConfig {
+	id: string;
+	name: string;
+	reasoning: boolean;
+	input: ("text" | "image")[];
+	cost: CostConfig;
+	contextWindow: number;
+	maxTokens: number;
+}
+
+// =============================================================================
+// models.dev schema types
+// =============================================================================
+
+export interface ModelsDevCost {
+	input: number;
+	output: number;
+	cache_read?: number;
+	cache_write?: number;
+}
+
+export interface ModelsDevLimit {
+	context: number;
+	output: number;
+}
+
+export interface ModelsDevModalities {
+	input?: string[];
+	output?: string[];
+}
+
+export interface ModelsDevModel {
+	id: string;
+	name: string;
+	reasoning: boolean;
+	cost?: ModelsDevCost;
+	limit: ModelsDevLimit;
+	modalities?: ModelsDevModalities;
+}
+
+export interface ModelsDevProvider {
+	id: string;
+	api: string;
+	models: Record<string, ModelsDevModel>;
+}
+
+// =============================================================================
+// OpenRouter API response types
+// =============================================================================
+
+export interface OpenRouterPricing {
+	prompt?: string | null;
+	completion?: string | null;
+	input_cache_write?: string | null;
+	input_cache_read?: string | null;
+}
+
+export interface OpenRouterArchitecture {
+	input_modalities?: string[] | null;
+	output_modalities?: string[] | null;
+}
+
+export interface OpenRouterTopProvider {
+	max_completion_tokens?: number | null;
+}
+
+export interface OpenRouterModel {
+	id: string;
+	name: string;
+	context_length: number;
+	max_completion_tokens?: number | null;
+	pricing?: OpenRouterPricing;
+	architecture?: OpenRouterArchitecture;
+	top_provider?: OpenRouterTopProvider;
+	supported_parameters?: string[];
+}
+
+// =============================================================================
+// Zen gateway types
+// =============================================================================
+
+export interface ZenGatewayModel {
+	id: string;
+	object?: string;
+}