drizzle-cube 0.4.4 → 0.4.7

package/dist/adapters/handler-DsNgnIPK.js

@@ -0,0 +1,458 @@
+import { h as I, Q as N, j, D as M } from "./mcp-transport-DFTCWene.js";
+import { handleDiscover as z, handleLoad as D } from "./utils.js";
+function O(r) {
+  if (r.length === 0)
+    return "No cubes are currently available.";
+  const o = ["## Available Cubes", ""];
+  for (const a of r) {
+    if (o.push(`### ${a.name}`), a.description && o.push(a.description), a.measures && a.measures.length > 0) {
+      o.push(""), o.push("**Measures:**");
+      for (const s of a.measures) {
+        const e = s.description ? ` - ${s.description}` : "";
+        o.push(`- \`${a.name}.${s.name}\` (${s.type})${e}`);
+      }
+    }
+    if (a.dimensions && a.dimensions.length > 0) {
+      o.push(""), o.push("**Dimensions:**");
+      for (const s of a.dimensions) {
+        const e = s.description ? ` - ${s.description}` : "";
+        o.push(`- \`${a.name}.${s.name}\` (${s.type})${e}`);
+      }
+    }
+    if (a.relationships && a.relationships.length > 0) {
+      o.push(""), o.push("**Joins:**");
+      for (const s of a.relationships)
+        o.push(`- → \`${s.targetCube}\` (${s.relationship})`);
+    }
+    o.push("");
+  }
+  return o.join(`
+`);
+}
+function f(r) {
+  return r.messages.map((o) => o.content.text).join(`
+
+`);
+}
+function U(r) {
+  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",
+    "",
+    "---",
+    "",
+    f(I),
+    "",
+    "---",
+    "",
+    f(N),
+    "",
+    "---",
+    "",
+    f(j),
+    "",
+    "---",
+    "",
+    f(M),
+    "",
+    "---",
+    "",
+    O(r)
+  ].join(`
+`);
+}
+function L() {
+  return [
+    // Tool 1: discover_cubes
+    {
+      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)" }
+        }
+      }
+    },
+    // Tool 2: get_cube_metadata
+    {
+      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: {}
+      }
+    },
+    // Tool 3: execute_query
+    {
+      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"
+          }
+        }
+      }
+    },
+    // Tool 4: add_portlet
+    {
+      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"]
+      }
+    },
+    // Tool 5: add_markdown
+    {
+      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 P(r) {
+  const { semanticLayer: o, securityContext: a } = r, s = /* @__PURE__ */ new Map();
+  return s.set("discover_cubes", async (e) => {
+    const n = await z(o, {
+      topic: e.topic,
+      intent: e.intent,
+      limit: e.limit,
+      minScore: e.minScore
+    });
+    return { result: JSON.stringify(n, null, 2) };
+  }), s.set("get_cube_metadata", async () => {
+    const e = o.getMetadata();
+    return { result: JSON.stringify(e, null, 2) };
+  }), s.set("execute_query", async (e) => {
+    try {
+      const n = {
+        measures: e.measures,
+        dimensions: e.dimensions,
+        filters: e.filters,
+        timeDimensions: e.timeDimensions,
+        order: e.order,
+        limit: e.limit
+      }, i = await D(o, a, { query: n });
+      return {
+        result: JSON.stringify({
+          rowCount: i.data.length,
+          data: i.data,
+          annotation: i.annotation
+        }, null, 2)
+      };
+    } catch (n) {
+      return {
+        result: `Query execution failed: ${n instanceof Error ? n.message : "Unknown error"}`,
+        isError: !0
+      };
+    }
+  }), s.set("add_portlet", async (e) => {
+    let n;
+    try {
+      n = 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 = o.validateQuery(n);
+    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)}`, b = {
+      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: b }
+    };
+  }), s.set("add_markdown", async (e) => {
+    const n = `markdown-${Date.now()}-${Math.random().toString(36).slice(2, 7)}`, i = {
+      id: n,
+      title: e.title,
+      content: e.content
+    };
+    return {
+      result: `Markdown block added to notebook (id: ${n}). [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* J(r) {
+  const { message: o, sessionId: a, semanticLayer: s, securityContext: e, agentConfig: n, apiKey: i } = r;
+  let y;
+  try {
+    const l = await import(
+      /* webpackIgnore: true */
+      "@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 b = new y({ apiKey: i }), E = L(), T = P({ semanticLayer: s, securityContext: e }), q = s.getMetadata(), C = U(q), R = n.model || "claude-sonnet-4-6", $ = n.maxTurns || 25, S = n.maxTokens || 4096, _ = [
+    { role: "user", content: o }
+  ];
+  try {
+    for (let l = 0; l < $; l++) {
+      const A = await b.messages.create({
+        model: R,
+        max_tokens: S,
+        system: C,
+        tools: E,
+        messages: _,
+        stream: !0
+      }), u = [];
+      let w = -1, m = "", x = "";
+      for await (const d of A)
+        switch (d.type) {
+          case "content_block_start": {
+            w++;
+            const t = d.content_block;
+            t.type === "tool_use" ? (u.push({ type: "tool_use", id: t.id, name: t.name, input: {} }), m = "", yield {
+              type: "tool_use_start",
+              data: { id: t.id, name: t.name, input: void 0 }
+            }) : t.type === "text" && u.push({ type: "text", text: "" });
+            break;
+          }
+          case "content_block_delta": {
+            const t = d.delta;
+            if (t.type === "text_delta" && t.text) {
+              const h = u[w];
+              h && (h.text = (h.text || "") + t.text), yield { type: "text_delta", data: t.text };
+            } else t.type === "input_json_delta" && t.partial_json && (m += t.partial_json);
+            break;
+          }
+          case "content_block_stop": {
+            const t = u[w];
+            if (t?.type === "tool_use" && m) {
+              try {
+                t.input = JSON.parse(m);
+              } catch {
+                t.input = {};
+              }
+              m = "";
+            }
+            break;
+          }
+          case "message_delta": {
+            const t = d.delta;
+            t.stop_reason && (x = t.stop_reason);
+            break;
+          }
+        }
+      if (_.push({ role: "assistant", content: u }), x !== "tool_use")
+        break;
+      const g = [];
+      for (const d of u) {
+        if (d.type !== "tool_use") continue;
+        const t = d.name, h = d.input || {}, p = d.id, k = T.get(t);
+        if (!k) {
+          g.push({
+            type: "tool_result",
+            tool_use_id: p,
+            content: `Unknown tool: ${t}`,
+            is_error: !0
+          }), yield {
+            type: "tool_use_result",
+            data: { id: p, name: t, result: `Unknown tool: ${t}` }
+          };
+          continue;
+        }
+        try {
+          const c = await k(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: t, result: c.result }
+          };
+        } catch (c) {
+          const v = c instanceof Error ? c.message : "Tool execution failed";
+          g.push({
+            type: "tool_result",
+            tool_use_id: p,
+            content: v,
+            is_error: !0
+          }), yield {
+            type: "tool_use_result",
+            data: { id: p, name: t, result: v }
+          };
+        }
+      }
+      yield { type: "turn_complete", data: {} }, _.push({ role: "user", content: g });
+    }
+    yield {
+      type: "done",
+      data: { sessionId: a || "" }
+    };
+  } catch (l) {
+    yield {
+      type: "error",
+      data: {
+        message: l instanceof Error ? l.message : "Agent execution failed"
+      }
+    };
+  }
+}
+export {
+  J as handleAgentChat
+};

package/dist/adapters/hono/index.cjs

@@ -1,7 +1,11 @@
-"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const T=require("hono"),s=require("../mcp-transport-BqLo4hKi.cjs"),w=require("../utils.cjs");var J=C=>{const d={...{origin:"*",allowMethods:["GET","HEAD","PUT","POST","DELETE","PATCH"],allowHeaders:[],exposeHeaders:[]},...C},P=(l=>typeof l=="string"?l==="*"?()=>l:a=>l===a?a:null:typeof l=="function"?l:a=>l.includes(a)?a:null)(d.origin),y=(l=>typeof l=="function"?l:Array.isArray(l)?()=>l:()=>[])(d.allowMethods);return async function(a,f){function q(u,o){a.res.headers.set(u,o)}const v=await P(a.req.header("origin")||"",a);if(v&&q("Access-Control-Allow-Origin",v),d.credentials&&q("Access-Control-Allow-Credentials","true"),d.exposeHeaders?.length&&q("Access-Control-Expose-Headers",d.exposeHeaders.join(",")),a.req.method==="OPTIONS"){d.origin!=="*"&&q("Vary","Origin"),d.maxAge!=null&&q("Access-Control-Max-Age",d.maxAge.toString());const u=await y(a.req.header("origin")||"",a);u.length&&q("Access-Control-Allow-Methods",u.join(","));let o=d.allowHeaders;if(!o?.length){const r=a.req.header("Access-Control-Request-Headers");r&&(o=r.split(/\s*,\s*/))}return o?.length&&(q("Access-Control-Allow-Headers",o.join(",")),a.res.headers.append("Vary","Access-Control-Request-Headers")),a.res.headers.delete("Content-Length"),a.res.headers.delete("Content-Type"),new Response(null,{headers:a.res.headers,status:204,statusText:"No Content"})}await f(),d.origin!=="*"&&a.header("Vary","Origin",{append:!0})}};function I(C){const{cubes:j,drizzle:d,schema:P,extractSecurityContext:y,engineType:l,cors:a,basePath:f="/cubejs-api/v1",cache:q,mcp:v={enabled:!0}}=C;if(!j||j.length===0)throw new Error("At least one cube must be provided in the cubes array");const u=new T.Hono;a&&u.use("/*",J(a));const o=new s.SemanticLayerCompiler({drizzle:d,schema:P,engineType:l,cache:q});if(j.forEach(r=>{o.registerCube(r)}),u.post(`${f}/load`,async r=>{try{const e=await r.req.json(),n=e.query||e,i=await y(r),t=o.validateQuery(n);if(!t.isValid)return r.json({error:`Query validation failed: ${t.errors.join(", ")}`},400);const c=r.req.header("x-cache-control")==="no-cache",p=await o.executeMultiCubeQuery(n,i,{skipCache:c});return r.json(w.formatCubeResponse(n,p,o))}catch(e){return console.error("Query execution error:",e),r.json({error:e instanceof Error?e.message:"Query execution failed"},500)}}),u.get(`${f}/load`,async r=>{try{const e=r.req.query("query");if(!e)return r.json({error:"Query parameter is required"},400);let n;try{n=JSON.parse(e)}catch{return r.json({error:"Invalid JSON in query parameter"},400)}const i=await y(r),t=o.validateQuery(n);if(!t.isValid)return r.json({error:`Query validation failed: ${t.errors.join(", ")}`},400);const c=r.req.header("x-cache-control")==="no-cache",p=await o.executeMultiCubeQuery(n,i,{skipCache:c});return r.json(w.formatCubeResponse(n,p,o))}catch(e){return console.error("Query execution error:",e),r.json({error:e instanceof Error?e.message:"Query execution failed"},500)}}),u.post(`${f}/batch`,async r=>{try{const e=await r.req.json(),{queries:n}=e;if(!n||!Array.isArray(n))return r.json({error:'Request body must contain a "queries" array'},400);if(n.length===0)return r.json({error:"Queries array cannot be empty"},400);const i=await y(r),t=r.req.header("x-cache-control")==="no-cache",c=await w.handleBatchRequest(n,i,o,{skipCache:t});return r.json(c)}catch(e){return console.error("Batch execution error:",e),r.json({error:e instanceof Error?e.message:"Batch execution failed"},500)}}),u.get(`${f}/meta`,r=>{try{const e=o.getMetadata();return r.json(w.formatMetaResponse(e))}catch(e){return console.error("Metadata error:",e),r.json({error:e instanceof Error?e.message:"Failed to fetch metadata"},500)}}),u.post(`${f}/sql`,async r=>{try{const e=await r.req.json(),n=await y(r),i=o.validateQuery(e);if(!i.isValid)return r.json({error:`Query validation failed: ${i.errors.join(", ")}`},400);const t=e.measures?.[0]||e.dimensions?.[0];if(!t)return r.json({error:"No measures or dimensions specified"},400);const c=t.split(".")[0],p=await o.generateSQL(c,e,n);return r.json(w.formatSqlResponse(e,p))}catch(e){return console.error("SQL generation error:",e),r.json({error:e instanceof Error?e.message:"SQL generation failed"},500)}}),u.get(`${f}/sql`,async r=>{try{const e=r.req.query("query");if(!e)return r.json({error:"Query parameter is required"},400);const n=JSON.parse(e),i=await y(r),t=o.validateQuery(n);if(!t.isValid)return r.json({error:`Query validation failed: ${t.errors.join(", ")}`},400);const c=n.measures?.[0]||n.dimensions?.[0];if(!c)return r.json({error:"No measures or dimensions specified"},400);const p=c.split(".")[0],g=await o.generateSQL(p,n,i);return r.json(w.formatSqlResponse(n,g))}catch(e){return console.error("SQL generation error:",e),r.json({error:e instanceof Error?e.message:"SQL generation failed"},500)}}),u.post(`${f}/dry-run`,async r=>{try{const e=await r.req.json(),n=e.query||e,i=await y(r),t=await w.handleDryRun(n,i,o);return r.json(t)}catch(e){return console.error("Dry-run error:",e),r.json({error:e instanceof Error?e.message:"Dry-run validation failed",valid:!1},400)}}),u.get(`${f}/dry-run`,async r=>{try{const e=r.req.query("query");if(!e)return r.json({error:"Query parameter is required",valid:!1},400);const n=JSON.parse(e),i=await y(r),t=await w.handleDryRun(n,i,o);return r.json(t)}catch(e){return console.error("Dry-run error:",e),r.json({error:e instanceof Error?e.message:"Dry-run validation failed",valid:!1},400)}}),u.post(`${f}/explain`,async r=>{try{const e=await r.req.json(),n=e.query||e,i=e.options||{},t=await y(r),c=o.validateQuery(n);if(!c.isValid)return r.json({error:`Query validation failed: ${c.errors.join(", ")}`},400);const p=await o.explainQuery(n,t,i);return r.json(p)}catch(e){return console.error("Explain error:",e),r.json({error:e instanceof Error?e.message:"Explain query failed"},500)}}),v.enabled!==!1){const r={uri:"drizzle-cube://schema",name:"Cube Schema",description:"Current cube metadata as JSON",mimeType:"application/json",text:JSON.stringify(o.getMetadata(),null,2)},e=[...s.getDefaultResources(),r],n=s.getDefaultPrompts(),i=v.basePath??"/mcp";u.post(`${i}`,async t=>{const c=s.validateOriginHeader(t.req.header("origin"),v.allowedOrigins?{allowedOrigins:v.allowedOrigins}:{});if(!c.valid)return t.json(s.buildJsonRpcError(null,-32600,c.reason),403);const p=t.req.header("accept");if(!s.validateAcceptHeader(p))return t.json(s.buildJsonRpcError(null,-32600,"Accept header must include both application/json and text/event-stream"),400);const g=s.negotiateProtocol(t.req.header());if(!g.ok)return t.json({error:"Unsupported MCP protocol version",supported:g.supported},426);const H=await t.req.json().catch(()=>null),m=s.parseJsonRpc(H);if(!m)return t.json(s.buildJsonRpcError(null,-32600,"Invalid JSON-RPC 2.0 request"),400);const $=s.wantsEventStream(p),O=m.method==="initialize";try{const h=await s.dispatchMcpMethod(m.method,m.params,{semanticLayer:o,extractSecurityContext:y,rawRequest:t,rawResponse:null,negotiatedProtocol:g.negotiated,resources:e,prompts:n});if(s.isNotification(m))return t.body(null,202);const S=s.buildJsonRpcResult(m.id??null,h),A=O&&h&&typeof h=="object"&&"sessionId"in h?h.sessionId:void 0,b={};if(A&&(b[s.MCP_SESSION_ID_HEADER]=A),$){const R=new TextEncoder,x=s.primeEventId(),Q=new ReadableStream({start(E){E.enqueue(R.encode(`id: ${x}
+"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const N=require("hono"),i=require("../mcp-transport-BB998cy5.cjs"),v=require("../utils.cjs");var D=w=>{const p={...{origin:"*",allowMethods:["GET","HEAD","PUT","POST","DELETE","PATCH"],allowHeaders:[],exposeHeaders:[]},...w},I=(y=>typeof y=="string"?y==="*"?()=>y:u=>y===u?u:null:typeof y=="function"?y:u=>y.includes(u)?u:null)(p.origin),m=(y=>typeof y=="function"?y:Array.isArray(y)?()=>y:()=>[])(p.allowMethods);return async function(u,h){function g(C,c){u.res.headers.set(C,c)}const b=await I(u.req.header("origin")||"",u);if(b&&g("Access-Control-Allow-Origin",b),p.credentials&&g("Access-Control-Allow-Credentials","true"),p.exposeHeaders?.length&&g("Access-Control-Expose-Headers",p.exposeHeaders.join(",")),u.req.method==="OPTIONS"){p.origin!=="*"&&g("Vary","Origin"),p.maxAge!=null&&g("Access-Control-Max-Age",p.maxAge.toString());const C=await m(u.req.header("origin")||"",u);C.length&&g("Access-Control-Allow-Methods",C.join(","));let c=p.allowHeaders;if(!c?.length){const s=u.req.header("Access-Control-Request-Headers");s&&(c=s.split(/\s*,\s*/))}return c?.length&&(g("Access-Control-Allow-Headers",c.join(",")),u.res.headers.append("Vary","Access-Control-Request-Headers")),u.res.headers.delete("Content-Length"),u.res.headers.delete("Content-Type"),new Response(null,{headers:u.res.headers,status:204,statusText:"No Content"})}await h(),p.origin!=="*"&&u.header("Vary","Origin",{append:!0})}};function O(w){const{cubes:j,drizzle:p,schema:I,extractSecurityContext:m,engineType:y,cors:u,basePath:h="/cubejs-api/v1",cache:g,mcp:b={enabled:!0},agent:C}=w;if(!j||j.length===0)throw new Error("At least one cube must be provided in the cubes array");const c=new N.Hono;u&&c.use("/*",D(u));const s=new i.SemanticLayerCompiler({drizzle:p,schema:I,engineType:y,cache:g});if(j.forEach(r=>{s.registerCube(r)}),c.post(`${h}/load`,async r=>{try{const e=await r.req.json(),n=e.query||e,o=await m(r),t=s.validateQuery(n);if(!t.isValid)return r.json({error:`Query validation failed: ${t.errors.join(", ")}`},400);const a=r.req.header("x-cache-control")==="no-cache",d=await s.executeMultiCubeQuery(n,o,{skipCache:a});return r.json(v.formatCubeResponse(n,d,s))}catch(e){return console.error("Query execution error:",e),r.json({error:e instanceof Error?e.message:"Query execution failed"},500)}}),c.get(`${h}/load`,async r=>{try{const e=r.req.query("query");if(!e)return r.json({error:"Query parameter is required"},400);let n;try{n=JSON.parse(e)}catch{return r.json({error:"Invalid JSON in query parameter"},400)}const o=await m(r),t=s.validateQuery(n);if(!t.isValid)return r.json({error:`Query validation failed: ${t.errors.join(", ")}`},400);const a=r.req.header("x-cache-control")==="no-cache",d=await s.executeMultiCubeQuery(n,o,{skipCache:a});return r.json(v.formatCubeResponse(n,d,s))}catch(e){return console.error("Query execution error:",e),r.json({error:e instanceof Error?e.message:"Query execution failed"},500)}}),c.post(`${h}/batch`,async r=>{try{const e=await r.req.json(),{queries:n}=e;if(!n||!Array.isArray(n))return r.json({error:'Request body must contain a "queries" array'},400);if(n.length===0)return r.json({error:"Queries array cannot be empty"},400);const o=await m(r),t=r.req.header("x-cache-control")==="no-cache",a=await v.handleBatchRequest(n,o,s,{skipCache:t});return r.json(a)}catch(e){return console.error("Batch execution error:",e),r.json({error:e instanceof Error?e.message:"Batch execution failed"},500)}}),c.get(`${h}/meta`,r=>{try{const e=s.getMetadata();return r.json(v.formatMetaResponse(e))}catch(e){return console.error("Metadata error:",e),r.json({error:e instanceof Error?e.message:"Failed to fetch metadata"},500)}}),c.post(`${h}/sql`,async r=>{try{const e=await r.req.json(),n=await m(r),o=s.validateQuery(e);if(!o.isValid)return r.json({error:`Query validation failed: ${o.errors.join(", ")}`},400);const t=e.measures?.[0]||e.dimensions?.[0];if(!t)return r.json({error:"No measures or dimensions specified"},400);const a=t.split(".")[0],d=await s.generateSQL(a,e,n);return r.json(v.formatSqlResponse(e,d))}catch(e){return console.error("SQL generation error:",e),r.json({error:e instanceof Error?e.message:"SQL generation failed"},500)}}),c.get(`${h}/sql`,async r=>{try{const e=r.req.query("query");if(!e)return r.json({error:"Query parameter is required"},400);const n=JSON.parse(e),o=await m(r),t=s.validateQuery(n);if(!t.isValid)return r.json({error:`Query validation failed: ${t.errors.join(", ")}`},400);const a=n.measures?.[0]||n.dimensions?.[0];if(!a)return r.json({error:"No measures or dimensions specified"},400);const d=a.split(".")[0],q=await s.generateSQL(d,n,o);return r.json(v.formatSqlResponse(n,q))}catch(e){return console.error("SQL generation error:",e),r.json({error:e instanceof Error?e.message:"SQL generation failed"},500)}}),c.post(`${h}/dry-run`,async r=>{try{const e=await r.req.json(),n=e.query||e,o=await m(r),t=await v.handleDryRun(n,o,s);return r.json(t)}catch(e){return console.error("Dry-run error:",e),r.json({error:e instanceof Error?e.message:"Dry-run validation failed",valid:!1},400)}}),c.get(`${h}/dry-run`,async r=>{try{const e=r.req.query("query");if(!e)return r.json({error:"Query parameter is required",valid:!1},400);const n=JSON.parse(e),o=await m(r),t=await v.handleDryRun(n,o,s);return r.json(t)}catch(e){return console.error("Dry-run error:",e),r.json({error:e instanceof Error?e.message:"Dry-run validation failed",valid:!1},400)}}),c.post(`${h}/explain`,async r=>{try{const e=await r.req.json(),n=e.query||e,o=e.options||{},t=await m(r),a=s.validateQuery(n);if(!a.isValid)return r.json({error:`Query validation failed: ${a.errors.join(", ")}`},400);const d=await s.explainQuery(n,t,o);return r.json(d)}catch(e){return console.error("Explain error:",e),r.json({error:e instanceof Error?e.message:"Explain query failed"},500)}}),C&&c.post(`${h}/agent/chat`,async r=>{try{const{handleAgentChat:e}=await Promise.resolve().then(()=>require("../handler-D6l8AbJV.cjs")),n=await r.req.json(),{message:o,sessionId:t}=n;if(!o||typeof o!="string")return r.json({error:"message is required and must be a string"},400);let a=(C.apiKey||"").trim();if(C.allowClientApiKey){const l=r.req.header("x-agent-api-key");l&&(a=l.trim())}if(!a)return r.json({error:"No API key configured. Set agent.apiKey in server config or send X-Agent-Api-Key header."},401);const d=await m(r),q=new TextEncoder,R=new ReadableStream({async start(l){try{const x=e({message:o,sessionId:t,semanticLayer:s,securityContext:d,agentConfig:C,apiKey:a});for await(const E of x){const f=`data: ${JSON.stringify(E)}
 
-`)),E.enqueue(R.encode(s.serializeSseEvent(S,x))),E.close()}});return new Response(Q,{status:200,headers:{"Content-Type":"text/event-stream","Cache-Control":"no-cache",Connection:"keep-alive",...b}})}return t.json(S,200,b)}catch(h){if(s.isNotification(m))return console.error("MCP notification processing error:",h),t.body(null,202);console.error("MCP RPC error:",h);const S=h?.code??-32603,A=h?.data,b=h.message||"MCP request failed",R=s.buildJsonRpcError(m.id??null,S,b,A);if($){const x=new TextEncoder,Q=s.primeEventId(),E=new ReadableStream({start(M){M.enqueue(x.encode(`id: ${Q}
+`;l.enqueue(q.encode(f))}}catch(x){const E={type:"error",data:{message:x instanceof Error?x.message:"Stream failed"}};l.enqueue(q.encode(`data: ${JSON.stringify(E)}
 
-`)),M.enqueue(x.encode(s.serializeSseEvent(R,Q))),M.close()}});return new Response(E,{status:200,headers:{"Content-Type":"text/event-stream","Cache-Control":"no-cache",Connection:"keep-alive"}})}return t.json(R,200)}}),u.delete(`${i}`,t=>t.json({error:"Session termination not supported"},405)),u.get(`${i}`,t=>{const c=new TextEncoder,p=s.primeEventId();let g;const H=new ReadableStream({start(m){m.enqueue(c.encode(s.serializeSseEvent({jsonrpc:"2.0",method:"mcp/ready",params:{protocol:"streamable-http"}},p,15e3))),g=setInterval(()=>{m.enqueue(c.encode(`: keep-alive
+`))}finally{l.close()}}});return new Response(R,{status:200,headers:{"Content-Type":"text/event-stream","Cache-Control":"no-cache",Connection:"keep-alive"}})}catch(e){return console.error("Agent chat error:",e),r.json({error:e instanceof Error?e.message:"Agent chat failed"},500)}}),b.enabled!==!1){const r={uri:"drizzle-cube://schema",name:"Cube Schema",description:"Current cube metadata as JSON",mimeType:"application/json",text:JSON.stringify(s.getMetadata(),null,2)},e=[...i.getDefaultResources(),r],n=i.getDefaultPrompts(),o=b.basePath??"/mcp";c.post(`${o}`,async t=>{const a=i.validateOriginHeader(t.req.header("origin"),b.allowedOrigins?{allowedOrigins:b.allowedOrigins}:{});if(!a.valid)return t.json(i.buildJsonRpcError(null,-32600,a.reason),403);const d=t.req.header("accept");if(!i.validateAcceptHeader(d))return t.json(i.buildJsonRpcError(null,-32600,"Accept header must include both application/json and text/event-stream"),400);const q=i.negotiateProtocol(t.req.header());if(!q.ok)return t.json({error:"Unsupported MCP protocol version",supported:q.supported},426);const R=await t.req.json().catch(()=>null),l=i.parseJsonRpc(R);if(!l)return t.json(i.buildJsonRpcError(null,-32600,"Invalid JSON-RPC 2.0 request"),400);const x=i.wantsEventStream(d),E=l.method==="initialize";try{const f=await i.dispatchMcpMethod(l.method,l.params,{semanticLayer:s,extractSecurityContext:m,rawRequest:t,rawResponse:null,negotiatedProtocol:q.negotiated,resources:e,prompts:n});if(i.isNotification(l))return t.body(null,202);const $=i.buildJsonRpcResult(l.id??null,f),T=E&&f&&typeof f=="object"&&"sessionId"in f?f.sessionId:void 0,S={};if(T&&(S[i.MCP_SESSION_ID_HEADER]=T),x){const A=new TextEncoder,Q=i.primeEventId(),H=new ReadableStream({start(P){P.enqueue(A.encode(`id: ${Q}
 
-`))},15e3)},cancel(){clearInterval(g)}});return new Response(H,{status:200,headers:{"Content-Type":"text/event-stream","Cache-Control":"no-cache",Connection:"keep-alive"}})})}return u}function N(C,j){const d=I(j);return C.route("/",d),C}function D(C){const j=new T.Hono;return N(j,C)}exports.createCubeApp=D;exports.createCubeRoutes=I;exports.mountCubeRoutes=N;
+`)),P.enqueue(A.encode(i.serializeSseEvent($,Q))),P.close()}});return new Response(H,{status:200,headers:{"Content-Type":"text/event-stream","Cache-Control":"no-cache",Connection:"keep-alive",...S}})}return t.json($,200,S)}catch(f){if(i.isNotification(l))return console.error("MCP notification processing error:",f),t.body(null,202);console.error("MCP RPC error:",f);const $=f?.code??-32603,T=f?.data,S=f.message||"MCP request failed",A=i.buildJsonRpcError(l.id??null,$,S,T);if(x){const Q=new TextEncoder,H=i.primeEventId(),P=new ReadableStream({start(M){M.enqueue(Q.encode(`id: ${H}
+
+`)),M.enqueue(Q.encode(i.serializeSseEvent(A,H))),M.close()}});return new Response(P,{status:200,headers:{"Content-Type":"text/event-stream","Cache-Control":"no-cache",Connection:"keep-alive"}})}return t.json(A,200)}}),c.delete(`${o}`,t=>t.json({error:"Session termination not supported"},405)),c.get(`${o}`,t=>{const a=new TextEncoder,d=i.primeEventId();let q;const R=new ReadableStream({start(l){l.enqueue(a.encode(i.serializeSseEvent({jsonrpc:"2.0",method:"mcp/ready",params:{protocol:"streamable-http"}},d,15e3))),q=setInterval(()=>{l.enqueue(a.encode(`: keep-alive
+
+`))},15e3)},cancel(){clearInterval(q)}});return new Response(R,{status:200,headers:{"Content-Type":"text/event-stream","Cache-Control":"no-cache",Connection:"keep-alive"}})})}return c}function J(w,j){const p=O(j);return w.route("/",p),w}function k(w){const j=new N.Hono;return J(j,w)}exports.createCubeApp=k;exports.createCubeRoutes=O;exports.mountCubeRoutes=J;

package/dist/adapters/hono/index.d.ts

@@ -1,5 +1,6 @@
 import { Hono } from 'hono';
 import { SemanticQuery, SecurityContext, DatabaseExecutor, DrizzleDatabase, Cube, CacheConfig } from '../../server';
+import { AgentConfig } from '../../server/agent/types';
 import { PostgresJsDatabase } from 'drizzle-orm/postgres-js';
 import { MySql2Database } from 'drizzle-orm/mysql2';
 import { BetterSQLite3Database } from 'drizzle-orm/better-sqlite3';
@@ -72,6 +73,12 @@ export interface HonoAdapterOptions {
      * @default { enabled: true }
      */
     mcp?: MCPOptions;
+    /**
+     * Agent configuration for the agentic AI notebook feature.
+     * When provided, enables the POST /agent/chat SSE endpoint.
+     * Requires `@anthropic-ai/sdk` as a peer dependency.
+     */
+    agent?: AgentConfig;
 }
 /**
  * Create Hono routes for Cube.js-compatible API