tokenfactory-pi 0.2.4 → 0.2.5

package/dist/index.d.ts

@@ -1,11 +1,11 @@
 /**
- * Nebius Token Factory — pi extension
+ * Nebius Token Factory - pi extension
  *
  * Fetches the current model catalog from the Token Factory API on startup
  * and registers all tool-capable text-generation models as a "nebius" provider.
  *
  * Environment:
- *   NEBIUS_API_KEY — required, Token Factory API key
+ *   NEBIUS_API_KEY - required, Token Factory API key
  *
  * Usage:
  *   pi -e /path/to/tokenfactory-pi

package/dist/index.js

@@ -1,11 +1,11 @@
 /**
- * Nebius Token Factory — pi extension
+ * Nebius Token Factory - pi extension
  *
  * Fetches the current model catalog from the Token Factory API on startup
  * and registers all tool-capable text-generation models as a "nebius" provider.
  *
  * Environment:
- *   NEBIUS_API_KEY — required, Token Factory API key
+ *   NEBIUS_API_KEY - required, Token Factory API key
  *
  * Usage:
  *   pi -e /path/to/tokenfactory-pi
@@ -18,10 +18,9 @@ const ENV_VAR = "NEBIUS_API_KEY";
 // ============================================================================
 // Helpers
 // ============================================================================
-function isToolCapableTextModel(m) {
-    const features = m.supported_features || [];
+function isTextModel(m) {
     const modality = m.architecture?.modality || "";
-    return features.includes("tools") && modality.includes("->text");
+    return modality.includes("->text");
 }
 function parseInputModalities(modality) {
     const input = ["text"];
@@ -30,7 +29,8 @@ function parseInputModalities(modality) {
     return input;
 }
 function parseCostPerMillion(raw) {
-    return parseFloat(raw || "0") * 1_000_000;
+    const parsed = parseFloat(raw || "0");
+    return isNaN(parsed) ? 0 : parsed * 1_000_000;
 }
 function isReasoningModel(id) {
     return /(-R1|-Thinking|QwQ)/.test(id);
@@ -50,6 +50,7 @@ export default async function (pi) {
         });
         if (!res.ok) {
             console.warn(`[${PROVIDER_NAME}] API returned ${res.status}: ${res.statusText}`);
+            console.warn(`[${PROVIDER_NAME}] Response headers:`, [...res.headers.entries()]);
             return;
         }
         response = (await res.json());
@@ -59,18 +60,34 @@ export default async function (pi) {
         return;
     }
     if (!Array.isArray(response.data)) {
-        console.warn(`[${PROVIDER_NAME}] Unexpected API response shape`);
+        console.warn(`[${PROVIDER_NAME}] Unexpected API response shape. Expected array, got:`, typeof response.data);
+        console.warn(`[${PROVIDER_NAME}] Response data:`, response);
         return;
     }
     const models = [];
     for (const m of response.data) {
-        if (!isToolCapableTextModel(m))
+        // Skip models without valid IDs
+        if (!m.id || m.id.trim() === "") {
+            console.log(`[${PROVIDER_NAME}] Skipping model with empty ID:`, m);
             continue;
+        }
+        if (!isTextModel(m)) {
+            continue;
+        }
         const modality = m.architecture?.modality || "";
+        // Validate and set defaults for critical fields
+        const contextLength = m.context_length && m.context_length > 0 ? m.context_length : 131072;
+        const modelName = m.name || m.id || "unknown-model";
+        const modelId = m.id || modelName;
+        // Skip models with zero context length even after defaults
+        if (contextLength <= 0) {
+            console.log(`[${PROVIDER_NAME}] Skipping model with invalid context length:`, m.id, m.context_length);
+            continue;
+        }
         models.push({
-            id: m.id,
-            name: m.name || m.id,
-            reasoning: isReasoningModel(m.id),
+            id: modelId,
+            name: modelName,
+            reasoning: isReasoningModel(modelId),
             input: parseInputModalities(modality),
             cost: {
                 input: parseCostPerMillion(m.pricing?.prompt),
@@ -78,8 +95,8 @@ export default async function (pi) {
                 cacheRead: 0,
                 cacheWrite: 0,
             },
-            contextWindow: m.context_length || 131072,
-            maxTokens: Math.min(m.context_length || 32768, 32768),
+            contextWindow: contextLength,
+            maxTokens: Math.min(contextLength, 32768),
             compat: {
                 supportsDeveloperRole: false,
                 maxTokensField: "max_tokens",
@@ -101,7 +118,7 @@ export default async function (pi) {
                 return;
             }
             const items = models
-                .sort((a, b) => a.id.localeCompare(b.id))
+                .sort((a, b) => b.cost.output - a.cost.output)
                 .map((m) => {
                 const tags = [];
                 if (m.reasoning)
@@ -111,7 +128,33 @@ export default async function (pi) {
                 const suffix = tags.length > 0 ? ` (${tags.join(", ")})` : "";
                 return `${m.id}${suffix}`;
             });
-            await ctx.ui.select(`Nebius Token Factory — ${models.length} models`, items);
+            // Show the selection dialog and capture the result
+            const selectedItem = await ctx.ui.select(`Nebius Token Factory - ${models.length} models`, items);
+            // If user cancelled (selectedItem is undefined), do nothing
+            if (!selectedItem) {
+                ctx.ui.notify("Model selection cancelled", "warning");
+                return;
+            }
+            // Extract the model ID from the selection (remove the suffix)
+            const selectedId = selectedItem.split(" (")[0];
+            const selectedModel = models.find(m => m.id === selectedId);
+            if (!selectedModel) {
+                ctx.ui.notify(`Model not found: ${selectedId}`, "error");
+                return;
+            }
+            // Find the model in the registry and switch to it
+            const model = ctx.modelRegistry.find(PROVIDER_NAME, selectedModel.id);
+            if (!model) {
+                ctx.ui.notify(`Model not found in registry: ${PROVIDER_NAME}/${selectedModel.id}`, "error");
+                return;
+            }
+            const success = await pi.setModel(model);
+            if (success) {
+                ctx.ui.notify(`Switched to model: ${model.provider}/${model.id}`, "info");
+            }
+            else {
+                ctx.ui.notify(`Failed to switch to model: ${model.provider}/${model.id}. Check API key?`, "error");
+            }
         },
     });
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tokenfactory-pi",
-  "version": "0.2.4",
+  "version": "0.2.5",
   "description": "Nebius Token Factory provider extension for pi coding agent. Requires `npm install -g @mariozechner/pi-coding-agent`. Install with `pi install npm:tokenfactory-pi`",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",