@marimo-team/frontend 0.15.1-dev30 → 0.15.1-dev33

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/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/infoDiagram-6WOFNB3A-DRPEjZ3d.js

@@ -1,2 +0,0 @@
-import{_ as a,l as i,I as s,d as p,K as n}from"./mermaid-Bu7PGvSJ.js";import{p as m}from"./treemap-75Q7IDZK-Bfo3G5Mq.js";import"./index-CTYfpi2J.js";import"./transform-B8bpuzxV.js";import"./timer-BwIYMJWC.js";import"./step-BwsUM5iJ.js";import"./_baseEach-D3nH_2xn.js";import"./_baseUniq-CQqCX6A5.js";import"./min-BIbDgP5-.js";import"./_baseMap-DLtMmtEG.js";import"./clone-BwvK2dmW.js";import"./_createAggregator-CcFqySRL.js";var d={parse:a(async r=>{const t=await m("info",r);i.debug(t)},"parse")},g={version:n.version+""},v={parser:d,db:{getVersion:a(()=>g.version,"getVersion")},renderer:{draw:a((r,t,o)=>{i.debug(`rendering info diagram
-`+r);const e=s(t);p(e,100,400,!0),e.append("g").append("text").attr("x",100).attr("y",40).attr("class","version").attr("font-size",32).style("text-anchor","middle").text(`v${o}`)},"draw")}};export{v as diagram};

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};