drizzle-cube 0.4.13 → 0.4.14

package/dist/adapters/fastify/index.js

@@ -1,19 +1,19 @@
 import { S as D, v as K, b as x, a as T, n as V, p as z, w as _, d as B, i as j, M as F, c as U, e as E, s as P } from "../mcp-transport-m1X1GtwG.js";
-import { formatErrorResponse as d, formatCubeResponse as A, handleBatchRequest as X, formatMetaResponse as G, formatSqlResponse as I, handleDryRun as M } from "../utils.js";
+import { formatErrorResponse as u, formatCubeResponse as A, handleBatchRequest as X, formatMetaResponse as G, formatSqlResponse as I, handleDryRun as M } from "../utils.js";
 const N = function(i, k, R) {
   const {
     cubes: S,
     drizzle: O,
     schema: J,
-    extractSecurityContext: p,
+    extractSecurityContext: y,
     engineType: L,
     cors: q,
-    basePath: m = "/cubejs-api/v1",
-    bodyLimit: g = 10485760,
+    basePath: p = "/cubejs-api/v1",
+    bodyLimit: m = 10485760,
     // 10MB
     cache: H,
-    mcp: b = { enabled: !0 },
-    agent: f
+    mcp: f = { enabled: !0 },
+    agent: v
   } = k;
   if (!S || S.length === 0)
     return R(new Error("At least one cube must be provided in the cubes array"));
@@ -28,8 +28,8 @@ const N = function(i, k, R) {
   });
   if (S.forEach((t) => {
     c.registerCube(t);
-  }), i.post(`${m}/load`, {
-    bodyLimit: g,
+  }), i.post(`${p}/load`, {
+    bodyLimit: m,
     schema: {
       body: {
         type: "object",
@@ -38,21 +38,21 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const e = t.body, a = e.query || e, n = await p(t), o = c.validateQuery(a);
-      if (!o.isValid)
-        return r.status(400).send(d(
-          `Query validation failed: ${o.errors.join(", ")}`,
+      const e = t.body, a = e.query || e, n = await y(t), s = c.validateQuery(a);
+      if (!s.isValid)
+        return r.status(400).send(u(
+          `Query validation failed: ${s.errors.join(", ")}`,
           400
         ));
-      const s = t.headers["x-cache-control"] === "no-cache", l = await c.executeMultiCubeQuery(a, n, { skipCache: s });
+      const o = t.headers["x-cache-control"] === "no-cache", l = await c.executeMultiCubeQuery(a, n, { skipCache: o });
       return A(a, l, c);
     } catch (e) {
-      return t.log.error(e, "Query execution error"), r.status(500).send(d(
+      return t.log.error(e, "Query execution error"), r.status(500).send(u(
         e instanceof Error ? e.message : "Query execution failed",
         500
       ));
     }
-  }), i.get(`${m}/load`, {
+  }), i.get(`${p}/load`, {
     schema: {
       querystring: {
         type: "object",
@@ -69,27 +69,27 @@ const N = function(i, k, R) {
       try {
         a = JSON.parse(e);
       } catch {
-        return r.status(400).send(d(
+        return r.status(400).send(u(
           "Invalid JSON in query parameter",
           400
         ));
       }
-      const n = await p(t), o = c.validateQuery(a);
-      if (!o.isValid)
-        return r.status(400).send(d(
-          `Query validation failed: ${o.errors.join(", ")}`,
+      const n = await y(t), s = c.validateQuery(a);
+      if (!s.isValid)
+        return r.status(400).send(u(
+          `Query validation failed: ${s.errors.join(", ")}`,
           400
         ));
-      const s = t.headers["x-cache-control"] === "no-cache", l = await c.executeMultiCubeQuery(a, n, { skipCache: s });
+      const o = t.headers["x-cache-control"] === "no-cache", l = await c.executeMultiCubeQuery(a, n, { skipCache: o });
       return A(a, l, c);
     } catch (e) {
-      return t.log.error(e, "Query execution error"), r.status(500).send(d(
+      return t.log.error(e, "Query execution error"), r.status(500).send(u(
         e instanceof Error ? e.message : "Query execution failed",
         500
       ));
     }
-  }), i.post(`${m}/batch`, {
-    bodyLimit: g,
+  }), i.post(`${p}/batch`, {
+    bodyLimit: m,
     schema: {
       body: {
         type: "object",
@@ -106,35 +106,35 @@ const N = function(i, k, R) {
     try {
       const { queries: e } = t.body;
       if (!e || !Array.isArray(e))
-        return r.status(400).send(d(
+        return r.status(400).send(u(
           'Request body must contain a "queries" array',
           400
         ));
       if (e.length === 0)
-        return r.status(400).send(d(
+        return r.status(400).send(u(
           "Queries array cannot be empty",
           400
         ));
-      const a = await p(t), n = t.headers["x-cache-control"] === "no-cache";
+      const a = await y(t), n = t.headers["x-cache-control"] === "no-cache";
       return await X(e, a, c, { skipCache: n });
     } catch (e) {
-      return t.log.error(e, "Batch execution error"), r.status(500).send(d(
+      return t.log.error(e, "Batch execution error"), r.status(500).send(u(
         e instanceof Error ? e.message : "Batch execution failed",
         500
       ));
     }
-  }), i.get(`${m}/meta`, async (t, r) => {
+  }), i.get(`${p}/meta`, async (t, r) => {
     try {
       const e = c.getMetadata();
       return G(e);
     } catch (e) {
-      return t.log.error(e, "Metadata error"), r.status(500).send(d(
+      return t.log.error(e, "Metadata error"), r.status(500).send(u(
         e instanceof Error ? e.message : "Failed to fetch metadata",
         500
       ));
     }
-  }), i.post(`${m}/sql`, {
-    bodyLimit: g,
+  }), i.post(`${p}/sql`, {
+    bodyLimit: m,
     schema: {
       body: {
         type: "object",
@@ -143,27 +143,27 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const e = t.body, a = await p(t), n = c.validateQuery(e);
+      const e = t.body, a = await y(t), n = c.validateQuery(e);
       if (!n.isValid)
-        return r.status(400).send(d(
+        return r.status(400).send(u(
           `Query validation failed: ${n.errors.join(", ")}`,
           400
         ));
-      const o = e.measures?.[0] || e.dimensions?.[0];
-      if (!o)
-        return r.status(400).send(d(
+      const s = e.measures?.[0] || e.dimensions?.[0];
+      if (!s)
+        return r.status(400).send(u(
           "No measures or dimensions specified",
           400
         ));
-      const s = o.split(".")[0], l = await c.generateSQL(s, e, a);
+      const o = s.split(".")[0], l = await c.generateSQL(o, e, a);
       return I(e, l);
     } catch (e) {
-      return t.log.error(e, "SQL generation error"), r.status(500).send(d(
+      return t.log.error(e, "SQL generation error"), r.status(500).send(u(
         e instanceof Error ? e.message : "SQL generation failed",
         500
       ));
     }
-  }), i.get(`${m}/sql`, {
+  }), i.get(`${p}/sql`, {
     schema: {
       querystring: {
         type: "object",
@@ -175,28 +175,28 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const { query: e } = t.query, a = JSON.parse(e), n = await p(t), o = c.validateQuery(a);
-      if (!o.isValid)
-        return r.status(400).send(d(
-          `Query validation failed: ${o.errors.join(", ")}`,
+      const { query: e } = t.query, a = JSON.parse(e), n = await y(t), s = c.validateQuery(a);
+      if (!s.isValid)
+        return r.status(400).send(u(
+          `Query validation failed: ${s.errors.join(", ")}`,
           400
         ));
-      const s = a.measures?.[0] || a.dimensions?.[0];
-      if (!s)
-        return r.status(400).send(d(
+      const o = a.measures?.[0] || a.dimensions?.[0];
+      if (!o)
+        return r.status(400).send(u(
           "No measures or dimensions specified",
           400
         ));
-      const l = s.split(".")[0], y = await c.generateSQL(l, a, n);
-      return I(a, y);
+      const l = o.split(".")[0], g = await c.generateSQL(l, a, n);
+      return I(a, g);
     } catch (e) {
-      return t.log.error(e, "SQL generation error"), r.status(500).send(d(
+      return t.log.error(e, "SQL generation error"), r.status(500).send(u(
         e instanceof Error ? e.message : "SQL generation failed",
         500
       ));
     }
-  }), i.post(`${m}/dry-run`, {
-    bodyLimit: g,
+  }), i.post(`${p}/dry-run`, {
+    bodyLimit: m,
     schema: {
       body: {
         type: "object",
@@ -205,7 +205,7 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const e = t.body, a = e.query || e, n = await p(t);
+      const e = t.body, a = e.query || e, n = await y(t);
       return await M(a, n, c);
     } catch (e) {
       return t.log.error(e, "Dry-run error"), r.status(400).send({
@@ -213,7 +213,7 @@ const N = function(i, k, R) {
         valid: !1
       });
     }
-  }), i.get(`${m}/dry-run`, {
+  }), i.get(`${p}/dry-run`, {
     schema: {
       querystring: {
         type: "object",
@@ -225,7 +225,7 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const { query: e } = t.query, a = JSON.parse(e), n = await p(t);
+      const { query: e } = t.query, a = JSON.parse(e), n = await y(t);
       return await M(a, n, c);
     } catch (e) {
       return t.log.error(e, "Dry-run error"), r.status(400).send({
@@ -233,8 +233,8 @@ const N = function(i, k, R) {
         valid: !1
       });
     }
-  }), i.post(`${m}/explain`, {
-    bodyLimit: g,
+  }), i.post(`${p}/explain`, {
+    bodyLimit: m,
     schema: {
       body: {
         type: "object",
@@ -243,17 +243,17 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const e = t.body, a = e.query || e, n = e.options || {}, o = await p(t), s = c.validateQuery(a);
-      return s.isValid ? await c.explainQuery(a, o, n) : r.status(400).send({
-        error: `Query validation failed: ${s.errors.join(", ")}`
+      const e = t.body, a = e.query || e, n = e.options || {}, s = await y(t), o = c.validateQuery(a);
+      return o.isValid ? await c.explainQuery(a, s, n) : r.status(400).send({
+        error: `Query validation failed: ${o.errors.join(", ")}`
       });
     } catch (e) {
       return t.log.error(e, "Explain error"), r.status(500).send({
         error: e instanceof Error ? e.message : "Explain query failed"
       });
     }
-  }), f && i.post(`${m}/agent/chat`, {
-    bodyLimit: g,
+  }), v && i.post(`${p}/agent/chat`, {
+    bodyLimit: m,
     schema: {
       body: {
         type: "object",
@@ -262,43 +262,44 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const { handleAgentChat: e } = await import("../handler-CQkIwtxp.js"), a = t.body, { message: n, sessionId: o } = a;
+      const { handleAgentChat: e } = await import("../handler-DZnCbydH.js"), a = t.body, { message: n, sessionId: s, history: o } = a;
       if (!n || typeof n != "string")
         return r.status(400).send({ error: "message is required and must be a string" });
-      let s = (f.apiKey || "").trim();
-      if (f.allowClientApiKey) {
-        const y = t.headers["x-agent-api-key"];
-        y && (s = y.trim());
+      let l = (v.apiKey || "").trim();
+      if (v.allowClientApiKey) {
+        const d = t.headers["x-agent-api-key"];
+        d && (l = d.trim());
       }
-      if (!s)
+      if (!l)
         return r.status(401).send({
           error: "No API key configured. Set agent.apiKey in server config or send X-Agent-Api-Key header."
         });
-      const l = await p(t);
+      const g = await y(t);
       r.raw.writeHead(200, {
         "Content-Type": "text/event-stream",
         "Cache-Control": "no-cache",
         Connection: "keep-alive"
       });
       try {
-        const y = e({
+        const d = e({
           message: n,
-          sessionId: o,
+          sessionId: s,
+          history: o,
           semanticLayer: c,
-          securityContext: l,
-          agentConfig: f,
-          apiKey: s
+          securityContext: g,
+          agentConfig: v,
+          apiKey: l
         });
-        for await (const u of y)
-          r.raw.write(`data: ${JSON.stringify(u)}
+        for await (const h of d)
+          r.raw.write(`data: ${JSON.stringify(h)}
 
 `);
-      } catch (y) {
-        const u = {
+      } catch (d) {
+        const h = {
           type: "error",
-          data: { message: y instanceof Error ? y.message : "Stream failed" }
+          data: { message: d instanceof Error ? d.message : "Stream failed" }
         };
-        r.raw.write(`data: ${JSON.stringify(u)}
+        r.raw.write(`data: ${JSON.stringify(h)}
 
 `);
       } finally {
@@ -310,10 +311,10 @@ const N = function(i, k, R) {
           error: e instanceof Error ? e.message : "Agent chat failed"
         });
     }
-  }), b.enabled !== !1) {
-    const t = b.basePath ?? "/mcp";
+  }), f.enabled !== !1) {
+    const t = f.basePath ?? "/mcp";
     i.post(`${t}`, {
-      bodyLimit: g,
+      bodyLimit: m,
       schema: {
         body: {
           type: "object",
@@ -323,40 +324,40 @@ const N = function(i, k, R) {
     }, async (r, e) => {
       const a = K(
         r.headers.origin,
-        b.allowedOrigins ? { allowedOrigins: b.allowedOrigins } : {}
+        f.allowedOrigins ? { allowedOrigins: f.allowedOrigins } : {}
       );
       if (!a.valid)
         return e.status(403).send(x(null, -32600, a.reason));
       const n = r.headers.accept;
       if (!T(n))
         return e.status(400).send(x(null, -32600, "Accept header must include both application/json and text/event-stream"));
-      const o = V(r.headers);
-      if (!o.ok)
+      const s = V(r.headers);
+      if (!s.ok)
         return e.status(426).send({
           error: "Unsupported MCP protocol version",
-          supported: o.supported
+          supported: s.supported
         });
-      const s = z(r.body);
-      if (!s)
+      const o = z(r.body);
+      if (!o)
         return e.status(400).send(x(null, -32600, "Invalid JSON-RPC 2.0 request"));
-      const l = _(n), y = s.method === "initialize";
+      const l = _(n), g = o.method === "initialize";
       try {
-        const u = await B(
-          s.method,
-          s.params,
+        const d = await B(
+          o.method,
+          o.params,
           {
             semanticLayer: c,
-            extractSecurityContext: p,
+            extractSecurityContext: y,
             rawRequest: r,
             rawResponse: e,
-            negotiatedProtocol: o.negotiated
+            negotiatedProtocol: s.negotiated
           }
         );
-        if (j(s))
+        if (j(o))
           return e.status(202).send();
-        const v = y && u && typeof u == "object" && "sessionId" in u ? u.sessionId : void 0;
-        v && e.header(F, v);
-        const C = U(s.id ?? null, u);
+        const h = g && d && typeof d == "object" && "sessionId" in d ? d.sessionId : void 0;
+        h && e.header(F, h);
+        const C = U(o.id ?? null, d);
         if (l) {
           const w = E();
           e.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${w}
@@ -365,11 +366,11 @@ ${P(C, w)}`);
           return;
         }
         return e.send(C);
-      } catch (u) {
-        if (j(s))
-          return r.log.error(u, "MCP notification processing error"), e.status(202).send();
-        r.log.error(u, "MCP RPC error");
-        const v = u?.code ?? -32603, C = u?.data, w = u.message || "MCP request failed", Q = x(s.id ?? null, v, w, C);
+      } catch (d) {
+        if (j(o))
+          return r.log.error(d, "MCP notification processing error"), e.status(202).send();
+        r.log.error(d, "MCP RPC error");
+        const h = d?.code ?? -32603, C = d?.data, w = d.message || "MCP request failed", Q = x(o.id ?? null, h, w, C);
         if (l) {
           const $ = E();
           e.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${$}
@@ -403,17 +404,17 @@ ${P(Q, $)}`);
   i.setErrorHandler(async (t, r, e) => {
     r.log.error(t, "Fastify cube adapter error"), e.statusCode < 400 && e.status(500);
     const a = t instanceof Error ? t : String(t);
-    return d(a, e.statusCode);
+    return u(a, e.statusCode);
   }), R();
 };
-async function Z(h, i) {
-  await h.register(N, i);
+async function Z(b, i) {
+  await b.register(N, i);
 }
-function ee(h) {
+function ee(b) {
   const i = require("fastify")({
     logger: !0
   });
-  return i.register(N, h), i;
+  return i.register(N, b), i;
 }
 export {
   ee as createCubeApp,