@arivie/mcp 0.1.0 → 1.0.1

package/CHANGELOG.md

@@ -1,3 +1,46 @@
+## 1.0.1
+
+### Patch Changes
+
+- @arivie/agent@0.2.1
+- @arivie/db-postgres@0.1.2
+
+## 1.0.0
+
+### Patch Changes
+
+- Arivie 20 June 2026 release
+
+  Mastra 1.45 upgrade, Agent Skills spec, and schedule configuration:
+
+  - Upgrade all `@mastra/*` dependencies to 1.45.0 / 1.21.0 / 1.11.0 / 1.14.0.
+  - Migrate every `SKILL.md` file to the official Agent Skills spec format.
+  - Add `defineSchedule` / `defineSchedules` configuration and wire schedules to Mastra Workflows with cron triggers.
+  - Add `arivie add schedule <name>` CLI command.
+  - Forward an optional Mastra `observability` instance through `ArivieConfig` to the Mastra runtime.
+  - Remove compatibility glue (`asMastraMcpServer`, `LooseGen`, `applyMaxStepsDefault`) in favor of native Mastra APIs.
+
+  Tool approval and HITL policy configuration:
+
+  - Add `ToolApprovalPolicy` type to `ArivieConfig.limits` with support for global, allow-list, deny-list, and function-based policies.
+  - Wire tool approval through to Mastra's native `requireToolApproval` / tool-suspension flow.
+  - Export `normalizeRequireToolApproval` from `@arivie/agent` for reusable policy normalization.
+
+  Dogfood eval harness using Mastra `runEvals` and PGlite:
+
+  - Add `@arivie/core/eval` subpath exporting SQL-semantic scorer helpers and a composite dogfood scorer.
+  - Migrate `scripts/run-eval.ts` from the legacy runner to Mastra `runEvals` with a composite scorer.
+  - Add PGlite-backed database adapter and adapter selection so `pnpm eval` works without Docker by default.
+  - Keep testcontainers path available via `USE_TESTCONTAINERS=1`.
+  - Use Mastra `RequestContext` for probe metadata in `runEvals` and register the scorer with Mastra to suppress warnings.
+  - Remove the broken `customers` join from the dogfood `orders` semantic entity.
+  - Upgrade `@electric-sql/pglite` to 0.5.3 for socket-server compatibility.
+
+- Updated dependencies
+  - @arivie/agent@0.2.0
+  - @arivie/ui-catalog@0.1.1
+  - @arivie/db-postgres@0.1.1
+
 ## 0.0.0
 
 - Initial `makeMcpServer` (Sprint 3 / S3-01): Mastra `MCPServer` with `ask`, `query`, `schema`, `memory` tools and `ask_arivie` agent bridge (REQ-26, RFC §4.7).

package/dist/index.d.ts

@@ -2,6 +2,9 @@ import { MCPServer } from '@mastra/mcp';
 import { Agent } from '@mastra/core/agent';
 import { PostgresAdapter } from '@arivie/db-postgres';
 import { SemanticLayer } from '@arivie/semantic';
+import { createMcpApp } from '@json-render/mcp';
+export { registerJsonRenderTool } from '@json-render/mcp';
+export { arivieUiCatalog } from '@arivie/ui-catalog';
 
 interface McpOptions {
     readonly agent: Agent;
@@ -14,4 +17,57 @@ interface McpOptions {
 
 declare function makeMcpServer(opts: McpOptions): MCPServer;
 
-export { type McpOptions, makeMcpServer };
+type CreateMcpAppReturn = Awaited<ReturnType<typeof createMcpApp>>;
+
+interface MakeMcpUiServerOptions extends McpOptions {
+    /**
+     * Override the default HTML shell. When omitted, a CDN-loaded shell is
+     * served that renders the Arivie catalog. Pass a pre-built bundle here
+     * for production deployments (no runtime esm.sh fetch, brand styling).
+     */
+    html?: string;
+}
+/**
+ * Build an MCP server that returns **renderable UI** to MCP UI-aware
+ * clients (Claude Desktop, Cursor, ChatGPT). Pairs Arivie's analytics
+ * tools (`ask`, `query`, `schema`) with `@json-render/mcp`'s catalog-
+ * driven render path.
+ *
+ * Wire shape:
+ *   1. `createMcpApp({ catalog, html })` registers the canonical
+ *      `render-ui` tool + the UI resource at `ui://render-ui/view.html`.
+ *      MCP clients use this when the agent emits a json-render spec.
+ *   2. We layer Arivie's `ask` / `query` / `schema` tools on top of the
+ *      same server. Their results are JSON-shaped (rows + SQL + metadata)
+ *      so the client can either render them as plain JSON OR pass them
+ *      back through `render-ui` with a spec.
+ *
+ * Catalog: 36 shadcn components + 4 Arivie-specific
+ * (`ArivieMetric`, `ArivieQueryResult`, `ArivieVerdict`,
+ * `ArivieSemanticEntity`).
+ *
+ * Smaller models (Gemini Flash, Grok) sometimes emit malformed specs;
+ * the json-render tool validates against the catalog's Zod schema and
+ * returns the partial spec rather than crashing.
+ */
+declare function makeMcpUiServer(opts: MakeMcpUiServerOptions): Promise<CreateMcpAppReturn>;
+
+/**
+ * Minimal self-contained HTML shell served as the json-render UI
+ * resource (`ui://render-ui/view.html`). MCP UI clients (Claude Desktop,
+ * Cursor, ChatGPT) open this in an iframe; the parent posts the tool
+ * result containing a json-render spec via `ontoolresult`, and this
+ * shell renders the spec.
+ *
+ * Implementation note: we load React + json-render packages from esm.sh
+ * at runtime instead of bundling them. This keeps `@arivie/mcp` from
+ * shipping a 200KB+ HTML blob and means the shell auto-tracks
+ * @json-render/* versions we install. The tradeoff is the shell does a
+ * network fetch on first render; for a dev/spike server this is fine.
+ *
+ * Customise this template (or replace via `makeMcpUiServer({ html })`)
+ * to ship a hand-tuned production shell with bundled assets + brand styling.
+ */
+declare const DEFAULT_UI_SHELL_HTML = "<!doctype html>\n<html lang=\"en\">\n  <head>\n    <meta charset=\"utf-8\" />\n    <meta name=\"viewport\" content=\"width=device-width, initial-scale=1\" />\n    <title>Arivie</title>\n    <style>\n      :root {\n        --bg: #0b1115;\n        --fg: #e7eef3;\n        --muted: #8aa0ad;\n        --accent: #0d9488;\n        --warn: #fb923c;\n      }\n      @media (prefers-color-scheme: light) {\n        :root {\n          --bg: #ffffff;\n          --fg: #0b1115;\n          --muted: #5b6770;\n          --accent: #0d9488;\n          --warn: #c2570a;\n        }\n      }\n      html, body, #root { height: 100%; margin: 0; }\n      body {\n        background: var(--bg);\n        color: var(--fg);\n        font-family: ui-sans-serif, -apple-system, BlinkMacSystemFont, \"Segoe UI\",\n                     Roboto, \"Helvetica Neue\", Arial, sans-serif;\n        line-height: 1.5;\n      }\n      .empty { padding: 24px; color: var(--muted); font-size: 14px; }\n      .err { padding: 16px; color: var(--warn); font-family: ui-monospace, monospace;\n             font-size: 12px; white-space: pre-wrap; }\n    </style>\n  </head>\n  <body>\n    <div id=\"root\">\n      <div class=\"empty\">Waiting for spec from the Arivie agent\u2026</div>\n    </div>\n    <script type=\"module\">\n      // Pulled at runtime \u2014 these versions match what @arivie/mcp depends on.\n      import * as React from \"https://esm.sh/react@19.1?bundle\";\n      import * as ReactDOM from \"https://esm.sh/react-dom@19.1/client?bundle\";\n      import { defineRegistry, Renderer } from \"https://esm.sh/@json-render/react@0.19?bundle\";\n      import { shadcnComponentDefinitions } from \"https://esm.sh/@json-render/shadcn@0.19/catalog?bundle\";\n      import { defineCatalog } from \"https://esm.sh/@json-render/core@0.19?bundle\";\n      import { schema } from \"https://esm.sh/@json-render/react@0.19/schema?bundle\";\n\n      // A bare-minimum mirror of the server-side Arivie catalog. We can't\n      // import the server catalog directly (this runs in the browser), so\n      // we re-declare the components the renderer needs. The Zod schema is\n      // only used for VALIDATION on the server; the renderer just looks up\n      // component names \u2192 React components in the registry.\n      const catalog = defineCatalog(schema, {\n        components: {\n          ...shadcnComponentDefinitions,\n          ArivieMetric: { props: schema.object({}), description: \"\" },\n          ArivieQueryResult: { props: schema.object({}), description: \"\" },\n          ArivieVerdict: { props: schema.object({}), description: \"\" },\n          ArivieSemanticEntity: { props: schema.object({}), description: \"\" },\n        },\n        actions: {},\n      });\n\n      const fmt = (v, format) => {\n        if (v == null) return \"\u2014\";\n        if (format === \"currency\") {\n          const n = Number(v);\n          if (!Number.isFinite(n)) return String(v);\n          return n.toLocaleString(undefined, { style: \"currency\", currency: \"USD\" });\n        }\n        if (format === \"percent\") return `${v}%`;\n        return String(v);\n      };\n\n      const { registry } = defineRegistry(catalog, {\n        components: {\n          ArivieMetric: ({ props }) =>\n            React.createElement(\"div\", {\n              style: {\n                padding: \"16px\", border: \"1px solid #2a3942\", borderRadius: 12,\n                display: \"inline-block\", minWidth: 220, margin: 8,\n              },\n            }, [\n              React.createElement(\"div\", { key: \"l\", style: { color: \"var(--muted)\", fontSize: 13 } }, props.label),\n              React.createElement(\"div\", { key: \"v\", style: { fontSize: 28, fontWeight: 600, marginTop: 4 } },\n                fmt(props.value, props.format)),\n              props.delta ? React.createElement(\"div\", { key: \"d\",\n                style: { fontSize: 13, marginTop: 4,\n                  color: props.deltaDirection === \"down\" ? \"var(--warn)\" : \"var(--accent)\" }\n              }, `${props.deltaDirection === \"down\" ? \"\u25BC\" : \"\u25B2\"} ${props.delta}`) : null,\n            ]),\n          ArivieQueryResult: ({ props }) =>\n            React.createElement(\"div\", { style: { margin: 12 } }, [\n              React.createElement(\"pre\", {\n                key: \"sql\",\n                style: {\n                  background: \"#0e1820\", color: \"#cde8ff\", padding: 12, borderRadius: 8,\n                  fontSize: 12, overflow: \"auto\", maxHeight: 240,\n                },\n              }, props.sql),\n              React.createElement(\"table\", {\n                key: \"t\",\n                style: { width: \"100%\", borderCollapse: \"collapse\", marginTop: 8, fontSize: 13 },\n              }, [\n                React.createElement(\"thead\", { key: \"h\" },\n                  React.createElement(\"tr\", null, Object.keys(props.rows?.[0] ?? {}).map(k =>\n                    React.createElement(\"th\", {\n                      key: k,\n                      style: { textAlign: \"left\", padding: \"6px 10px\", borderBottom: \"1px solid #2a3942\" },\n                    }, k)))),\n                React.createElement(\"tbody\", { key: \"b\" },\n                  (props.rows ?? []).map((row, i) =>\n                    React.createElement(\"tr\", { key: i },\n                      Object.values(row).map((v, j) =>\n                        React.createElement(\"td\", {\n                          key: j,\n                          style: { padding: \"6px 10px\", borderBottom: \"1px solid #1a242c\" },\n                        }, String(v)))))),\n              ]),\n              React.createElement(\"div\", {\n                key: \"meta\",\n                style: { fontSize: 12, color: \"var(--muted)\", marginTop: 6 },\n              }, `${props.rows?.length ?? 0} row(s)${props.durationMs ? ` \u00B7 ${props.durationMs}ms` : \"\"}${props.truncated ? \" \u00B7 truncated\" : \"\"}`),\n            ]),\n          ArivieVerdict: ({ props }) => {\n            const colors = {\n              healthy: { bg: \"#0e2a22\", fg: \"#10b981\" },\n              watch:   { bg: \"#2a230e\", fg: \"#fbbf24\" },\n              breached:{ bg: \"#2a1010\", fg: \"#f87171\" },\n              info:    { bg: \"#0e1c2a\", fg: \"#60a5fa\" },\n            };\n            const c = colors[props.status] ?? colors.info;\n            return React.createElement(\"div\", {\n              style: {\n                display: \"inline-flex\", alignItems: \"center\", gap: 8, padding: \"6px 14px\",\n                borderRadius: 999, background: c.bg, color: c.fg, fontSize: 13, margin: 8,\n              },\n            }, [\n              React.createElement(\"strong\", { key: \"s\" }, props.status.toUpperCase()),\n              React.createElement(\"span\", { key: \"m\" }, props.message),\n              props.threshold ? React.createElement(\"span\", { key: \"t\", style: { opacity: 0.7 } }, `(${props.threshold})`) : null,\n            ]);\n          },\n          ArivieSemanticEntity: ({ props }) =>\n            React.createElement(\"div\", {\n              style: {\n                margin: 12, padding: 16, border: \"1px solid #2a3942\", borderRadius: 12,\n              },\n            }, [\n              React.createElement(\"h3\", { key: \"n\", style: { margin: 0, fontSize: 16 } }, props.entityName),\n              props.description ? React.createElement(\"p\", { key: \"d\",\n                style: { color: \"var(--muted)\", fontSize: 13 } }, props.description) : null,\n              [\"measures\", \"dimensions\", \"segments\"].map(kind =>\n                (props[kind]?.length ?? 0) > 0\n                  ? React.createElement(\"div\", { key: kind, style: { marginTop: 8 } }, [\n                      React.createElement(\"div\", { key: \"h\",\n                        style: { fontSize: 12, color: \"var(--muted)\", textTransform: \"uppercase\" } }, kind),\n                      React.createElement(\"ul\", { key: \"u\", style: { margin: \"4px 0 0\", paddingLeft: 18 } },\n                        props[kind].map((x, i) => React.createElement(\"li\", { key: i, style: { fontSize: 13 } },\n                          [React.createElement(\"strong\", { key: \"n\" }, x.name),\n                           x.description ? ` \u2014 ${x.description}` : \"\"]))),\n                    ])\n                  : null\n              ),\n            ]),\n        },\n      });\n\n      const root = ReactDOM.createRoot(document.getElementById(\"root\"));\n\n      function render(spec) {\n        try {\n          root.render(React.createElement(Renderer, { spec, registry }));\n        } catch (err) {\n          root.render(React.createElement(\"div\", { className: \"err\" },\n            `Render error: ${err?.message ?? String(err)}`));\n        }\n      }\n\n      // MCP UI clients post the tool result via window.ontoolresult.\n      window.ontoolresult = (result) => {\n        const text = result?.content?.[0]?.text;\n        if (!text) return;\n        try {\n          const spec = JSON.parse(text);\n          render(spec);\n        } catch (err) {\n          root.render(React.createElement(\"div\", { className: \"err\" },\n            `Bad spec JSON: ${err?.message ?? String(err)}`));\n        }\n      };\n    </script>\n  </body>\n</html>\n";
+
+export { DEFAULT_UI_SHELL_HTML, type MakeMcpUiServerOptions, type McpOptions, makeMcpServer, makeMcpUiServer };

package/dist/index.js

@@ -2,6 +2,10 @@ import { validateExecuteSql } from '@arivie/db-postgres';
 import { createTool } from '@mastra/core/tools';
 import { MCPServer } from '@mastra/mcp';
 import { z } from 'zod';
+import { createMcpApp } from '@json-render/mcp';
+export { registerJsonRenderTool } from '@json-render/mcp';
+import { arivieUiCatalog } from '@arivie/ui-catalog';
+export { arivieUiCatalog } from '@arivie/ui-catalog';
 
 // src/make-server.ts
 var ASK_DESCRIPTION = "Ask Arivie a question; runs the agent's full conversational round-trip.";
@@ -100,4 +104,293 @@ function makeMcpServer(opts) {
   });
 }
 
-export { makeMcpServer };
+// src/ui-shell.ts
+var DEFAULT_UI_SHELL_HTML = `<!doctype html>
+<html lang="en">
+  <head>
+    <meta charset="utf-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1" />
+    <title>Arivie</title>
+    <style>
+      :root {
+        --bg: #0b1115;
+        --fg: #e7eef3;
+        --muted: #8aa0ad;
+        --accent: #0d9488;
+        --warn: #fb923c;
+      }
+      @media (prefers-color-scheme: light) {
+        :root {
+          --bg: #ffffff;
+          --fg: #0b1115;
+          --muted: #5b6770;
+          --accent: #0d9488;
+          --warn: #c2570a;
+        }
+      }
+      html, body, #root { height: 100%; margin: 0; }
+      body {
+        background: var(--bg);
+        color: var(--fg);
+        font-family: ui-sans-serif, -apple-system, BlinkMacSystemFont, "Segoe UI",
+                     Roboto, "Helvetica Neue", Arial, sans-serif;
+        line-height: 1.5;
+      }
+      .empty { padding: 24px; color: var(--muted); font-size: 14px; }
+      .err { padding: 16px; color: var(--warn); font-family: ui-monospace, monospace;
+             font-size: 12px; white-space: pre-wrap; }
+    </style>
+  </head>
+  <body>
+    <div id="root">
+      <div class="empty">Waiting for spec from the Arivie agent\u2026</div>
+    </div>
+    <script type="module">
+      // Pulled at runtime \u2014 these versions match what @arivie/mcp depends on.
+      import * as React from "https://esm.sh/react@19.1?bundle";
+      import * as ReactDOM from "https://esm.sh/react-dom@19.1/client?bundle";
+      import { defineRegistry, Renderer } from "https://esm.sh/@json-render/react@0.19?bundle";
+      import { shadcnComponentDefinitions } from "https://esm.sh/@json-render/shadcn@0.19/catalog?bundle";
+      import { defineCatalog } from "https://esm.sh/@json-render/core@0.19?bundle";
+      import { schema } from "https://esm.sh/@json-render/react@0.19/schema?bundle";
+
+      // A bare-minimum mirror of the server-side Arivie catalog. We can't
+      // import the server catalog directly (this runs in the browser), so
+      // we re-declare the components the renderer needs. The Zod schema is
+      // only used for VALIDATION on the server; the renderer just looks up
+      // component names \u2192 React components in the registry.
+      const catalog = defineCatalog(schema, {
+        components: {
+          ...shadcnComponentDefinitions,
+          ArivieMetric: { props: schema.object({}), description: "" },
+          ArivieQueryResult: { props: schema.object({}), description: "" },
+          ArivieVerdict: { props: schema.object({}), description: "" },
+          ArivieSemanticEntity: { props: schema.object({}), description: "" },
+        },
+        actions: {},
+      });
+
+      const fmt = (v, format) => {
+        if (v == null) return "\u2014";
+        if (format === "currency") {
+          const n = Number(v);
+          if (!Number.isFinite(n)) return String(v);
+          return n.toLocaleString(undefined, { style: "currency", currency: "USD" });
+        }
+        if (format === "percent") return \`\${v}%\`;
+        return String(v);
+      };
+
+      const { registry } = defineRegistry(catalog, {
+        components: {
+          ArivieMetric: ({ props }) =>
+            React.createElement("div", {
+              style: {
+                padding: "16px", border: "1px solid #2a3942", borderRadius: 12,
+                display: "inline-block", minWidth: 220, margin: 8,
+              },
+            }, [
+              React.createElement("div", { key: "l", style: { color: "var(--muted)", fontSize: 13 } }, props.label),
+              React.createElement("div", { key: "v", style: { fontSize: 28, fontWeight: 600, marginTop: 4 } },
+                fmt(props.value, props.format)),
+              props.delta ? React.createElement("div", { key: "d",
+                style: { fontSize: 13, marginTop: 4,
+                  color: props.deltaDirection === "down" ? "var(--warn)" : "var(--accent)" }
+              }, \`\${props.deltaDirection === "down" ? "\u25BC" : "\u25B2"} \${props.delta}\`) : null,
+            ]),
+          ArivieQueryResult: ({ props }) =>
+            React.createElement("div", { style: { margin: 12 } }, [
+              React.createElement("pre", {
+                key: "sql",
+                style: {
+                  background: "#0e1820", color: "#cde8ff", padding: 12, borderRadius: 8,
+                  fontSize: 12, overflow: "auto", maxHeight: 240,
+                },
+              }, props.sql),
+              React.createElement("table", {
+                key: "t",
+                style: { width: "100%", borderCollapse: "collapse", marginTop: 8, fontSize: 13 },
+              }, [
+                React.createElement("thead", { key: "h" },
+                  React.createElement("tr", null, Object.keys(props.rows?.[0] ?? {}).map(k =>
+                    React.createElement("th", {
+                      key: k,
+                      style: { textAlign: "left", padding: "6px 10px", borderBottom: "1px solid #2a3942" },
+                    }, k)))),
+                React.createElement("tbody", { key: "b" },
+                  (props.rows ?? []).map((row, i) =>
+                    React.createElement("tr", { key: i },
+                      Object.values(row).map((v, j) =>
+                        React.createElement("td", {
+                          key: j,
+                          style: { padding: "6px 10px", borderBottom: "1px solid #1a242c" },
+                        }, String(v)))))),
+              ]),
+              React.createElement("div", {
+                key: "meta",
+                style: { fontSize: 12, color: "var(--muted)", marginTop: 6 },
+              }, \`\${props.rows?.length ?? 0} row(s)\${props.durationMs ? \` \xB7 \${props.durationMs}ms\` : ""}\${props.truncated ? " \xB7 truncated" : ""}\`),
+            ]),
+          ArivieVerdict: ({ props }) => {
+            const colors = {
+              healthy: { bg: "#0e2a22", fg: "#10b981" },
+              watch:   { bg: "#2a230e", fg: "#fbbf24" },
+              breached:{ bg: "#2a1010", fg: "#f87171" },
+              info:    { bg: "#0e1c2a", fg: "#60a5fa" },
+            };
+            const c = colors[props.status] ?? colors.info;
+            return React.createElement("div", {
+              style: {
+                display: "inline-flex", alignItems: "center", gap: 8, padding: "6px 14px",
+                borderRadius: 999, background: c.bg, color: c.fg, fontSize: 13, margin: 8,
+              },
+            }, [
+              React.createElement("strong", { key: "s" }, props.status.toUpperCase()),
+              React.createElement("span", { key: "m" }, props.message),
+              props.threshold ? React.createElement("span", { key: "t", style: { opacity: 0.7 } }, \`(\${props.threshold})\`) : null,
+            ]);
+          },
+          ArivieSemanticEntity: ({ props }) =>
+            React.createElement("div", {
+              style: {
+                margin: 12, padding: 16, border: "1px solid #2a3942", borderRadius: 12,
+              },
+            }, [
+              React.createElement("h3", { key: "n", style: { margin: 0, fontSize: 16 } }, props.entityName),
+              props.description ? React.createElement("p", { key: "d",
+                style: { color: "var(--muted)", fontSize: 13 } }, props.description) : null,
+              ["measures", "dimensions", "segments"].map(kind =>
+                (props[kind]?.length ?? 0) > 0
+                  ? React.createElement("div", { key: kind, style: { marginTop: 8 } }, [
+                      React.createElement("div", { key: "h",
+                        style: { fontSize: 12, color: "var(--muted)", textTransform: "uppercase" } }, kind),
+                      React.createElement("ul", { key: "u", style: { margin: "4px 0 0", paddingLeft: 18 } },
+                        props[kind].map((x, i) => React.createElement("li", { key: i, style: { fontSize: 13 } },
+                          [React.createElement("strong", { key: "n" }, x.name),
+                           x.description ? \` \u2014 \${x.description}\` : ""]))),
+                    ])
+                  : null
+              ),
+            ]),
+        },
+      });
+
+      const root = ReactDOM.createRoot(document.getElementById("root"));
+
+      function render(spec) {
+        try {
+          root.render(React.createElement(Renderer, { spec, registry }));
+        } catch (err) {
+          root.render(React.createElement("div", { className: "err" },
+            \`Render error: \${err?.message ?? String(err)}\`));
+        }
+      }
+
+      // MCP UI clients post the tool result via window.ontoolresult.
+      window.ontoolresult = (result) => {
+        const text = result?.content?.[0]?.text;
+        if (!text) return;
+        try {
+          const spec = JSON.parse(text);
+          render(spec);
+        } catch (err) {
+          root.render(React.createElement("div", { className: "err" },
+            \`Bad spec JSON: \${err?.message ?? String(err)}\`));
+        }
+      };
+    </script>
+  </body>
+</html>
+`;
+
+// src/make-ui-server.ts
+async function makeMcpUiServer(opts) {
+  const html = opts.html ?? DEFAULT_UI_SHELL_HTML;
+  const server = await createMcpApp({
+    name: `Arivie (${opts.ownerName})`,
+    version: "0.3.0",
+    catalog: arivieUiCatalog,
+    html,
+    tool: {
+      name: "render_arivie_ui",
+      title: "Render Arivie UI",
+      description: "Render an interactive UI for an analytics result. Pass a json-render `spec` built from the Arivie catalog (ArivieMetric, ArivieQueryResult, ArivieVerdict, ArivieSemanticEntity, plus shadcn components). The MCP client renders the spec inside its UI surface."
+    }
+  });
+  const s = server;
+  s.registerTool(
+    "ask",
+    {
+      title: "Ask Arivie",
+      description: "Ask Arivie a natural-language question. Runs the full agent loop (semantic-layer-grounded SQL + optional file artifacts). Returns the agent's answer text plus the tool trace. For UI rendering, also consider calling `render_arivie_ui` with a spec built from the result.",
+      inputSchema: {
+        prompt: z.string().describe("The question to ask Arivie")
+      }
+    },
+    async ({ prompt }) => {
+      const result = await opts.agent.generate(prompt);
+      const text = result != null && typeof result === "object" && "text" in result ? String(result.text) : String(result);
+      return { content: [{ type: "text", text }] };
+    }
+  );
+  s.registerTool(
+    "query",
+    {
+      title: "Execute SQL",
+      description: "Execute a read-only SQL query against the owner's database. Returns rows + SQL + timing. To render as a table card, pass the result into `render_arivie_ui` with an ArivieQueryResult spec.",
+      inputSchema: {
+        sql: z.string().describe("A SELECT or WITH SQL query")
+      }
+    },
+    async ({ sql }) => {
+      const trimmed = sql.trim();
+      validateExecuteSql(trimmed);
+      const start = Date.now();
+      const result = await opts.db.execute({
+        query: trimmed,
+        runAsRole: "arivie_reader",
+        userId: "mcp",
+        rowLimit: 50,
+        timeoutMs: 3e4
+      });
+      const durationMs = Date.now() - start;
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify({
+              rows: result.rows,
+              rowCount: result.rowCount,
+              truncated: result.truncated,
+              durationMs,
+              sql: trimmed,
+              source: "postgres"
+            })
+          }
+        ]
+      };
+    }
+  );
+  s.registerTool(
+    "schema",
+    {
+      title: "Semantic Layer",
+      description: "Return the semantic-layer catalog + entities for this owner. Each entity surfaces measures, dimensions, segments, joins. To render one entity as a card, pass it into `render_arivie_ui` with an ArivieSemanticEntity spec.",
+      inputSchema: {}
+    },
+    async () => ({
+      content: [
+        {
+          type: "text",
+          text: JSON.stringify({
+            catalog: opts.semantic.catalog,
+            entities: [...opts.semantic.entities.values()]
+          })
+        }
+      ]
+    })
+  );
+  return server;
+}
+
+export { DEFAULT_UI_SHELL_HTML, makeMcpServer, makeMcpUiServer };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arivie/mcp",
-  "version": "0.1.0",
+  "version": "1.0.1",
   "private": false,
   "type": "module",
   "description": "Arivie Mastra MCP server — ask, query, schema, and memory tools over MCPServer.",
@@ -27,25 +27,31 @@
     "CHANGELOG.md"
   ],
   "dependencies": {
-    "@mastra/core": "^1.35.0",
-    "@mastra/mcp": "^1.7.0",
+    "@json-render/core": "^0.19.0",
+    "@json-render/mcp": "^0.19.0",
+    "@json-render/react": "^0.19.0",
+    "@json-render/shadcn": "^0.19.0",
+    "@mastra/core": "^1.45.0",
+    "@mastra/mcp": "^1.11.0",
+    "@modelcontextprotocol/sdk": "^1.29.0",
     "zod": "^4.4.0",
-    "@arivie/db-postgres": "0.1.0",
-    "@arivie/semantic": "0.1.0"
+    "@arivie/semantic": "0.1.0",
+    "@arivie/ui-catalog": "0.1.1",
+    "@arivie/db-postgres": "0.1.2"
   },
   "peerDependencies": {
-    "@arivie/agent": "0.1.0"
+    "@arivie/agent": "0.2.1"
   },
   "devDependencies": {
-    "@mastra/core": "^1.35.0",
+    "@mastra/core": "^1.45.0",
     "@types/node": "^22.0.0",
     "ai": "^6.0.0",
     "tsup": "^8.5.1",
     "typescript": "^6.0.0",
     "vitest": "^4.1.0",
     "zod": "^4.4.0",
-    "@arivie/agent": "0.1.0",
-    "@arivie/workspace": "0.1.0"
+    "@arivie/agent": "0.2.1",
+    "@arivie/workspace": "0.1.1"
   },
   "publishConfig": {
     "access": "public",