pi-free 2.0.6 → 2.0.8

package/providers/cline/cline-models.ts

@@ -43,8 +43,8 @@ function extractNameFromId(id: string): string {
  */
 function parsePricing(pricingStr: string | undefined): number {
 	if (!pricingStr || pricingStr === "0") return 0;
-	const parsed = parseFloat(pricingStr);
-	return isNaN(parsed) ? 0 : parsed * 1_000_000; // Convert to per-million
+	const parsed = Number.parseFloat(pricingStr);
+	return Number.isNaN(parsed) ? 0 : parsed * 1_000_000; // Convert to per-million
 }
 
 /**

package/providers/cline/cline.ts

@@ -210,7 +210,7 @@ function shapeMessagesForCline(messages: any[]): any[] {
 // Extension entry point
 // =============================================================================
 
-export default async function (pi: ExtensionAPI) {
+export default async function clineProvider(pi: ExtensionAPI) {
 	let allModels = await fetchClineModels(false).catch((err) => {
 		logWarning("cline", "Failed to fetch models at startup", err);
 		return [];

package/providers/codestral/codestral.ts

@@ -0,0 +1,139 @@
+/**
+ * Codestral Provider Extension
+ *
+ * Codestral is Mistral AI's code-focused model. This provider registers it
+ * through the Codestral-specific endpoint (codestral.mistral.ai) using
+ * the Mistral SDK (api: "mistral-conversations") — separate from the built-in
+ * "mistral" provider which uses api.mistral.ai.
+ *
+ * NOTE: Do NOT use api: "openai-completions" here. Codestral's API is
+ * Mistral-format (camelCase fields, maxTokens, no stream_options/store).
+ * The OpenAI completions adapter sends OpenAI-specific fields that Mistral
+ * rejects with HTTP 422 "Extra inputs are not permitted".
+ *
+ * Free tier (Experiment plan):
+ *   - 2 req/min, 500K tokens/min, 1B tokens/month
+ *   - No credit card — phone verification only
+ *   - Sign up at https://console.mistral.ai/codestral
+ *
+ * Endpoints:
+ *   Chat:  https://codestral.mistral.ai/v1/chat/completions
+ *   FIM:   https://codestral.mistral.ai/v1/fim/completions (not used by pi)
+ *
+ * Setup:
+ *   1. Get API key from https://console.mistral.ai/codestral
+ *   2. Set CODESTRAL_API_KEY env var (or MISTRAL_API_KEY as fallback)
+ *
+ * Usage:
+ *   pi install git:github.com/apmantza/pi-free
+ *   # Set CODESTRAL_API_KEY env var
+ *   # Models appear in /model selector as "codestral/codestral-latest"
+ */
+
+import type {
+	ExtensionAPI,
+	ProviderModelConfig,
+} from "@mariozechner/pi-coding-agent";
+import {
+	getCodestralApiKey,
+	getCodestralShowPaid,
+	getMistralApiKey,
+} from "../../config.ts";
+import { BASE_URL_CODESTRAL, PROVIDER_CODESTRAL } from "../../constants.ts";
+import { createLogger } from "../../lib/logger.ts";
+import { registerWithGlobalToggle } from "../../lib/registry.ts";
+import { enhanceWithCI, setupProvider } from "../../provider-helper.ts";
+
+const _logger = createLogger("codestral");
+
+// =============================================================================
+// Model Definition
+// =============================================================================
+
+const CODESTRAL_MODEL: ProviderModelConfig = {
+	id: "codestral-latest",
+	name: "Codestral",
+	reasoning: false,
+	input: ["text"],
+	cost: {
+		input: 0.3,
+		output: 0.9,
+		cacheRead: 0,
+		cacheWrite: 0,
+	},
+	contextWindow: 256_000,
+	maxTokens: 4_096,
+};
+
+// =============================================================================
+// Extension Entry Point
+// =============================================================================
+
+export default async function codestralProvider(pi: ExtensionAPI) {
+	// Try CODESTRAL_API_KEY first, fall back to MISTRAL_API_KEY
+	const apiKey = getCodestralApiKey() || getMistralApiKey();
+
+	if (!apiKey) {
+		_logger.info(
+			"[codestral] Skipping — neither CODESTRAL_API_KEY nor MISTRAL_API_KEY set",
+		);
+		return;
+	}
+
+	const keySource = getCodestralApiKey()
+		? "CODESTRAL_API_KEY"
+		: "MISTRAL_API_KEY";
+	_logger.info(`[codestral] Using key from ${keySource}`);
+
+	const allModels = [CODESTRAL_MODEL];
+	const freeModels = allModels; // All $0.30/$0.90 — still accessible via Experiment free tier
+	const stored = { free: freeModels, all: allModels };
+
+	// Re-register function — uses mistral-conversations API (Mistral SDK)
+	// NOT openai-completions: Codestral uses the same API format as Mistral
+	// and rejects OpenAI-specific fields (stream_options, store, max_completion_tokens) with 422.
+	const reRegister = (models: typeof freeModels) => {
+		pi.registerProvider(PROVIDER_CODESTRAL, {
+			baseUrl: BASE_URL_CODESTRAL,
+			apiKey,
+			api: "mistral-conversations" as const,
+			models: enhanceWithCI(models, PROVIDER_CODESTRAL),
+		});
+	};
+
+	// Register with global toggle
+	registerWithGlobalToggle(PROVIDER_CODESTRAL, stored, reRegister, true);
+
+	// Setup provider (toggle command, status bar, error handling)
+	setupProvider(
+		pi,
+		{
+			providerId: PROVIDER_CODESTRAL,
+			initialShowPaid: getCodestralShowPaid(),
+			skipToggle: true, // Only one model, no toggle needed
+			reRegister: (models) => {
+				stored.free = models;
+				stored.all = models;
+				reRegister(models);
+			},
+		},
+		stored,
+	);
+
+	// Initial registration — uses mistral-conversations API (Mistral SDK)
+	reRegister(freeModels);
+
+	_logger.info(`[codestral] Registered codestral-latest via ${keySource}`);
+
+	// Status bar
+	pi.on("model_select", (_event, ctx) => {
+		if (_event.model?.provider !== PROVIDER_CODESTRAL) {
+			ctx.ui.setStatus(`${PROVIDER_CODESTRAL}-status`, undefined);
+			return;
+		}
+		ctx.ui.setStatus(
+			`${PROVIDER_CODESTRAL}-status`,
+			`codestral: 1 model (free tier) 🔑`,
+		);
+	});
+}

package/providers/crofai/crofai.ts

@@ -15,94 +15,16 @@
  *   # Models appear in /model selector
  */
 
-import type {
-	ExtensionAPI,
-	ProviderModelConfig,
-} from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 import { getCrofaiApiKey, getCrofaiShowPaid } from "../../config.ts";
-import {
-	BASE_URL_CROFAI,
-	DEFAULT_FETCH_TIMEOUT_MS,
-	PROVIDER_CROFAI,
-} from "../../constants.ts";
+import { BASE_URL_CROFAI, PROVIDER_CROFAI } from "../../constants.ts";
 import { createLogger } from "../../lib/logger.ts";
-import {
-	getProxyModelCompat,
-	isLikelyReasoningModel,
-} from "../../lib/provider-compat.ts";
 import { isFreeModel, registerWithGlobalToggle } from "../../lib/registry.ts";
-import { fetchWithRetry } from "../../lib/util.ts";
+import { fetchOpenAICompatibleModels } from "../../lib/util.ts";
 import { createReRegister, setupProvider } from "../../provider-helper.ts";
 
 const _logger = createLogger("crofai");
 
-// =============================================================================
-// Fetch CrofAI models
-// =============================================================================
-
-interface CrofaiModel {
-	id: string;
-	object?: string;
-	created?: number;
-	owned_by?: string;
-}
-
-async function fetchCrofaiModels(
-	apiKey: string,
-): Promise<ProviderModelConfig[]> {
-	_logger.info("[crofai] Fetching models from CrofAI API...");
-
-	try {
-		const response = await fetchWithRetry(
-			`${BASE_URL_CROFAI}/models`,
-			{
-				headers: {
-					Authorization: `Bearer ${apiKey}`,
-					"Content-Type": "application/json",
-				},
-			},
-			3,
-			1000,
-			DEFAULT_FETCH_TIMEOUT_MS,
-		);
-
-		if (!response.ok) {
-			throw new Error(`CrofAI API error: ${response.status}`);
-		}
-
-		const data = (await response.json()) as { data?: CrofaiModel[] };
-		const models = data.data ?? [];
-
-		_logger.info(`[crofai] Fetched ${models.length} models`);
-
-		return models
-			.filter((m) => m.id) // Filter out any empty entries
-			.map((m) => {
-				const name = m.id.split("/").pop() || m.id;
-				return {
-					id: m.id,
-					name,
-					reasoning: isLikelyReasoningModel({ id: m.id, name }),
-					input: ["text"],
-					cost: {
-						input: 0, // CrofAI doesn't expose pricing via API
-						output: 0,
-						cacheRead: 0,
-						cacheWrite: 0,
-					},
-					contextWindow: 128000, // Default, varies by model
-					maxTokens: 4096,
-					compat: getProxyModelCompat({ id: m.id, name }),
-				} satisfies ProviderModelConfig;
-			});
-	} catch (error) {
-		_logger.error("[crofai] Failed to fetch models:", {
-			error: error instanceof Error ? error.message : String(error),
-		});
-		return [];
-	}
-}
-
 // =============================================================================
 // Extension Entry Point
 // =============================================================================
@@ -117,8 +39,12 @@ export default async function crofaiProvider(pi: ExtensionAPI) {
 		return;
 	}
 
-	// Fetch models
-	const allModels = await fetchCrofaiModels(apiKey);
+	// Fetch models via shared OpenAI-compatible helper
+	const allModels = await fetchOpenAICompatibleModels(
+		"crofai",
+		BASE_URL_CROFAI,
+		apiKey,
+	);
 
 	if (allModels.length === 0) {
 		_logger.warn("[crofai] No models available");
@@ -165,6 +91,9 @@ export default async function crofaiProvider(pi: ExtensionAPI) {
 		stored,
 	);
 
-	// Initial registration
-	reRegister(freeModels);
+	// Initial registration — respect persisted toggle state
+	const showPaid = getCrofaiShowPaid();
+	const initialModels =
+		showPaid && stored.all.length > 0 ? stored.all : freeModels;
+	reRegister(initialModels);
 }

package/providers/deepinfra/deepinfra.ts

@@ -0,0 +1,109 @@
+/**
+ * DeepInfra Provider Extension
+ *
+ * DeepInfra is an AI inference cloud with an OpenAI-compatible API for
+ * 100+ open-source models (Llama, DeepSeek, Mistral, Qwen, Mixtral, etc.).
+ *
+ * Free tier:
+ *   - $5 one-time credit on signup (no credit card)
+ *   - ~5M tokens, expires after 90 days
+ *   - 60 RPM (varies by model)
+ *
+ * Paid: pay-per-token after credits exhaust
+ *
+ * Endpoint:
+ *   Chat: https://api.deepinfra.com/v1/openai/chat/completions
+ *
+ * Setup:
+ *   1. Sign up at https://deepinfra.com/ (GitHub or email)
+ *   2. Get API key from https://deepinfra.com/dash/api_keys
+ *   3. Set DEEPINFRA_TOKEN env var (or add to ~/.pi/free.json)
+ *
+ * Usage:
+ *   pi install git:github.com/apmantza/pi-free
+ *   # Set DEEPINFRA_TOKEN env var
+ *   # Models appear in /model selector as "deepinfra/meta-llama/..."
+ */
+
+import type {
+	ExtensionAPI,
+	ProviderModelConfig,
+} from "@mariozechner/pi-coding-agent";
+import { getDeepinfraApiKey } from "../../config.ts";
+import { BASE_URL_DEEPINFRA, PROVIDER_DEEPINFRA } from "../../constants.ts";
+import { createLogger } from "../../lib/logger.ts";
+import { registerWithGlobalToggle } from "../../lib/registry.ts";
+import { fetchOpenAICompatibleModels } from "../../lib/util.ts";
+import { createReRegister, setupProvider } from "../../provider-helper.ts";
+
+const _logger = createLogger("deepinfra");
+
+// =============================================================================
+// Extension Entry Point
+// =============================================================================
+
+export default async function deepinfraProvider(pi: ExtensionAPI) {
+	const apiKey = getDeepinfraApiKey();
+
+	if (!apiKey) {
+		_logger.info(
+			"[deepinfra] Skipping — DEEPINFRA_TOKEN not set. Sign up at https://deepinfra.com/",
+		);
+		return;
+	}
+
+	// Fetch models via shared OpenAI-compatible helper
+	const allModels = await fetchOpenAICompatibleModels(
+		"deepinfra",
+		BASE_URL_DEEPINFRA,
+		apiKey,
+		{ cost: { input: 0.3, output: 0.9 } },
+	);
+
+	if (allModels.length === 0) {
+		_logger.warn("[deepinfra] No models available");
+		return;
+	}
+
+	// DeepInfra is a trial credit provider — $5 one-time credit, no truly free models.
+	// All models are marked as paid. When free-only mode is ON, no models are shown.
+	// Toggle free-only OFF to see all models.
+	const freeModels: ProviderModelConfig[] = [];
+	const stored = { free: freeModels, all: allModels };
+
+	_logger.info(
+		`[deepinfra] Registered ${allModels.length} models (trial credit, 0 free)`,
+	);
+
+	// Create re-register function
+	const reRegister = createReRegister(pi, {
+		providerId: PROVIDER_DEEPINFRA,
+		baseUrl: BASE_URL_DEEPINFRA,
+		apiKey,
+	});
+
+	// Register with global toggle
+	registerWithGlobalToggle(PROVIDER_DEEPINFRA, stored, reRegister, true);
+
+	// Setup provider with toggle command
+	setupProvider(
+		pi,
+		{
+			providerId: PROVIDER_DEEPINFRA,
+			initialShowPaid: true, // trial credit: default to showing all models
+			tosUrl: "https://deepinfra.com/pricing",
+			reRegister: (models, _stored) => {
+				if (_stored) {
+					stored.free = _stored.free;
+					stored.all = _stored.all;
+				}
+				reRegister(models);
+			},
+		},
+		stored,
+	);
+
+	// Initial registration — DeepInfra is a trial-credit provider,
+	// so always show all models. Users see them immediately on setup.
+	reRegister(allModels);
+}