@blockrun/clawrouter 0.5.9 → 0.6.2

package/dist/index.js

@@ -1996,9 +1996,13 @@ var InsufficientFundsError = class extends Error {
   requiredUSD;
   walletAddress;
   constructor(opts) {
-    super(
-      `Insufficient USDC balance. Current: ${opts.currentBalanceUSD}, Required: ${opts.requiredUSD}. Fund wallet: ${opts.walletAddress}`
-    );
+    const msg = [
+      `Insufficient balance. Current: ${opts.currentBalanceUSD}, Required: ${opts.requiredUSD}`,
+      `Options:`,
+      `  1. Fund wallet: ${opts.walletAddress}`,
+      `  2. Use free model: /model free`
+    ].join("\n");
+    super(msg);
     this.name = "InsufficientFundsError";
     this.currentBalanceUSD = opts.currentBalanceUSD;
     this.requiredUSD = opts.requiredUSD;
@@ -2009,7 +2013,14 @@ var EmptyWalletError = class extends Error {
   code = "EMPTY_WALLET";
   walletAddress;
   constructor(walletAddress) {
-    super(`No USDC balance. Fund wallet to use ClawRouter: ${walletAddress}`);
+    const msg = [
+      `No USDC balance.`,
+      `Options:`,
+      `  1. Fund wallet: ${walletAddress}`,
+      `  2. Use free model: /model free`,
+      `  3. Uninstall: bash ~/.openclaw/extensions/clawrouter/scripts/uninstall.sh`
+    ].join("\n");
+    super(msg);
     this.name = "EmptyWalletError";
     this.walletAddress = walletAddress;
   }
@@ -2309,6 +2320,34 @@ var DEFAULT_REQUEST_TIMEOUT_MS = 18e4;
 var DEFAULT_PORT = 8402;
 var MAX_FALLBACK_ATTEMPTS = 3;
 var HEALTH_CHECK_TIMEOUT_MS = 2e3;
+var RATE_LIMIT_COOLDOWN_MS = 6e4;
+var rateLimitedModels = /* @__PURE__ */ new Map();
+function isRateLimited(modelId) {
+  const hitTime = rateLimitedModels.get(modelId);
+  if (!hitTime) return false;
+  const elapsed = Date.now() - hitTime;
+  if (elapsed >= RATE_LIMIT_COOLDOWN_MS) {
+    rateLimitedModels.delete(modelId);
+    return false;
+  }
+  return true;
+}
+function markRateLimited(modelId) {
+  rateLimitedModels.set(modelId, Date.now());
+  console.log(`[ClawRouter] Model ${modelId} rate-limited, will deprioritize for 60s`);
+}
+function prioritizeNonRateLimited(models) {
+  const available = [];
+  const rateLimited = [];
+  for (const model of models) {
+    if (isRateLimited(model)) {
+      rateLimited.push(model);
+    } else {
+      available.push(model);
+    }
+  }
+  return [...available, ...rateLimited];
+}
 var BALANCE_CHECK_BUFFER = 1.5;
 function getProxyPort() {
   const envPort = process.env.BLOCKRUN_PROXY_PORT;
@@ -2431,6 +2470,18 @@ function normalizeMessagesForGoogle(messages) {
 function isGoogleModel(modelId) {
   return modelId.startsWith("google/") || modelId.startsWith("gemini");
 }
+function normalizeMessagesForThinking(messages) {
+  if (!messages || messages.length === 0) return messages;
+  let hasChanges = false;
+  const normalized = messages.map((msg) => {
+    if (msg.role === "assistant" && msg.tool_calls && Array.isArray(msg.tool_calls) && msg.tool_calls.length > 0 && msg.reasoning_content === void 0) {
+      hasChanges = true;
+      return { ...msg, reasoning_content: "" };
+    }
+    return msg;
+  });
+  return hasChanges ? normalized : messages;
+}
 var KIMI_BLOCK_RE = /<[｜|][^<>]*begin[^<>]*[｜|]>[\s\S]*?<[｜|][^<>]*end[^<>]*[｜|]>/gi;
 var KIMI_TOKEN_RE = /<[｜|][^<>]*[｜|]>/g;
 var THINKING_TAG_RE = /<\s*\/?\s*(?:think(?:ing)?|thought|antthinking)\b[^>]*>/gi;
@@ -2642,6 +2693,9 @@ async function tryModelRequest(upstreamUrl, method, headers, body, modelId, maxT
     if (isGoogleModel(modelId) && Array.isArray(parsed.messages)) {
       parsed.messages = normalizeMessagesForGoogle(parsed.messages);
     }
+    if (parsed.thinking && Array.isArray(parsed.messages)) {
+      parsed.messages = normalizeMessagesForThinking(parsed.messages);
+    }
     requestBody = Buffer.from(JSON.stringify(parsed));
   } catch {
   }
@@ -2801,42 +2855,18 @@ async function proxyRequest(req, res, apiBase, payFetch, options, routerOpts, de
       const bufferedCostMicros = estimatedCostMicros * BigInt(Math.ceil(BALANCE_CHECK_BUFFER * 100)) / 100n;
       const sufficiency = await balanceMonitor.checkSufficient(bufferedCostMicros);
       if (sufficiency.info.isEmpty || !sufficiency.sufficient) {
-        if (routingDecision) {
-          console.log(
-            `[ClawRouter] Wallet ${sufficiency.info.isEmpty ? "empty" : "insufficient"} ($${sufficiency.info.balanceUSD}), falling back to free model: ${FREE_MODEL}`
-          );
-          modelId = FREE_MODEL;
-          const parsed = JSON.parse(body.toString());
-          parsed.model = FREE_MODEL;
-          body = Buffer.from(JSON.stringify(parsed));
-          options.onLowBalance?.({
-            balanceUSD: sufficiency.info.balanceUSD,
-            walletAddress: sufficiency.info.walletAddress
-          });
-        } else {
-          deduplicator.removeInflight(dedupKey);
-          if (sufficiency.info.isEmpty) {
-            const error = new EmptyWalletError(sufficiency.info.walletAddress);
-            options.onInsufficientFunds?.({
-              balanceUSD: sufficiency.info.balanceUSD,
-              requiredUSD: balanceMonitor.formatUSDC(bufferedCostMicros),
-              walletAddress: sufficiency.info.walletAddress
-            });
-            throw error;
-          } else {
-            const error = new InsufficientFundsError({
-              currentBalanceUSD: sufficiency.info.balanceUSD,
-              requiredUSD: balanceMonitor.formatUSDC(bufferedCostMicros),
-              walletAddress: sufficiency.info.walletAddress
-            });
-            options.onInsufficientFunds?.({
-              balanceUSD: sufficiency.info.balanceUSD,
-              requiredUSD: balanceMonitor.formatUSDC(bufferedCostMicros),
-              walletAddress: sufficiency.info.walletAddress
-            });
-            throw error;
-          }
-        }
+        const originalModel = modelId;
+        console.log(
+          `[ClawRouter] Wallet ${sufficiency.info.isEmpty ? "empty" : "insufficient"} ($${sufficiency.info.balanceUSD}), falling back to free model: ${FREE_MODEL} (requested: ${originalModel})`
+        );
+        modelId = FREE_MODEL;
+        const parsed = JSON.parse(body.toString());
+        parsed.model = FREE_MODEL;
+        body = Buffer.from(JSON.stringify(parsed));
+        options.onLowBalance?.({
+          balanceUSD: sufficiency.info.balanceUSD,
+          walletAddress: sufficiency.info.walletAddress
+        });
       } else if (sufficiency.info.isLow) {
         options.onLowBalance?.({
           balanceUSD: sufficiency.info.balanceUSD,
@@ -2907,6 +2937,7 @@ async function proxyRequest(req, res, apiBase, payFetch, options, routerOpts, de
         );
       }
       modelsToTry = contextFiltered.slice(0, MAX_FALLBACK_ATTEMPTS);
+      modelsToTry = prioritizeNonRateLimited(modelsToTry);
     } else {
       modelsToTry = modelId ? [modelId] : [];
     }
@@ -2939,6 +2970,9 @@ async function proxyRequest(req, res, apiBase, payFetch, options, routerOpts, de
         status: result.errorStatus || 500
       };
       if (result.isProviderError && !isLastAttempt) {
+        if (result.errorStatus === 429) {
+          markRateLimited(tryModel);
+        }
         console.log(
           `[ClawRouter] Provider error from ${tryModel}, trying fallback: ${result.errorBody?.slice(0, 100)}`
         );
@@ -3265,6 +3299,18 @@ function isRetryable(errorOrResponse, config) {
 }
 
 // src/index.ts
+async function waitForProxyHealth(port, timeoutMs = 3e3) {
+  const start = Date.now();
+  while (Date.now() - start < timeoutMs) {
+    try {
+      const res = await fetch(`http://127.0.0.1:${port}/health`);
+      if (res.ok) return true;
+    } catch {
+    }
+    await new Promise((r) => setTimeout(r, 100));
+  }
+  return false;
+}
 function isCompletionMode() {
   const args = process.argv;
   return args.some((arg, i) => arg === "completion" && i >= 1 && i <= 3);
@@ -3314,20 +3360,34 @@ function injectModelsConfig(logger) {
       config.agents.defaults.model.primary = "blockrun/auto";
       needsWrite = true;
     }
-    if (config.agents.defaults.models && typeof config.agents.defaults.models === "object") {
-      const allowlist = config.agents.defaults.models;
-      for (const model of OPENCLAW_MODELS) {
-        const fullId = `blockrun/${model.id}`;
-        if (!allowlist[fullId]) {
-          const alias = model.id.includes("/") ? model.id.split("/").pop() : model.id;
-          allowlist[fullId] = { alias };
-          needsWrite = true;
-        }
+    const KEY_MODEL_ALIASES = [
+      { id: "auto", alias: "auto" },
+      { id: "free", alias: "free" },
+      { id: "sonnet", alias: "sonnet" },
+      { id: "opus", alias: "opus" },
+      { id: "haiku", alias: "haiku" },
+      { id: "grok", alias: "grok" },
+      { id: "deepseek", alias: "deepseek" },
+      { id: "kimi", alias: "kimi" },
+      { id: "gemini", alias: "gemini" },
+      { id: "flash", alias: "flash" },
+      { id: "gpt", alias: "gpt" },
+      { id: "reasoner", alias: "reasoner" }
+    ];
+    if (!config.agents) config.agents = {};
+    if (!config.agents.defaults) config.agents.defaults = {};
+    if (!config.agents.defaults.models) config.agents.defaults.models = {};
+    const allowlist = config.agents.defaults.models;
+    for (const m of KEY_MODEL_ALIASES) {
+      const fullId = `blockrun/${m.id}`;
+      if (!allowlist[fullId]) {
+        allowlist[fullId] = { alias: m.alias };
+        needsWrite = true;
       }
     }
     if (needsWrite) {
       writeFileSync(configPath, JSON.stringify(config, null, 2));
-      logger.info("Set default model to blockrun/auto (smart routing enabled)");
+      logger.info("Smart routing enabled (blockrun/auto)");
     }
   } catch {
   }
@@ -3399,29 +3459,11 @@ async function startProxyInBackground(api) {
   const { key: walletKey, address, source } = await resolveOrGenerateWalletKey();
   if (source === "generated") {
     api.logger.info(`Generated new wallet: ${address}`);
-    api.logger.info(`Fund with USDC on Base to start using ClawRouter.`);
   } else if (source === "saved") {
     api.logger.info(`Using saved wallet: ${address}`);
   } else {
     api.logger.info(`Using wallet from BLOCKRUN_WALLET_KEY: ${address}`);
   }
-  const startupMonitor = new BalanceMonitor(address);
-  try {
-    const startupBalance = await startupMonitor.checkBalance();
-    if (startupBalance.isEmpty) {
-      api.logger.warn(`[!] No USDC balance. Fund wallet to use ClawRouter: ${address}`);
-    } else if (startupBalance.isLow) {
-      api.logger.warn(
-        `[!] Low balance: ${startupBalance.balanceUSD} remaining. Fund wallet: ${address}`
-      );
-    } else {
-      api.logger.info(`Wallet balance: ${startupBalance.balanceUSD}`);
-    }
-  } catch (err) {
-    api.logger.warn(
-      `Could not check wallet balance: ${err instanceof Error ? err.message : String(err)}`
-    );
-  }
   const routingConfig = api.pluginConfig?.routing;
   const proxy = await startProxy({
     walletKey,
@@ -3450,7 +3492,21 @@ async function startProxyInBackground(api) {
   });
   setActiveProxy(proxy);
   activeProxyHandle = proxy;
-  api.logger.info(`BlockRun provider active \u2014 ${proxy.baseUrl}/v1 (smart routing enabled)`);
+  api.logger.info(`ClawRouter ready \u2014 smart routing enabled`);
+  api.logger.info(`Pricing: Simple ~$0.001 | Code ~$0.01 | Complex ~$0.05 | Free: $0`);
+  const startupMonitor = new BalanceMonitor(address);
+  startupMonitor.checkBalance().then((balance) => {
+    if (balance.isEmpty) {
+      api.logger.info(`Wallet: ${address} | Balance: $0.00`);
+      api.logger.info(`Using FREE model. Fund wallet for premium models.`);
+    } else if (balance.isLow) {
+      api.logger.info(`Wallet: ${address} | Balance: ${balance.balanceUSD} (low)`);
+    } else {
+      api.logger.info(`Wallet: ${address} | Balance: ${balance.balanceUSD}`);
+    }
+  }).catch(() => {
+    api.logger.info(`Wallet: ${address} | Balance: (checking...)`);
+  });
 }
 async function createStatsCommand() {
   return {
@@ -3556,7 +3612,7 @@ var plugin = {
   name: "ClawRouter",
   description: "Smart LLM router \u2014 30+ models, x402 micropayments, 78% cost savings",
   version: VERSION,
-  register(api) {
+  async register(api) {
     const isDisabled = process.env.CLAWROUTER_DISABLED === "true" || process.env.CLAWROUTER_DISABLED === "1";
     if (isDisabled) {
       api.logger.info("ClawRouter disabled (CLAWROUTER_DISABLED=true). Using default routing.");
@@ -3622,11 +3678,18 @@ var plugin = {
         }
       }
     });
-    startProxyInBackground(api).catch((err) => {
+    try {
+      await startProxyInBackground(api);
+      const port = getProxyPort();
+      const healthy = await waitForProxyHealth(port);
+      if (!healthy) {
+        api.logger.warn(`Proxy health check timed out, commands may not work immediately`);
+      }
+    } catch (err) {
       api.logger.error(
         `Failed to start BlockRun proxy: ${err instanceof Error ? err.message : String(err)}`
       );
-    });
+    }
   }
 };
 var index_default = plugin;