initial relay scaffold

2026-05-11 20:03:27 -05:00
commit b9d86fa303
58 changed files with 7609 additions and 0 deletions
@@ -0,0 +1,114 @@
+// POST /relay/analyze — forwards an analysis prompt to the chosen
+// backend and returns the standard envelope.
+//
+// Request body (application/json):
+//   { prompt: string }
+//
+// Headers: same as /relay/transcribe (X-Recap-Install-Id required,
+// X-Recap-Job-Id optional, Authorization optional Bearer license).
+//
+// Same charge-once-per-job semantics: a Recap summarize job pairs
+// transcribe + analyze with the same X-Recap-Job-Id. The first call
+// (whichever endpoint) charges 1 credit; the second is free.
+
+import express from "express";
+import { resolveLicense } from "../keysat-client.js";
+import { getOrCreateRow, planBackend, commitCredit } from "../credits.js";
+import { lookupJob, markJobCharged, refundJob } from "../job-credits.js";
+import { getConfigSnapshot, getTierQuotas } from "../config.js";
+import { createGeminiBackend } from "../backends/gemini.js";
+import { createHardwareBackend } from "../backends/hardware.js";
+import { envelope, errorEnvelope } from "./envelope.js";
+
+export function analyzeRouter() {
+  const router = express.Router();
+
+  router.post("/analyze", express.json({ limit: "10mb" }), async (req, res) => {
+    const installId = req.header("X-Recap-Install-Id");
+    const jobId = req.header("X-Recap-Job-Id") || null;
+    const auth = req.header("Authorization");
+
+    if (!installId) {
+      const e = await errorEnvelope({
+        error: "missing X-Recap-Install-Id header",
+        statusHint: 400,
+      });
+      return res.status(400).json(e.body);
+    }
+    const prompt = req.body?.prompt;
+    if (!prompt || typeof prompt !== "string") {
+      const e = await errorEnvelope({
+        error: "missing or non-string body.prompt",
+        installId,
+        statusHint: 400,
+      });
+      return res.status(400).json(e.body);
+    }
+
+    const license = await resolveLicense(auth);
+    const tier = license.tier;
+
+    const row = await getOrCreateRow(installId);
+    row.tier_snapshot = tier;
+
+    let reusedJob = false;
+    let chosenBackend = null;
+    const existingJob = lookupJob(installId, jobId);
+    if (existingJob) {
+      reusedJob = true;
+      chosenBackend = existingJob.backend;
+    } else {
+      const cfg = await getConfigSnapshot();
+      const hasHardware = !!cfg.relay_gemma_base_url;
+      const quota = await getTierQuotas();
+      const plan = planBackend(row, quota, { hasHardware });
+      if (!plan.allowed) {
+        const e = await errorEnvelope({
+          error: plan.reason,
+          installId,
+          tier,
+          statusHint: 402,
+        });
+        return res.status(402).json(e.body);
+      }
+      chosenBackend = plan.backend;
+    }
+
+    const cfg = await getConfigSnapshot();
+    let result;
+    try {
+      if (chosenBackend === "gemini") {
+        const backend = createGeminiBackend({ apiKey: cfg.relay_gemini_api_key });
+        result = await backend.analyzeText({ prompt });
+      } else {
+        const backend = createHardwareBackend({
+          parakeetBaseURL: cfg.relay_parakeet_base_url,
+          gemmaBaseURL: cfg.relay_gemma_base_url,
+        });
+        result = await backend.analyzeText({ prompt });
+      }
+    } catch (err) {
+      if (reusedJob) refundJob(installId, jobId);
+      console.error(`[relay/analyze] backend error: ${err?.message}`);
+      const e = await errorEnvelope({
+        error: err?.message || "backend_error",
+        installId,
+        tier,
+        statusHint: err?.status || 502,
+      });
+      return res.status(e.statusHint).json(e.body);
+    }
+
+    let creditCharged = 0;
+    if (!reusedJob) {
+      await commitCredit(installId, { backend: chosenBackend, tier });
+      markJobCharged(installId, jobId, { backend: chosenBackend, tier });
+      creditCharged = 1;
+    }
+
+    const body = await envelope({ result, installId, tier, creditCharged });
+    res.json(body);
+  });
+
+  return router;
+}