mtok-bridge 0.2.0 → 0.3.0

package/README.md

@@ -35,6 +35,16 @@ want to get PAID for a model (on-chain, per call, in USDC on Base) and be discov
 board instead of handing out keys, the market relay wraps this exact bridge with settlement. Same
 tool, one layer on top.
 
+## the shared serve core (for market hosts)
+
+`mtok-bridge` also exports `createServeCore`, the paid-serve state machine the market relay
+(and the workers-ai house seller) run: validate request => verify the on-chain DrawPaid
+(request-hash bound) => bound both legs against the payment => claim => upstream => complete,
+fail-closed. The redemption store is a pluggable interface (`{ state, get, claim, complete,
+retentionMs }`, each method sync or async: the core awaits every call), so a filesystem store and a Workers KV store drive the same code path. If you
+are just serving a model for a key, you never need it; it is here so every mtok seller host
+shares one money path.
+
 
 ---
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mtok-bridge",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "description": "Serve any model as an OpenAI-compatible API with a key. No payment, no market, runs anywhere node runs. The transport core behind mtok.market's seller relay.",
   "type": "module",
   "bin": {

package/src/bridge.mjs

@@ -4,6 +4,10 @@
 // later, the market relay on top) adapts its req/res to them. The market layer (on-chain verify,
 // fee, redemption, discovery) is a SEPARATE wrapper that calls the same serveChat guts; it is not
 // in here. (#566)
+//
+// The PAID-serve state machine those market hosts share lives in serve-core.mjs (#603) and is
+// re-exported here so `import { createServeCore } from 'mtok-bridge'` just works.
+export * from './serve-core.mjs';
 
 // Bearer-key auth. No key configured (null/'') = OPEN on purpose (keyless mode is an explicit
 // opt-in the CLI announces loudly). A configured key must match exactly.

package/src/serve-core.mjs

@@ -0,0 +1,345 @@
+// mtok-bridge serve core (#603): the PAID-serve state machine shared by every mtok seller host.
+// validate request => verify DrawPaid (request-hash bound) => screen payer => bound both legs =>
+// claim => upstream => complete, with the exact fail-closed semantics hardened in #597: a
+// post-claim failure leaves the draw durably pending (409 terminal on retry), so one payment can
+// never buy a second upstream spend. Host-agnostic and dependency-free (web crypto only): the
+// node relay composes it with its fs redemption store; the workers-ai house seller composes the
+// same core with a KV-backed store. The http shell, booking locks, platform config fetch, and
+// payer-screen POLICY stay in the host.
+//
+// The redemption dependency is an INTERFACE the caller passes:
+//   { state(key), get(key), claim(key, markerKey), complete(key, payload), retentionMs }
+// state(key) returns 'pending' | 'complete' | null. claim(key, markerKey) returns false when the
+// draw is already claimed and THROWS when it cannot claim durably (the core then refuses before
+// upstream spend). complete(key, payload) throws when the payload cannot be persisted; the claim
+// then stays pending on purpose so every replay fails closed instead of spending upstream again.
+
+const BALANCE_EPSILON = 1e-6;
+const REQUEST_NONCE_RE = /^0x[0-9a-fA-F]{32}$/;
+const CHAT_ROLES = new Set(['developer', 'system', 'user', 'assistant']);
+const REQUEST_KEYS = new Set(['model', 'messages', 'max_tokens', 'temperature', 'response_format', 'stream', 'n']);
+
+// sha256 over the request commitment, via web crypto so the identical bytes come out on node
+// (>=20 has globalThis.crypto) and on workers. Same output as node's createHash('sha256').
+async function hash32(v) {
+  const bytes = new TextEncoder().encode(typeof v === 'string' ? v : JSON.stringify(v ?? null));
+  const digest = await crypto.subtle.digest('SHA-256', bytes);
+  return '0x' + [...new Uint8Array(digest)].map((b) => b.toString(16).padStart(2, '0')).join('');
+}
+
+function legacyContentText(content) {
+  const render = (part) => {
+    if (typeof part === 'string') return part;
+    if (Array.isArray(part)) return part.map(render).filter(Boolean).join('\n');
+    if (part == null) return '';
+    if (typeof part !== 'object') return String(part);
+    if (typeof part.text === 'string') return part.text;
+    if (typeof part.content === 'string') return part.content;
+    const type = typeof part.type === 'string' && /^[a-z0-9_-]{1,32}$/i.test(part.type) ? part.type : 'non-text';
+    return `[${type} omitted]`;
+  };
+  return render(content) || '[empty legacy content]';
+}
+
+function sanitizeLegacyMessage(message) {
+  const source = message && typeof message === 'object' && !Array.isArray(message) ? message : { content: message };
+  if (CHAT_ROLES.has(source.role)) return { role: source.role, content: legacyContentText(source.content) };
+  const label = source.role === 'tool' || source.role === 'function' ? source.role : 'legacy';
+  return { role: 'user', content: `[${label} message]\n${legacyContentText(source.content)}` };
+}
+
+function validateRequest(request, model, { legacy = false } = {}) {
+  if (!request || typeof request !== 'object' || Array.isArray(request)) {
+    return { error: 'request must be an object' };
+  }
+  const unknown = Object.keys(request).find((key) => !REQUEST_KEYS.has(key));
+  if (unknown && !legacy) return { error: `unsupported request field: ${unknown}` };
+  if (!legacy && request.model != null && String(request.model) !== String(model)) {
+    return { error: `request model ${request.model} is not served here` };
+  }
+  if (!Array.isArray(request.messages) || request.messages.length === 0) {
+    return { error: 'request.messages must be a nonempty array' };
+  }
+  if (!legacy) {
+    for (const message of request.messages) {
+      if (!message || typeof message !== 'object' || Array.isArray(message)) {
+        return { error: 'each message must be an object' };
+      }
+      const extra = Object.keys(message).find((key) => key !== 'role' && key !== 'content');
+      if (extra) return { error: `unsupported message field: ${extra}` };
+      if (!CHAT_ROLES.has(message.role)) return { error: `unsupported message role: ${message.role}` };
+      if (typeof message.content !== 'string') return { error: 'message content must be plain text' };
+    }
+  }
+  if (!legacy && request.stream != null && request.stream !== false) return { error: 'streaming is not supported' };
+  if (!legacy && request.n != null && request.n !== 1) return { error: 'request.n must be 1' };
+  const validMaxTokens = Number.isInteger(request.max_tokens) && request.max_tokens > 0;
+  if (!legacy && request.max_tokens != null && !validMaxTokens) {
+    return { error: 'max_tokens must be a positive integer' };
+  }
+  const validTemperature = Number.isFinite(request.temperature) && request.temperature >= 0 && request.temperature <= 2;
+  if (!legacy && request.temperature != null && !validTemperature) {
+    return { error: 'temperature must be between 0 and 2' };
+  }
+  let validResponseFormat = false;
+  if (request.response_format != null) {
+    const format = request.response_format;
+    validResponseFormat = !!format && typeof format === 'object' && !Array.isArray(format)
+      && Object.keys(format).length === 1
+      && ['json_object', 'text'].includes(format.type);
+    if (!legacy && !validResponseFormat) {
+      return { error: 'response_format must be exactly { type: "json_object" } or { type: "text" }' };
+    }
+  }
+  return {
+    safeRequest: {
+      model,
+      messages: legacy
+        ? request.messages.map(sanitizeLegacyMessage)
+        : request.messages.map(({ role, content }) => ({ role, content })),
+      ...(validMaxTokens ? { max_tokens: request.max_tokens } : {}),
+      ...(validTemperature ? { temperature: request.temperature } : {}),
+      ...(validResponseFormat ? { response_format: { type: request.response_format.type } } : {}),
+    },
+  };
+}
+
+/**
+ * Throws if the upstream model doesn't match the offer model.
+ * Protects against cheap-swap accusations: echo the real model you delivered.
+ */
+export function enforceModelEcho(upstreamModel, offerModel) {
+  if (String(upstreamModel) !== String(offerModel))
+    throw new Error(`model mismatch: upstream ${upstreamModel} != offer ${offerModel}`);
+}
+
+export function configuredFeeAtomic({ sellerUsdAtomic, feeAddress, feeBps }) {
+  const bps = BigInt(Math.trunc(Math.max(0, Number(feeBps) || 0)));
+  if (!feeAddress || bps === 0n) return 0n;
+  return (BigInt(sellerUsdAtomic || 0) * bps + 5000n) / 10000n;
+}
+
+// Conservative tokenizer-independent upper estimate: a tokenizer cannot consume
+// more text tokens than UTF-8 bytes, plus the chat envelope around each message.
+// The core accepts plain-text messages only, so no unpriced multimodal parts
+// can bypass this bound. Pure, dependency-free, no I/O.
+export const MESSAGE_OVERHEAD_TOKENS = 4;
+export function estimateInputTokens(messages) {
+  const utf8 = new TextEncoder();
+  let tokens = 3; // reply priming
+  for (const m of messages ?? []) {
+    tokens += MESSAGE_OVERHEAD_TOKENS;
+    tokens += utf8.encode(String(m?.role ?? '')).length;
+    tokens += utf8.encode(typeof m?.content === 'string' ? m.content : JSON.stringify(m?.content ?? null)).length;
+  }
+  return tokens;
+}
+
+// Bound a serve against the paid budget in BOTH legs (#495/#460). The relay used
+// to cap only OUTPUT, so a dust draw + a huge prompt got its output capped but the
+// whole prompt forwarded, making the seller eat unbounded upstream INPUT compute.
+// Estimate the input cost and REFUSE before any upstream call if it alone meets or
+// exceeds the payment; otherwise cap output over the budget LEFT after input. inPrice
+// and outPrice are USD per MTok. The estimate gates the refuse ONLY; real billing
+// still meters the upstream's reported token counts, so this never over-charges.
+export function boundServe({ messages, budgetUsd, inPrice, outPrice, reqMax, contextCeil = 4096 }) {
+  const estIn = estimateInputTokens(messages);
+  const estInCostUsd = estIn * (Number(inPrice) > 0 ? Number(inPrice) : 0) / 1e6;
+  if (estInCostUsd >= budgetUsd) return { refuse: true, reason: 'input', estIn, estInCostUsd };
+  const outBudgetUsd = budgetUsd - estInCostUsd;
+  let maxTok = contextCeil;
+  if (Number(reqMax) > 0) maxTok = Math.min(maxTok, Math.floor(Number(reqMax)));
+  if (!Number.isFinite(Number(outPrice)) || Number(outPrice) <= 0) {
+    return { refuse: true, reason: 'output_price', estIn, estInCostUsd };
+  }
+  maxTok = Math.min(maxTok, Math.floor(outBudgetUsd / Number(outPrice) * 1e6));
+  if (maxTok < 1) return { refuse: true, reason: 'output', estIn, estInCostUsd };
+  return { refuse: false, maxTok, estIn, estInCostUsd };
+}
+
+// The core factory. `verifier` is an mtok-verify createOnchainVerifier instance (or anything
+// with the same verifyDrawPaid contract); `redemption` is the store interface documented at the
+// top; `upstream(payload)` returns an OpenAI-shaped completion or throws (httpUpstream /
+// workersAiUpstream from bridge.mjs both satisfy it); `screenPayer(payerLower)` is an optional
+// async predicate the host composes (denylist, hook), truthy = refuse, throw = refuse (fail
+// closed). serve(body) returns { status, body } for the host to serialize; the host owns
+// per-booking serialization (locks) around it.
+export function createServeCore({
+  model, inPrice, outPrice,
+  verifier, redemption, upstream, log,
+  offerId, sellerAgentId, sellerWallet,
+  dripContractAddress, feeRecipient, feeBps,
+  screenPayer,
+}) {
+  const serve = async (body) => {
+    const { bookingId, n, buyerId, request, requestNonce, drawPaidTxHash } = body;
+    const hasRequestNonce = Object.hasOwn(body, 'requestNonce');
+    if (!bookingId) return { status: 400, body: { error: 'bad_request', detail: 'DRAW needs bookingId' } };
+    if (n == null) return { status: 400, body: { error: 'bad_request', detail: 'DRAW needs a delivery index n (per-booking idempotency key)' } };
+    if (!Number.isSafeInteger(n) || n < 0 || n > 0xffffffff) {
+      return { status: 400, body: { error: 'bad_request', detail: 'DRAW delivery index n must be a nonnegative uint32 integer' } };
+    }
+    if (hasRequestNonce && !REQUEST_NONCE_RE.test(requestNonce)) {
+      return { status: 400, body: { error: 'bad_request', detail: 'DRAW needs requestNonce as 16 random bytes encoded as 0x-prefixed hex' } };
+    }
+    const checked = validateRequest(request, model, { legacy: !hasRequestNonce });
+    if (checked.error) return { status: 400, body: { error: 'bad_request', detail: checked.error } };
+
+    // Legacy SDKs paid for sha256(JSON.stringify(request)) and sent no nonce.
+    // Keep those already-paid draws redeemable while current offers advertise
+    // nonce-v1 so current SDKs can require the private commitment before paying.
+    const requestHashScheme = hasRequestNonce ? 'nonce-v1' : 'legacy-v0';
+    const requestHash = hasRequestNonce ? await hash32({ request, requestNonce }) : await hash32(request);
+    // Prefix the commitment scheme so legacy and nonce-v1 entries cannot alias.
+    // nonce-v1 binds the request to a buyer-held random nonce, preventing a chain
+    // observer from guessing a common prompt and deriving its completion key.
+    // Legacy redemption remains only to honor draws already paid by old SDKs.
+    const cacheKey = `${requestHashScheme}:${bookingId}:${n}:${requestHash}`;
+    const oldLegacyKey = hasRequestNonce ? null : `${bookingId}:${n}:${requestHash}`;
+
+    // Contract mode is the ONLY mode (#487): the legacy direct-transfer FUND
+    // lane and its /api/bookings/:id balance read are gone. If the platform is
+    // not running the drip contract, REFUSE the draw with a clear error rather
+    // than fall back to a lane that no longer exists.
+    if (!dripContractAddress) {
+      return { status: 402, body: { error: 'contract_mode_required', detail: 'this relay only serves contract-mode draws; the platform is not reporting a dripContractAddress' } };
+    }
+    if (!drawPaidTxHash) return { status: 402, body: { error: 'draw_payment_required', detail: 'contract mode requires drawPaidTxHash before upstream delivery' } };
+    let paid;
+    try {
+      paid = await verifier.verifyDrawPaid(drawPaidTxHash, {
+        contractAddress: dripContractAddress,
+        buyerAgentId: buyerId,
+        sellerAgentId, // when set, enforces the offer-owner match (#codex review)
+        bookingId,
+        offerId,
+        model,
+        n,
+        requestHash,
+        sellerWallet,
+        feeRecipient,
+        // #580: refuse a payment older than the redemption window. The JSONL
+        // payload cache AND the claim markers are both aged out at boot (#600),
+        // so past retention this age bound is the sole replay defense (its
+        // skip-on-unreadable-block residual is named in redemption.mjs). An
+        // honest retry is seconds-to-minutes old, never days.
+        maxPaidAgeMs: redemption.retentionMs,
+      });
+    } catch (e) {
+      return { status: 402, body: { error: 'payment_unverified', detail: e.message } };
+    }
+    if (!paid?.ok) return { status: 402, body: { error: 'payment_unverified', detail: paid?.reason || 'unknown' } };
+    const expectedFee = configuredFeeAtomic({
+      sellerUsdAtomic: paid.event.sellerUsdAtomic,
+      feeAddress: feeRecipient,
+      feeBps,
+    });
+    if (BigInt(paid.event.feeUsdAtomic || 0) < expectedFee) {
+      return { status: 402, body: { error: 'payment_unverified', detail: 'fee_amount_too_low' } };
+    }
+    // Screen the verified payer before spending upstream capacity. The
+    // payment already settled on-chain (that money is the buyer's loss);
+    // this refuses the SERVICE, which is the only refusal an edge can
+    // still make in contract mode.
+    if (screenPayer) {
+      try {
+        if (await screenPayer(String(paid.from || '').toLowerCase())) {
+          return { status: 403, body: { error: 'payer_denied', detail: 'the verified payer wallet is denylisted by this relay' } };
+        }
+      } catch (e) {
+        // A broken screen hook fails CLOSED: do not serve on an unscreenable payer.
+        return { status: 403, body: { error: 'payer_denied', detail: 'payer screening failed: ' + e.message } };
+      }
+    }
+    let storedKey = cacheKey;
+    // The redemption interface may be sync (fs store) or async (a Workers KV
+    // store): await normalizes both, and identity-awaits cost nothing under the
+    // host's per-booking lock.
+    let redemptionState = await redemption.state(storedKey);
+    // Preserve an upgrade's pre-scheme redemption log; missing this alias
+    // would let a legacy paid draw run upstream again after relay upgrade.
+    if (!redemptionState && oldLegacyKey) {
+      const oldLegacyState = await redemption.state(oldLegacyKey);
+      if (oldLegacyState) {
+        storedKey = oldLegacyKey;
+        redemptionState = oldLegacyState;
+      } else {
+        // Checking the legacy marker may have refreshed a prefixed record
+        // appended by another upgraded process.
+        redemptionState = await redemption.state(cacheKey);
+      }
+    }
+    if (redemptionState === 'complete') return { status: 200, body: await redemption.get(storedKey) };
+    if (redemptionState === 'pending') {
+      return { status: 409, body: { error: 'draw_pending', detail: 'this paid draw was already claimed; refusing to run upstream again', _bookingId: bookingId } };
+    }
+    const paidEvent = paid.event;
+    const remainingUsd = Number(paid.event.sellerUsdAtomic || 0) / 1e6;
+    if (remainingUsd < BALANCE_EPSILON) {
+      return { status: 402, body: { error: 'balance_exhausted', detail: `remainingUsd=${remainingUsd}`, _bookingId: bookingId, remainingUsd } };
+    }
+
+    // Bound BOTH legs against what the draw paid for (#495/#460). The relay used
+    // to cap only OUTPUT, so a dust draw + a huge prompt got its output capped but
+    // the whole prompt forwarded => the seller ate unbounded upstream INPUT compute.
+    // Now: REFUSE before any upstream call if the estimated input cost alone meets
+    // the payment, else cap output over the budget LEFT after input. Input is priced
+    // at the higher of our offer price and the buyer's committed event price, so the
+    // buyer can't zero the input leg to sneak a big prompt.
+    const eventInPriceUsd = Number(paidEvent.inputPricePerMTokAtomic || 0) / 1e6;
+    const eventOutPriceUsd = Number(paidEvent.outputPricePerMTokAtomic || 0) / 1e6;
+    const boundInPrice = Math.max(Number(inPrice) || 0, eventInPriceUsd);
+    const boundOutPrice = Math.max(Number(outPrice) || 0, eventOutPriceUsd);
+    const bound = boundServe({ messages: checked.safeRequest.messages, budgetUsd: remainingUsd, inPrice: boundInPrice, outPrice: boundOutPrice, reqMax: checked.safeRequest.max_tokens });
+    if (bound.refuse) {
+      const error = bound.reason === 'input' ? 'input_too_large' : 'output_unfunded';
+      return { status: 402, body: { error, detail: `estimated input (~${bound.estIn} tokens, $${bound.estInCostUsd.toFixed(6)}) leaves no safely funded output in the paid amount ($${remainingUsd})`, _bookingId: bookingId, remainingUsd } };
+    }
+    const safeRequest = { ...checked.safeRequest, model, max_tokens: bound.maxTok };
+
+    try {
+      // Legacy uses its old unprefixed identity only for the atomic marker so
+      // parallel upgraded stores and an existing log converge on one claim.
+      if (!(await redemption.claim(cacheKey, oldLegacyKey ?? cacheKey))) {
+        return { status: 409, body: { error: 'draw_pending', detail: 'this paid draw was already claimed; refusing to run upstream again', _bookingId: bookingId } };
+      }
+    } catch (e) {
+      return { status: 503, body: { error: 'redemption_unavailable', detail: `could not durably claim the paid draw: ${e.message}`, _bookingId: bookingId } };
+    }
+
+    let completion;
+    try {
+      completion = await upstream(safeRequest);
+    } catch (e) {
+      return { status: 502, body: { error: 'upstream_error', detail: e.message } };
+    }
+
+    try { enforceModelEcho(completion.model, model); }
+    catch (e) { return { status: 502, body: { error: 'model_mismatch', detail: e.message } }; }
+
+    // ── Contract mode is REPORT-FREE (chain-native phase 2 stage 3, #387) ──
+    // The verified DrawPaid event IS the record: the platform indexes the
+    // draw from MtokDripLedger logs, so there is nothing to report (the
+    // platform deleted POST /api/chunks/report in #487). The flow is
+    // verify => cap => claim => serve => complete, and the seller holds no platform
+    // secret. remainingUsd echoes what is left of THIS draw's paid amount
+    // after metering at the event's committed per-MTok prices.
+    const usage = completion.usage ?? {};
+    const inTok = Number(usage.prompt_tokens ?? usage.input_tokens ?? 0) || 0;
+    const outTok = Number(usage.completion_tokens ?? usage.output_tokens ?? 0) || 0;
+    // price is atomic USD per MTok (1e6 tokens): usd = tokens * priceAtomic / 1e12
+    const usedUsd = (inTok * Number(paidEvent.inputPricePerMTokAtomic || 0) + outTok * Number(paidEvent.outputPricePerMTokAtomic || 0)) / 1e12;
+    const payload = { ...completion, _bookingId: bookingId, remainingUsd: Math.max(0, Math.round((remainingUsd - usedUsd) * 1e6) / 1e6) };
+    try {
+      await redemption.complete(cacheKey, payload);
+    } catch (e) {
+      // The caller is already here and inference already ran: return its result.
+      // The durable pending claim remains authoritative, so every replay fails
+      // closed instead of spending upstream again.
+      (log ?? console).error?.(`mtok serve core: completion for ${cacheKey} could not be persisted (${e.message}); retries will remain pending`);
+    }
+    return { status: 200, body: payload };
+  };
+
+  return { serve };
+}