drizzle-cube 0.4.15 → 0.4.17

package/dist/adapters/{handler-DZnCbydH.js → handler-nTycPMWF.js}

@@ -1,6 +1,6 @@
-import { h as N, Q as P, j as L, D as R } from "./mcp-transport-m1X1GtwG.js";
-import { handleDiscover as Y, handleLoad as M } from "./utils.js";
-function z(o) {
+import { h as $, Q as B, j, D as U } from "./mcp-transport-m1X1GtwG.js";
+import { handleDiscover as Z, handleLoad as G } from "./utils.js";
+function K(o) {
   if (o.length === 0)
     return "No cubes are currently available.";
   const t = ["## Available Cubes", ""];
@@ -29,12 +29,12 @@ function z(o) {
   return t.join(`
 `);
 }
-function V(o) {
+function I(o) {
   return o.messages.map((t) => t.content.text).join(`
 
 `);
 }
-function H(o) {
+function W(o) {
   return [
     "# Drizzle Cube Analytics Agent",
     "",
@@ -160,19 +160,19 @@ function H(o) {
     "",
     "---",
     "",
-    V(N),
+    I($),
     "",
     "---",
     "",
-    V(P),
+    I(B),
     "",
     "---",
     "",
-    V(L),
+    I(j),
     "",
     "---",
     "",
-    V(R),
+    I(U),
     "",
     "---",
     "",
@@ -226,11 +226,11 @@ function H(o) {
     "",
     "---",
     "",
-    z(o)
+    K(o)
   ].join(`
 `);
 }
-const B = {
+const X = {
   label: "Bar Chart",
   description: "Compare values across categories",
   useCase: "Best for comparing discrete categories, showing rankings, or displaying changes over time",
@@ -296,7 +296,7 @@ const B = {
       description: "Number formatting for right Y-axis"
     }
   ]
-}, $ = {
+}, J = {
   label: "Line Chart",
   description: "Show trends and changes over time",
   useCase: "Best for continuous data, trends, time series, and showing relationships between multiple series",
@@ -379,7 +379,7 @@ const B = {
       description: "Number formatting for right Y-axis"
     }
   ]
-}, j = {
+}, Q = {
   label: "Area Chart",
   description: "Emphasize magnitude of change over time",
   useCase: "Best for showing cumulative totals, volume changes, or stacked comparisons over time",
@@ -451,7 +451,7 @@ const B = {
       description: "Number formatting for right Y-axis"
     }
   ]
-}, G = {
+}, ee = {
   label: "Pie Chart",
   description: "Show proportions of a whole",
   useCase: "Best for showing percentage distribution or composition of a total (limit to 5-7 slices)",
@@ -485,7 +485,7 @@ const B = {
       description: "Number formatting for values"
     }
   ]
-}, K = {
+}, te = {
   label: "Scatter Plot",
   description: "Reveal correlations between variables",
   useCase: "Best for identifying patterns, correlations, outliers, and relationships between two measures",
@@ -533,7 +533,7 @@ const B = {
       description: "Number formatting for Y-axis"
     }
   ]
-}, Z = {
+}, ae = {
   label: "Bubble Chart",
   description: "Compare three dimensions of data",
   useCase: "Best for showing relationships between three variables (X, Y, and size), market analysis",
@@ -599,7 +599,7 @@ const B = {
       description: "Number formatting for Y-axis and values"
     }
   ]
-}, U = {
+}, se = {
   label: "Radar Chart",
   description: "Compare multiple metrics across categories",
   useCase: "Best for multivariate comparisons, performance metrics, strengths/weaknesses analysis",
@@ -638,7 +638,7 @@ const B = {
       description: "Number formatting for values"
     }
   ]
-}, W = {
+}, ie = {
   label: "Radial Bar Chart",
   description: "Circular progress and KPI visualization",
   useCase: "Best for showing progress toward goals, KPIs, or comparing percentages in a compact form",
@@ -670,7 +670,7 @@ const B = {
       description: "Number formatting for values"
     }
   ]
-}, X = {
+}, oe = {
   label: "TreeMap",
   description: "Visualize hierarchical data with nested rectangles",
   useCase: "Best for showing part-to-whole relationships in hierarchical data, disk usage, budget allocation",
@@ -712,7 +712,7 @@ const B = {
     }
   ],
   clickableElements: { cell: !0 }
-}, J = {
+}, re = {
   label: "Data Table",
   description: "Display detailed tabular data",
   useCase: "Best for precise values, detailed analysis, sortable/filterable data exploration",
@@ -735,7 +735,7 @@ const B = {
       description: "Number formatting for numeric values"
     }
   ]
-}, Q = {
+}, ne = {
   label: "Activity Grid",
   description: "GitHub-style activity grid showing temporal patterns across different time scales",
   useCase: "Best for visualizing activity patterns over time. Supports hour (3hr blocks × days), day (days × weeks), week (weeks × months), month (months × quarters), and quarter (quarters × years) granularities",
@@ -780,7 +780,7 @@ const B = {
     } : { isValid: !0 };
   },
   clickableElements: { cell: !0 }
-}, ee = {
+}, le = {
   label: "KPI Number",
   description: "Display key performance indicators as large numbers",
   useCase: "Perfect for showing important metrics like revenue, user count, or other key business metrics in a prominent, easy-to-read format",
@@ -850,7 +850,7 @@ const B = {
     }
   ],
   displayOptions: ["hideHeader"]
-}, te = {
+}, de = {
   label: "KPI Delta",
   description: "Display change between latest and previous values with trend indicators",
   useCase: "Perfect for showing performance changes over time, such as revenue growth, user acquisition changes, or other metrics where the trend and delta are more important than the absolute value",
@@ -942,7 +942,7 @@ const B = {
     isValid: !1,
     message: "A measure is required for KPI Delta charts"
   } : { isValid: !0 }
-}, ae = {
+}, pe = {
   label: "KPI Text",
   description: "Display key performance indicators as customizable text",
   useCase: "Perfect for showing metrics with custom formatting, combining multiple values, or displaying contextual KPI information using templates",
@@ -984,7 +984,7 @@ const B = {
     }
   ],
   displayOptions: ["hideHeader"]
-}, ie = {
+}, ue = {
   label: "Markdown",
   description: "Display custom markdown content with formatting",
   useCase: "Perfect for adding documentation, notes, section headers, instructions, or formatted text to dashboards",
@@ -1076,7 +1076,7 @@ Use --- for horizontal rules.`,
       description: "Add an accent-colored border on one side of the content"
     }
   ]
-}, se = {
+}, ce = {
   label: "Funnel Chart",
   description: "Show conversion through sequential steps",
   useCase: "Best for visualizing user journey funnels, sales pipelines, or multi-step processes",
@@ -1163,7 +1163,7 @@ Use --- for horizontal rules.`,
       description: "Display 90th percentile time to convert"
     }
   ]
-}, oe = {
+}, me = {
   label: "Sankey Chart",
   description: "Show flow between states or steps",
   useCase: "Best for visualizing user journey flows, path analysis, or state transitions",
@@ -1219,7 +1219,7 @@ Use --- for horizontal rules.`,
       description: "Hide the statistics footer below the chart"
     }
   ]
-}, re = {
+}, ye = {
   label: "Sunburst Chart",
   description: "Show hierarchical flow as radial rings",
   useCase: "Best for visualizing forward paths from a starting event in a compact radial layout",
@@ -1264,7 +1264,7 @@ Use --- for horizontal rules.`,
       description: "Hide the statistics footer below the chart"
     }
   ]
-}, ne = {
+}, he = {
   label: "Heatmap",
   description: "Visualize intensity across two dimensions",
   useCase: "Best for showing patterns in matrix data like correlations, schedules, or category comparisons",
@@ -1336,7 +1336,7 @@ Use --- for horizontal rules.`,
     }
   ],
   validate: (o) => o.xAxis?.length ? o.yAxis?.length ? o.valueField?.length ? { isValid: !0 } : { isValid: !1, message: "Value measure required" } : { isValid: !1, message: "Y-axis dimension required" } : { isValid: !1, message: "X-axis dimension required" }
-}, le = {
+}, fe = {
   label: "Retention Matrix",
   // RetentionHeatmap auto-configures from the retention data structure
   // No drop zones needed as the chart maps directly to cohort × period matrix
@@ -1360,7 +1360,7 @@ Use --- for horizontal rules.`,
   ],
   description: "Cohort retention matrix visualization",
   useCase: "Visualize user retention over time by cohort"
-}, de = {
+}, be = {
   label: "Retention Chart",
   // RetentionCombinedChart auto-configures from the retention data structure
   // No drop zones needed as the chart maps directly to retention result data
@@ -1403,7 +1403,7 @@ Use --- for horizontal rules.`,
   ],
   description: "Combined retention visualization with line chart and heatmap modes",
   useCase: "Visualize user retention over time with optional breakdown segmentation"
-}, pe = {
+}, ge = {
   label: "Box Plot",
   description: "Show statistical distribution (median, IQR, whiskers) across categories",
   useCase: "Best for P&L spread per symbol, trade size distribution, latency distribution across platforms",
@@ -1436,7 +1436,7 @@ Use --- for horizontal rules.`,
       description: "Number formatting for the value axis"
     }
   ]
-}, ue = {
+}, xe = {
   label: "Waterfall Chart",
   description: "Show cumulative effect of sequential positive and negative values",
   useCase: "Best for P&L decomposition, cash flow analysis, budget variance, or any sequential contribution breakdown",
@@ -1491,7 +1491,7 @@ Use --- for horizontal rules.`,
       description: "Number formatting for the Y-axis"
     }
   ]
-}, ce = {
+}, we = {
   label: "Candlestick Chart",
   description: "Financial candlestick chart showing open/close body and high/low wicks",
   useCase: "Best for EOD quotes (bid/ask spread per date/symbol), markout distribution bands, or OHLC price data",
@@ -1556,7 +1556,7 @@ Use --- for horizontal rules.`,
       description: "Number formatting for the price axis"
     }
   ]
-}, me = {
+}, ke = {
   label: "Measure Profile",
   description: "Plot N measures as sequential X-axis points to visualise a profile or shape across intervals",
   useCase: "Best for markout interval analysis (e.g. avgMinus2m → avgAtEvent → avgPlus2h), metric profiles, or any pattern across ordered measures",
@@ -1621,7 +1621,7 @@ Use --- for horizontal rules.`,
       description: "Number formatting for the Y-axis"
     }
   ]
-}, ye = {
+}, ve = {
   label: "Gauge Chart",
   description: "Half-circle arc gauge for a single KPI value versus a maximum target",
   useCase: "Best for high-water marks vs equity, margin utilisation, or any single value progress toward a goal",
@@ -1680,36 +1680,36 @@ Use --- for horizontal rules.`,
       description: "Number formatting for the displayed value and axis labels"
     }
   ]
-}, q = {
-  bar: B,
-  line: $,
-  area: j,
-  pie: G,
-  scatter: K,
-  bubble: Z,
-  radar: U,
-  radialBar: W,
-  treemap: X,
-  table: J,
-  activityGrid: Q,
-  kpiNumber: ee,
-  kpiDelta: te,
-  kpiText: ae,
-  markdown: ie,
-  funnel: se,
-  sankey: oe,
-  sunburst: re,
-  heatmap: ne,
-  retentionHeatmap: le,
-  retentionCombined: de,
-  boxPlot: pe,
-  waterfall: ue,
-  candlestick: ce,
-  measureProfile: me,
-  gauge: ye
+}, P = {
+  bar: X,
+  line: J,
+  area: Q,
+  pie: ee,
+  scatter: te,
+  bubble: ae,
+  radar: se,
+  radialBar: ie,
+  treemap: oe,
+  table: re,
+  activityGrid: ne,
+  kpiNumber: le,
+  kpiDelta: de,
+  kpiText: pe,
+  markdown: ue,
+  funnel: ce,
+  sankey: me,
+  sunburst: ye,
+  heatmap: he,
+  retentionHeatmap: fe,
+  retentionCombined: be,
+  boxPlot: ge,
+  waterfall: xe,
+  candlestick: we,
+  measureProfile: ke,
+  gauge: ve
 };
-function he(o, t, a) {
-  const i = q[o];
+function Ce(o, t, a) {
+  const i = P[o];
   if (!i)
     return { isValid: !0, errors: [] };
   if (i.skipQuery)
@@ -1717,19 +1717,19 @@ function he(o, t, a) {
   const e = [];
   for (const s of i.dropZones) {
     if (!s.mandatory) continue;
-    const r = t?.[s.key];
-    if (!(Array.isArray(r) ? r.length > 0 : !!r)) {
-      const m = s.acceptTypes?.join("/") ?? "fields";
+    const l = t?.[s.key];
+    if (!(Array.isArray(l) ? l.length > 0 : !!l)) {
+      const u = s.acceptTypes?.join("/") ?? "fields";
       e.push(
-        `chartConfig.${s.key} is required for ${o} chart (${s.label}). Accepts: ${m}.`
+        `chartConfig.${s.key} is required for ${o} chart (${s.label}). Accepts: ${u}.`
       );
     }
   }
   if (o === "bar") {
     const s = t?.xAxis;
     if (!(Array.isArray(s) ? s.length > 0 : !!s)) {
-      const p = a.dimensions ?? [], m = a.timeDimensions ?? [];
-      p.length > 0 || m.length > 0 ? e.push(
+      const d = a.dimensions ?? [], u = a.timeDimensions ?? [];
+      d.length > 0 || u.length > 0 ? e.push(
         "chartConfig.xAxis is required for bar charts. Put a dimension in xAxis so bars have category labels."
       ) : e.push(
         'Bar charts need an xAxis dimension for category labels. Add a dimension to the query or use "table" chart type instead.'
@@ -1739,74 +1739,74 @@ function he(o, t, a) {
   if (t?.xAxis && t?.series) {
     const s = new Set(
       Array.isArray(t.xAxis) ? t.xAxis : [t.xAxis]
-    ), p = (Array.isArray(t.series) ? t.series : [t.series]).filter((m) => s.has(m));
-    p.length > 0 && e.push(
-      `chartConfig.series must not contain the same field as xAxis (found: ${p.join(", ")}). The series field is only for splitting into grouped/stacked sub-series by a DIFFERENT dimension. Remove the duplicate from series.`
+    ), d = (Array.isArray(t.series) ? t.series : [t.series]).filter((u) => s.has(u));
+    d.length > 0 && e.push(
+      `chartConfig.series must not contain the same field as xAxis (found: ${d.join(", ")}). The series field is only for splitting into grouped/stacked sub-series by a DIFFERENT dimension. Remove the duplicate from series.`
     );
   }
   return { isValid: e.length === 0, errors: e };
 }
-function fe(o, t, a) {
-  const i = q[o];
+function Te(o, t, a) {
+  const i = P[o];
   if (!i)
     return t ?? {};
-  const e = { ...t }, s = a.measures ?? [], r = a.dimensions ?? [], m = (a.timeDimensions ?? []).map((c) => c.dimension);
-  for (const c of i.dropZones) {
-    const y = e[c.key];
-    if (Array.isArray(y) ? y.length > 0 : !!y) continue;
-    const g = c.acceptTypes ?? [];
-    if (c.key === "sizeField" || c.key === "colorField") {
-      if (g.includes("measure")) {
-        const h = /* @__PURE__ */ new Set();
-        for (const x of i.dropZones) {
-          if (x.key === c.key) continue;
-          const u = e[x.key];
-          Array.isArray(u) ? u.forEach((A) => h.add(A)) : typeof u == "string" && h.add(u);
+  const e = { ...t }, s = a.measures ?? [], l = a.dimensions ?? [], u = (a.timeDimensions ?? []).map((p) => p.dimension);
+  for (const p of i.dropZones) {
+    const m = e[p.key];
+    if (Array.isArray(m) ? m.length > 0 : !!m) continue;
+    const b = p.acceptTypes ?? [];
+    if (p.key === "sizeField" || p.key === "colorField") {
+      if (b.includes("measure")) {
+        const g = /* @__PURE__ */ new Set();
+        for (const S of i.dropZones) {
+          if (S.key === p.key) continue;
+          const x = e[S.key];
+          Array.isArray(x) ? x.forEach((V) => g.add(V)) : typeof x == "string" && g.add(x);
         }
-        const d = s.filter((x) => !h.has(x));
-        d.length > 0 && (e[c.key] = d[0]);
+        const C = s.filter((S) => !g.has(S));
+        C.length > 0 && (e[p.key] = C[0]);
       }
       continue;
     }
-    const f = [];
-    if (g.includes("dimension") && f.push(...r), g.includes("timeDimension") && f.push(...m), g.includes("measure") && f.push(...s), f.length === 0) continue;
-    let b = f;
-    if (c.key === "series") {
-      const h = new Set(
+    const y = [];
+    if (b.includes("dimension") && y.push(...l), b.includes("timeDimension") && y.push(...u), b.includes("measure") && y.push(...s), y.length === 0) continue;
+    let f = y;
+    if (p.key === "series") {
+      const g = new Set(
         Array.isArray(e.xAxis) ? e.xAxis : e.xAxis ? [e.xAxis] : []
       );
-      if (b = f.filter((d) => !h.has(d)), b.length === 0) continue;
+      if (f = y.filter((C) => !g.has(C)), f.length === 0) continue;
     }
-    const w = c.maxItems ?? 1 / 0, C = b.slice(0, w);
-    C.length > 0 && (e[c.key] = C);
+    const v = p.maxItems ?? 1 / 0, D = f.slice(0, v);
+    D.length > 0 && (e[p.key] = D);
   }
   return e;
 }
-function be(o) {
+function Ae(o) {
   const t = [`
 Chart config requirements by type:`];
   for (const a of o) {
-    const i = q[a];
+    const i = P[a];
     if (!i) continue;
-    const e = i.description ?? "", s = i.useCase ?? "", r = [e, s].filter(Boolean).join(". "), p = r ? ` — ${r}.` : "", m = i.dropZones.filter((y) => y.mandatory);
-    if (m.length === 0 && !i.skipQuery) {
-      t.push(`  ${a}${p} chartConfig auto-inferred from query.`);
+    const e = i.description ?? "", s = i.useCase ?? "", l = [e, s].filter(Boolean).join(". "), d = l ? ` — ${l}.` : "", u = i.dropZones.filter((m) => m.mandatory);
+    if (u.length === 0 && !i.skipQuery) {
+      t.push(`  ${a}${d} chartConfig auto-inferred from query.`);
       continue;
     }
     if (i.skipQuery) {
-      t.push(`  ${a}${p} No query needed.`);
+      t.push(`  ${a}${d} No query needed.`);
       continue;
     }
-    const c = m.map((y) => {
-      const l = y.acceptTypes?.join("/") ?? "any", g = y.maxItems ? ` (max ${y.maxItems})` : "";
-      return `${y.key}=[${l}]${g}`;
+    const p = u.map((m) => {
+      const n = m.acceptTypes?.join("/") ?? "any", b = m.maxItems ? ` (max ${m.maxItems})` : "";
+      return `${m.key}=[${n}]${b}`;
     });
-    t.push(`  ${a}${p} Requires ${c.join(", ")}.`);
+    t.push(`  ${a}${d} Requires ${p.join(", ")}.`);
   }
   return t.join(`
 `);
 }
-const _ = [
+const N = [
   "bar",
   "line",
   "area",
@@ -1826,7 +1826,7 @@ const _ = [
   "boxPlot",
   "markdown"
 ];
-function ge() {
+function De() {
   return [
     // Tool 1: discover_cubes
     {
@@ -1860,13 +1860,13 @@ function ge() {
         properties: {
           measures: {
             type: "array",
-            items: { type: "string" },
-            description: 'Aggregation measures (e.g., ["Employees.count"])'
+            items: { type: "string", pattern: "^[A-Z][a-zA-Z0-9]*\\.[a-zA-Z][a-zA-Z0-9]*$" },
+            description: 'Aggregation measures — MUST be "CubeName.measureName" format (e.g., ["PullRequests.count", "Issues.openCount"]). NEVER use just the cube name.'
           },
           dimensions: {
             type: "array",
-            items: { type: "string" },
-            description: 'Grouping dimensions (e.g., ["Employees.name"])'
+            items: { type: "string", pattern: "^[A-Z][a-zA-Z0-9]*\\.[a-zA-Z][a-zA-Z0-9]*$" },
+            description: 'Grouping dimensions — MUST be "CubeName.dimensionName" format (e.g., ["Teams.name", "Employees.department"]). NEVER use just the cube name.'
           },
           filters: {
             type: "array",
@@ -1978,7 +1978,7 @@ function ge() {
     {
       name: "add_portlet",
       description: `Add a chart visualization to the notebook.
-` + be(_) + `
+` + Ae(N) + `
 The query is validated before adding. The portlet fetches its own data.`,
       input_schema: {
         type: "object",
@@ -1990,7 +1990,7 @@ The query is validated before adding. The portlet fetches its own data.`,
           },
           chartType: {
             type: "string",
-            enum: _,
+            enum: N,
             description: "Chart type to render"
           },
           chartConfig: {
@@ -2050,7 +2050,7 @@ The query is validated before adding. The portlet fetches its own data.`,
                 title: { type: "string", description: "Portlet title" },
                 chartType: {
                   type: "string",
-                  enum: _,
+                  enum: N,
                   description: 'Chart type. Use "markdown" for section headers.'
                 },
                 query: {
@@ -2121,10 +2121,10 @@ The query is validated before adding. The portlet fetches its own data.`,
     }
   ];
 }
-function xe(o) {
+function Se(o) {
   const { semanticLayer: t, securityContext: a } = o, i = /* @__PURE__ */ new Map();
   return i.set("discover_cubes", async (e) => {
-    const s = await Y(t, {
+    const s = await Z(t, {
       topic: e.topic,
       intent: e.intent,
       limit: e.limit,
@@ -2136,8 +2136,17 @@ function xe(o) {
     return { result: JSON.stringify(e, null, 2) };
   }), i.set("execute_query", async (e) => {
     try {
-      let s;
-      e.funnel ? s = { funnel: e.funnel } : e.flow ? s = { flow: e.flow } : e.retention ? s = { retention: e.retention } : s = {
+      const s = (u, p) => {
+        if (!Array.isArray(u)) return;
+        const m = u.filter((n) => typeof n == "string" && !n.includes("."));
+        if (m.length > 0)
+          throw new Error(
+            `Invalid ${p}: ${m.map((n) => `"${n}"`).join(", ")}. All ${p} must use "CubeName.fieldName" format (e.g., "PullRequests.count", "Teams.name"). Use discover_cubes or get_cube_metadata to find valid field names.`
+          );
+      };
+      s(e.measures, "measures"), s(e.dimensions, "dimensions");
+      let l;
+      e.funnel ? l = { funnel: e.funnel } : e.flow ? l = { flow: e.flow } : e.retention ? l = { retention: e.retention } : l = {
         measures: e.measures,
         dimensions: e.dimensions,
         filters: e.filters,
@@ -2145,12 +2154,12 @@ function xe(o) {
         order: e.order,
         limit: e.limit
       };
-      const r = await M(t, a, { query: s });
+      const d = await G(t, a, { query: l });
       return {
         result: JSON.stringify({
-          rowCount: r.data.length,
-          data: r.data,
-          annotation: r.annotation
+          rowCount: d.data.length,
+          data: d.data,
+          annotation: d.annotation
         }, null, 2)
       };
     } catch (s) {
@@ -2160,138 +2169,138 @@ function xe(o) {
       };
     }
   }), i.set("add_portlet", async (e) => {
-    const r = {
+    const l = {
       number: "kpiNumber",
       retention: "retentionHeatmap"
     }[e.chartType] ?? e.chartType;
-    let p;
+    let d;
     try {
-      p = JSON.parse(e.query);
+      d = JSON.parse(e.query);
     } catch {
       return {
         result: "Invalid query: could not parse JSON string. Ensure `query` is a valid JSON string.",
         isError: !0
       };
     }
-    const m = t.validateQuery(p);
-    if (!m.isValid)
+    const u = t.validateQuery(d);
+    if (!u.isValid)
       return {
         result: `Invalid query — fix these errors and retry:
-${m.errors.join(`
+${u.errors.join(`
 `)}`,
         isError: !0
       };
-    const c = !!(p.funnel || p.flow || p.retention);
-    let y;
-    if (c)
-      y = e.chartConfig ?? {};
+    const p = !!(d.funnel || d.flow || d.retention);
+    let m;
+    if (p)
+      m = e.chartConfig ?? {};
     else {
-      const f = fe(r, e.chartConfig, p), b = he(r, f, p);
-      if (!b.isValid)
+      const y = Te(l, e.chartConfig, d), f = Ce(l, y, d);
+      if (!f.isValid)
         return {
           result: `Chart config invalid — fix these errors and retry:
-${b.errors.join(`
+${f.errors.join(`
 `)}`,
           isError: !0
         };
-      y = f;
+      m = y;
     }
-    const l = `portlet-${Date.now()}-${Math.random().toString(36).slice(2, 7)}`, g = {
-      id: l,
+    const n = `portlet-${Date.now()}-${Math.random().toString(36).slice(2, 7)}`, b = {
+      id: n,
       title: e.title,
       query: e.query,
-      chartType: r,
-      chartConfig: y,
+      chartType: l,
+      chartConfig: m,
       displayConfig: e.displayConfig
     };
     return {
-      result: `Portlet "${e.title}" added to notebook (id: ${l}, chart: ${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_portlet", data: g }
+      result: `Portlet "${e.title}" added to notebook (id: ${n}, chart: ${l}). [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 }
     };
   }), i.set("add_markdown", async (e) => {
-    const s = `markdown-${Date.now()}-${Math.random().toString(36).slice(2, 7)}`, r = {
+    const s = `markdown-${Date.now()}-${Math.random().toString(36).slice(2, 7)}`, l = {
       id: s,
       title: e.title,
       content: e.content
     };
     return {
       result: `Markdown block added to notebook (id: ${s}). [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: r }
+      sideEffect: { type: "add_markdown", data: l }
     };
   }), i.set("save_as_dashboard", async (e) => {
     try {
       const s = e.portlets;
       if (!s || s.length === 0)
         return { result: "Dashboard must contain at least one portlet.", isError: !0 };
-      const r = [];
-      for (const l of s) {
-        if (l.chartType === "markdown") continue;
-        const f = l.query;
-        if (!f) {
-          r.push(`Portlet "${l.title}": missing query`);
+      const l = [];
+      for (const n of s) {
+        if (n.chartType === "markdown") continue;
+        const y = n.query;
+        if (!y) {
+          l.push(`Portlet "${n.title}": missing query`);
           continue;
         }
-        let b;
+        let f;
         try {
-          b = JSON.parse(f);
+          f = JSON.parse(y);
         } catch {
-          r.push(`Portlet "${l.title}": invalid JSON query`);
+          l.push(`Portlet "${n.title}": invalid JSON query`);
           continue;
         }
-        const w = t.validateQuery(b);
-        w.isValid || r.push(`Portlet "${l.title}": ${w.errors.join(", ")}`);
+        const v = t.validateQuery(f);
+        v.isValid || l.push(`Portlet "${n.title}": ${v.errors.join(", ")}`);
       }
-      if (r.length > 0)
+      if (l.length > 0)
         return {
           result: `Dashboard has invalid portlets — fix these errors and retry:
-${r.join(`
+${l.join(`
 `)}`,
           isError: !0
         };
-      const p = {
-        portlets: s.map((l) => {
-          const g = l.chartType, f = g === "markdown", b = f ? "query" : l.analysisType || "query", w = b === "funnel" ? "funnel" : b === "flow" ? "flow" : b === "retention" ? "retention" : "query", C = l.query || "{}";
-          let h;
+      const d = {
+        portlets: s.map((n) => {
+          const b = n.chartType, y = b === "markdown", f = y ? "query" : n.analysisType || "query", v = f === "funnel" ? "funnel" : f === "flow" ? "flow" : f === "retention" ? "retention" : "query", D = n.query || "{}";
+          let g;
           try {
-            h = JSON.parse(C);
+            g = JSON.parse(D);
           } catch {
-            h = {};
+            g = {};
           }
-          const d = {
+          const C = {
             version: 1,
-            analysisType: w,
+            analysisType: v,
             activeView: "chart",
             charts: {
-              [w]: {
-                chartType: g,
-                chartConfig: l.chartConfig || {},
-                displayConfig: l.displayConfig || {}
+              [v]: {
+                chartType: b,
+                chartConfig: n.chartConfig || {},
+                displayConfig: n.displayConfig || {}
               }
             },
-            query: f ? {} : h
+            query: y ? {} : g
           };
           return {
-            id: l.id,
-            title: l.title,
-            analysisConfig: d,
-            dashboardFilterMapping: l.dashboardFilterMapping,
-            w: l.w,
-            h: l.h,
-            x: l.x,
-            y: l.y
+            id: n.id,
+            title: n.title,
+            analysisConfig: C,
+            dashboardFilterMapping: n.dashboardFilterMapping,
+            w: n.w,
+            h: n.h,
+            x: n.x,
+            y: n.y
           };
         }),
         filters: e.filters,
         colorPalette: e.colorPalette
-      }, m = e.title, c = p.portlets.length, y = p.filters?.length || 0;
+      }, u = e.title, p = d.portlets.length, m = d.filters?.length || 0;
       return {
-        result: `Dashboard "${m}" created with ${c} portlets and ${y} filters.`,
+        result: `Dashboard "${u}" created with ${p} portlets and ${m} filters.`,
         sideEffect: {
           type: "dashboard_saved",
           data: {
-            title: m,
+            title: u,
             description: e.description,
-            dashboardConfig: p
+            dashboardConfig: d
           }
         }
       };
@@ -2303,15 +2312,15 @@ ${r.join(`
     }
   }), i;
 }
-async function* Ce(o) {
-  const { message: t, sessionId: a, history: i, semanticLayer: e, securityContext: s, agentConfig: r, apiKey: p } = o;
-  let m;
+async function* Ee(o) {
+  const { message: t, history: a, semanticLayer: i, securityContext: e, agentConfig: s, apiKey: l } = o, d = o.sessionId || crypto.randomUUID(), u = s.observability, p = crypto.randomUUID(), m = Date.now();
+  let n;
   try {
-    const d = await import(
+    const c = await import(
       /* webpackIgnore: true */
       "@anthropic-ai/sdk"
     );
-    m = d.default || d.Anthropic || d;
+    n = c.default || c.Anthropic || c;
   } catch {
     yield {
       type: "error",
@@ -2321,140 +2330,213 @@ async function* Ce(o) {
     };
     return;
   }
-  const c = new m({ apiKey: p }), y = ge(), l = xe({ semanticLayer: e, securityContext: s }), g = e.getMetadata(), f = H(g), b = r.model || "claude-sonnet-4-6", w = r.maxTurns || 25, C = r.maxTokens || 4096, h = [];
-  if (i && i.length > 0) {
-    for (const d of i)
-      if (d.role === "user")
-        h.push({ role: "user", content: d.content });
-      else if (d.role === "assistant") {
-        const x = [];
-        if (d.content && x.push({ type: "text", text: d.content }), d.toolCalls && d.toolCalls.length > 0) {
-          for (const u of d.toolCalls)
-            x.push({
+  const b = new n({ apiKey: l }), y = De(), f = Se({ semanticLayer: i, securityContext: e }), v = i.getMetadata(), D = W(v), g = s.model || "claude-sonnet-4-6", C = s.maxTurns || 25, S = s.maxTokens || 4096;
+  try {
+    u?.onChatStart?.({
+      traceId: p,
+      sessionId: d,
+      message: t,
+      model: g,
+      historyLength: a?.length ?? 0
+    });
+  } catch {
+  }
+  const x = [];
+  if (a && a.length > 0) {
+    for (const c of a)
+      if (c.role === "user")
+        x.push({ role: "user", content: c.content });
+      else if (c.role === "assistant") {
+        const F = [];
+        if (c.content && F.push({ type: "text", text: c.content }), c.toolCalls && c.toolCalls.length > 0) {
+          for (const h of c.toolCalls)
+            F.push({
               type: "tool_use",
-              id: u.id,
-              name: u.name,
-              input: u.input || {}
+              id: h.id,
+              name: h.name,
+              input: h.input || {}
             });
-          h.push({ role: "assistant", content: x }), h.push({
+          x.push({ role: "assistant", content: F }), x.push({
             role: "user",
-            content: d.toolCalls.map((u) => ({
+            content: c.toolCalls.map((h) => ({
               type: "tool_result",
-              tool_use_id: u.id,
-              content: typeof u.result == "string" ? u.result : JSON.stringify(u.result ?? ""),
-              ...u.status === "error" ? { is_error: !0 } : {}
+              tool_use_id: h.id,
+              content: typeof h.result == "string" ? h.result : JSON.stringify(h.result ?? ""),
+              ...h.status === "error" ? { is_error: !0 } : {}
             }))
           });
-        } else x.length > 0 && h.push({ role: "assistant", content: d.content });
+        } else F.length > 0 && x.push({ role: "assistant", content: c.content });
       }
   }
-  h.push({ role: "user", content: t });
+  x.push({ role: "user", content: t });
+  let V = 0;
   try {
-    for (let d = 0; d < w; d++) {
-      const x = await c.messages.create({
-        model: b,
-        max_tokens: C,
-        system: f,
+    for (let c = 0; c < C; c++) {
+      V = c + 1;
+      const F = await b.messages.create({
+        model: g,
+        max_tokens: S,
+        system: D,
         tools: y,
-        messages: h,
+        messages: x,
         stream: !0
-      }), u = [];
-      let A = -1, D = "", E = "";
-      for await (const k of x)
-        switch (k.type) {
+      }), h = [];
+      let q = -1, _ = "", O = "", L, R;
+      const H = Date.now();
+      for await (const w of F)
+        switch (w.type) {
           case "content_block_start": {
-            A++;
-            const n = k.content_block;
-            n.type === "tool_use" ? (u.push({ type: "tool_use", id: n.id, name: n.name, input: {} }), D = "", yield {
+            q++;
+            const r = w.content_block;
+            r.type === "tool_use" ? (h.push({ type: "tool_use", id: r.id, name: r.name, input: {} }), _ = "", yield {
               type: "tool_use_start",
-              data: { id: n.id, name: n.name, input: void 0 }
-            }) : n.type === "text" && u.push({ type: "text", text: "" });
+              data: { id: r.id, name: r.name, input: void 0 }
+            }) : r.type === "text" && h.push({ type: "text", text: "" });
             break;
           }
           case "content_block_delta": {
-            const n = k.delta;
-            if (n.type === "text_delta" && n.text) {
-              const S = u[A];
-              S && (S.text = (S.text || "") + n.text), yield { type: "text_delta", data: n.text };
-            } else n.type === "input_json_delta" && n.partial_json && (D += n.partial_json);
+            const r = w.delta;
+            if (r.type === "text_delta" && r.text) {
+              const T = h[q];
+              T && (T.text = (T.text || "") + r.text), yield { type: "text_delta", data: r.text };
+            } else r.type === "input_json_delta" && r.partial_json && (_ += r.partial_json);
             break;
           }
           case "content_block_stop": {
-            const n = u[A];
-            if (n?.type === "tool_use" && D) {
+            const r = h[q];
+            if (r?.type === "tool_use" && _) {
               try {
-                n.input = JSON.parse(D);
+                r.input = JSON.parse(_);
               } catch {
-                n.input = {};
+                r.input = {};
               }
-              D = "";
+              _ = "";
             }
             break;
           }
+          case "message_start": {
+            const r = w.message;
+            r?.usage?.input_tokens != null && (L = r.usage.input_tokens);
+            break;
+          }
           case "message_delta": {
-            const n = k.delta;
-            n.stop_reason && (E = n.stop_reason);
+            const r = w.delta, T = w.usage;
+            T?.output_tokens != null && (R = T.output_tokens), r.stop_reason && (O = r.stop_reason);
             break;
           }
         }
-      if (h.push({ role: "assistant", content: u }), E !== "tool_use")
+      try {
+        u?.onGenerationEnd?.({
+          traceId: p,
+          turn: c,
+          model: g,
+          stopReason: O,
+          inputTokens: L,
+          outputTokens: R,
+          durationMs: Date.now() - H
+        });
+      } catch {
+      }
+      if (x.push({ role: "assistant", content: h }), O !== "tool_use")
         break;
-      const F = [];
-      for (const k of u) {
-        if (k.type !== "tool_use") continue;
-        const n = k.name, S = k.input || {}, T = k.id, I = l.get(n);
-        if (!I) {
-          F.push({
+      const E = [];
+      for (const w of h) {
+        if (w.type !== "tool_use") continue;
+        const r = w.name, T = w.input || {}, A = w.id, M = f.get(r);
+        if (!M) {
+          E.push({
             type: "tool_result",
-            tool_use_id: T,
-            content: `Unknown tool: ${n}`,
+            tool_use_id: A,
+            content: `Unknown tool: ${r}`,
             is_error: !0
           }), yield {
             type: "tool_use_result",
-            data: { id: T, name: n, result: `Unknown tool: ${n}`, isError: !0 }
+            data: { id: A, name: r, result: `Unknown tool: ${r}`, isError: !0 }
           };
           continue;
         }
+        const Y = Date.now();
         try {
-          const v = await I(S);
-          v.sideEffect && (yield v.sideEffect), F.push({
+          const k = await M(T);
+          k.sideEffect && (yield k.sideEffect), E.push({
             type: "tool_result",
-            tool_use_id: T,
-            content: v.result,
-            ...v.isError ? { is_error: !0 } : {}
+            tool_use_id: A,
+            content: k.result,
+            ...k.isError ? { is_error: !0 } : {}
           }), yield {
             type: "tool_use_result",
-            data: { id: T, name: n, result: v.result, ...v.isError ? { isError: !0 } : {} }
+            data: { id: A, name: r, result: k.result, ...k.isError ? { isError: !0 } : {} }
           };
-        } catch (v) {
-          const O = v instanceof Error ? v.message : "Tool execution failed";
-          F.push({
+          try {
+            u?.onToolEnd?.({
+              traceId: p,
+              turn: c,
+              toolName: r,
+              toolUseId: A,
+              isError: !!k.isError,
+              durationMs: Date.now() - Y
+            });
+          } catch {
+          }
+        } catch (k) {
+          const z = k instanceof Error ? k.message : "Tool execution failed";
+          E.push({
             type: "tool_result",
-            tool_use_id: T,
-            content: O,
+            tool_use_id: A,
+            content: z,
             is_error: !0
           }), yield {
             type: "tool_use_result",
-            data: { id: T, name: n, result: O, isError: !0 }
+            data: { id: A, name: r, result: z, isError: !0 }
           };
+          try {
+            u?.onToolEnd?.({
+              traceId: p,
+              turn: c,
+              toolName: r,
+              toolUseId: A,
+              isError: !0,
+              durationMs: Date.now() - Y
+            });
+          } catch {
+          }
         }
       }
-      yield { type: "turn_complete", data: {} }, h.push({ role: "user", content: F });
+      yield { type: "turn_complete", data: {} }, x.push({ role: "user", content: E });
+    }
+    try {
+      u?.onChatEnd?.({
+        traceId: p,
+        sessionId: d,
+        totalTurns: V,
+        durationMs: Date.now() - m
+      });
+    } catch {
     }
     yield {
       type: "done",
-      data: { sessionId: a || "" }
+      data: { sessionId: d || "", traceId: p }
     };
-  } catch (d) {
+  } catch (c) {
+    try {
+      u?.onChatEnd?.({
+        traceId: p,
+        sessionId: d,
+        totalTurns: 0,
+        durationMs: Date.now() - m,
+        error: c instanceof Error ? c.message : "Unknown error"
+      });
+    } catch {
+    }
     yield {
       type: "error",
       data: {
-        message: we(d)
+        message: Fe(c)
       }
     };
   }
 }
-function we(o) {
+function Fe(o) {
   if (!o || !(o instanceof Error))
     return "Something went wrong. Please try again.";
   const t = o.message || "", a = {
@@ -2481,5 +2563,5 @@ function we(o) {
   return t;
 }
 export {
-  Ce as handleAgentChat
+  Ee as handleAgentChat
 };