tokenfactory-pi 0.2.5 → 0.2.6

package/dist/index.js

@@ -12,9 +12,13 @@
  *   pi -e /path/to/tokenfactory-pi --provider nebius
  *   pi -e /path/to/tokenfactory-pi --provider nebius --model Qwen/Qwen3-32B
  */
+import { gunzipSync } from "node:zlib";
 const PROVIDER_NAME = "nebius";
 const BASE_URL = "https://api.tokenfactory.nebius.com/v1";
 const ENV_VAR = "NEBIUS_API_KEY";
+function isGzip(bytes) {
+    return bytes.length >= 2 && bytes[0] === 0x1f && bytes[1] === 0x8b;
+}
 // ============================================================================
 // Helpers
 // ============================================================================
@@ -35,6 +39,19 @@ function parseCostPerMillion(raw) {
 function isReasoningModel(id) {
     return /(-R1|-Thinking|QwQ)/.test(id);
 }
+async function readTokenFactoryResponse(res) {
+    const bytes = Buffer.from(await res.arrayBuffer());
+    const body = (isGzip(bytes) ? gunzipSync(bytes) : bytes).toString("utf8");
+    try {
+        return JSON.parse(body);
+    }
+    catch {
+        const preview = body.slice(0, 200).replace(/\s+/g, " ");
+        throw new Error(`Invalid Token Factory JSON response (${res.status} ${res.statusText}, ` +
+            `content-type=${res.headers.get("content-type") || "unknown"}, ` +
+            `content-encoding=${res.headers.get("content-encoding") || "none"}): ${preview}`);
+    }
+}
 // ============================================================================
 // Extension entry point
 // ============================================================================
@@ -46,14 +63,17 @@ export default async function (pi) {
     let response;
     try {
         const res = await fetch(`${BASE_URL}/models?verbose=true`, {
-            headers: { Authorization: `Bearer ${apiKey}` },
+            headers: {
+                Authorization: `Bearer ${apiKey}`,
+                "Accept-Encoding": "identity",
+            },
         });
         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());
+        response = await readTokenFactoryResponse(res);
     }
     catch (error) {
         console.warn(`[${PROVIDER_NAME}] Failed to fetch models:`, error);
@@ -107,6 +127,9 @@ export default async function (pi) {
         baseUrl: BASE_URL,
         apiKey: ENV_VAR,
         api: "openai-completions",
+        headers: {
+            "Accept-Encoding": "identity",
+        },
         models,
     });
     // /nebius-models command to list and select a model

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tokenfactory-pi",
-  "version": "0.2.5",
+  "version": "0.2.6",
   "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",
@@ -43,4 +43,4 @@
     "check": "tsc --noEmit",
     "prepare": "npm run build"
   }
-}
+}