@tokenzero/mcp 0.1.4 → 0.1.6

package/bin/tokenzero-mcp.mjs

@@ -68,7 +68,18 @@ var PRICES = {
   "gemini-3-flash": P("gemini", "fast", 0.5, 3, 0.05),
   "gemini-2.5-pro": P("gemini", "frontier", 1.25, 10, 0.125, void 0, { longCtxAt: 2e5, longCtx: { input: 2.5, output: 15, cachedInput: 0.25 } }),
   "gemini-2.5-flash": P("gemini", "fast", 0.3, 2.5, 0.03),
-  "gemini-2.5-flash-lite": P("gemini", "fast", 0.1, 0.4, 0.01)
+  "gemini-2.5-flash-lite": P("gemini", "fast", 0.1, 0.4, 0.01),
+  // ---- xAI Grok ----
+  "grok-4": P("xai", "frontier", 3, 15, 0.75),
+  "grok-4-fast": P("xai", "fast", 0.2, 0.8, 0.05),
+  "grok-4-mini": P("xai", "fast", 0.2, 0.8, 0.05),
+  "grok-3": P("xai", "mid", 3, 15, 0.75),
+  "grok-3-mini": P("xai", "fast", 0.3, 0.5, 0.075),
+  // ---- DeepSeek ----
+  "deepseek-v3": P("deepseek", "fast", 0.27, 1.1, 0.07),
+  "deepseek-chat": P("deepseek", "fast", 0.27, 1.1, 0.07),
+  "deepseek-r1": P("deepseek", "mid", 0.55, 2.19, 0.14),
+  "deepseek-reasoner": P("deepseek", "mid", 0.55, 2.19, 0.14)
 };
 var CASCADE_TARGET = {
   "gpt-6-astra": "gpt-6.1-sol",
@@ -99,13 +110,19 @@ var CASCADE_TARGET = {
   "gemini-3.1-pro": "gemini-3.8-flash",
   "gemini-3.8-flash": "gemini-3.5-flash-lite",
   "gemini-2.5-pro": "gemini-2.5-flash",
-  "gemini-2.5-flash": "gemini-2.5-flash-lite"
+  "gemini-2.5-flash": "gemini-2.5-flash-lite",
+  "grok-4": "grok-4-fast",
+  "grok-3": "grok-3-mini",
+  "deepseek-r1": "deepseek-v3",
+  "deepseek-reasoner": "deepseek-chat"
 };
 var KEYS_BY_LENGTH = Object.keys(PRICES).sort((a, b) => b.length - a.length);
 function normalizeModel(model) {
   if (typeof model !== "string" || !model) return null;
   let m = model.toLowerCase().trim();
-  m = m.replace(/^(openai|anthropic|google|models|vertex_ai)\//, "");
+  m = m.replace(/^(openai|anthropic|google|gemini|x-ai|xai|deepseek|bedrock|models|vertex_ai)\//, "");
+  m = m.replace(/^(?:us\.|eu\.|apac\.)?anthropic\./, "");
+  m = m.replace(/-v\d+:\d+$/, "");
   m = m.replace(/@\d{8}$/, "");
   m = m.replace(/-(\d{8}|\d{4}-\d{2}-\d{2})$/, "");
   m = m.replace(/-(latest|preview|customtools|thinking)$/, "");
@@ -667,7 +684,7 @@ function manifests(root) {
 }
 
 // mcp/src/main.ts
-var VERSION = "0.1.4";
+var VERSION = "0.1.6";
 var BAKED_URL = true ? "https://tokenzero.sh" : "__TOKENZERO_DEFAULT_URL__";
 var SUPPORTED = ["2026-07-28", "2025-11-25", "2025-06-18", "2025-03-26"];
 var serviceUrl = () => (process.env.TOKENZERO_URL || (BAKED_URL.startsWith("http") ? BAKED_URL : "")).replace(/\/+$/, "");
@@ -686,6 +703,11 @@ var TOOLS = [
     name: "optimize_prompt_payload",
     description: "Rewrite an OpenAI Chat/Responses or Anthropic Messages request body so its static prefix is provider-cacheable, and project the saving. Sends the body you pass to the TokenZero service (set TOKENZERO_URL); do not pass secrets in it.",
     inputSchema: { type: "object", properties: { body: { type: "object" }, endpoint: { type: "string", enum: ["/v1/chat/completions", "/v1/responses", "/v1/messages"] }, strategy: { type: "string", enum: ["static", "conversation"] } }, required: ["body"], additionalProperties: false }
+  },
+  {
+    name: "provision_workspace",
+    description: "Zero-click autonomous workspace and API key provisioning on TokenZero (POST /v1/workspaces). Returns a live tz_live_... API key, workspace_id, 1-click auto-unlock dashboard URL, gateway base URLs, and badge SVG URL with zero signup form required.",
+    inputSchema: { type: "object", properties: { name: { type: "string", description: "Optional workspace name (defaults to mcp-agent)" } }, additionalProperties: false }
   }
 ];
 async function callTool(name, args) {
@@ -714,6 +736,38 @@ async function callTool(name, args) {
       if (j.result?.isError) throw new Error(j.result.content?.[0]?.text ?? "remote tool error");
       return j.result?.structuredContent ?? j.result;
     }
+    case "provision_workspace": {
+      const base = serviceUrl();
+      if (!base) throw new Error("set TOKENZERO_URL to use this tool");
+      const wsName = typeof args?.name === "string" && args.name.trim() ? args.name.trim() : "mcp-agent";
+      const r = await fetch(base + "/v1/workspaces", {
+        method: "POST",
+        redirect: "manual",
+        signal: AbortSignal.timeout(15e3),
+        headers: { "content-type": "application/json", "user-agent": `tokenzero-mcp/${VERSION}` },
+        body: JSON.stringify({ name: wsName })
+      }).catch((e) => {
+        throw new Error(`could not reach the TokenZero service at ${base} (${e?.cause?.code ?? e?.message ?? e}); set TOKENZERO_URL to use a different one`);
+      });
+      if (r.status >= 300 && r.status < 400) throw new Error(`${base} is not publicly reachable yet (redirected)`);
+      const j = await r.json();
+      if (r.status >= 400) throw new Error(j?.error?.message ?? `HTTP ${r.status}`);
+      const id = j.workspace?.id ?? j.workspace_id ?? j.id;
+      const key = j.key ?? j.api_key;
+      return {
+        workspace_id: id,
+        api_key: key,
+        name: wsName,
+        dashboard_url: `${base}/dashboard?workspace=${id}`,
+        one_click_dashboard_url: `${base}/dashboard?workspace=${id}#key=${key}`,
+        gateway_base_url: `${base}/v1`,
+        anthropic_base_url: base,
+        gemini_base_url: `${base}/gemini`,
+        zero_code_base_url: `${base}/w/${key}/v1`,
+        mcp_url: `${base}/mcp`,
+        badge_svg_url: `${base}/v1/badge/${id}.svg`
+      };
+    }
     default:
       throw new Error(`unknown tool ${name}`);
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tokenzero/mcp",
-  "version": "0.1.4",
+  "version": "0.1.6",
   "description": "TokenZero MCP server (stdio): audit LLM token waste in your project locally, price model alternatives, and make prompts cache-friendly from Claude Code, Cursor, Windsurf or Cline.",
   "type": "module",
   "license": "Apache-2.0",