From 3d599f2c25754b496b7e19d04286ab158e63b068 Mon Sep 17 00:00:00 2001
From: Mark Miro <contact@markmiro.com>
Date: Fri, 6 Feb 2026 15:12:51 -0800
Subject: [PATCH] Add error boundary

---
 packages/components/src/index.ts              |  3 +
 .../notebook-preview/src/NotebookRenderer.tsx | 79 +++++++++++--------
 2 files changed, 48 insertions(+), 34 deletions(-)

diff --git a/packages/components/src/index.ts b/packages/components/src/index.ts
index bd5714b2..bc3fc991 100644
--- a/packages/components/src/index.ts
+++ b/packages/components/src/index.ts
@@ -76,6 +76,9 @@ export { groupConsecutiveStreamOutputs } from "./utils/output-grouping.js";
 export { OutputTypesDemoPage } from "./OutputTypesDemoPage.js";
 export { Incrementor } from "./Incrementor.js";
 
+// Error boundary
+export { ErrorBoundary } from "react-error-boundary";
+
 // Re-export types from schema for convenience
 export type {
   OutputData,
diff --git a/packages/notebook-preview/src/NotebookRenderer.tsx b/packages/notebook-preview/src/NotebookRenderer.tsx
index 32e1d250..9682c53a 100644
--- a/packages/notebook-preview/src/NotebookRenderer.tsx
+++ b/packages/notebook-preview/src/NotebookRenderer.tsx
@@ -5,6 +5,7 @@ import {
   SuspenseSpinner,
   ExecutionCount,
   SyntaxHighlighter,
+  ErrorBoundary,
 } from "@runtimed/components";
 
 // Jupyter notebook types
@@ -158,25 +159,31 @@ function CodeCell({ cell, language }: { cell: JupyterCell; language: string }) {
         <div className="min-w-0 flex-1">
           {/* Input */}
           <div className="overflow-hidden rounded border border-gray-200">
-            <SyntaxHighlighter
-              language={language}
-              enableCopy={true}
-              customStyle={{ fontSize: "0.8rem" }}
+            <ErrorBoundary
+              fallback={<div>Error rendering syntax highlighter</div>}
             >
-              {source}
-            </SyntaxHighlighter>
+              <SyntaxHighlighter
+                language={language}
+                enableCopy={true}
+                customStyle={{ fontSize: "0.8rem" }}
+              >
+                {source}
+              </SyntaxHighlighter>
+            </ErrorBoundary>
           </div>
 
           {/* Outputs */}
           {outputs.length > 0 && (
             <div className="mt-2 rounded border border-gray-100 bg-white">
-              <SuspenseSpinner>
-                <OutputsContainer>
-                  {outputs.map((output) => (
-                    <SingleOutput key={output.id} output={output} />
-                  ))}
-                </OutputsContainer>
-              </SuspenseSpinner>
+              <ErrorBoundary fallback={<div>Error rendering outputs</div>}>
+                <SuspenseSpinner>
+                  <OutputsContainer>
+                    {outputs.map((output) => (
+                      <SingleOutput key={output.id} output={output} />
+                    ))}
+                  </OutputsContainer>
+                </SuspenseSpinner>
+              </ErrorBoundary>
             </div>
           )}
         </div>
@@ -210,9 +217,11 @@ function MarkdownCell({ cell }: { cell: JupyterCell }) {
       <div className="flex items-start gap-2">
         <div className="w-12 flex-shrink-0" />
         <div className="min-w-0 flex-1 p-3">
-          <SuspenseSpinner>
-            <SingleOutput output={markdownOutput} />
-          </SuspenseSpinner>
+          <ErrorBoundary fallback={<div>Error rendering markdown</div>}>
+            <SuspenseSpinner>
+              <SingleOutput output={markdownOutput} />
+            </SuspenseSpinner>
+          </ErrorBoundary>
         </div>
       </div>
     </div>
@@ -243,24 +252,26 @@ export function NotebookRenderer({ notebook }: { notebook: JupyterNotebook }) {
 
   return (
     <div className="notebook-preview py-4 pr-4 pl-2">
-      {notebook.cells.map((cell, index) => {
-        switch (cell.cell_type) {
-          case "code":
-            return (
-              <CodeCell
-                key={cell.id || index}
-                cell={cell}
-                language={language}
-              />
-            );
-          case "markdown":
-            return <MarkdownCell key={cell.id || index} cell={cell} />;
-          case "raw":
-            return <RawCell key={cell.id || index} cell={cell} />;
-          default:
-            return null;
-        }
-      })}
+      <ErrorBoundary fallback={<div>Error rendering notebook</div>}>
+        {notebook.cells.map((cell, index) => {
+          switch (cell.cell_type) {
+            case "code":
+              return (
+                <CodeCell
+                  key={cell.id || index}
+                  cell={cell}
+                  language={language}
+                />
+              );
+            case "markdown":
+              return <MarkdownCell key={cell.id || index} cell={cell} />;
+            case "raw":
+              return <RawCell key={cell.id || index} cell={cell} />;
+            default:
+              return null;
+          }
+        })}
+      </ErrorBoundary>
     </div>
   );
 }