tokenmix 0.4.13 → 0.4.14

package/README.md

@@ -102,6 +102,14 @@ TOKENMIX_DEFAULT_MODEL=qwen-flash       npx tokenmix aider
 
 (`tokenmix claude` and `tokenmix codex` speak the Anthropic / Responses protocols, so they need a Claude-family model; the other agents accept any chat model.)
 
+## Slow or restricted networks
+
+Requests auto-retry transient network failures (so a single hiccup won't fail a command) and time out after 20s. On a slow, proxied, or firewalled connection, raise the timeout with `TOKENMIX_TIMEOUT_MS`:
+
+```bash
+TOKENMIX_TIMEOUT_MS=60000 npx tokenmix login
+```
+
 ## Configuration Location
 
 Your TokenMix credentials are stored locally at:

package/dist/api/client.js

@@ -22,7 +22,28 @@ function handleAxios(err) {
             e.message;
         throw new ApiError(e.response.status, msg);
     }
-    throw new ApiError(0, `Could not reach the TokenMix API (${e.message || 'network error'}). Check your internet connection.`);
+    throw new ApiError(0, `Could not reach the TokenMix API (${e.message || 'network error'}). Check your internet connection or proxy.`);
+}
+// Network resilience for flaky / slow / GFW'd connections: a generous default
+// timeout (override with TOKENMIX_TIMEOUT_MS) and automatic retry of transient
+// TRANSPORT failures (no HTTP response) with exponential backoff. HTTP errors
+// (4xx/5xx) are real answers and are NEVER retried.
+export const REQUEST_TIMEOUT_MS = Number(process.env.TOKENMIX_TIMEOUT_MS) || 20000;
+const MAX_RETRIES = 2;
+export async function withRetry(fn) {
+    let lastErr;
+    for (let attempt = 0; attempt <= MAX_RETRIES; attempt++) {
+        try {
+            return await fn();
+        }
+        catch (err) {
+            if (err.response || attempt === MAX_RETRIES)
+                throw err;
+            lastErr = err;
+            await new Promise((r) => setTimeout(r, 300 * 2 ** attempt));
+        }
+    }
+    throw lastErr;
 }
 // Public endpoint, no auth required.
 // Note: backend pagination uses `per_page` (NOT `page_size`); max 500 (anything >500 falls back to 20).
@@ -30,10 +51,10 @@ function handleAxios(err) {
 export async function listPublicModels(cfg) {
     const c = cfg || (await readConfig());
     try {
-        const r = await axios.get(`${apiBaseUrl(c)}/api/models`, {
+        const r = await withRetry(() => axios.get(`${apiBaseUrl(c)}/api/models`, {
             params: { per_page: 500 },
-            timeout: 15000,
-        });
+            timeout: REQUEST_TIMEOUT_MS,
+        }));
         const list = unwrap(r);
         return Array.isArray(list) ? list : list?.list ?? [];
     }
@@ -49,11 +70,11 @@ export async function verifyApiKey(apiKey, baseUrl) {
     // then raises a clear "could not reach the API" ApiError, letting callers tell a
     // network problem apart from a genuinely bad key.
     try {
-        const r = await axios.get(`${baseUrl || DEFAULT_API_BASE}/v1/models`, {
+        const r = await withRetry(() => axios.get(`${baseUrl || DEFAULT_API_BASE}/v1/models`, {
             headers: { Authorization: `Bearer ${apiKey}` },
-            timeout: 15000,
+            timeout: REQUEST_TIMEOUT_MS,
             validateStatus: () => true,
-        });
+        }));
         return r.status === 200;
     }
     catch (err) {
@@ -62,10 +83,10 @@ export async function verifyApiKey(apiKey, baseUrl) {
 }
 export async function fetchWallet(apiKey, baseUrl) {
     try {
-        const r = await axios.get(`${baseUrl || DEFAULT_API_BASE}/v1/wallet`, {
+        const r = await withRetry(() => axios.get(`${baseUrl || DEFAULT_API_BASE}/v1/wallet`, {
             headers: { Authorization: `Bearer ${apiKey}` },
-            timeout: 15000,
-        });
+            timeout: REQUEST_TIMEOUT_MS,
+        }));
         return unwrap(r);
     }
     catch (err) {
@@ -81,7 +102,7 @@ export class DeviceFlowError extends Error {
 }
 export async function startDeviceAuthorization(baseUrl, clientName = 'tokenmix-cli') {
     try {
-        const r = await axios.post(`${baseUrl}/api/auth/device/code`, { client_name: clientName }, { timeout: 15000 });
+        const r = await withRetry(() => axios.post(`${baseUrl}/api/auth/device/code`, { client_name: clientName }, { timeout: REQUEST_TIMEOUT_MS }));
         return unwrap(r);
     }
     catch (err) {
@@ -106,7 +127,7 @@ export async function pollDeviceToken(baseUrl, auth, onTick) {
             onTick(Math.max(0, Math.round((deadline - Date.now()) / 1000)));
         }
         try {
-            const r = await axios.post(`${baseUrl}/api/auth/device/token`, { device_code: auth.device_code }, { timeout: 15000, validateStatus: () => true });
+            const r = await axios.post(`${baseUrl}/api/auth/device/token`, { device_code: auth.device_code }, { timeout: REQUEST_TIMEOUT_MS, validateStatus: () => true });
             if (r.status === 200 && r.data?.code === 0) {
                 const body = r.data.data;
                 return {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tokenmix",
-  "version": "0.4.13",
+  "version": "0.4.14",
   "description": "Zero-config CLI to use any open-source coding agent with TokenMix as the unified LLM backend.",
   "type": "module",
   "bin": {