drizzle-cube 0.4.4 → 0.4.7

package/dist/adapters/fastify/index.js

@@ -1,57 +1,58 @@
-import { S as D, v as H, b as C, a as V, n as z, p as T, w as _, d as B, i as $, M as F, c as U, e as q, s as R } from "../mcp-transport-YHDZWKOi.js";
-import { formatErrorResponse as u, formatCubeResponse as j, handleBatchRequest as G, formatMetaResponse as K, formatSqlResponse as M, handleDryRun as I } from "../utils.js";
-const N = function(n, O, E) {
+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-DFTCWene.js";
+import { formatErrorResponse as d, 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: w,
-    drizzle: k,
-    schema: A,
-    extractSecurityContext: y,
+    cubes: S,
+    drizzle: O,
+    schema: J,
+    extractSecurityContext: p,
     engineType: L,
-    cors: Q,
-    basePath: p = "/cubejs-api/v1",
-    bodyLimit: m = 10485760,
+    cors: q,
+    basePath: m = "/cubejs-api/v1",
+    bodyLimit: g = 10485760,
     // 10MB
-    cache: J,
-    mcp: g = { enabled: !0 }
-  } = O;
-  if (!w || w.length === 0)
-    return E(new Error("At least one cube must be provided in the cubes array"));
-  Q && n.register(import("@fastify/cors"), Q), n.addHook("onRequest", async (r, t) => {
-    r.method === "POST" && (r.body = void 0);
+    cache: H,
+    mcp: b = { enabled: !0 },
+    agent: f
+  } = k;
+  if (!S || S.length === 0)
+    return R(new Error("At least one cube must be provided in the cubes array"));
+  q && i.register(import("@fastify/cors"), q), i.addHook("onRequest", async (t, r) => {
+    t.method === "POST" && (t.body = void 0);
   });
   const c = new D({
-    drizzle: k,
-    schema: A,
+    drizzle: O,
+    schema: J,
     engineType: L,
-    cache: J
+    cache: H
   });
-  if (w.forEach((r) => {
-    c.registerCube(r);
-  }), n.post(`${p}/load`, {
-    bodyLimit: m,
+  if (S.forEach((t) => {
+    c.registerCube(t);
+  }), i.post(`${m}/load`, {
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
         additionalProperties: !0
       }
     }
-  }, async (r, t) => {
+  }, async (t, r) => {
     try {
-      const e = r.body, a = e.query || e, s = await y(r), o = c.validateQuery(a);
+      const e = t.body, a = e.query || e, n = await p(t), o = c.validateQuery(a);
       if (!o.isValid)
-        return t.status(400).send(u(
+        return r.status(400).send(d(
           `Query validation failed: ${o.errors.join(", ")}`,
           400
         ));
-      const i = r.headers["x-cache-control"] === "no-cache", d = await c.executeMultiCubeQuery(a, s, { skipCache: i });
-      return j(a, d, c);
+      const s = t.headers["x-cache-control"] === "no-cache", l = await c.executeMultiCubeQuery(a, n, { skipCache: s });
+      return A(a, l, c);
     } catch (e) {
-      return r.log.error(e, "Query execution error"), t.status(500).send(u(
+      return t.log.error(e, "Query execution error"), r.status(500).send(d(
         e instanceof Error ? e.message : "Query execution failed",
         500
       ));
     }
-  }), n.get(`${p}/load`, {
+  }), i.get(`${m}/load`, {
     schema: {
       querystring: {
         type: "object",
@@ -61,34 +62,34 @@ const N = function(n, O, E) {
         required: ["query"]
       }
     }
-  }, async (r, t) => {
+  }, async (t, r) => {
     try {
-      const { query: e } = r.query;
+      const { query: e } = t.query;
       let a;
       try {
         a = JSON.parse(e);
       } catch {
-        return t.status(400).send(u(
+        return r.status(400).send(d(
           "Invalid JSON in query parameter",
           400
         ));
       }
-      const s = await y(r), o = c.validateQuery(a);
+      const n = await p(t), o = c.validateQuery(a);
       if (!o.isValid)
-        return t.status(400).send(u(
+        return r.status(400).send(d(
           `Query validation failed: ${o.errors.join(", ")}`,
           400
         ));
-      const i = r.headers["x-cache-control"] === "no-cache", d = await c.executeMultiCubeQuery(a, s, { skipCache: i });
-      return j(a, d, c);
+      const s = t.headers["x-cache-control"] === "no-cache", l = await c.executeMultiCubeQuery(a, n, { skipCache: s });
+      return A(a, l, c);
     } catch (e) {
-      return r.log.error(e, "Query execution error"), t.status(500).send(u(
+      return t.log.error(e, "Query execution error"), r.status(500).send(d(
         e instanceof Error ? e.message : "Query execution failed",
         500
       ));
     }
-  }), n.post(`${p}/batch`, {
-    bodyLimit: m,
+  }), i.post(`${m}/batch`, {
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
@@ -101,68 +102,68 @@ const N = function(n, O, E) {
         }
       }
     }
-  }, async (r, t) => {
+  }, async (t, r) => {
     try {
-      const { queries: e } = r.body;
+      const { queries: e } = t.body;
       if (!e || !Array.isArray(e))
-        return t.status(400).send(u(
+        return r.status(400).send(d(
           'Request body must contain a "queries" array',
           400
         ));
       if (e.length === 0)
-        return t.status(400).send(u(
+        return r.status(400).send(d(
           "Queries array cannot be empty",
           400
         ));
-      const a = await y(r), s = r.headers["x-cache-control"] === "no-cache";
-      return await G(e, a, c, { skipCache: s });
+      const a = await p(t), n = t.headers["x-cache-control"] === "no-cache";
+      return await X(e, a, c, { skipCache: n });
     } catch (e) {
-      return r.log.error(e, "Batch execution error"), t.status(500).send(u(
+      return t.log.error(e, "Batch execution error"), r.status(500).send(d(
         e instanceof Error ? e.message : "Batch execution failed",
         500
       ));
     }
-  }), n.get(`${p}/meta`, async (r, t) => {
+  }), i.get(`${m}/meta`, async (t, r) => {
     try {
       const e = c.getMetadata();
-      return K(e);
+      return G(e);
     } catch (e) {
-      return r.log.error(e, "Metadata error"), t.status(500).send(u(
+      return t.log.error(e, "Metadata error"), r.status(500).send(d(
         e instanceof Error ? e.message : "Failed to fetch metadata",
         500
       ));
     }
-  }), n.post(`${p}/sql`, {
-    bodyLimit: m,
+  }), i.post(`${m}/sql`, {
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
         additionalProperties: !0
       }
     }
-  }, async (r, t) => {
+  }, async (t, r) => {
     try {
-      const e = r.body, a = await y(r), s = c.validateQuery(e);
-      if (!s.isValid)
-        return t.status(400).send(u(
-          `Query validation failed: ${s.errors.join(", ")}`,
+      const e = t.body, a = await p(t), n = c.validateQuery(e);
+      if (!n.isValid)
+        return r.status(400).send(d(
+          `Query validation failed: ${n.errors.join(", ")}`,
           400
         ));
       const o = e.measures?.[0] || e.dimensions?.[0];
       if (!o)
-        return t.status(400).send(u(
+        return r.status(400).send(d(
           "No measures or dimensions specified",
           400
         ));
-      const i = o.split(".")[0], d = await c.generateSQL(i, e, a);
-      return M(e, d);
+      const s = o.split(".")[0], l = await c.generateSQL(s, e, a);
+      return I(e, l);
     } catch (e) {
-      return r.log.error(e, "SQL generation error"), t.status(500).send(u(
+      return t.log.error(e, "SQL generation error"), r.status(500).send(d(
         e instanceof Error ? e.message : "SQL generation failed",
         500
       ));
     }
-  }), n.get(`${p}/sql`, {
+  }), i.get(`${m}/sql`, {
     schema: {
       querystring: {
         type: "object",
@@ -172,47 +173,47 @@ const N = function(n, O, E) {
         required: ["query"]
       }
     }
-  }, async (r, t) => {
+  }, async (t, r) => {
     try {
-      const { query: e } = r.query, a = JSON.parse(e), s = await y(r), o = c.validateQuery(a);
+      const { query: e } = t.query, a = JSON.parse(e), n = await p(t), o = c.validateQuery(a);
       if (!o.isValid)
-        return t.status(400).send(u(
+        return r.status(400).send(d(
           `Query validation failed: ${o.errors.join(", ")}`,
           400
         ));
-      const i = a.measures?.[0] || a.dimensions?.[0];
-      if (!i)
-        return t.status(400).send(u(
+      const s = a.measures?.[0] || a.dimensions?.[0];
+      if (!s)
+        return r.status(400).send(d(
           "No measures or dimensions specified",
           400
         ));
-      const d = i.split(".")[0], x = await c.generateSQL(d, a, s);
-      return M(a, x);
+      const l = s.split(".")[0], y = await c.generateSQL(l, a, n);
+      return I(a, y);
     } catch (e) {
-      return r.log.error(e, "SQL generation error"), t.status(500).send(u(
+      return t.log.error(e, "SQL generation error"), r.status(500).send(d(
         e instanceof Error ? e.message : "SQL generation failed",
         500
       ));
     }
-  }), n.post(`${p}/dry-run`, {
-    bodyLimit: m,
+  }), i.post(`${m}/dry-run`, {
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
         additionalProperties: !0
       }
     }
-  }, async (r, t) => {
+  }, async (t, r) => {
     try {
-      const e = r.body, a = e.query || e, s = await y(r);
-      return await I(a, s, c);
+      const e = t.body, a = e.query || e, n = await p(t);
+      return await M(a, n, c);
     } catch (e) {
-      return r.log.error(e, "Dry-run error"), t.status(400).send({
+      return t.log.error(e, "Dry-run error"), r.status(400).send({
         error: e instanceof Error ? e.message : "Dry-run validation failed",
         valid: !1
       });
     }
-  }), n.get(`${p}/dry-run`, {
+  }), i.get(`${m}/dry-run`, {
     schema: {
       querystring: {
         type: "object",
@@ -222,142 +223,200 @@ const N = function(n, O, E) {
         required: ["query"]
       }
     }
-  }, async (r, t) => {
+  }, async (t, r) => {
     try {
-      const { query: e } = r.query, a = JSON.parse(e), s = await y(r);
-      return await I(a, s, c);
+      const { query: e } = t.query, a = JSON.parse(e), n = await p(t);
+      return await M(a, n, c);
     } catch (e) {
-      return r.log.error(e, "Dry-run error"), t.status(400).send({
+      return t.log.error(e, "Dry-run error"), r.status(400).send({
         error: e instanceof Error ? e.message : "Dry-run validation failed",
         valid: !1
       });
     }
-  }), n.post(`${p}/explain`, {
-    bodyLimit: m,
+  }), i.post(`${m}/explain`, {
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
         additionalProperties: !0
       }
     }
-  }, async (r, t) => {
+  }, async (t, r) => {
     try {
-      const e = r.body, a = e.query || e, s = e.options || {}, o = await y(r), i = c.validateQuery(a);
-      return i.isValid ? await c.explainQuery(a, o, s) : t.status(400).send({
-        error: `Query validation failed: ${i.errors.join(", ")}`
+      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(", ")}`
       });
     } catch (e) {
-      return r.log.error(e, "Explain error"), t.status(500).send({
+      return t.log.error(e, "Explain error"), r.status(500).send({
         error: e instanceof Error ? e.message : "Explain query failed"
       });
     }
-  }), g.enabled !== !1) {
-    const r = g.basePath ?? "/mcp";
-    n.post(`${r}`, {
-      bodyLimit: m,
+  }), f && i.post(`${m}/agent/chat`, {
+    bodyLimit: g,
+    schema: {
+      body: {
+        type: "object",
+        additionalProperties: !0
+      }
+    }
+  }, async (t, r) => {
+    try {
+      const { handleAgentChat: e } = await import("../handler-DsNgnIPK.js"), a = t.body, { message: n, sessionId: 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());
+      }
+      if (!s)
+        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);
+      r.raw.writeHead(200, {
+        "Content-Type": "text/event-stream",
+        "Cache-Control": "no-cache",
+        Connection: "keep-alive"
+      });
+      try {
+        const y = e({
+          message: n,
+          sessionId: o,
+          semanticLayer: c,
+          securityContext: l,
+          agentConfig: f,
+          apiKey: s
+        });
+        for await (const u of y)
+          r.raw.write(`data: ${JSON.stringify(u)}
+
+`);
+      } catch (y) {
+        const u = {
+          type: "error",
+          data: { message: y instanceof Error ? y.message : "Stream failed" }
+        };
+        r.raw.write(`data: ${JSON.stringify(u)}
+
+`);
+      } finally {
+        r.raw.end();
+      }
+    } catch (e) {
+      if (t.log.error(e, "Agent chat error"), !r.raw.headersSent)
+        return r.status(500).send({
+          error: e instanceof Error ? e.message : "Agent chat failed"
+        });
+    }
+  }), b.enabled !== !1) {
+    const t = b.basePath ?? "/mcp";
+    i.post(`${t}`, {
+      bodyLimit: g,
       schema: {
         body: {
           type: "object",
           additionalProperties: !0
         }
       }
-    }, async (t, e) => {
-      const a = H(
-        t.headers.origin,
-        g.allowedOrigins ? { allowedOrigins: g.allowedOrigins } : {}
+    }, async (r, e) => {
+      const a = K(
+        r.headers.origin,
+        b.allowedOrigins ? { allowedOrigins: b.allowedOrigins } : {}
       );
       if (!a.valid)
-        return e.status(403).send(C(null, -32600, a.reason));
-      const s = t.headers.accept;
-      if (!V(s))
-        return e.status(400).send(C(null, -32600, "Accept header must include both application/json and text/event-stream"));
-      const o = z(t.headers);
+        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)
         return e.status(426).send({
           error: "Unsupported MCP protocol version",
           supported: o.supported
         });
-      const i = T(t.body);
-      if (!i)
-        return e.status(400).send(C(null, -32600, "Invalid JSON-RPC 2.0 request"));
-      const d = _(s), x = i.method === "initialize";
+      const s = z(r.body);
+      if (!s)
+        return e.status(400).send(x(null, -32600, "Invalid JSON-RPC 2.0 request"));
+      const l = _(n), y = s.method === "initialize";
       try {
-        const l = await B(
-          i.method,
-          i.params,
+        const u = await B(
+          s.method,
+          s.params,
           {
             semanticLayer: c,
-            extractSecurityContext: y,
-            rawRequest: t,
+            extractSecurityContext: p,
+            rawRequest: r,
             rawResponse: e,
             negotiatedProtocol: o.negotiated
           }
         );
-        if ($(i))
+        if (j(s))
           return e.status(202).send();
-        const b = x && l && typeof l == "object" && "sessionId" in l ? l.sessionId : void 0;
-        b && e.header(F, b);
-        const v = U(i.id ?? null, l);
-        if (d) {
-          const f = q();
-          e.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${f}
+        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);
+        if (l) {
+          const w = E();
+          e.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${w}
 
-${R(v, f)}`);
+${P(C, w)}`);
           return;
         }
-        return e.send(v);
-      } catch (l) {
-        if ($(i))
-          return t.log.error(l, "MCP notification processing error"), e.status(202).send();
-        t.log.error(l, "MCP RPC error");
-        const b = l?.code ?? -32603, v = l?.data, f = l.message || "MCP request failed", S = C(i.id ?? null, b, f, v);
-        if (d) {
-          const P = q();
-          e.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${P}
+        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);
+        if (l) {
+          const $ = E();
+          e.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${$}
 
-${R(S, P)}`);
+${P(Q, $)}`);
           return;
         }
-        return e.send(S);
+        return e.send(Q);
       }
-    }), n.get(`${r}`, async (t, e) => {
-      const a = q();
+    }), i.get(`${t}`, async (r, e) => {
+      const a = E();
       e.raw.writeHead(200, {
         "Content-Type": "text/event-stream",
         "Cache-Control": "no-cache",
         Connection: "keep-alive"
-      }), e.raw.write(R({
+      }), e.raw.write(P({
         jsonrpc: "2.0",
         method: "mcp/ready",
         params: { protocol: "streamable-http" }
       }, a, 15e3));
-      const s = setInterval(() => {
+      const n = setInterval(() => {
         e.raw.write(`: keep-alive
 
 `);
       }, 15e3);
-      t.raw.on("close", () => {
-        clearInterval(s);
+      r.raw.on("close", () => {
+        clearInterval(n);
       });
-    }), n.delete(`${r}`, async (t, e) => e.status(405).send({ error: "Session termination not supported" }));
+    }), i.delete(`${t}`, async (r, e) => e.status(405).send({ error: "Session termination not supported" }));
   }
-  n.setErrorHandler(async (r, t, e) => {
-    t.log.error(r, "Fastify cube adapter error"), e.statusCode < 400 && e.status(500);
-    const a = r instanceof Error ? r : String(r);
-    return u(a, e.statusCode);
-  }), E();
+  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);
+  }), R();
 };
-async function Y(h, n) {
-  await h.register(N, n);
+async function Z(h, i) {
+  await h.register(N, i);
 }
-function Z(h) {
-  const n = require("fastify")({
+function ee(h) {
+  const i = require("fastify")({
     logger: !0
   });
-  return n.register(N, h), n;
+  return i.register(N, h), i;
 }
 export {
-  Z as createCubeApp,
+  ee as createCubeApp,
   N as cubePlugin,
-  Y as registerCubeRoutes
+  Z as registerCubeRoutes
 };

package/dist/adapters/handler-D6l8AbJV.cjs

@@ -0,0 +1,7 @@
+"use strict";var j=Object.create;var T=Object.defineProperty;var O=Object.getOwnPropertyDescriptor;var z=Object.getOwnPropertyNames;var D=Object.getPrototypeOf,U=Object.prototype.hasOwnProperty;var P=(a,t,n,s)=>{if(t&&typeof t=="object"||typeof t=="function")for(let e of z(t))!U.call(a,e)&&e!==n&&T(a,e,{get:()=>t[e],enumerable:!(s=O(t,e))||s.enumerable});return a};var L=(a,t,n)=>(n=a!=null?j(D(a)):{},P(t||!a||!a.__esModule?T(n,"default",{value:a,enumerable:!0}):n,a));Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const f=require("./mcp-transport-BB998cy5.cjs"),q=require("./utils.cjs");function Y(a){if(a.length===0)return"No cubes are currently available.";const t=["## Available Cubes",""];for(const n of a){if(t.push(`### ${n.name}`),n.description&&t.push(n.description),n.measures&&n.measures.length>0){t.push(""),t.push("**Measures:**");for(const s of n.measures){const e=s.description?` - ${s.description}`:"";t.push(`- \`${n.name}.${s.name}\` (${s.type})${e}`)}}if(n.dimensions&&n.dimensions.length>0){t.push(""),t.push("**Dimensions:**");for(const s of n.dimensions){const e=s.description?` - ${s.description}`:"";t.push(`- \`${n.name}.${s.name}\` (${s.type})${e}`)}}if(n.relationships&&n.relationships.length>0){t.push(""),t.push("**Joins:**");for(const s of n.relationships)t.push(`- → \`${s.targetCube}\` (${s.relationship})`)}t.push("")}return t.join(`
+`)}function b(a){return a.messages.map(t=>t.content.text).join(`
+
+`)}function F(a){return["# Drizzle Cube Analytics Agent","","You are an analytics agent that helps users explore and visualize data.","You have access to a semantic layer with cubes (data models) that you can query.","","## Your Workflow","","1. **Discover** available cubes using `discover_cubes` to understand the data","2. **Explain your approach** by calling `add_markdown` to describe what you plan to analyze and why","3. **Query** data using `execute_query` with properly structured queries","4. **Explain results** by calling `add_markdown` to describe what the data shows, key insights, and methodology","5. **Visualize** results by calling `add_portlet` to add charts/tables to the notebook","","## Important Guidelines","","- ALWAYS discover cubes first before attempting queries","- Use `add_portlet` to create visualizations after getting query results","- Use `add_markdown` to explain your findings, methodology, and insights","- Choose appropriate chart types: bar for categories, line for trends, table for detailed data","- When showing data, always add both a portlet (visualization) and markdown (explanation)","- If a query fails, explain the error and try an alternative approach","","## Output Format Rules","","### CRITICAL: Always think before acting","- EVERY single turn MUST begin with a text message (1-2 sentences) BEFORE any tool calls. This is your #1 rule — never violate it.","- This applies to EVERY turn, including turns where you are adding visualizations or explanations to the notebook.",`- Even when adding multiple portlets/markdowns in sequence, each turn must start with text like "Now I'll add the productivity chart." or "Next, let me visualize the department breakdown."`,'- Example good turn: "Let me discover what data is available." → discover_cubes',`- Example good turn: "I'll add a bar chart showing the top employees." → add_markdown → add_portlet`,"- Example bad turn: (no text) → add_portlet ← NEVER do this","","### Text vs Notebook","- ALL analysis, findings, methodology, and insights MUST go through `add_markdown` tool calls — never in your text responses","- Your text responses must be 1-2 short sentences (under 50 words) summarizing what you are about to do next — status updates only","- Never use markdown formatting (headers, bullets, bold, code blocks) in text responses — plain sentences only","","### Notebook content rules","- Before each `add_portlet`, ALWAYS call `add_markdown` first to explain WHY you are adding this visualization and what it shows","- Before calling `add_portlet`, verify the query is valid: all fields in `order` must also appear in `measures` or `dimensions`",'- Never put data tables in markdown blocks — use `add_portlet` with chartType "table" instead',"- Think out loud in the notebook: use `add_markdown` to share your reasoning at each step so users can follow along","","---","",b(f.MCP_GUIDE_PROMPT),"","---","",b(f.QUERY_RULES_PROMPT),"","---","",b(f.QUERY_BUILDING_PROMPT),"","---","",b(f.DATE_FILTERING_PROMPT),"","---","",Y(a)].join(`
+`)}function J(){return[{name:"discover_cubes",description:"Search for available data cubes by topic or intent. Call this FIRST to understand what data is available. Returns cube names, measures, dimensions, and relationships.",input_schema:{type:"object",properties:{topic:{type:"string",description:'Keyword to search (e.g., "sales", "employees")'},intent:{type:"string",description:'Natural language goal (e.g., "analyze productivity trends")'},limit:{type:"number",description:"Max results (default: 10)"},minScore:{type:"number",description:"Min relevance 0-1 (default: 0.1)"}}}},{name:"get_cube_metadata",description:"Get full metadata for all registered cubes including all measures, dimensions, types, and relationships. Use this for detailed schema information.",input_schema:{type:"object",properties:{}}},{name:"execute_query",description:"Execute a semantic query and return data results. The query follows the Cube.js query format with measures, dimensions, filters, timeDimensions, order, and limit.",input_schema:{type:"object",properties:{measures:{type:"array",items:{type:"string"},description:'Aggregation measures (e.g., ["Employees.count"])'},dimensions:{type:"array",items:{type:"string"},description:'Grouping dimensions (e.g., ["Employees.name"])'},filters:{type:"array",items:{type:"object",properties:{member:{type:"string"},operator:{type:"string"},values:{type:"array",items:{}}},required:["member","operator"]},description:"Filter conditions"},timeDimensions:{type:"array",items:{type:"object",properties:{dimension:{type:"string"},granularity:{type:"string"},dateRange:{}},required:["dimension"]},description:"Time dimensions with optional granularity"},order:{type:"object",description:'Sort order (e.g., {"Employees.count": "desc"})'},limit:{type:"number",description:"Row limit"}}}},{name:"add_portlet",description:"Add a chart or table visualization to the notebook. The query is validated before adding — if invalid, an error is returned so you can fix the query. The portlet fetches its own data - you do NOT need to pass data to it.",input_schema:{type:"object",properties:{title:{type:"string",description:"Title for the visualization"},query:{type:"string",description:"JSON string of the CubeQuery to visualize"},chartType:{type:"string",enum:["bar","line","area","pie","scatter","radar","bubble","table","number","funnel","heatmap","sankey","retention"],description:"Chart type to render"},chartConfig:{type:"object",properties:{xAxis:{type:"array",items:{type:"string"}},yAxis:{type:"array",items:{type:"string"}},series:{type:"array",items:{type:"string"}},sizeField:{type:"string"},colorField:{type:"string"}},description:"Chart axis configuration"},displayConfig:{type:"object",properties:{showLegend:{type:"boolean"},showGrid:{type:"boolean"},showTooltip:{type:"boolean"},stacked:{type:"boolean"},orientation:{type:"string",enum:["horizontal","vertical"]}},description:"Chart display configuration"}},required:["title","query","chartType"]}},{name:"add_markdown",description:"Add an explanation or analysis text block to the notebook. Use markdown formatting. Use this to explain findings, methodology, and insights alongside visualizations.",input_schema:{type:"object",properties:{title:{type:"string",description:"Optional title for the text block"},content:{type:"string",description:"Markdown content to display"}},required:["content"]}}]}function B(a){const{semanticLayer:t,securityContext:n}=a,s=new Map;return s.set("discover_cubes",async e=>{const r=await q.handleDiscover(t,{topic:e.topic,intent:e.intent,limit:e.limit,minScore:e.minScore});return{result:JSON.stringify(r,null,2)}}),s.set("get_cube_metadata",async()=>{const e=t.getMetadata();return{result:JSON.stringify(e,null,2)}}),s.set("execute_query",async e=>{try{const r={measures:e.measures,dimensions:e.dimensions,filters:e.filters,timeDimensions:e.timeDimensions,order:e.order,limit:e.limit},i=await q.handleLoad(t,n,{query:r});return{result:JSON.stringify({rowCount:i.data.length,data:i.data,annotation:i.annotation},null,2)}}catch(r){return{result:`Query execution failed: ${r instanceof Error?r.message:"Unknown error"}`,isError:!0}}}),s.set("add_portlet",async e=>{let r;try{r=JSON.parse(e.query)}catch{return{result:"Invalid query: could not parse JSON string. Ensure `query` is a valid JSON string.",isError:!0}}const i=t.validateQuery(r);if(!i.isValid)return{result:`Invalid query — fix these errors and retry:
+${i.errors.join(`
+`)}`,isError:!0};const y=`portlet-${Date.now()}-${Math.random().toString(36).slice(2,7)}`,_={id:y,title:e.title,query:e.query,chartType:e.chartType,chartConfig:e.chartConfig,displayConfig:e.displayConfig};return{result:`Portlet "${e.title}" added to notebook (id: ${y}, chart: ${e.chartType}). [Reminder: in your next response, start with a brief sentence about what you will do next BEFORE making any tool calls.]`,sideEffect:{type:"add_portlet",data:_}}}),s.set("add_markdown",async e=>{const r=`markdown-${Date.now()}-${Math.random().toString(36).slice(2,7)}`,i={id:r,title:e.title,content:e.content};return{result:`Markdown block added to notebook (id: ${r}). [Reminder: in your next response, start with a brief sentence about what you will do next BEFORE making any tool calls.]`,sideEffect:{type:"add_markdown",data:i}}}),s}async function*G(a){const{message:t,sessionId:n,semanticLayer:s,securityContext:e,agentConfig:r,apiKey:i}=a;let y;try{const l=await import("@anthropic-ai/sdk");y=l.default||l.Anthropic||l}catch{yield{type:"error",data:{message:"@anthropic-ai/sdk is required. Install it with: npm install @anthropic-ai/sdk"}};return}const _=new y({apiKey:i}),C=J(),R=B({semanticLayer:s,securityContext:e}),S=s.getMetadata(),$=F(S),A=r.model||"claude-sonnet-4-6",I=r.maxTurns||25,N=r.maxTokens||4096,w=[{role:"user",content:t}];try{for(let l=0;l<I;l++){const M=await _.messages.create({model:A,max_tokens:N,system:$,tools:C,messages:w,stream:!0}),u=[];let x=-1,m="",k="";for await(const d of M)switch(d.type){case"content_block_start":{x++;const o=d.content_block;o.type==="tool_use"?(u.push({type:"tool_use",id:o.id,name:o.name,input:{}}),m="",yield{type:"tool_use_start",data:{id:o.id,name:o.name,input:void 0}}):o.type==="text"&&u.push({type:"text",text:""});break}case"content_block_delta":{const o=d.delta;if(o.type==="text_delta"&&o.text){const h=u[x];h&&(h.text=(h.text||"")+o.text),yield{type:"text_delta",data:o.text}}else o.type==="input_json_delta"&&o.partial_json&&(m+=o.partial_json);break}case"content_block_stop":{const o=u[x];if(o?.type==="tool_use"&&m){try{o.input=JSON.parse(m)}catch{o.input={}}m=""}break}case"message_delta":{const o=d.delta;o.stop_reason&&(k=o.stop_reason);break}}if(w.push({role:"assistant",content:u}),k!=="tool_use")break;const g=[];for(const d of u){if(d.type!=="tool_use")continue;const o=d.name,h=d.input||{},p=d.id,v=R.get(o);if(!v){g.push({type:"tool_result",tool_use_id:p,content:`Unknown tool: ${o}`,is_error:!0}),yield{type:"tool_use_result",data:{id:p,name:o,result:`Unknown tool: ${o}`}};continue}try{const c=await v(h);c.sideEffect&&(yield c.sideEffect),g.push({type:"tool_result",tool_use_id:p,content:c.result,...c.isError?{is_error:!0}:{}}),yield{type:"tool_use_result",data:{id:p,name:o,result:c.result}}}catch(c){const E=c instanceof Error?c.message:"Tool execution failed";g.push({type:"tool_result",tool_use_id:p,content:E,is_error:!0}),yield{type:"tool_use_result",data:{id:p,name:o,result:E}}}}yield{type:"turn_complete",data:{}},w.push({role:"user",content:g})}yield{type:"done",data:{sessionId:n||""}}}catch(l){yield{type:"error",data:{message:l instanceof Error?l.message:"Agent execution failed"}}}}exports.handleAgentChat=G;