solrouter 0.3.3 → 0.3.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "solrouter",
-  "version": "0.3.3",
+  "version": "0.3.4",
   "description": "Local privacy router for coding tools: pseudonymises every request before it leaves your machine, and sends work a frontier model would refuse to an uncensored open-weight model you control.",
   "type": "module",
   "bin": {

package/src/providers.js

@@ -107,6 +107,20 @@ export async function callSolrouterAgent(leg, prompt) {
   return { ok: res.ok && json?.success !== false, status: res.status, reply: json?.reply ?? "", raw: json };
 }
 
+/**
+ * Transparent passthrough to the frontier for endpoints we don't transform —
+ * e.g. Claude Code's /v1/messages/count_tokens preflight. Returns the raw body so
+ * the response is relayed verbatim.
+ */
+export async function forward(leg, { method = "POST", path, body, clientHeaders }) {
+  const base = leg.baseUrl.replace(/\/$/, "");
+  const init = { method, headers: authHeaders(leg, clientHeaders) };
+  if (body !== undefined) init.body = JSON.stringify(withModel(body, leg));
+  const res = await fetch(base + path, init);
+  const text = await res.text();
+  return { ok: res.ok, status: res.status, text, contentType: res.headers.get("content-type") };
+}
+
 /** Streaming call. Returns { ok, status, stream } (raw SSE ReadableStream). */
 export async function streamLeg(leg, body, clientHeaders) {
   const res = await fetch(endpoint(leg), {

package/src/server.js

@@ -7,7 +7,9 @@
 // localhost only — nothing here should be reachable off the machine.
 
 import { createServer } from "node:http";
-import { handle, handleStream } from "./index.js";
+import { handle, handleStream, createMasker } from "./index.js";
+import { forward } from "./providers.js";
+import { dialect, rewriteText } from "./wire.js";
 import { writeReceipt } from "./receipt.js";
 
 function readBody(req) {
@@ -26,6 +28,27 @@ export function startServer(cfg) {
     const send = (code, obj) => { res.writeHead(code, { "content-type": "application/json" }); res.end(JSON.stringify(obj)); };
 
     if (req.method === "GET" && req.url === "/health") return send(200, { ok: true, proxy: "solrouter", mode: cfg.mode });
+
+    // Transparent passthrough for other Anthropic endpoints the client probes —
+    // notably /v1/messages/count_tokens, which Claude Code calls before every
+    // request and treats a 404 on as "model unavailable". Forward to the frontier.
+    if (/^\/v1\//.test(req.url) && !/^\/(v1\/chat\/completions|v1\/messages)$/.test(req.url)) {
+      let body;
+      if (req.method === "POST") {
+        try { body = await readBody(req); } catch { return send(400, { error: "invalid JSON body" }); }
+        if (body && cfg.masking?.enabled !== false) {
+          const m = createMasker(cfg.masking);
+          body = rewriteText(body, dialect(body, req.url), s => m.mask(s).text);
+        }
+      }
+      try {
+        const r = await forward(cfg.frontier, { method: req.method, path: req.url, body, clientHeaders: lower(req.headers) });
+        res.writeHead(r.status, { "content-type": r.contentType || "application/json" });
+        res.end(r.text);
+      } catch (e) { send(502, { error: "proxy error", detail: String(e?.message || e) }); }
+      return;
+    }
+
     if (req.method !== "POST" || !/^\/(v1\/chat\/completions|v1\/messages)$/.test(req.url)) {
       return send(404, { error: "not found" });
     }