nous-token 0.3.1 → 0.3.3

package/cli/setup.ts

@@ -156,10 +156,10 @@ const tools: Tool[] = [
 // ── Shell env helper ──
 
 // Collect env vars to set — don't write to shell rc file
-const envCommands: string[] = [];
+const envCommands: Map<string, string> = new Map();
 
 function setShellEnv(key: string, value: string, toolName: string): ConfigResult {
-  envCommands.push(`export ${key}="${value}"`);
+  envCommands.set(key, `export ${key}="${value}"`);
   return { success: true, message: `${key}` };
 }
 
@@ -233,13 +233,14 @@ for (const tool of tools) {
 console.log("");
 
 if (configured > 0) {
+  const cmds = [...envCommands.values()];
   console.log(`  \x1b[32m${configured} tool(s) detected.\x1b[0m\n`);
   console.log("  \x1b[1mThis terminal only:\x1b[0m");
-  console.log(`  \x1b[36m${envCommands.join(" && ")}\x1b[0m\n`);
+  console.log(`  \x1b[36m${cmds.join(" && ")}\x1b[0m\n`);
   console.log("  \x1b[1mAll terminals (permanent):\x1b[0m");
   const shell = process.env.SHELL || "/bin/bash";
   const rcFile = shell.includes("zsh") ? "~/.zshrc" : "~/.bashrc";
-  console.log(`  \x1b[36mecho '${envCommands.join("\\n")}' >> ${rcFile} && source ${rcFile}\x1b[0m\n`);
+  console.log(`  \x1b[36mecho '${cmds.join("\\n")}' >> ${rcFile} && source ${rcFile}\x1b[0m\n`);
   console.log(`  See your usage at \x1b[4mhttps://token.nousai.cc\x1b[0m`);
 } else {
   console.log("  No AI tools found. Install Claude Code, Cursor, or any OpenAI-compatible tool.");

package/index.ts

@@ -1,4 +1,7 @@
 import type { OpenClawPluginApi } from "openclaw/plugin-sdk";
+import { readFileSync } from "fs";
+import { homedir } from "os";
+import { join } from "path";
 
 // nous-token plugin: routes LLM calls through the nous-token gateway
 // for real usage tracking on the global leaderboard.
@@ -12,6 +15,10 @@ import type { OpenClawPluginApi } from "openclaw/plugin-sdk";
 
 const GATEWAY = "https://gateway.nousai.cc";
 
+function loadWallet(): string {
+  try { return readFileSync(join(homedir(), ".nous-token"), "utf-8").trim(); } catch { return ""; }
+}
+
 // Built-in providers: prefix maps to gateway shortcut route
 const BUILTIN_PROVIDERS: Record<string, { prefix: string; api: string; envVars: string[] }> = {
   openai:     { prefix: "openai",     api: "openai-completions",  envVars: ["OPENAI_API_KEY"] },
@@ -34,11 +41,13 @@ interface PluginConfig {
 
 export default function plugin(api: OpenClawPluginApi): void {
   const cfg = (api.pluginConfig ?? {}) as PluginConfig;
+  const wallet = loadWallet();
+  const walletSegment = wallet ? `/w/${wallet}` : "";
 
   // ── Register built-in providers ──
   for (const [providerKey, { prefix, api: apiType, envVars }] of Object.entries(BUILTIN_PROVIDERS)) {
     registerProvider(api, providerKey, {
-      baseUrl: `${GATEWAY}/${prefix}/v1`,
+      baseUrl: `${GATEWAY}/${prefix}${walletSegment}/v1`,
       apiType,
       envVars,
     });
@@ -48,8 +57,7 @@ export default function plugin(api: OpenClawPluginApi): void {
   if (cfg.customProviders) {
     for (const [name, { upstream, envVar, api: apiType }] of Object.entries(cfg.customProviders)) {
       registerProvider(api, name, {
-        // Custom providers use X-Nous-Upstream header instead of shortcut prefix
-        baseUrl: `${GATEWAY}/v1`,
+        baseUrl: `${GATEWAY}${walletSegment}/v1`,
         apiType: apiType || "openai-completions",
         envVars: [envVar],
         upstreamHeader: upstream,
@@ -58,7 +66,8 @@ export default function plugin(api: OpenClawPluginApi): void {
   }
 
   const total = Object.keys(BUILTIN_PROVIDERS).length + Object.keys(cfg.customProviders || {}).length;
-  api.logger.info(`[nous-token] ready — ${total} providers, usage tracked via gateway`);
+  if (!wallet) api.logger.warn(`[nous-token] no wallet found — run "npx nous-token setup 0xYOUR_WALLET" to link usage to your address`);
+  api.logger.info(`[nous-token] ready — ${total} providers${wallet ? `, wallet ${wallet.slice(0, 6)}...${wallet.slice(-4)}` : ""}`);
 }
 
 // ── Provider registration ──

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "nous-token",
-  "version": "0.3.1",
+  "version": "0.3.3",
   "description": "Open-source AI usage tracker. One command to track all your AI spending.",
   "type": "module",
   "main": "index.ts",