grain-tools 0.10.16 → 0.10.18

package/dist/connection-setup.html

@@ -12745,7 +12745,10 @@ function ConnectionSetupPage({
         /* @__PURE__ */ jsxRuntimeExports.jsx("strong", { children: "Connection failed" }),
         /* @__PURE__ */ jsxRuntimeExports.jsx("p", { children: error })
       ] }),
-      tested && !error && /* @__PURE__ */ jsxRuntimeExports.jsx("p", { className: "grain-connection-success", role: "status", children: "Connection test successful." }),
+      tested && !error && /* @__PURE__ */ jsxRuntimeExports.jsxs("p", { className: "grain-connection-success", role: "status", children: [
+        /* @__PURE__ */ jsxRuntimeExports.jsx("span", { "aria-hidden": "true", children: "✓ " }),
+        "Connection test successful."
+      ] }),
       /* @__PURE__ */ jsxRuntimeExports.jsxs("div", { className: "grain-connection-grid", children: [
         /* @__PURE__ */ jsxRuntimeExports.jsxs("div", { className: "grain-connection-field", children: [
           /* @__PURE__ */ jsxRuntimeExports.jsx("label", { htmlFor: "grain-connection-type", children: "Database type" }),
@@ -12833,7 +12836,10 @@ function ConnectionSetupPage({
         ] })
       ] }),
       /* @__PURE__ */ jsxRuntimeExports.jsxs("div", { className: "grain-connection-footer", children: [
-        /* @__PURE__ */ jsxRuntimeExports.jsx("p", { children: "Grain will test the connection before saving it." }),
+        tested && !error ? /* @__PURE__ */ jsxRuntimeExports.jsxs("p", { className: "grain-connection-success", children: [
+          /* @__PURE__ */ jsxRuntimeExports.jsx("span", { "aria-hidden": "true", children: "✓ " }),
+          "Connection test successful."
+        ] }) : error ? /* @__PURE__ */ jsxRuntimeExports.jsx("p", { className: "grain-connection-footer-error", children: error }) : /* @__PURE__ */ jsxRuntimeExports.jsx("p", { children: "Grain will test the connection before saving it." }),
         /* @__PURE__ */ jsxRuntimeExports.jsxs("div", { className: "grain-connection-actions", children: [
           /* @__PURE__ */ jsxRuntimeExports.jsx(
             "button",
@@ -12940,7 +12946,12 @@ function ConnectionSetupApp() {
   if (saved)
     return /* @__PURE__ */ jsxRuntimeExports.jsxs("main", { className: "grain-setup-state", role: "status", children: [
       /* @__PURE__ */ jsxRuntimeExports.jsx("h1", { children: "Saved." }),
-      /* @__PURE__ */ jsxRuntimeExports.jsx("p", { children: "Go back to your agent — it will retry." })
+      /* @__PURE__ */ jsxRuntimeExports.jsx("p", { children: "Go back to your agent, and it will retry." }),
+      /* @__PURE__ */ jsxRuntimeExports.jsxs("p", { className: "grain-setup-workspace", children: [
+        "To run queries yourself, run ",
+        /* @__PURE__ */ jsxRuntimeExports.jsx("code", { children: "grain ui" }),
+        " in a terminal to open the Grain workspace (Beta). The workspace is in beta and may be unstable."
+      ] })
     ] });
   if (!state || !actions)
     return /* @__PURE__ */ jsxRuntimeExports.jsx("main", { className: "grain-setup-state", role: "status", children: /* @__PURE__ */ jsxRuntimeExports.jsx("p", { children: message }) });
@@ -12949,13 +12960,11 @@ function ConnectionSetupApp() {
     {
       value: {
         mode: "standalone",
-        sendMessage: () => {
-        },
+        sendMessage: () => void 0,
         callServerTool: async () => {
           throw new Error("Unavailable");
         },
-        onMessage: () => () => {
-        },
+        onMessage: () => () => void 0,
         isReady: true
       },
       children: /* @__PURE__ */ jsxRuntimeExports.jsx(
@@ -13129,11 +13138,18 @@ clientExports.createRoot(document.getElementById("root")).render(/* @__PURE__ */
   margin: 5px 0 0;
 }
 .grain-connection-success {
-  color: var(--grain-ink);
-  background: var(--grain-surface);
+  color: var(--vscode-testing-iconPassed, var(--sp-success));
+  border: 1px solid var(--vscode-testing-iconPassed, var(--sp-success));
+  background: color-mix(in srgb, var(--sp-success) 8%, var(--grain-surface));
   padding: 10px 14px;
   border-radius: var(--grain-radius-sm);
 }
+.grain-connection-footer .grain-connection-success {
+  color: var(--vscode-testing-iconPassed, var(--sp-success));
+}
+.grain-connection-footer-error {
+  color: var(--vscode-errorForeground, var(--sp-error));
+}
 .grain-connection-page :is(button, input, select, summary):focus-visible {
   outline: 2px solid var(--grain-accent);
   outline-offset: 2px;
@@ -17468,6 +17484,31 @@ body {
 .sp-web-sidebar-header .sp-web-sidebar-title {
   flex: 1;
 }
+.sp-web-section-toggle,
+.sp-web-agent-session-toggle {
+  border: 0;
+  background: transparent;
+  color: var(--grain-ink);
+  font: inherit;
+  text-align: left;
+  cursor: pointer;
+}
+.sp-web-section-toggle {
+  padding: 0;
+}
+.sp-web-agent-session-toggle {
+  width: 100%;
+  padding: 7px 9px;
+  color: var(--grain-ink-muted);
+  font-size: 11px;
+}
+.sp-web-section-toggle:focus-visible,
+.sp-web-agent-session-toggle:focus-visible {
+  outline: 1px solid var(--grain-accent);
+}
+[aria-current='page'].sp-web-connection-item {
+  background: var(--grain-surface-raised);
+}
 .sp-web-section-add,
 .sp-web-sidebar-close {
   display: inline-flex;
@@ -17659,6 +17700,14 @@ body {
   gap: 4px;
   padding: 18px 24px 14px;
 }
+.sp-web-query-heading h1 {
+  margin: 0;
+  font-size: 22px;
+}
+.sp-web-query-provenance {
+  color: var(--grain-ink-muted);
+  font-size: 11px;
+}
 .sp-web-query-heading input,
 .sp-web-query-heading textarea {
   width: 100%;
@@ -17874,6 +17923,18 @@ body {
 .sp-web-connection-detail p {
   color: var(--grain-ink-muted);
 }
+.sp-web-connection-detail .sp-web-connection-test-success {
+  color: var(--vscode-testing-iconPassed, var(--sp-success));
+}
+.sp-web-connection-detail .sp-web-connection-test-error {
+  color: var(--vscode-errorForeground, var(--sp-error));
+}
+.sp-web-connection-delete-confirm {
+  margin-top: 16px;
+}
+.sp-web-connection-delete-confirm p {
+  color: var(--grain-ink);
+}
 .sp-web-connection-actions {
   display: flex;
   gap: 8px;
@@ -17913,6 +17974,15 @@ body {
   .grain-connection-readonly {
     grid-template-columns: 1fr;
   }
+}
+
+.grain-setup-workspace {
+  color: var(--grain-ink-muted);
+  font-size: 13px;
+}
+
+.grain-setup-workspace code {
+  font-family: var(--grain-font-mono);
 }</style>
   </head>
   <body>

package/dist/credential-page.html

@@ -4351,6 +4351,31 @@ body {
 .sp-web-sidebar-header .sp-web-sidebar-title {
   flex: 1;
 }
+.sp-web-section-toggle,
+.sp-web-agent-session-toggle {
+  border: 0;
+  background: transparent;
+  color: var(--grain-ink);
+  font: inherit;
+  text-align: left;
+  cursor: pointer;
+}
+.sp-web-section-toggle {
+  padding: 0;
+}
+.sp-web-agent-session-toggle {
+  width: 100%;
+  padding: 7px 9px;
+  color: var(--grain-ink-muted);
+  font-size: 11px;
+}
+.sp-web-section-toggle:focus-visible,
+.sp-web-agent-session-toggle:focus-visible {
+  outline: 1px solid var(--grain-accent);
+}
+[aria-current='page'].sp-web-connection-item {
+  background: var(--grain-surface-raised);
+}
 .sp-web-section-add,
 .sp-web-sidebar-close {
   display: inline-flex;
@@ -4542,6 +4567,14 @@ body {
   gap: 4px;
   padding: 18px 24px 14px;
 }
+.sp-web-query-heading h1 {
+  margin: 0;
+  font-size: 22px;
+}
+.sp-web-query-provenance {
+  color: var(--grain-ink-muted);
+  font-size: 11px;
+}
 .sp-web-query-heading input,
 .sp-web-query-heading textarea {
   width: 100%;
@@ -4757,6 +4790,18 @@ body {
 .sp-web-connection-detail p {
   color: var(--grain-ink-muted);
 }
+.sp-web-connection-detail .sp-web-connection-test-success {
+  color: var(--vscode-testing-iconPassed, var(--sp-success));
+}
+.sp-web-connection-detail .sp-web-connection-test-error {
+  color: var(--vscode-errorForeground, var(--sp-error));
+}
+.sp-web-connection-delete-confirm {
+  margin-top: 16px;
+}
+.sp-web-connection-delete-confirm p {
+  color: var(--grain-ink);
+}
 .sp-web-connection-actions {
   display: flex;
   gap: 8px;
@@ -4917,11 +4962,18 @@ body {
   margin: 5px 0 0;
 }
 .grain-connection-success {
-  color: var(--grain-ink);
-  background: var(--grain-surface);
+  color: var(--vscode-testing-iconPassed, var(--sp-success));
+  border: 1px solid var(--vscode-testing-iconPassed, var(--sp-success));
+  background: color-mix(in srgb, var(--sp-success) 8%, var(--grain-surface));
   padding: 10px 14px;
   border-radius: var(--grain-radius-sm);
 }
+.grain-connection-footer .grain-connection-success {
+  color: var(--vscode-testing-iconPassed, var(--sp-success));
+}
+.grain-connection-footer-error {
+  color: var(--vscode-errorForeground, var(--sp-error));
+}
 .grain-connection-page :is(button, input, select, summary):focus-visible {
   outline: 2px solid var(--grain-accent);
   outline-offset: 2px;

package/dist/mcp-app-shell.html

@@ -0,0 +1,31 @@
+<!doctype html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1">
+  <title>Grain Results</title>
+  <style>
+    body { margin: 0; padding: 12px; color: var(--vscode-foreground, #222); font: 13px var(--vscode-font-family, sans-serif); }
+    button { font: inherit; }
+    .row { margin: 0 0 8px; }
+    #columns { overflow-wrap: anywhere; }
+    #error { color: var(--vscode-errorForeground, #b00); }
+    #grid { width: 100%; height: 480px; border: 0; }
+    label { display: inline-flex; gap: 6px; align-items: center; }
+  </style>
+  <script type="module" crossorigin>(function(){const e=document.createElement("link").relList;if(e&&e.supports&&e.supports("modulepreload"))return;for(const t of document.querySelectorAll('link[rel="modulepreload"]'))s(t);new MutationObserver(t=>{for(const r of t)if(r.type==="childList")for(const u of r.addedNodes)u.tagName==="LINK"&&u.rel==="modulepreload"&&s(u)}).observe(document,{childList:!0,subtree:!0});function n(t){const r={};return t.integrity&&(r.integrity=t.integrity),t.referrerPolicy&&(r.referrerPolicy=t.referrerPolicy),t.crossOrigin==="use-credentials"?r.credentials="include":t.crossOrigin==="anonymous"?r.credentials="omit":r.credentials="same-origin",r}function s(t){if(t.ep)return;t.ep=!0;const r=n(t);fetch(t.href,r)}})();const l=new Map;let O=0,c,m=!1,i,f="",d=!1;function p(o,e={}){const n=++O;return new Promise(s=>{l.set(n,s),window.parent.postMessage({jsonrpc:"2.0",id:n,method:o,params:e},"*"),window.setTimeout(()=>{l.get(n)&&(l.delete(n),s({jsonrpc:"2.0",id:n,error:{message:"The host did not respond"}}))},1e4)})}function a(o){const e=document.querySelector("#error");e&&(e.textContent=o)}function y(o){const e=document.querySelector("#open");e&&(e.textContent=o)}function h(o){c?.contentWindow?.postMessage(o,"*")}function v(){!i||!m||h({jsonrpc:"2.0",method:"ui/notifications/tool-result",params:{structuredContent:i}})}window.addEventListener("message",o=>{const e=o.data;if(!(!e||e.jsonrpc!=="2.0")){if(o.source===c?.contentWindow){if(e.id!==void 0){const n=`frame-${String(e.id)}`;l.set(n,s=>{c?.contentWindow?.postMessage({...s,id:e.id},"*")}),window.parent.postMessage({...e,id:n},"*")}else window.parent.postMessage(e,"*"),e.method==="ui/notifications/initialized"&&(m=!0,v());return}if(e.id!==void 0&&l.has(e.id)){l.get(e.id)?.(e),l.delete(e.id);return}typeof e.method=="string"&&e.method.startsWith("ui/notifications/")&&(e.method==="ui/notifications/tool-result"?C(e.params):h(e))}});function C(o){if(!o||!o.structuredContent||typeof o.structuredContent!="object")return;i=o.structuredContent,v();const n=i.route==="editor",s=document.querySelector("#toggle-wrap");s&&(s.hidden=!n),y(n?"Open in Grain":"Open in workspace"),f=typeof i.cardPreferenceKey=="string"?i.cardPreferenceKey:"";const t=document.querySelector("#title"),r=document.querySelector("#count"),u=document.querySelector("#columns"),g=document.querySelector("#status");t&&(t.textContent=String(i.title||i.tabId||"Query result")),r&&(r.textContent=`${Number(i.rowCount??0).toLocaleString()} rows`),u&&(u.textContent=`Columns: ${(i.columns||[]).map(S=>S.name||"").join(", ")||"none"}`),g&&(g.textContent=`Status: ${i.status||"unknown"}`);try{d=f?window.localStorage.getItem(f)==="on":!1}catch{d=!1}const w=document.querySelector("#toggle");w&&(w.checked=d),(!n||d)&&b()}async function b(){if(c||!i)return;a("");const o=await p("resources/read",{uri:"ui://sql-preview/results-grid-full"}),n=o.result?.contents?.find(s=>typeof s.text=="string")?.text;if(o.error||!n){const s=i.route==="editor";a(s?"The full results view is unavailable. Open in Grain to see the result.":"The full results view is unavailable. Open in workspace to see the result."),y(s?"Open in Grain":"Open in workspace");return}c=document.createElement("iframe"),m=!1,c.id="grid",c.title="Full query results",c.srcdoc=n,document.body.append(c),document.querySelector("#card")?.classList.add("expanded")}document.querySelector("#open")?.addEventListener("click",async()=>{if(!i?.sessionId||!i.tabId){a("This result cannot be opened yet.");return}const o=i.route==="editor"?"open_in_grain":"open_in_workspace",e=await p("tools/call",{name:"open_result_view",arguments:{sessionId:i.sessionId,tabId:i.tabId,button:o}}),n=e.result,t=(n?.content||[]).map(u=>u.text).filter(Boolean).join(`
+`),r=n?.structuredContent?.url||t.match(/https?:\/\/[^\s)]+/)?.[0];if(r){if(!(await p("ui/open-link",{url:r})).error){a(t&&t!==r?t:"Opened the result workspace.");return}p("tools/call",{name:"record_ui_event",arguments:{event:"mcp_result_view_opened",button:o}}).catch(()=>{}),a(`${t||"Open this result in your browser:"}
+${r}`);return}t&&a(t),e.error&&a(`${o==="open_in_grain"?"Open in Grain":"Open in workspace"} is not available yet.`)});document.querySelector("#toggle")?.addEventListener("change",o=>{d=o.currentTarget.checked;try{f&&window.localStorage.setItem(f,d?"on":"off")}catch{d=o.currentTarget.checked}d?b():(c?.remove(),c=void 0,document.querySelector("#card")?.classList.remove("expanded"))});p("ui/initialize",{appInfo:{name:"Grain Results Card",version:"1.0.0"},appCapabilities:{},protocolVersion:"2025-11-25"}).then(()=>window.parent.postMessage({jsonrpc:"2.0",method:"ui/notifications/initialized"},"*"));</script>
+</head>
+<body>
+  <main id="card">
+    <h2 id="title" class="row">Query result</h2>
+    <div id="count" class="row"></div>
+    <div id="columns" class="row"></div>
+    <div id="status" class="row"></div>
+    <div class="row"><button id="open">Open in Grain</button></div>
+    <label id="toggle-wrap"><input id="toggle" type="checkbox">Show in chat</label>
+    <div id="error" role="status" aria-live="polite"></div>
+  </main>
+</body>
+</html>

package/dist/mcp-app.html

@@ -12521,6 +12521,72 @@ function BaseHostProvider({
 }) {
   return /* @__PURE__ */ jsxRuntimeExports.jsx(HostContext.Provider, { value, children });
 }
+function useWorkspaceResultTarget() {
+  const [targets, setTargets] = reactExports.useState({});
+  const [resetToken, setResetToken] = reactExports.useState(0);
+  const remember = reactExports.useCallback((tabId, structured) => {
+    setResetToken((current) => current + 1);
+    const sessionId = structured?.["sessionId"];
+    const resultTabId = structured?.["tabId"];
+    if (structured?.["route"] === "chat" && typeof sessionId === "string" && typeof resultTabId === "string") {
+      setTargets((current) => ({ ...current, [tabId]: { sessionId, tabId: resultTabId } }));
+    }
+  }, []);
+  return { targetForTab: (tabId) => targets[tabId], remember, resetToken };
+}
+function WorkspaceResultAction({
+  target,
+  resetToken,
+  callServerTool,
+  sendMessage
+}) {
+  const [linkError, setLinkError] = reactExports.useState("");
+  reactExports.useEffect(() => setLinkError(""), [resetToken]);
+  const openWorkspace = async () => {
+    if (!target) {
+      return;
+    }
+    setLinkError("");
+    try {
+      const response = await callServerTool("open_result_view", {
+        ...target,
+        button: "open_in_workspace"
+      });
+      const structured = response.structuredContent;
+      if (typeof structured?.url === "string") {
+        try {
+          await sendMessage("openLink", { url: structured.url });
+        } catch {
+          void callServerTool("record_ui_event", {
+            event: "mcp_result_view_opened",
+            button: "open_in_workspace"
+          }).catch(() => void 0);
+          setLinkError(structured.url);
+        }
+      } else if (structured?.openedInEditor === true) {
+        setLinkError("Opened the result in Grain.");
+      } else {
+        throw new Error("The workspace link is unavailable");
+      }
+    } catch {
+      setLinkError("Could not open this result in the workspace.");
+    }
+  };
+  return /* @__PURE__ */ jsxRuntimeExports.jsxs(jsxRuntimeExports.Fragment, { children: [
+    target && /* @__PURE__ */ jsxRuntimeExports.jsx("button", { className: "sp-btn", onClick: () => void openWorkspace(), children: "Open in workspace" }),
+    linkError && /* @__PURE__ */ jsxRuntimeExports.jsx("pre", { className: "sp-error-bar", role: "status", "aria-live": "polite", children: linkError })
+  ] });
+}
+function copyResultToClipboard(result, copyText) {
+  if (!result || result.rows.length === 0) {
+    return;
+  }
+  const header = result.columns.map((column) => column.name).join("	");
+  const rows = result.rows.map(
+    (row) => result.columns.map((_, index) => String(row[index] ?? "")).join("	")
+  );
+  copyText([header, ...rows].join("\n"));
+}
 const createStoreImpl = (createState) => {
   let state;
   const listeners = /* @__PURE__ */ new Set();
@@ -73210,7 +73276,6 @@ const DataGrid = reactExports.forwardRef(function DataGrid2({
   nativeTheme = false,
   disableFilters = false,
   autoSizeStrategy,
-  fillWidth = false,
   onCellDoubleClick
 }, ref) {
   const trackUi = useTrackUi();
@@ -73379,21 +73444,6 @@ const DataGrid = reactExports.forwardRef(function DataGrid2({
     }),
     [data, getNodeValue, copyIncludesHeaders]
   );
-  const userResizedRef = reactExports.useRef(false);
-  const fillSpareWidth = reactExports.useCallback((e2) => {
-    if (userResizedRef.current) return;
-    const body = e2.api.getHorizontalPixelRange?.();
-    const columns = (e2.api.getAllDisplayedColumns?.() ?? []).filter((c) => !c.getPinned());
-    if (!body || columns.length === 0) return;
-    const used = columns.reduce((sum2, c) => sum2 + c.getActualWidth(), 0);
-    if (used < body.right - body.left) e2.api.sizeColumnsToFit();
-  }, []);
-  const noteUserResize = reactExports.useCallback((e2) => {
-    if (e2.source === "uiColumnResized" && e2.finished) userResizedRef.current = true;
-  }, []);
-  reactExports.useEffect(() => {
-    userResizedRef.current = false;
-  }, [data?.columns]);
   const columnDefs = reactExports.useMemo(() => {
     if (!data?.columns) return [];
     const isInRange = (rowIndex, displayedColIndex) => {
@@ -73578,11 +73628,6 @@ const DataGrid = reactExports.forwardRef(function DataGrid2({
           onGridReady: (p2) => {
             apiRef.current = p2.api;
           },
-          ...fillWidth ? {
-            onFirstDataRendered: fillSpareWidth,
-            onGridSizeChanged: fillSpareWidth,
-            onColumnResized: noteUserResize
-          } : {},
           rowData: data.rows,
           columnDefs,
           defaultColDef: {
@@ -147891,6 +147936,7 @@ function App() {
     updateTab
   } = useAppStore();
   const [isAppMode, setIsAppMode] = reactExports.useState(true);
+  const { targetForTab, remember, resetToken } = useWorkspaceResultTarget();
   const hasReceivedToolCall = reactExports.useRef(false);
   const hasFallbackFired = reactExports.useRef(false);
   const gridRef = reactExports.useRef(null);
@@ -148005,6 +148051,7 @@ function App() {
       const structured = params?.structuredContent && typeof params.structuredContent === "object" ? params.structuredContent : void 0;
       const resultSql = entry?.sql ?? (typeof rawCallKey === "string" ? rawCallKey : typeof structured?.["query"] === "string" ? structured["query"] : "");
       const tabId = entry?.tabId ?? createTab(resultSql, false);
+      remember(tabId, structured);
       setIsAppMode(true);
       if (params?.isError) {
         const text = params.content?.find((c) => c.type === "text")?.text;
@@ -148230,14 +148277,8 @@ function App() {
       copyTextSafe(csvString);
     }
   };
-  const handleCopy = () => {
-    if (!result || result.rows.length === 0) return;
-    const header = result.columns.map((c) => c.name).join("	");
-    const tsvRows = result.rows.map(
-      (row) => result.columns.map((_, i2) => String(row[i2] ?? "")).join("	")
-    );
-    copyTextSafe([header, ...tsvRows].join("\n"));
-  };
+  const workspaceTarget = targetForTab(activeTab?.id ?? "");
+  const handleCopy = () => copyResultToClipboard(result, copyTextSafe);
   if (!isReady || !activeTab)
     return /* @__PURE__ */ jsxRuntimeExports.jsx("div", { style: { display: "flex", justifyContent: "center", padding: "40px" }, children: /* @__PURE__ */ jsxRuntimeExports.jsx("div", { className: "spinner" }) });
   const showTabs = tabs.length > 1 || !isAppMode;
@@ -148297,7 +148338,8 @@ function App() {
             result && /* @__PURE__ */ jsxRuntimeExports.jsxs("span", { className: "sp-row-count", children: [
               /* @__PURE__ */ jsxRuntimeExports.jsx("strong", { children: result.rowCount ?? result.rows.length }),
               " rows"
-            ] })
+            ] }),
+            result && /* @__PURE__ */ jsxRuntimeExports.jsx(WorkspaceResultAction, { target: workspaceTarget, resetToken, callServerTool, sendMessage })
           ] }),
           /* @__PURE__ */ jsxRuntimeExports.jsx("div", { className: "sp-result-header-right", children: result && result.rows.length > 0 && /* @__PURE__ */ jsxRuntimeExports.jsxs(jsxRuntimeExports.Fragment, { children: [
             /* @__PURE__ */ jsxRuntimeExports.jsx("button", { className: "sp-btn", onClick: handleCopy, "aria-label": "Copy results", children: "Copy" }),
@@ -161960,11 +162002,15 @@ function MCPHostProvider({ children }) {
       };
     }
   });
-  const sendMessage = reactExports.useCallback((command, payload) => {
+  const sendMessage = reactExports.useCallback(async (command, payload) => {
     if (!appRef.current) return;
     switch (command) {
       case "openLink":
-        if (appRef.current.openLink) appRef.current.openLink(payload);
+        if (appRef.current.openLink) {
+          await appRef.current.openLink(payload);
+        } else {
+          throw new Error("The host cannot open links");
+        }
         break;
       default:
         console.warn("[MCPHost] Unknown command sent:", command);