pi-free 2.0.6 → 2.0.8

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (36) hide show
  1. package/CHANGELOG.md +540 -421
  2. package/README.md +572 -495
  3. package/config.ts +58 -11
  4. package/constants.ts +12 -0
  5. package/index.ts +66 -2
  6. package/lib/model-detection.ts +1 -0
  7. package/lib/model-enhancer.ts +20 -20
  8. package/lib/open-browser.ts +1 -1
  9. package/lib/quota-monitor.ts +123 -0
  10. package/lib/types.ts +101 -101
  11. package/lib/util.ts +460 -351
  12. package/package.json +68 -68
  13. package/provider-failover/benchmark-lookup.ts +743 -702
  14. package/provider-failover/benchmarks-chunk-0.ts +48 -48
  15. package/provider-failover/benchmarks-chunk-1.ts +44 -44
  16. package/provider-failover/benchmarks-chunk-2.ts +39 -39
  17. package/provider-failover/benchmarks-chunk-3.ts +41 -41
  18. package/provider-failover/benchmarks-chunk-4.ts +33 -33
  19. package/providers/cline/cline-auth.ts +473 -473
  20. package/providers/cline/cline-models.ts +2 -2
  21. package/providers/cline/cline.ts +1 -1
  22. package/providers/codestral/codestral.ts +139 -0
  23. package/providers/crofai/crofai.ts +14 -85
  24. package/providers/deepinfra/deepinfra.ts +109 -0
  25. package/providers/kilo/kilo-auth.ts +155 -155
  26. package/providers/kilo/kilo.ts +1 -1
  27. package/providers/llm7/llm7.ts +156 -0
  28. package/providers/model-fetcher.ts +2 -2
  29. package/providers/nvidia/nvidia.ts +4 -4
  30. package/providers/ollama/ollama.ts +1 -1
  31. package/providers/opencode-session.ts +1 -1
  32. package/providers/qwen/qwen-models.ts +101 -101
  33. package/providers/qwen/qwen.ts +1 -1
  34. package/providers/sambanova/sambanova.ts +109 -0
  35. package/providers/zenmux/zenmux.ts +5 -2
  36. package/scripts/check-extensions.mjs +6 -4
package/lib/util.ts CHANGED
@@ -1,351 +1,460 @@
1
- import { createLogger } from "./logger.ts";
2
- import type { ProviderModelConfig } from "./types.ts";
3
-
4
- const _logger = createLogger("util");
5
-
6
- // =============================================================================
7
- // Shared Utilities
8
- // =============================================================================
9
-
10
- /**
11
- * Log a warning message for provider operations
12
- */
13
- export function logWarning(
14
- provider: string,
15
- message: string,
16
- error?: unknown,
17
- ): void {
18
- _logger.warn(
19
- `[${provider}] ${message}`,
20
- error ? { error: String(error) } : undefined,
21
- );
22
- }
23
-
24
- /**
25
- * Fetch with timeout using AbortController
26
- */
27
- export async function fetchWithTimeout(
28
- url: string,
29
- options: RequestInit,
30
- timeoutMs = 30000,
31
- ): Promise<Response> {
32
- const controller = new AbortController();
33
- const timeoutId = setTimeout(() => controller.abort(), timeoutMs);
34
-
35
- try {
36
- const response = await fetch(url, {
37
- ...options,
38
- signal: controller.signal,
39
- });
40
- return response;
41
- } finally {
42
- clearTimeout(timeoutId);
43
- }
44
- }
45
-
46
- /**
47
- * Fetch with retry logic and timeout
48
- */
49
- export async function fetchWithRetry(
50
- url: string,
51
- options: RequestInit,
52
- retries = 3,
53
- delayMs = 1000,
54
- timeoutMs = 30000,
55
- ): Promise<Response> {
56
- let lastError: unknown;
57
-
58
- for (let i = 0; i < retries; i++) {
59
- try {
60
- const response = await fetchWithTimeout(url, options, timeoutMs);
61
- if (response.ok) return response;
62
-
63
- // If it's a rate limit, throw immediately
64
- if (response.status === 429) {
65
- throw new Error(`Rate limited (429)`);
66
- }
67
-
68
- // For server errors, retry
69
- if (response.status >= 500) {
70
- lastError = new Error(`Server error ${response.status}`);
71
- if (i < retries - 1) {
72
- await new Promise((r) => setTimeout(r, delayMs * (i + 1)));
73
- continue;
74
- }
75
- // Last retry exhausted - throw the error
76
- throw lastError;
77
- }
78
-
79
- return response; // Return non-ok but non-retryable responses
80
- } catch (error) {
81
- lastError = error;
82
- if (i < retries - 1) {
83
- await new Promise((r) => setTimeout(r, delayMs * (i + 1)));
84
- }
85
- }
86
- }
87
-
88
- throw lastError;
89
- }
90
-
91
- // =============================================================================
92
- // Shared API Response Parsing
93
- // =============================================================================
94
-
95
- /**
96
- * Parse and validate model list API response
97
- * Shared between Kilo, OpenRouter, and other providers
98
- */
99
- export async function parseModelResponse<T>(
100
- response: Response,
101
- providerName: string,
102
- ): Promise<{ data: T[] }> {
103
- if (!response.ok) {
104
- throw new Error(
105
- `Failed to fetch ${providerName} models: ${response.status} ${response.statusText}`,
106
- );
107
- }
108
-
109
- const json = (await response.json()) as { data?: T[] };
110
-
111
- if (!json.data || !Array.isArray(json.data)) {
112
- throw new Error(
113
- `Invalid ${providerName} models response: missing data array`,
114
- );
115
- }
116
-
117
- return { data: json.data };
118
- }
119
-
120
- // =============================================================================
121
- // Model Filtering Utilities
122
- // =============================================================================
123
-
124
- // Models known to be small (no "Xb" in their ID) that should be filtered.
125
- // Updated as new small free models appear on OpenRouter/Kilo.
126
- const KNOWN_SMALL_MODELS: ReadonlySet<string> = new Set([
127
- // Microsoft Phi models (1.5B–14B)
128
- "microsoft/phi-3-mini-128k-instruct",
129
- "microsoft/phi-3-mini-4k-instruct",
130
- "microsoft/phi-3-small-128k-instruct",
131
- "microsoft/phi-3-small-8k-instruct",
132
- "microsoft/phi-3-medium-128k-instruct",
133
- "microsoft/phi-3-medium-4k-instruct",
134
- "microsoft/phi-3.5-mini-instruct",
135
- "microsoft/phi-4-mini-instruct",
136
- "microsoft/phi-4-mini-reasoning",
137
- "microsoft/phi-4-reasoning-plus",
138
- // OpenChat (7B)
139
- "openchat/openchat-3.5-0106",
140
- "openchat/openchat-3.5-1210",
141
- // Mistral 7B variants
142
- "mistralai/mistral-7b-instruct-v0.1",
143
- "mistralai/mistral-7b-instruct-v0.2",
144
- "mistralai/mistral-7b-instruct-v0.3",
145
- // Gemma small variants
146
- "google/gemma-2b-it",
147
- "google/gemma-1.1-2b-it",
148
- // DeepSeek small variants
149
- "deepseek/deepseek-r1-distill-qwen-1.5b",
150
- "deepseek/deepseek-r1-distill-llama-8b",
151
- "deepseek/deepseek-r1-distill-qwen-7b",
152
- "deepseek/deepseek-r1-distill-qwen-14b",
153
- // Stripe Hyena (2.7B)
154
- "togethercomputer/stripedhy-2.7b",
155
- // TinyLlama
156
- "tinyllama/tinyllama-1.1b-chat-v1.0",
157
- ]);
158
-
159
- /**
160
- * Check if model is usable based on size constraints and naming.
161
- * Extracts model size from ID (e.g., "llama-3-70b" -> 70) and compares to minSizeB.
162
- * Falls back to a blocklist for models that don't encode size in the name.
163
- */
164
- export function isUsableModel(modelId: string, minSizeB?: number): boolean {
165
- // Filter out models that are likely test or debug models
166
- if (modelId.includes("test") || modelId.includes("debug")) {
167
- return false;
168
- }
169
-
170
- // Filter by minimum size if specified
171
- if (minSizeB !== undefined) {
172
- // Known-small blocklist (models without "Xb" in the name)
173
- // Strip :free suffix used by OpenRouter/Kilo
174
- const baseId = modelId.replace(/:free$/, "");
175
- if (KNOWN_SMALL_MODELS.has(baseId)) return false;
176
-
177
- // Check Mixture-of-Experts models first (e.g., "8x22b" = 176b total)
178
- const parsed = parseModelSize(modelId);
179
- if (parsed?.type === "moe") {
180
- if (parsed.experts * parsed.sizePerExpert < minSizeB) return false;
181
- return true; // MoE model passed size check
182
- }
183
-
184
- // Standard model size (e.g., "70b", "8b")
185
- if (parsed?.type === "standard" && parsed.size < minSizeB) return false;
186
- }
187
-
188
- return true;
189
- }
190
-
191
- // =============================================================================
192
- // Model Size Parsing (no regex — avoids SonarCloud S5852 flags)
193
- // =============================================================================
194
-
195
- interface MoeSize {
196
- type: "moe";
197
- experts: number;
198
- sizePerExpert: number;
199
- }
200
-
201
- interface StandardSize {
202
- type: "standard";
203
- size: number;
204
- }
205
-
206
- /**
207
- * Extract model size from a model ID without using regex.
208
- * Handles both MoE ("8x22b") and standard ("70b", "8b") formats.
209
- */
210
- function parseModelSize(modelId: string): MoeSize | StandardSize | null {
211
- const lower = modelId.toLowerCase();
212
-
213
- // MoE: digits "x" digits "b" (e.g., "8x22b", "a35b" is NOT MoE)
214
- // Walk through each 'x' to find one preceded by a digit and followed by digits then 'b'
215
- let searchPos = 0;
216
- while (true) {
217
- const xIdx = lower.indexOf("x", searchPos);
218
- if (xIdx <= 0) break;
219
- const beforeChar = lower[xIdx - 1];
220
- if (beforeChar >= "0" && beforeChar <= "9") {
221
- const bIdx = lower.indexOf("b", xIdx + 1);
222
- if (bIdx > xIdx + 1) {
223
- // Walk backwards from x to find start of expert-count number
224
- let countStart = xIdx - 1;
225
- while (
226
- countStart > 0 &&
227
- lower[countStart - 1] >= "0" &&
228
- lower[countStart - 1] <= "9"
229
- ) {
230
- countStart--;
231
- }
232
- const experts = Number.parseInt(lower.slice(countStart, xIdx), 10);
233
- const size = Number.parseFloat(lower.slice(xIdx + 1, bIdx));
234
- if (
235
- !Number.isNaN(experts) &&
236
- !Number.isNaN(size) &&
237
- experts > 0 &&
238
- size > 0
239
- ) {
240
- const afterB = lower.slice(bIdx + 1);
241
- if (
242
- afterB.length === 0 ||
243
- ((afterB[0] < "0" || afterB[0] > "9") && afterB[0] !== ".")
244
- ) {
245
- return { type: "moe", experts, sizePerExpert: size };
246
- }
247
- }
248
- }
249
- }
250
- searchPos = xIdx + 1;
251
- }
252
-
253
- // Standard: "digits" "b" not followed by digit/dot (e.g., "70b", "8b")
254
- for (let i = 0; i < lower.length; i++) {
255
- if (lower[i] === "b") {
256
- const afterB = lower.slice(i + 1);
257
- if (
258
- afterB.length > 0 &&
259
- ((afterB[0] >= "0" && afterB[0] <= "9") || afterB[0] === ".")
260
- ) {
261
- continue; // b followed by digit or dot — not our match
262
- }
263
- // Walk backwards to find start of number
264
- let start = i;
265
- while (
266
- start > 0 &&
267
- ((lower[start - 1] >= "0" && lower[start - 1] <= "9") ||
268
- lower[start - 1] === ".")
269
- ) {
270
- start--;
271
- }
272
- if (start < i) {
273
- const numStr = lower.slice(start, i);
274
- const size = Number.parseFloat(numStr);
275
- if (!Number.isNaN(size) && size > 0) {
276
- return { type: "standard", size };
277
- }
278
- }
279
- break;
280
- }
281
- }
282
-
283
- return null;
284
- }
285
-
286
- // =============================================================================
287
- // Model Name Cleaning
288
- // =============================================================================
289
-
290
- /**
291
- * Strip provider prefix from model names.
292
- * OpenRouter/Kilo return names like "Provider : Model Name" or "Provider / Model Name".
293
- * We only want the model name part.
294
- */
295
- export function cleanModelName(name: string): string {
296
- // Handle patterns like "Provider : Model Name" or "Provider / Model Name"
297
- const colonIdx = name.indexOf(":");
298
- const slashIdx = name.indexOf("/");
299
- const idx =
300
- colonIdx === -1
301
- ? slashIdx
302
- : slashIdx === -1
303
- ? colonIdx
304
- : Math.min(colonIdx, slashIdx);
305
- if (idx > 0) {
306
- return name.slice(idx + 1).trim();
307
- }
308
- return name.trim();
309
- }
310
-
311
- // =============================================================================
312
- // Model Mapping
313
- // =============================================================================
314
-
315
- /**
316
- * Map OpenRouter/Kilo API model to ProviderModelConfig
317
- * Shared between OpenRouter and Kilo providers
318
- */
319
- export function mapOpenRouterModel(m: {
320
- id: string;
321
- name: string;
322
- context_length?: number;
323
- max_completion_tokens?: number | null;
324
- top_provider?: { max_completion_tokens?: number | null };
325
- pricing?: { prompt?: string | null; completion?: string | null };
326
- architecture?: {
327
- input_modalities?: string[] | null;
328
- output_modalities?: string[] | null;
329
- };
330
- }): ProviderModelConfig {
331
- const promptPrice = parseFloat(m.pricing?.prompt ?? "0");
332
- const completionPrice = parseFloat(m.pricing?.completion ?? "0");
333
-
334
- return {
335
- id: m.id,
336
- name: cleanModelName(m.name),
337
- reasoning: false, // OpenRouter doesn't expose reasoning flag directly
338
- input: m.architecture?.input_modalities?.includes("image")
339
- ? (["text", "image"] as const)
340
- : (["text"] as const),
341
- cost: {
342
- input: promptPrice,
343
- output: completionPrice,
344
- cacheRead: 0,
345
- cacheWrite: 0,
346
- },
347
- contextWindow: m.context_length ?? 4096,
348
- maxTokens:
349
- m.max_completion_tokens ?? m.top_provider?.max_completion_tokens ?? 4096,
350
- };
351
- }
1
+ import { createLogger } from "./logger.ts";
2
+ import {
3
+ getProxyModelCompat,
4
+ isLikelyReasoningModel,
5
+ } from "./provider-compat.ts";
6
+ import type { ProviderModelConfig as PiProviderModelConfig } from "@mariozechner/pi-coding-agent";
7
+ import type { ProviderModelConfig } from "./types.ts";
8
+
9
+ const _logger = createLogger("util");
10
+
11
+ // =============================================================================
12
+ // Shared Utilities
13
+ // =============================================================================
14
+
15
+ /**
16
+ * Log a warning message for provider operations
17
+ */
18
+ export function logWarning(
19
+ provider: string,
20
+ message: string,
21
+ error?: unknown,
22
+ ): void {
23
+ _logger.warn(
24
+ `[${provider}] ${message}`,
25
+ error ? { error: String(error) } : undefined,
26
+ );
27
+ }
28
+
29
+ /**
30
+ * Fetch with timeout using AbortController
31
+ */
32
+ export async function fetchWithTimeout(
33
+ url: string,
34
+ options: RequestInit,
35
+ timeoutMs = 30000,
36
+ ): Promise<Response> {
37
+ const controller = new AbortController();
38
+ const timeoutId = setTimeout(() => controller.abort(), timeoutMs);
39
+
40
+ try {
41
+ const response = await fetch(url, {
42
+ ...options,
43
+ signal: controller.signal,
44
+ });
45
+ return response;
46
+ } finally {
47
+ clearTimeout(timeoutId);
48
+ }
49
+ }
50
+
51
+ /**
52
+ * Fetch with retry logic and timeout
53
+ */
54
+ export async function fetchWithRetry(
55
+ url: string,
56
+ options: RequestInit,
57
+ retries = 3,
58
+ delayMs = 1000,
59
+ timeoutMs = 30000,
60
+ ): Promise<Response> {
61
+ let lastError: unknown;
62
+
63
+ for (let i = 0; i < retries; i++) {
64
+ try {
65
+ const response = await fetchWithTimeout(url, options, timeoutMs);
66
+ if (response.ok) return response;
67
+
68
+ // If it's a rate limit, throw immediately
69
+ if (response.status === 429) {
70
+ throw new Error(`Rate limited (429)`);
71
+ }
72
+
73
+ // For server errors, retry
74
+ if (response.status >= 500) {
75
+ lastError = new Error(`Server error ${response.status}`);
76
+ if (i < retries - 1) {
77
+ await new Promise((r) => setTimeout(r, delayMs * (i + 1)));
78
+ continue;
79
+ }
80
+ // Last retry exhausted - throw the error
81
+ throw lastError;
82
+ }
83
+
84
+ return response; // Return non-ok but non-retryable responses
85
+ } catch (error) {
86
+ lastError = error;
87
+ if (i < retries - 1) {
88
+ await new Promise((r) => setTimeout(r, delayMs * (i + 1)));
89
+ }
90
+ }
91
+ }
92
+
93
+ throw lastError;
94
+ }
95
+
96
+ // =============================================================================
97
+ // Shared API Response Parsing
98
+ // =============================================================================
99
+
100
+ /**
101
+ * Parse and validate model list API response
102
+ * Shared between Kilo, OpenRouter, and other providers
103
+ */
104
+ export async function parseModelResponse<T>(
105
+ response: Response,
106
+ providerName: string,
107
+ ): Promise<{ data: T[] }> {
108
+ if (!response.ok) {
109
+ throw new Error(
110
+ `Failed to fetch ${providerName} models: ${response.status} ${response.statusText}`,
111
+ );
112
+ }
113
+
114
+ const json = (await response.json()) as { data?: T[] };
115
+
116
+ if (!json.data || !Array.isArray(json.data)) {
117
+ throw new Error(
118
+ `Invalid ${providerName} models response: missing data array`,
119
+ );
120
+ }
121
+
122
+ return { data: json.data };
123
+ }
124
+
125
+ // =============================================================================
126
+ // Model Filtering Utilities
127
+ // =============================================================================
128
+
129
+ // Models known to be small (no "Xb" in their ID) that should be filtered.
130
+ // Updated as new small free models appear on OpenRouter/Kilo.
131
+ const KNOWN_SMALL_MODELS: ReadonlySet<string> = new Set([
132
+ // Microsoft Phi models (1.5B–14B)
133
+ "microsoft/phi-3-mini-128k-instruct",
134
+ "microsoft/phi-3-mini-4k-instruct",
135
+ "microsoft/phi-3-small-128k-instruct",
136
+ "microsoft/phi-3-small-8k-instruct",
137
+ "microsoft/phi-3-medium-128k-instruct",
138
+ "microsoft/phi-3-medium-4k-instruct",
139
+ "microsoft/phi-3.5-mini-instruct",
140
+ "microsoft/phi-4-mini-instruct",
141
+ "microsoft/phi-4-mini-reasoning",
142
+ "microsoft/phi-4-reasoning-plus",
143
+ // OpenChat (7B)
144
+ "openchat/openchat-3.5-0106",
145
+ "openchat/openchat-3.5-1210",
146
+ // Mistral 7B variants
147
+ "mistralai/mistral-7b-instruct-v0.1",
148
+ "mistralai/mistral-7b-instruct-v0.2",
149
+ "mistralai/mistral-7b-instruct-v0.3",
150
+ // Gemma small variants
151
+ "google/gemma-2b-it",
152
+ "google/gemma-1.1-2b-it",
153
+ // DeepSeek small variants
154
+ "deepseek/deepseek-r1-distill-qwen-1.5b",
155
+ "deepseek/deepseek-r1-distill-llama-8b",
156
+ "deepseek/deepseek-r1-distill-qwen-7b",
157
+ "deepseek/deepseek-r1-distill-qwen-14b",
158
+ // Stripe Hyena (2.7B)
159
+ "togethercomputer/stripedhy-2.7b",
160
+ // TinyLlama
161
+ "tinyllama/tinyllama-1.1b-chat-v1.0",
162
+ ]);
163
+
164
+ /**
165
+ * Check if model is usable based on size constraints and naming.
166
+ * Extracts model size from ID (e.g., "llama-3-70b" -> 70) and compares to minSizeB.
167
+ * Falls back to a blocklist for models that don't encode size in the name.
168
+ */
169
+ export function isUsableModel(modelId: string, minSizeB?: number): boolean {
170
+ // Filter out models that are likely test or debug models
171
+ if (modelId.includes("test") || modelId.includes("debug")) {
172
+ return false;
173
+ }
174
+
175
+ // Filter by minimum size if specified
176
+ if (minSizeB !== undefined) {
177
+ // Known-small blocklist (models without "Xb" in the name)
178
+ // Strip :free suffix used by OpenRouter/Kilo
179
+ const baseId = modelId.replace(/:free$/, "");
180
+ if (KNOWN_SMALL_MODELS.has(baseId)) return false;
181
+
182
+ // Check Mixture-of-Experts models first (e.g., "8x22b" = 176b total)
183
+ const parsed = parseModelSize(modelId);
184
+ if (parsed?.type === "moe") {
185
+ if (parsed.experts * parsed.sizePerExpert < minSizeB) return false;
186
+ return true; // MoE model passed size check
187
+ }
188
+
189
+ // Standard model size (e.g., "70b", "8b")
190
+ if (parsed?.type === "standard" && parsed.size < minSizeB) return false;
191
+ }
192
+
193
+ return true;
194
+ }
195
+
196
+ // =============================================================================
197
+ // Model Size Parsing (no regex — avoids SonarCloud S5852 flags)
198
+ // =============================================================================
199
+
200
+ interface MoeSize {
201
+ type: "moe";
202
+ experts: number;
203
+ sizePerExpert: number;
204
+ }
205
+
206
+ interface StandardSize {
207
+ type: "standard";
208
+ size: number;
209
+ }
210
+
211
+ /**
212
+ * Extract model size from a model ID without using regex.
213
+ * Handles both MoE ("8x22b") and standard ("70b", "8b") formats.
214
+ */
215
+ /**
216
+ * Parse MoE (Mixture of Experts) model size like "8x22b".
217
+ */
218
+ function parseMoeSize(lower: string): MoeSize | null {
219
+ let searchPos = 0;
220
+ while (true) {
221
+ const xIdx = lower.indexOf("x", searchPos);
222
+ if (xIdx <= 0) break;
223
+ const beforeChar = lower[xIdx - 1];
224
+ if (!(beforeChar >= "0" && beforeChar <= "9")) {
225
+ searchPos = xIdx + 1;
226
+ continue;
227
+ }
228
+ const bIdx = lower.indexOf("b", xIdx + 1);
229
+ if (bIdx <= xIdx + 1) {
230
+ searchPos = xIdx + 1;
231
+ continue;
232
+ }
233
+ let countStart = xIdx - 1;
234
+ while (
235
+ countStart > 0 &&
236
+ lower[countStart - 1] >= "0" &&
237
+ lower[countStart - 1] <= "9"
238
+ ) {
239
+ countStart--;
240
+ }
241
+ const experts = Number.parseInt(lower.slice(countStart, xIdx), 10);
242
+ const size = Number.parseFloat(lower.slice(xIdx + 1, bIdx));
243
+ if (
244
+ !Number.isNaN(experts) &&
245
+ !Number.isNaN(size) &&
246
+ experts > 0 &&
247
+ size > 0
248
+ ) {
249
+ const afterB = lower.slice(bIdx + 1);
250
+ if (
251
+ afterB.length === 0 ||
252
+ ((afterB[0] < "0" || afterB[0] > "9") && afterB[0] !== ".")
253
+ ) {
254
+ return { type: "moe", experts, sizePerExpert: size };
255
+ }
256
+ }
257
+ searchPos = xIdx + 1;
258
+ }
259
+ return null;
260
+ }
261
+
262
+ /**
263
+ * Parse standard model size like "70b" or "8b".
264
+ */
265
+ function parseStandardSize(lower: string): StandardSize | null {
266
+ for (let i = 0; i < lower.length; i++) {
267
+ if (lower[i] !== "b") continue;
268
+ const afterB = lower.slice(i + 1);
269
+ if (
270
+ afterB.length > 0 &&
271
+ ((afterB[0] >= "0" && afterB[0] <= "9") || afterB[0] === ".")
272
+ ) {
273
+ continue; // b followed by digit or dot — not our match
274
+ }
275
+ let start = i;
276
+ while (
277
+ start > 0 &&
278
+ ((lower[start - 1] >= "0" && lower[start - 1] <= "9") ||
279
+ lower[start - 1] === ".")
280
+ ) {
281
+ start--;
282
+ }
283
+ if (start < i) {
284
+ const numStr = lower.slice(start, i);
285
+ const size = Number.parseFloat(numStr);
286
+ if (!Number.isNaN(size) && size > 0) {
287
+ return { type: "standard", size };
288
+ }
289
+ }
290
+ break;
291
+ }
292
+ return null;
293
+ }
294
+
295
+ function parseModelSize(modelId: string): MoeSize | StandardSize | null {
296
+ const lower = modelId.toLowerCase();
297
+ return parseMoeSize(lower) ?? parseStandardSize(lower) ?? null;
298
+ }
299
+
300
+ // =============================================================================
301
+ // Model Name Cleaning
302
+ // =============================================================================
303
+
304
+ /**
305
+ * Strip provider prefix from model names.
306
+ * OpenRouter/Kilo return names like "Provider : Model Name" or "Provider / Model Name".
307
+ * We only want the model name part.
308
+ */
309
+ export function cleanModelName(name: string): string {
310
+ // Handle patterns like "Provider : Model Name" or "Provider / Model Name"
311
+ const colonIdx = name.indexOf(":");
312
+ const slashIdx = name.indexOf("/");
313
+ const idx =
314
+ colonIdx === -1
315
+ ? slashIdx
316
+ : slashIdx === -1
317
+ ? colonIdx
318
+ : Math.min(colonIdx, slashIdx);
319
+ if (idx > 0) {
320
+ return name.slice(idx + 1).trim();
321
+ }
322
+ return name.trim();
323
+ }
324
+
325
+ // =============================================================================
326
+ // Model Mapping
327
+ // =============================================================================
328
+
329
+ /**
330
+ * Map OpenRouter/Kilo API model to ProviderModelConfig
331
+ * Shared between OpenRouter and Kilo providers
332
+ */
333
+ export function mapOpenRouterModel(m: {
334
+ id: string;
335
+ name: string;
336
+ context_length?: number;
337
+ max_completion_tokens?: number | null;
338
+ top_provider?: { max_completion_tokens?: number | null };
339
+ pricing?: { prompt?: string | null; completion?: string | null };
340
+ architecture?: {
341
+ input_modalities?: string[] | null;
342
+ output_modalities?: string[] | null;
343
+ };
344
+ }): ProviderModelConfig {
345
+ const promptPrice = Number.parseFloat(m.pricing?.prompt ?? "0");
346
+ const completionPrice = Number.parseFloat(m.pricing?.completion ?? "0");
347
+
348
+ return {
349
+ id: m.id,
350
+ name: cleanModelName(m.name),
351
+ reasoning: false, // OpenRouter doesn't expose reasoning flag directly
352
+ input: m.architecture?.input_modalities?.includes("image")
353
+ ? (["text", "image"] as const)
354
+ : (["text"] as const),
355
+ cost: {
356
+ input: promptPrice,
357
+ output: completionPrice,
358
+ cacheRead: 0,
359
+ cacheWrite: 0,
360
+ },
361
+ contextWindow: m.context_length ?? 4096,
362
+ maxTokens:
363
+ m.max_completion_tokens ?? m.top_provider?.max_completion_tokens ?? 4096,
364
+ };
365
+ }
366
+
367
+ // =============================================================================
368
+ // OpenAI-Compatible Provider Helpers
369
+ // =============================================================================
370
+
371
+ /**
372
+ * Defaults for mapping models from OpenAI-compatible /v1/models endpoints.
373
+ */
374
+ export interface OpenAIModelDefaults {
375
+ /** Per-model cost defaults (set to 0 if provider is free-tier). */
376
+ cost?: { input: number; output: number };
377
+ /** Default context window (tokens). */
378
+ contextWindow?: number;
379
+ /** Default max output tokens. */
380
+ maxTokens?: number;
381
+ /** Default input modalities. */
382
+ input?: string[];
383
+ }
384
+
385
+ /**
386
+ * Generic model shape returned by OpenAI-compatible /v1/models endpoints.
387
+ */
388
+ export interface OpenAIModelEntry {
389
+ id: string;
390
+ object?: string;
391
+ created?: number;
392
+ owned_by?: string;
393
+ }
394
+
395
+ /**
396
+ * Fetch and map models from an OpenAI-compatible /v1/models endpoint.
397
+ *
398
+ * Eliminates ~40 lines of duplicated fetch→parse→map boilerplate
399
+ * that was repeated in CrofAI, DeepInfra, and SambaNova providers.
400
+ */
401
+ export async function fetchOpenAICompatibleModels(
402
+ providerId: string,
403
+ baseUrl: string,
404
+ apiKey: string,
405
+ defaults: OpenAIModelDefaults = {},
406
+ ): Promise<PiProviderModelConfig[]> {
407
+ const logger = createLogger(providerId);
408
+
409
+ logger.info(`[${providerId}] Fetching models...`);
410
+
411
+ try {
412
+ const response = await fetchWithRetry(
413
+ `${baseUrl}/models`,
414
+ {
415
+ headers: {
416
+ Authorization: `Bearer ${apiKey}`,
417
+ "Content-Type": "application/json",
418
+ },
419
+ },
420
+ 3,
421
+ 1000,
422
+ 30000,
423
+ );
424
+
425
+ if (!response.ok) {
426
+ throw new Error(`${providerId} API error: ${response.status}`);
427
+ }
428
+
429
+ const data = (await response.json()) as { data?: OpenAIModelEntry[] };
430
+ const models = data.data ?? [];
431
+
432
+ logger.info(`[${providerId}] Fetched ${models.length} models`);
433
+
434
+ return models
435
+ .filter((m) => m.id)
436
+ .map((m): PiProviderModelConfig => {
437
+ const name = m.id.split("/").pop() || m.id;
438
+ return {
439
+ id: m.id,
440
+ name,
441
+ reasoning: isLikelyReasoningModel({ id: m.id, name }),
442
+ input: (defaults.input as PiProviderModelConfig["input"]) ?? ["text"],
443
+ cost: {
444
+ input: defaults.cost?.input ?? 0,
445
+ output: defaults.cost?.output ?? 0,
446
+ cacheRead: 0,
447
+ cacheWrite: 0,
448
+ },
449
+ contextWindow: defaults.contextWindow ?? 128_000,
450
+ maxTokens: defaults.maxTokens ?? 4_096,
451
+ compat: getProxyModelCompat({ id: m.id, name }),
452
+ };
453
+ });
454
+ } catch (error) {
455
+ logger.error(`[${providerId}] Failed to fetch models:`, {
456
+ error: error instanceof Error ? error.message : String(error),
457
+ });
458
+ return [];
459
+ }
460
+ }