Refine rendering of equations in chat pane

Author: faysou

frontend/package.json

@@ -92,6 +92,7 @@
     "@radix-ui/react-use-callback-ref": "~1.1.1",
     "@radix-ui/react-use-controllable-state": "~1.2.2",
     "@replit/codemirror-vim": "^6.3.0",
+    "@streamdown/math": "^1.0.1",
     "@tailwindcss/postcss": "^4.1.18",
     "@tailwindcss/typography": "^0.5.19",
     "@tanstack/react-table": "^8.21.3",
@@ -165,7 +166,7 @@
     "remark-gfm": "^4.0.1",
     "rpc-anywhere": "^1.7.0",
     "sql-formatter": "^15.6.12",
-    "streamdown": "^1.6.10",
+    "streamdown": "^2.1.0",
     "string-dedent": "^3.0.2",
     "swiper": "^11.2.10",
     "tailwind-merge": "^2.6.0",
@@ -181,6 +182,7 @@
     "vega-tooltip": "^1.1.0",
     "vscode-jsonrpc": "^8.2.1",
     "vscode-languageserver-protocol": "^3.17.5",
+    "vscode-languageserver-types": "^3.17.5",
     "web-vitals": "^4.2.4",
     "zod": "^4.3.4"
   },

frontend/src/components/markdown/markdown-renderer.tsx

@@ -1,6 +1,7 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 
 import { EditorView } from "@codemirror/view";
+import { math } from "@streamdown/math";
 import { useAtomValue } from "jotai";
 import { BetweenHorizontalStartIcon } from "lucide-react";
 import { memo, Suspense, useState } from "react";
@@ -184,7 +185,11 @@ const COMPONENTS: Components = {
 
 export const MarkdownRenderer = memo(({ content }: { content: string }) => {
   return (
-    <Streamdown components={COMPONENTS} className="mo-markdown-renderer">
+    <Streamdown
+      components={COMPONENTS}
+      plugins={{ math }}
+      className="mo-markdown-renderer"
+    >
       {content}
     </Streamdown>
   );

marimo/_server/ai/prompts.py

@@ -27,7 +27,11 @@
         "If an import already exists, do not import it again.",
         "If a variable is already defined, use another name, or make it private by adding an underscore at the beginning.",
     ],
-    "markdown": [],
+    "markdown": [
+        "Use double dollar signs ($$) for ALL mathematical expressions (both inline and block) in markdown responses.",
+        "Example: $$E=mc^2$$",
+        "Do NOT use single dollar signs or square brackets for math.",
+    ],
     "sql": [
         "The SQL must use duckdb syntax.",
     ],