@marimo-team/frontend 0.15.1-dev32 → 0.15.1-dev34

package/src/components/editor/ai/add-cell-with-ai.tsx

@@ -9,6 +9,7 @@ import {
 import { markdown } from "@codemirror/lang-markdown";
 import { sql } from "@codemirror/lang-sql";
 import { Prec } from "@codemirror/state";
+import { promptHistory, storePrompt } from "@marimo-team/codemirror-ai";
 import ReactCodeMirror, {
   EditorView,
   keymap,
@@ -26,8 +27,9 @@ import {
   SparklesIcon,
   XIcon,
 } from "lucide-react";
-import { useMemo, useState } from "react";
+import { useMemo, useRef, useState } from "react";
 import useEvent from "react-use-event-hook";
+import { z } from "zod";
 import { AIModelDropdown } from "@/components/ai/ai-model-dropdown";
 import { Button } from "@/components/ui/button";
 import {
@@ -48,6 +50,7 @@ import { useRuntimeManager } from "@/core/runtime/config";
 import { useTheme } from "@/theme/useTheme";
 import { cn } from "@/utils/cn";
 import { prettyError } from "@/utils/errors";
+import { ZodLocalStorage } from "@/utils/localStorage";
 import { useCellActions } from "../../../core/cells/cells";
 import { PythonIcon } from "../cell/code/icons";
 import {
@@ -71,6 +74,10 @@ const languageAtom = atomWithStorage<"python" | "sql">(
   "python",
 );
 
+const KEY = "marimo:ai-prompt-history";
+// Store the prompt history in local storage
+const promptHistoryStorage = new ZodLocalStorage(z.array(z.string()), () => []);
+
 /**
  * Add a cell with AI.
  */
@@ -86,6 +93,7 @@ export const AddCellWithAI: React.FC<{
   const ai = useAtomValue(aiAtom);
   const editModel = ai?.models?.edit_model || DEFAULT_AI_MODEL;
   const { saveModelChange } = useModelChange();
+  const inputRef = useRef<ReactCodeMirrorRef>(null);
 
   const {
     completion,
@@ -120,6 +128,9 @@ export const AddCellWithAI: React.FC<{
 
   const submit = () => {
     if (!isLoading) {
+      if (inputRef.current?.view) {
+        storePrompt(inputRef.current.view);
+      }
       handleSubmit();
     }
   };
@@ -185,6 +196,7 @@ export const AddCellWithAI: React.FC<{
     <div className="flex items-center px-3">
       <SparklesIcon className="size-4 text-(--blue-11) mr-2" />
       <PromptInput
+        inputRef={inputRef}
         onClose={() => {
           setCompletion("");
           onClose();
@@ -338,6 +350,12 @@ export const PromptInput = ({
       markdownLanguage.language.data.of({
         autocomplete: additionalCompletionsSource,
       }),
+      promptHistory({
+        storage: {
+          load: () => promptHistoryStorage.get(KEY),
+          save: (prompts) => promptHistoryStorage.set(KEY, prompts),
+        },
+      }),
       EditorView.lineWrapping,
       minimalSetup(),
       Prec.highest(
@@ -392,21 +410,6 @@ export const PromptInput = ({
           },
         },
       ]),
-      // Trap arrow up/down to prevent them from being used to navigate the editor
-      keymap.of([
-        {
-          key: "ArrowUp",
-          preventDefault: true,
-          stopPropagation: true,
-        },
-      ]),
-      keymap.of([
-        {
-          key: "ArrowDown",
-          preventDefault: true,
-          stopPropagation: true,
-        },
-      ]),
     ];
   }, [store, additionalCompletionsSource, handleSubmit, handleEscape]);
 

package/src/components/editor/ai/ai-completion-editor.tsx

@@ -9,6 +9,7 @@ import { Button } from "@/components/ui/button";
 import { customPythonLanguageSupport } from "@/core/codemirror/language/languages/python";
 
 import "./merge-editor.css";
+import { storePrompt } from "@marimo-team/codemirror-ai";
 import type { ReactCodeMirrorRef } from "@uiw/react-codemirror";
 import { useAtom } from "jotai";
 import { Checkbox } from "@/components/ui/checkbox";
@@ -175,6 +176,9 @@ export const AiCompletionEditor: React.FC<Props> = ({
               }}
               onSubmit={() => {
                 if (!isLoading) {
+                  if (inputRef.current?.view) {
+                    storePrompt(inputRef.current.view);
+                  }
                   handleSubmit();
                 }
               }}

package/src/core/dom/outline.ts

@@ -8,9 +8,23 @@ import type { OutputMessage } from "../kernel/messages";
 // Tags that we don't want to include in the outline
 const excludedTags = ["marimo-carousel", "marimo-tabs", "marimo-accordion"];
 
-function getOutline(html: string): Outline {
-  const items: Outline["items"] = [];
+/**
+ * Extracts a table of contents {@link Outline} from an HTML string.
+ *
+ * Each {@link OutlineItem} corresponds to a heading (h1–h6) found in the input.
+ *
+ * @param html - The HTML content to parse.
+ * @returns An {@link Outline}, otherwise `null` if parsing via isn't supported in JS runtime (e.g., Node).
+ */
+function getOutline(html: string): Outline | null {
+  // Some JS runtimes (e.g. Node.js for the VSCode extension) do not
+  // expose DOMParser globally. In those environments parsing isn't
+  // possible (without a polyfill), so return just return `null`.
+  if (typeof DOMParser === "undefined") {
+    return null;
+  }
 
+  const items: Outline["items"] = [];
   const parser = new DOMParser();
   const document = parser.parseFromString(html, "text/html");
 

package/dist/assets/_baseMap-DLtMmtEG.js

@@ -1 +0,0 @@
-import{b as m}from"./_baseEach-D3nH_2xn.js";import{x as s}from"./index-CTYfpi2J.js";function e(r,o){var a=-1,t=s(r)?Array(r.length):[];return m(r,function(n,f,i){t[++a]=o(n,f,i)}),t}export{e as b};

package/dist/assets/channel-7kaxt7IF.js

@@ -1 +0,0 @@
-import{U as s,C as o}from"./mermaid-Bu7PGvSJ.js";const n=(a,r)=>s.lang.round(o.parse(a)[r]);export{n as c};

package/dist/assets/classDiagram-3BZAVTQC-Djy0cY4g.js

@@ -1 +0,0 @@
-import{s as a,c as s,a as t,C as o}from"./chunk-JBRWN2VN-bXiYdBHm.js";import{_ as e}from"./mermaid-Bu7PGvSJ.js";import"./transform-B8bpuzxV.js";import"./chunk-GLLZNHP4-Cit8A-EP.js";import"./chunk-WVR4S24B-BdZs7vKR.js";import"./chunk-NRVI72HA-Caq7CUfs.js";import"./index-CTYfpi2J.js";import"./step-BwsUM5iJ.js";import"./timer-BwIYMJWC.js";var i={parser:t,get db(){return new o},renderer:s,styles:a,init:e(r=>{r.class||(r.class={}),r.class.arrowMarkerAbsolute=r.arrowMarkerAbsolute},"init")};export{i as diagram};

package/dist/assets/classDiagram-v2-QTMF73CY-Djy0cY4g.js

@@ -1 +0,0 @@
-import{s as a,c as s,a as t,C as o}from"./chunk-JBRWN2VN-bXiYdBHm.js";import{_ as e}from"./mermaid-Bu7PGvSJ.js";import"./transform-B8bpuzxV.js";import"./chunk-GLLZNHP4-Cit8A-EP.js";import"./chunk-WVR4S24B-BdZs7vKR.js";import"./chunk-NRVI72HA-Caq7CUfs.js";import"./index-CTYfpi2J.js";import"./step-BwsUM5iJ.js";import"./timer-BwIYMJWC.js";var i={parser:t,get db(){return new o},renderer:s,styles:a,init:e(r=>{r.class||(r.class={}),r.class.arrowMarkerAbsolute=r.arrowMarkerAbsolute},"init")};export{i as diagram};

package/dist/assets/clone-BwvK2dmW.js

@@ -1 +0,0 @@
-import{b as n}from"./_baseUniq-CQqCX6A5.js";function o(r){return n(r,4)}export{o as c};

package/dist/assets/stateDiagram-v2-EYPG3UTE-CT0Gpq7L.js

@@ -1 +0,0 @@
-import{s as t,b as a,a as e,S as o}from"./chunk-LXBSTHXV-BP4Soimz.js";import{_ as s}from"./mermaid-Bu7PGvSJ.js";import"./transform-B8bpuzxV.js";import"./chunk-WVR4S24B-BdZs7vKR.js";import"./chunk-NRVI72HA-Caq7CUfs.js";import"./index-CTYfpi2J.js";import"./step-BwsUM5iJ.js";import"./timer-BwIYMJWC.js";var i={parser:e,get db(){return new o(2)},renderer:a,styles:t,init:s(r=>{r.state||(r.state={}),r.state.arrowMarkerAbsolute=r.arrowMarkerAbsolute},"init")};export{i as diagram};