@nxuss/lemma 0.8.1 → 0.9.0

package/dist/cjs/proxy/SseRelay.js

@@ -1,64 +1,34 @@
 "use strict";
-/**
- * Lemma SseRelay
- *
- * Handles Server-Sent Events (SSE) streaming in two modes:
- *
- * MISS mode: Pipes upstream SSE response from OpenAI/Anthropic to the client
- *            in real-time (transparent relay).
- *
- * HIT mode:  Simulates an SSE stream from a cached response by chunking the
- *            text into ~5-word pieces and emitting them with a small delay.
- *            Visually identical to a real stream for IDEs like Cursor.
- *
- * Both modes:
- *   - Set correct SSE headers (text/event-stream, no-cache)
- *   - Inject X-Lemma-Cache, X-Lemma-Similarity, X-Lemma-Tier headers
- *   - Append the `data: [DONE]` sentinel
- *   - Support both OpenAI and Anthropic SSE envelope formats
- */
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.SseRelay = void 0;
-const http_1 = __importDefault(require("http"));
-const https_1 = __importDefault(require("https"));
-// ─── Constants ────────────────────────────────────────────────────────────────
-const CHUNK_WORDS = 5; // words per simulated chunk
-const CHUNK_DELAY_MS = 15; // ms between simulated chunks
-// ─── SseRelay ─────────────────────────────────────────────────────────────────
+const CHUNK_SIZE = 30;
+const CHUNK_DELAY_MS = 15;
 class SseRelay {
-    /**
-     * Simulate an SSE stream from a cached response.
-     */
     static async simulateHit(res, cachedResponse, opts) {
         SseRelay.setSseHeaders(res, opts);
         const text = SseRelay.extractTextFromResponse(cachedResponse, opts.provider);
-        const model = cachedResponse?.model || (opts.provider === 'openai' ? 'gpt-4' : 'claude-3-5-sonnet-20241022');
-        const id = `lemma-hit-${Date.now()}`;
-        const words = text.split(' ');
-        // Emit chunks
-        for (let i = 0; i < words.length; i += CHUNK_WORDS) {
-            const chunk = words.slice(i, i + CHUNK_WORDS).join(' ');
-            const sseData = SseRelay.buildChunkEvent(chunk, id, model, opts.provider, false);
-            res.write(`data: ${JSON.stringify(sseData)}\n\n`);
+        const model = cachedResponse?.model || SseRelay.defaultModel(opts.provider);
+        const id = `lemma-${Date.now()}`;
+        for (let i = 0; i < text.length; i += CHUNK_SIZE) {
+            const chunk = text.slice(i, i + CHUNK_SIZE);
+            const evt = SseRelay.buildChunkEvent(chunk, id, model, opts.provider, false);
+            res.write(SseRelay.formatSSE(evt, opts.provider));
             await sleep(CHUNK_DELAY_MS);
         }
-        // Emit final chunk (finish_reason: stop)
-        const stopEvent = SseRelay.buildChunkEvent('', id, model, opts.provider, true);
-        res.write(`data: ${JSON.stringify(stopEvent)}\n\n`);
-        res.write('data: [DONE]\n\n');
+        const stopEvt = SseRelay.buildChunkEvent('', id, model, opts.provider, true);
+        res.write(SseRelay.formatSSE(stopEvt, opts.provider));
+        if (opts.provider !== 'gemini') {
+            res.write('data: [DONE]\n\n');
+        }
         res.end();
     }
-    /**
-     * Relay a live upstream SSE stream transparently.
-     */
     static async relayMiss(req, res, upstreamBody, provider, opts) {
         SseRelay.setSseHeaders(res, opts);
         const apiKey = provider === 'openai'
             ? process.env.OPENAI_API_KEY
-            : process.env.ANTHROPIC_API_KEY;
+            : provider === 'anthropic'
+                ? process.env.ANTHROPIC_API_KEY
+                : process.env.GEMINI_API_KEY;
         if (!apiKey) {
             res.write(`data: ${JSON.stringify({ error: `${provider.toUpperCase()}_API_KEY not set` })}\n\n`);
             res.write('data: [DONE]\n\n');
@@ -68,7 +38,7 @@ class SseRelay {
         const { hostname, path, headers } = SseRelay.buildUpstreamRequest(provider, apiKey);
         const body = JSON.stringify({ ...upstreamBody, stream: true });
         await new Promise((resolve, reject) => {
-            const lib = hostname.startsWith('localhost') ? http_1.default : https_1.default;
+            const lib = hostname.startsWith('localhost') ? require('http') : require('https');
             const reqOpts = {
                 hostname,
                 port: hostname.startsWith('localhost') ? 80 : 443,
@@ -101,19 +71,25 @@ class SseRelay {
             upstreamReq.end();
         });
     }
-    // ── Private helpers ─────────────────────────────────────────────────────────
     static setSseHeaders(res, opts) {
         res.setHeader('Content-Type', 'text/event-stream');
         res.setHeader('Cache-Control', 'no-cache');
         res.setHeader('Connection', 'keep-alive');
         res.setHeader('X-Accel-Buffering', 'no');
-        res.setHeader('X-Lemma-Cache', 'HIT');
+        res.setHeader('X-Lemma-Cache', opts.cacheHeader || 'HIT');
         res.setHeader('X-Lemma-Similarity', (opts.similarity ?? 1.0).toFixed(3));
         res.setHeader('X-Lemma-Tier', opts.tier);
         if (opts.latencyMs !== undefined) {
             res.setHeader('X-Lemma-Latency', `${opts.latencyMs}ms`);
         }
     }
+    static defaultModel(provider) {
+        if (provider === 'openai')
+            return 'gpt-4';
+        if (provider === 'anthropic')
+            return 'claude-3-5-sonnet-20241022';
+        return 'gemini-1.5-pro';
+    }
     static buildUpstreamRequest(provider, apiKey) {
         if (provider === 'openai') {
             return {
@@ -122,10 +98,17 @@ class SseRelay {
                 headers: { Authorization: `Bearer ${apiKey}` },
             };
         }
+        if (provider === 'anthropic') {
+            return {
+                hostname: 'api.anthropic.com',
+                path: '/v1/messages',
+                headers: { 'x-api-key': apiKey, 'anthropic-version': '2023-06-01' },
+            };
+        }
         return {
-            hostname: 'api.anthropic.com',
-            path: '/v1/messages',
-            headers: { 'x-api-key': apiKey, 'anthropic-version': '2023-06-01' },
+            hostname: 'generativelanguage.googleapis.com',
+            path: `/v1beta/models/${apiKey}`,
+            headers: {},
         };
     }
     static buildChunkEvent(text, id, model, provider, isStop) {
@@ -142,27 +125,41 @@ class SseRelay {
                     }],
             };
         }
-        // Anthropic format
-        if (isStop) {
-            return { type: 'message_stop' };
+        if (provider === 'anthropic') {
+            if (isStop)
+                return { type: 'message_stop' };
+            return {
+                type: 'content_block_delta',
+                index: 0,
+                delta: { type: 'text_delta', text },
+            };
         }
         return {
-            type: 'content_block_delta',
-            index: 0,
-            delta: { type: 'text_delta', text },
+            candidates: [{
+                    content: { parts: [{ text }] },
+                    finishReason: isStop ? 'STOP' : undefined,
+                }],
         };
     }
+    static formatSSE(data, provider) {
+        if (provider === 'gemini') {
+            return `${JSON.stringify(data)}\n`;
+        }
+        return `data: ${JSON.stringify(data)}\n\n`;
+    }
     static extractTextFromResponse(response, provider) {
         try {
             if (provider === 'openai') {
                 return response?.choices?.[0]?.message?.content || '';
             }
-            // Anthropic
-            const content = response?.content;
-            if (Array.isArray(content)) {
-                return content.map((c) => c.text || '').join('');
+            if (provider === 'anthropic') {
+                const content = response?.content;
+                if (Array.isArray(content)) {
+                    return content.map((c) => c.text || '').join('');
+                }
+                return String(content || '');
             }
-            return String(content || '');
+            return response?.candidates?.[0]?.content?.parts?.[0]?.text || '';
         }
         catch {
             return '';

package/dist/cjs/proxy/SseRelay.js.map

@@ -1 +1 @@
-{"version":3,"file":"SseRelay.js","sourceRoot":"","sources":["../../../src/proxy/SseRelay.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;;;GAiBG;;;;;;AAEH,gDAAwB;AACxB,kDAA0B;AAG1B,iFAAiF;AAEjF,MAAM,WAAW,GAAG,CAAC,CAAC,CAAQ,4BAA4B;AAC1D,MAAM,cAAc,GAAG,EAAE,CAAC,CAAI,8BAA8B;AAa5D,iFAAiF;AAEjF,MAAa,QAAQ;IACnB;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,WAAW,CACtB,GAAa,EACb,cAAmB,EACnB,IAAqB;QAErB,QAAQ,CAAC,aAAa,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAElC,MAAM,IAAI,GAAG,QAAQ,CAAC,uBAAuB,CAAC,cAAc,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7E,MAAM,KAAK,GAAG,cAAc,EAAE,KAAK,IAAI,CAAC,IAAI,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,4BAA4B,CAAC,CAAC;QAC7G,MAAM,EAAE,GAAG,aAAa,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;QACrC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAE9B,cAAc;QACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,WAAW,EAAE,CAAC;YACnD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACxD,MAAM,OAAO,GAAG,QAAQ,CAAC,eAAe,CAAC,KAAK,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YACjF,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAClD,MAAM,KAAK,CAAC,cAAc,CAAC,CAAC;QAC9B,CAAC;QAED,yCAAyC;QACzC,MAAM,SAAS,GAAG,QAAQ,CAAC,eAAe,CAAC,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC/E,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACpD,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;QAC9B,GAAG,CAAC,GAAG,EAAE,CAAC;IACZ,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,SAAS,CACpB,GAAY,EACZ,GAAa,EACb,YAAiB,EACjB,QAAkB,EAClB,IAAqB;QAErB,QAAQ,CAAC,aAAa,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAElC,MAAM,MAAM,GAAG,QAAQ,KAAK,QAAQ;YAClC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc;YAC5B,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;QAElC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,GAAG,QAAQ,CAAC,WAAW,EAAE,kBAAkB,EAAE,CAAC,MAAM,CAAC,CAAC;YACjG,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;YAC9B,GAAG,CAAC,GAAG,EAAE,CAAC;YACV,OAAO;QACT,CAAC;QAED,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC,oBAAoB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAEpF,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QAE/D,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC1C,MAAM,GAAG,GAAG,QAAQ,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,cAAI,CAAC,CAAC,CAAC,eAAK,CAAC;YAC5D,MAAM,OAAO,GAAG;gBACd,QAAQ;gBACR,IAAI,EAAE,QAAQ,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG;gBACjD,IAAI;gBACJ,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,GAAG,OAAO;oBACV,cAAc,EAAE,kBAAkB;oBAClC,gBAAgB,EAAE,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC;iBAC1C;aACF,CAAC;YAEF,MAAM,WAAW,GAAI,GAAoB,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,WAAW,EAAE,EAAE;gBACzE,WAAW,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;oBACvC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBACnB,CAAC,CAAC,CAAC;gBACH,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;oBACzB,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;oBAC9B,GAAG,CAAC,GAAG,EAAE,CAAC;oBACV,OAAO,EAAE,CAAC;gBACZ,CAAC,CAAC,CAAC;gBACH,WAAW,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAClC,CAAC,CAAC,CAAC;YAEH,WAAW,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBAC9B,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;gBACjE,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;gBAC9B,GAAG,CAAC,GAAG,EAAE,CAAC;gBACV,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YAEH,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACxB,WAAW,CAAC,GAAG,EAAE,CAAC;QACpB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,+EAA+E;IAEvE,MAAM,CAAC,aAAa,CAAC,GAAa,EAAE,IAAqB;QAC/D,GAAG,CAAC,SAAS,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;QACnD,GAAG,CAAC,SAAS,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;QAC3C,GAAG,CAAC,SAAS,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;QAC1C,GAAG,CAAC,SAAS,CAAC,mBAAmB,EAAE,IAAI,CAAC,CAAC;QACzC,GAAG,CAAC,SAAS,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;QACtC,GAAG,CAAC,SAAS,CAAC,oBAAoB,EAAE,CAAC,IAAI,CAAC,UAAU,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;QACzE,GAAG,CAAC,SAAS,CAAC,cAAc,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACjC,GAAG,CAAC,SAAS,CAAC,iBAAiB,EAAE,GAAG,IAAI,CAAC,SAAS,IAAI,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,oBAAoB,CAAC,QAAkB,EAAE,MAAc;QAKpE,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC1B,OAAO;gBACL,QAAQ,EAAE,gBAAgB;gBAC1B,IAAI,EAAE,sBAAsB;gBAC5B,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE;aAC/C,CAAC;QACJ,CAAC;QACD,OAAO;YACL,QAAQ,EAAE,mBAAmB;YAC7B,IAAI,EAAE,cAAc;YACpB,OAAO,EAAE,EAAE,WAAW,EAAE,MAAM,EAAE,mBAAmB,EAAE,YAAY,EAAE;SACpE,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,eAAe,CAC5B,IAAY,EACZ,EAAU,EACV,KAAa,EACb,QAAkB,EAClB,MAAe;QAEf,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC1B,OAAO;gBACL,EAAE;gBACF,MAAM,EAAE,uBAAuB;gBAC/B,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;gBACtC,KAAK;gBACL,OAAO,EAAE,CAAC;wBACR,KAAK,EAAE,CAAC;wBACR,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE;wBACtC,aAAa,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI;qBACtC,CAAC;aACH,CAAC;QACJ,CAAC;QACD,mBAAmB;QACnB,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;QAClC,CAAC;QACD,OAAO;YACL,IAAI,EAAE,qBAAqB;YAC3B,KAAK,EAAE,CAAC;YACR,KAAK,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;SACpC,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAAC,QAAa,EAAE,QAAkB;QACtE,IAAI,CAAC;YACH,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBAC1B,OAAO,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;YACxD,CAAC;YACD,YAAY;YACZ,MAAM,OAAO,GAAG,QAAQ,EAAE,OAAO,CAAC;YAClC,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC3B,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACxD,CAAC;YACD,OAAO,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;QAC/B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;CACF;AAhLD,4BAgLC;AAED,SAAS,KAAK,CAAC,EAAU;IACvB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC"}
+{"version":3,"file":"SseRelay.js","sourceRoot":"","sources":["../../../src/proxy/SseRelay.ts"],"names":[],"mappings":";;;AAEA,MAAM,UAAU,GAAG,EAAE,CAAC;AACtB,MAAM,cAAc,GAAG,EAAE,CAAC;AAY1B,MAAa,QAAQ;IACnB,MAAM,CAAC,KAAK,CAAC,WAAW,CACtB,GAAa,EACb,cAAmB,EACnB,IAAqB;QAErB,QAAQ,CAAC,aAAa,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAElC,MAAM,IAAI,GAAG,QAAQ,CAAC,uBAAuB,CAAC,cAAc,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7E,MAAM,KAAK,GAAG,cAAc,EAAE,KAAK,IAAI,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5E,MAAM,EAAE,GAAG,SAAS,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;QAEjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC;YACjD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,CAAC;YAC5C,MAAM,GAAG,GAAG,QAAQ,CAAC,eAAe,CAAC,KAAK,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YAC7E,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;YAClD,MAAM,KAAK,CAAC,cAAc,CAAC,CAAC;QAC9B,CAAC;QAED,MAAM,OAAO,GAAG,QAAQ,CAAC,eAAe,CAAC,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC7E,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;QACtD,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC/B,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;QAChC,CAAC;QACD,GAAG,CAAC,GAAG,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,SAAS,CACpB,GAAY,EACZ,GAAa,EACb,YAAiB,EACjB,QAAkB,EAClB,IAAqB;QAErB,QAAQ,CAAC,aAAa,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAElC,MAAM,MAAM,GAAG,QAAQ,KAAK,QAAQ;YAClC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc;YAC5B,CAAC,CAAC,QAAQ,KAAK,WAAW;gBACxB,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB;gBAC/B,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAEjC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,GAAG,QAAQ,CAAC,WAAW,EAAE,kBAAkB,EAAE,CAAC,MAAM,CAAC,CAAC;YACjG,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;YAC9B,GAAG,CAAC,GAAG,EAAE,CAAC;YACV,OAAO;QACT,CAAC;QAED,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC,oBAAoB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACpF,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QAE/D,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC1C,MAAM,GAAG,GAAG,QAAQ,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAClF,MAAM,OAAO,GAAG;gBACd,QAAQ;gBACR,IAAI,EAAE,QAAQ,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG;gBACjD,IAAI;gBACJ,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,GAAG,OAAO;oBACV,cAAc,EAAE,kBAAkB;oBAClC,gBAAgB,EAAE,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC;iBAC1C;aACF,CAAC;YAEF,MAAM,WAAW,GAAG,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,WAAgB,EAAE,EAAE;gBAC5D,WAAW,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;oBACvC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBACnB,CAAC,CAAC,CAAC;gBACH,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;oBACzB,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;oBAC9B,GAAG,CAAC,GAAG,EAAE,CAAC;oBACV,OAAO,EAAE,CAAC;gBACZ,CAAC,CAAC,CAAC;gBACH,WAAW,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAClC,CAAC,CAAC,CAAC;YAEH,WAAW,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAU,EAAE,EAAE;gBACrC,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;gBACjE,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;gBAC9B,GAAG,CAAC,GAAG,EAAE,CAAC;gBACV,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YAEH,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACxB,WAAW,CAAC,GAAG,EAAE,CAAC;QACpB,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,MAAM,CAAC,aAAa,CAAC,GAAa,EAAE,IAAqB;QAC/D,GAAG,CAAC,SAAS,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;QACnD,GAAG,CAAC,SAAS,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;QAC3C,GAAG,CAAC,SAAS,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;QAC1C,GAAG,CAAC,SAAS,CAAC,mBAAmB,EAAE,IAAI,CAAC,CAAC;QACzC,GAAG,CAAC,SAAS,CAAC,eAAe,EAAE,IAAI,CAAC,WAAW,IAAI,KAAK,CAAC,CAAC;QAC1D,GAAG,CAAC,SAAS,CAAC,oBAAoB,EAAE,CAAC,IAAI,CAAC,UAAU,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;QACzE,GAAG,CAAC,SAAS,CAAC,cAAc,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACjC,GAAG,CAAC,SAAS,CAAC,iBAAiB,EAAE,GAAG,IAAI,CAAC,SAAS,IAAI,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,YAAY,CAAC,QAAkB;QAC5C,IAAI,QAAQ,KAAK,QAAQ;YAAE,OAAO,OAAO,CAAC;QAC1C,IAAI,QAAQ,KAAK,WAAW;YAAE,OAAO,4BAA4B,CAAC;QAClE,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAEO,MAAM,CAAC,oBAAoB,CAAC,QAAkB,EAAE,MAAc;QAKpE,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC1B,OAAO;gBACL,QAAQ,EAAE,gBAAgB;gBAC1B,IAAI,EAAE,sBAAsB;gBAC5B,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE;aAC/C,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;YAC7B,OAAO;gBACL,QAAQ,EAAE,mBAAmB;gBAC7B,IAAI,EAAE,cAAc;gBACpB,OAAO,EAAE,EAAE,WAAW,EAAE,MAAM,EAAE,mBAAmB,EAAE,YAAY,EAAE;aACpE,CAAC;QACJ,CAAC;QACD,OAAO;YACL,QAAQ,EAAE,mCAAmC;YAC7C,IAAI,EAAE,kBAAkB,MAAM,EAAE;YAChC,OAAO,EAAE,EAAE;SACZ,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,eAAe,CAC5B,IAAY,EACZ,EAAU,EACV,KAAa,EACb,QAAkB,EAClB,MAAe;QAEf,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC1B,OAAO;gBACL,EAAE;gBACF,MAAM,EAAE,uBAAuB;gBAC/B,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;gBACtC,KAAK;gBACL,OAAO,EAAE,CAAC;wBACR,KAAK,EAAE,CAAC;wBACR,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE;wBACtC,aAAa,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI;qBACtC,CAAC;aACH,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;YAC7B,IAAI,MAAM;gBAAE,OAAO,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;YAC5C,OAAO;gBACL,IAAI,EAAE,qBAAqB;gBAC3B,KAAK,EAAE,CAAC;gBACR,KAAK,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;aACpC,CAAC;QACJ,CAAC;QACD,OAAO;YACL,UAAU,EAAE,CAAC;oBACX,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE;oBAC9B,YAAY,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;iBAC1C,CAAC;SACH,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,SAAS,CAAC,IAAY,EAAE,QAAkB;QACvD,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC1B,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;QACrC,CAAC;QACD,OAAO,SAAS,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC;IAC7C,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAAC,QAAa,EAAE,QAAkB;QACtE,IAAI,CAAC;YACH,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBAC1B,OAAO,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;YACxD,CAAC;YACD,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;gBAC7B,MAAM,OAAO,GAAG,QAAQ,EAAE,OAAO,CAAC;gBAClC,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC3B,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACxD,CAAC;gBACD,OAAO,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;YAC/B,CAAC;YACD,OAAO,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;QACpE,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;CACF;AAnMD,4BAmMC;AAED,SAAS,KAAK,CAAC,EAAU;IACvB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC"}

package/dist/esm/cli/lemma-proxy.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"lemma-proxy.d.ts","sourceRoot":"","sources":["../../../src/cli/lemma-proxy.ts"],"names":[],"mappings":";AA0bA,wBAAgB,+BAA+B,CAAC,OAAO,CAAC,EAAE;IAAE,SAAS,CAAC,EAAE,OAAO,CAAA;CAAE,GAAG,IAAI,CA4DvF;AA2jED,wBAAsB,eAAe,CAAC,KAAK,EAAE,OAAO,GAAG,OAAO,CAAC;IAAE,OAAO,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CA6KhL"}
+{"version":3,"file":"lemma-proxy.d.ts","sourceRoot":"","sources":["../../../src/cli/lemma-proxy.ts"],"names":[],"mappings":";AAsbA,wBAAgB,+BAA+B,CAAC,OAAO,CAAC,EAAE;IAAE,SAAS,CAAC,EAAE,OAAO,CAAA;CAAE,GAAG,IAAI,CA4DvF;AAmgED,wBAAsB,eAAe,CAAC,KAAK,EAAE,OAAO,GAAG,OAAO,CAAC;IAAE,OAAO,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CA6KhL"}

package/dist/esm/cli/lemma-proxy.js

@@ -38,25 +38,22 @@ import SemanticScrubber from '../security/SemanticScrubber';
 import CloudSyncClient from '../cloud/CloudSyncClient';
 import { pruneHistoryMessages, squeezePrompt } from '../utils/ContextSqueezer';
 import { AgentMultiplexer } from '../proxy/AgentMultiplexer';
+import { SseRelay } from '../proxy/SseRelay';
+import { KeyManager } from '../proxy/KeyManager';
+import { Gatekeeper } from '../proxy/Gatekeeper';
 import { ProjectRegistry } from '../proxy/ProjectRegistry';
 import { shouldInject, buildMinimalContext, buildDeepContext, sniffProjectFromMessages } from '../proxy/ContextInjector';
 import { savingsLedger } from '../utils/SavingsLedger';
+import { CACHE_DIR, PID_FILE, PORT_FILE, STATS_FILE, USAGE_FILE, LICENSE_FILE, CLIPBOARD_PID_FILE, FREE_TIER_LIMIT } from '../proxy/ProjectStore';
 const chroma = new ChromaClient({ host: 'localhost', port: 8000 });
 const dummyEmbeddingFunction = { generate: (texts) => Promise.resolve([]) };
 const complexityRouter = new ComplexityRouter();
 const semanticScrubber = new SemanticScrubber();
 const cloudSync = new CloudSyncClient();
 const agentMultiplexer = new AgentMultiplexer();
+const keyManager = new KeyManager();
+const gatekeeper = new Gatekeeper(keyManager);
 // ── Paths ──────────────────────────────────────────────────────────────────────
-const HOME = os.homedir();
-const CACHE_DIR = path.join(HOME, '.lemma-cache');
-const PID_FILE = path.join(CACHE_DIR, 'proxy.pid');
-const PORT_FILE = path.join(CACHE_DIR, 'proxy.port');
-const STATS_FILE = path.join(CACHE_DIR, 'stats.json');
-const USAGE_FILE = path.join(CACHE_DIR, 'usage.json');
-const LICENSE_FILE = path.join(CACHE_DIR, 'license.json');
-const CLIPBOARD_PID_FILE = path.join(CACHE_DIR, 'clipboard.pid');
-const FREE_LIMIT = 300;
 const VALIDATE_URL = 'https://lemma.nxus.studio/api/validate-license';
 async function checkDependency(url) {
     try {
@@ -504,21 +501,8 @@ export function startBackgroundClipboardWatcher(cliOpts) {
     }
 }
 // ── License helpers ────────────────────────────────────────────────────────────
-let isProCached = null;
-let lastLicenseCheck = 0;
-const LICENSE_CACHE_TTL = 10000; // 10 seconds cache
-async function loadLicense() {
-    return readJson(LICENSE_FILE, { isPro: false });
-}
-async function isPro() {
-    const now = Date.now();
-    if (isProCached !== null && (now - lastLicenseCheck) < LICENSE_CACHE_TTL) {
-        return isProCached;
-    }
-    const l = await loadLicense();
-    isProCached = !!(l && l.isPro);
-    lastLicenseCheck = now;
-    return isProCached;
+function isPro() {
+    return gatekeeper.getMode() === 'pro';
 }
 async function validateKeyRemote(key) {
     return new Promise((resolve, reject) => {
@@ -642,7 +626,7 @@ async function getEmbedding(text) {
     }
 }
 async function semanticGet(provider, prompt, projectName = 'global') {
-    if (!await isPro())
+    if (!isPro())
         return null;
     const emb = await getEmbedding(prompt);
     if (!emb)
@@ -735,7 +719,7 @@ async function semanticGet(provider, prompt, projectName = 'global') {
     return null;
 }
 async function semanticSet(provider, prompt, data, projectName = 'global') {
-    if (!await isPro())
+    if (!isPro())
         return;
     const emb = await getEmbedding(prompt);
     if (!emb)
@@ -838,53 +822,6 @@ function setSseHeaders(res, fromCache, similarity, tier) {
     res.setHeader('X-Lemma-Similarity', (similarity || 1.0).toFixed(3));
     res.setHeader('X-Lemma-Tier', tier);
 }
-function buildOaiChunk(id, model, text, stop) {
-    return { id, object: 'chat.completion.chunk', created: Math.floor(Date.now() / 1000), model,
-        choices: [{ index: 0, delta: stop ? {} : { content: text }, finish_reason: stop ? 'stop' : null }] };
-}
-function buildAntChunk(text, stop) {
-    if (stop)
-        return { type: 'message_stop' };
-    return { type: 'content_block_delta', index: 0, delta: { type: 'text_delta', text } };
-}
-function buildGemChunk(text, stop) {
-    return { candidates: [{ content: { parts: [{ text }] }, finishReason: stop ? 'STOP' : undefined }] };
-}
-async function simulateHitStream(res, cachedData, provider, model, tier, similarity, cacheHeader = 'HIT') {
-    setSseHeaders(res, cacheHeader, similarity, tier);
-    const id = `lemma-${Date.now()}`;
-    let text = '';
-    try {
-        if (provider === 'openai')
-            text = cachedData?.choices?.[0]?.message?.content || '';
-        else if (provider === 'anthropic') {
-            const c = cachedData?.content;
-            text = Array.isArray(c) ? c.map((x) => x.text || '').join('') : String(c || '');
-        }
-        else if (provider === 'gemini') {
-            text = cachedData?.candidates?.[0]?.content?.parts?.[0]?.text || '';
-        }
-    }
-    catch { }
-    const chunkSize = 30;
-    for (let i = 0; i < text.length; i += chunkSize) {
-        const chunk = text.slice(i, i + chunkSize);
-        let evt;
-        if (provider === 'openai')
-            evt = buildOaiChunk(id, model, chunk, false);
-        else if (provider === 'anthropic')
-            evt = buildAntChunk(chunk, false);
-        else if (provider === 'gemini')
-            evt = buildGemChunk(chunk, false);
-        res.write(provider === 'gemini' ? `${JSON.stringify(evt)}\n` : `data: ${JSON.stringify(evt)}\n\n`);
-        await new Promise(r => setTimeout(r, 15));
-    }
-    const stopEvt = provider === 'openai' ? buildOaiChunk(id, model, '', true) : (provider === 'anthropic' ? buildAntChunk('', true) : buildGemChunk('', true));
-    res.write(provider === 'gemini' ? `${JSON.stringify(stopEvt)}\n` : `data: ${JSON.stringify(stopEvt)}\n\n`);
-    if (provider !== 'gemini')
-        res.write('data: [DONE]\n\n');
-    res.end();
-}
 async function relayMissStream(res, body, provider, tier) {
     setSseHeaders(res, false, 1.0, tier);
     const apiKey = provider === 'openai' ? process.env.OPENAI_API_KEY : process.env.ANTHROPIC_API_KEY;
@@ -994,8 +931,8 @@ class LemmaServer {
     }
     setupRoutes() {
         this.app.get('/health', async (req, res) => {
-            const pro = await isPro();
-            res.json({ status: 'ok', project: this.projectName, mode: pro ? 'pro' : 'standard', tier: pro ? 'pro' : 'free', port: this.port });
+            const gs = gatekeeper.getStatus();
+            res.json({ status: 'ok', project: this.projectName, mode: gs.mode, tier: gs.isPro ? 'pro' : 'free', port: this.port });
         });
         // MCP Status endpoint
         this.app.get('/api/mcp-status', (req, res) => {
@@ -1094,7 +1031,7 @@ class LemmaServer {
                     activeAgents,
                     totalAgents,
                     uptime: process.uptime(),
-                    tier: (await isPro()) ? 'pro' : 'free'
+                    tier: (isPro()) ? 'pro' : 'free'
                 });
             });
             this.app.get('/api/agents', async (req, res) => {
@@ -1681,14 +1618,14 @@ class LemmaServer {
     }
     async handleCompletion(req, res, provider) {
         const t0 = Date.now();
-        const pro = await isPro();
+        const pro = isPro();
         const tier = pro ? 'pro' : 'standard';
         const isStream = !!req.body.stream;
         let historyTokensSaved = 0;
         if (!pro) {
             const usage = await loadUsage();
-            if (usage.count >= FREE_LIMIT) {
-                return res.status(429).json({ error: { message: `Free tier limit (${FREE_LIMIT}/mo). Upgrade: https://lemma.nxus.studio/upgrade` } });
+            if (usage.count >= FREE_TIER_LIMIT) {
+                return res.status(429).json({ error: { message: `Free tier limit (${FREE_TIER_LIMIT}/mo). Upgrade: https://lemma.nxus.studio/upgrade` } });
             }
         }
         const rawPrompt = extractPrompt(req.body, provider);
@@ -1707,7 +1644,7 @@ class LemmaServer {
             await recordStat(this.stats, this.projectName, true, Date.now() - t0, provider, cacheTokensSaved);
             const unmaskedData = semanticScrubber.unmask(hit.data, tokenMap);
             if (isStream)
-                return simulateHitStream(res, unmaskedData, provider, originalModel, tier, 1.0);
+                return SseRelay.simulateHit(res, unmaskedData, { provider: provider, tier, similarity: 1.0 });
             res.setHeader('X-Lemma-Cache', 'HIT');
             res.setHeader('X-Lemma-Similarity', '1.000');
             res.setHeader('X-Lemma-Tier', tier);
@@ -1722,7 +1659,7 @@ class LemmaServer {
                     const unmaskedData = semanticScrubber.unmask(semHit.data, tokenMap);
                     const cacheHeader = semHit.hiveMind ? 'HIT-HIVE-MIND' : 'HIT';
                     if (isStream)
-                        return simulateHitStream(res, unmaskedData, provider, originalModel, tier, semHit.similarity, cacheHeader);
+                        return SseRelay.simulateHit(res, unmaskedData, { provider: provider, tier, similarity: semHit.similarity, cacheHeader });
                     res.setHeader('X-Lemma-Cache', cacheHeader);
                     res.setHeader('X-Lemma-Similarity', semHit.similarity.toFixed(3));
                     res.setHeader('X-Lemma-Tier', tier);
@@ -1772,7 +1709,7 @@ Adjusted Answer:`;
                                 await recordStat(this.stats, this.projectName, true, Date.now() - t0, provider, 2000);
                                 const unmaskedData = semanticScrubber.unmask(synthesizedData, tokenMap);
                                 if (isStream) {
-                                    return simulateHitStream(res, unmaskedData, provider, originalModel, tier, semHit.similarity);
+                                    return SseRelay.simulateHit(res, unmaskedData, { provider: provider, tier, similarity: semHit.similarity });
                                 }
                                 res.setHeader('X-Lemma-Cache', 'HIT-CARS');
                                 res.setHeader('X-Lemma-Similarity', semHit.similarity.toFixed(3));
@@ -1792,7 +1729,7 @@ Adjusted Answer:`;
                 await recordStat(this.stats, this.projectName, true, Date.now() - t0, provider, 2000);
                 const unmaskedData = semanticScrubber.unmask(cloudHit.data, tokenMap);
                 if (isStream)
-                    return simulateHitStream(res, unmaskedData, provider, originalModel, tier, cloudHit.similarity || 0.95);
+                    return SseRelay.simulateHit(res, unmaskedData, { provider: provider, tier, similarity: cloudHit.similarity || 0.95 });
                 res.setHeader('X-Lemma-Cache', 'HIT');
                 res.setHeader('X-Lemma-Similarity', (cloudHit.similarity || 0.95).toFixed(3));
                 res.setHeader('X-Lemma-Tier', tier);
@@ -2011,7 +1948,7 @@ Adjusted Answer:`;
                     console.warn(`   Set AUTH_ENABLED=true and AUTH_REQUIRE_API_KEY=true before exposing to any network.\x1b[0m\n`);
                 }
             }
-            const pro = await isPro();
+            const pro = isPro();
             console.log(`\x1b[1m\x1b[35m┌──────────────────────────────────────────────────────────┐\x1b[0m`);
             console.log(`\x1b[1m\x1b[35m│   🚀 LEMMA PROXY v${VERSION} — THE UNIVERSAL AI GATEWAY      │\x1b[0m`);
             console.log(`\x1b[1m\x1b[35m└──────────────────────────────────────────────────────────┘\x1b[0m`);
@@ -2105,7 +2042,7 @@ program
     .version(VERSION)
     .addHelpText('after', `
 \x1b[1mPlan Comparison\x1b[0m
-  \x1b[32mFree\x1b[0m    ${FREE_LIMIT} req/mo · Exact cache · Privacy firewall · MCP tools
+  \x1b[32mFree\x1b[0m    ${FREE_TIER_LIMIT} req/mo · Exact cache · Privacy firewall · MCP tools
   \x1b[35mPro\x1b[0m     Unlimited  · Semantic cache · Cloud sync · Autopilot · Heal
          \x1b[36mhttps://lemma.nxus.studio/upgrade\x1b[0m
 
@@ -2181,7 +2118,7 @@ program.command('start')
     const runAutopilot = opts.autopilot || opts.stack;
     if (runAutopilot) {
         try {
-            const pro = await isPro();
+            const pro = isPro();
             if (!pro) {
                 if (opts.autopilot) {
                     console.log(`\n🛸 \x1b[35m[Autopilot]\x1b[0m Active background auto-healing watcher is a Lemma Pro feature.`);
@@ -2325,9 +2262,9 @@ program.command('stats')
     console.log(`Cache Hits     : ${s.hits} (${((s.hits / s.total || 0) * 100).toFixed(1)}%)`);
     console.log(`Tokens Saved   : ${s.totalTokensSaved.toLocaleString()}`);
     console.log(`Avg Latency    : ${(s.totalLatency / (s.total || 1)).toFixed(2)}ms`);
-    console.log(`Tier           : ${(await isPro()) ? 'PRO' : 'FREE'}`);
-    if (!await isPro())
-        console.log(`Free Usage     : ${usage.count}/${FREE_LIMIT} requests`);
+    console.log(`Tier           : ${(isPro()) ? 'PRO' : 'FREE'}`);
+    if (!isPro())
+        console.log(`Free Usage     : ${usage.count}/${FREE_TIER_LIMIT} requests`);
     console.log(`──────────────────────────────────\n`);
 });
 program.command('activate <key>')
@@ -2343,6 +2280,7 @@ program.command('activate <key>')
         const res = await validateKeyRemote(key.trim());
         if (res.valid) {
             await writeJson(LICENSE_FILE, { key, isPro: true, activatedAt: new Date().toISOString(), validatedRemotely: true });
+            gatekeeper.reload();
             console.log('✅ License activated! Enjoy Pro features (Server-verified).');
         }
         else {
@@ -2583,7 +2521,7 @@ program.command('heal')
     .option('--apply', 'Apply the auto-generated code fix automatically', false)
     .action(async (opts) => {
     try {
-        const pro = await isPro();
+        const pro = isPro();
         if (!pro) {
             console.log(`\n🩺 \x1b[35m[Heal]\x1b[0m Guided code diagnostics and auto-healing is a Lemma Pro feature.`);
             console.log(`💡 Upgrade to Pro in your Dashboard or visit: \x1b[4mhttps://lemma.nxus.studio/upgrade\x1b[0m\n`);
@@ -2612,7 +2550,7 @@ program.command('autopilot')
     .option('--dir <directory>', 'Directory to monitor for changes', 'src')
     .action(async (opts) => {
     try {
-        const pro = await isPro();
+        const pro = isPro();
         if (!pro) {
             console.log(`\n🛸 \x1b[35m[Autopilot]\x1b[0m Active background auto-healing watcher is a Lemma Pro feature.`);
             console.log(`💡 Upgrade to Pro in your Dashboard or visit: \x1b[4mhttps://lemma.nxus.studio/upgrade\x1b[0m\n`);