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

package/src/core/ai/context/providers/error.ts

@@ -0,0 +1,165 @@
+/* Copyright 2024 Marimo. All rights reserved. */
+
+import type { Completion } from "@codemirror/autocomplete";
+import { cellErrorsAtom } from "@/core/cells/cells";
+import type { CellId } from "@/core/cells/ids";
+import type { MarimoError } from "@/core/kernel/messages";
+import type { JotaiStore } from "@/core/state/jotai";
+import { PluralWord } from "@/utils/pluralize";
+import { type AIContextItem, AIContextProvider } from "../registry";
+import { contextToXml } from "../utils";
+import { Boosts } from "./common";
+
+export interface ErrorContextItem extends AIContextItem {
+  type: "error";
+  data: {
+    type: "all-errors";
+    errors: Array<{
+      cellId: CellId;
+      cellName: string;
+      errorData: MarimoError[];
+    }>;
+  };
+}
+
+function describeError(error: MarimoError): string {
+  if (error.type === "setup-refs") {
+    return "The setup cell cannot have references";
+  }
+  if (error.type === "cycle") {
+    return "This cell is in a cycle";
+  }
+  if (error.type === "multiple-defs") {
+    return `The variable '${error.name}' was defined by another cell`;
+  }
+  if (error.type === "import-star") {
+    return error.msg;
+  }
+  if (error.type === "ancestor-stopped") {
+    return error.msg;
+  }
+  if (error.type === "ancestor-prevented") {
+    return error.msg;
+  }
+  if (error.type === "exception") {
+    return error.msg;
+  }
+  if (error.type === "strict-exception") {
+    return error.msg;
+  }
+  if (error.type === "interruption") {
+    return "This cell was interrupted and needs to be re-run";
+  }
+  if (error.type === "syntax") {
+    return error.msg;
+  }
+  if (error.type === "unknown") {
+    return error.msg;
+  }
+  return "Unknown error";
+}
+
+const errorsTxt = new PluralWord("error", "errors");
+export class ErrorContextProvider extends AIContextProvider<ErrorContextItem> {
+  readonly title = "Errors";
+  readonly mentionPrefix = "@";
+  readonly contextType = "error";
+
+  constructor(private store: JotaiStore) {
+    super();
+  }
+
+  getItems(): ErrorContextItem[] {
+    const errors = this.store.get(cellErrorsAtom);
+    if (errors.length === 0) {
+      return [];
+    }
+
+    return [
+      {
+        uri: this.asURI("all"),
+        name: "Errors",
+        type: this.contextType,
+        data: {
+          type: "all-errors",
+          errors: errors.map((error) => ({
+            cellId: error.cellId,
+            cellName: error.cellName,
+            errorData: error.output.data,
+          })),
+        },
+        description: "All errors in the notebook",
+      },
+    ];
+
+    // TODO: maybe handle single errors or grouped by types
+  }
+
+  formatCompletion(item: ErrorContextItem): Completion {
+    if (item.data.type === "all-errors") {
+      return {
+        label: "@Errors",
+        displayLabel: "Errors",
+        detail: `${item.data.errors.length} ${errorsTxt.pluralize(item.data.errors.length)}`,
+        boost: Boosts.ERROR,
+        type: "error",
+        apply: "@Errors",
+        section: "Errors",
+        info: () => {
+          const infoContainer = document.createElement("div");
+          infoContainer.classList.add(
+            "mo-cm-tooltip",
+            "docs-documentation",
+            "min-w-[200px]",
+            "flex",
+            "flex-col",
+            "gap-1",
+            "p-2",
+          );
+
+          const headerDiv = document.createElement("div");
+          headerDiv.classList.add("flex", "flex-col", "gap-1");
+
+          const nameDiv = document.createElement("div");
+          nameDiv.classList.add("font-bold", "text-base");
+          nameDiv.textContent = "Errors";
+          headerDiv.append(nameDiv);
+
+          const descriptionDiv = document.createElement("div");
+          descriptionDiv.classList.add("text-sm", "text-muted-foreground");
+          descriptionDiv.textContent = `${item.data.errors.length} ${errorsTxt.pluralize(item.data.errors.length)}`;
+          headerDiv.append(descriptionDiv);
+
+          infoContainer.append(headerDiv);
+
+          return infoContainer;
+        },
+      };
+    }
+
+    return {
+      label: "Error",
+      displayLabel: "Error",
+      boost: Boosts.ERROR,
+    };
+  }
+
+  formatContext(item: ErrorContextItem): string {
+    const { data } = item;
+
+    const xmls = data.errors.map((err) => {
+      return contextToXml({
+        type: this.contextType,
+        data: {
+          name: err.cellName || `Cell ${err.cellId}`,
+          description: err.errorData
+            .map((err) => describeError(err))
+            .join("\n"),
+        },
+      });
+    });
+
+    const xml = xmls.join("\n\n");
+    return xml;
+  }
+}

package/src/core/ai/context/providers/tables.ts

@@ -5,6 +5,7 @@ import type { DatasetTablesMap } from "@/core/datasets/data-source-connections";
 import type { DataTable } from "@/core/kernel/messages";
 import type { AIContextItem } from "../registry";
 import { AIContextProvider } from "../registry";
+import { contextToXml } from "../utils";
 import { Boosts } from "./common";
 
 export interface TableContextItem extends AIContextItem {
@@ -32,8 +33,8 @@ export class TableContextProvider extends AIContextProvider<TableContextItem> {
   }
 
   formatContext(item: TableContextItem): string {
-    const { data, uri: id } = item;
-    const { columns, source, num_rows, num_columns } = data;
+    const { data } = item;
+    const { columns, source, num_rows, num_columns, name } = data;
     const shape = [
       num_rows == null ? undefined : `${num_rows} rows`,
       num_columns == null ? undefined : `${num_columns} columns`,
@@ -41,16 +42,23 @@ export class TableContextProvider extends AIContextProvider<TableContextItem> {
       .filter(Boolean)
       .join(", ");
 
-    let context = `Table: ${id}\nSource: ${source || "unknown"}`;
+    let details = "";
     if (shape) {
-      context += `\nShape: ${shape}`;
+      details += `\nShape: ${shape}`;
     }
 
     if (columns && columns.length > 0) {
-      context += `\nColumns:\n${columns.map((col) => `  - ${col.name}: ${col.type}`).join("\n")}`;
+      details += `\nColumns:\n${columns.map((col) => `  - ${col.name}: ${col.type}`).join("\n")}`;
     }
 
-    return context;
+    return contextToXml({
+      type: this.contextType,
+      data: {
+        name: name,
+        source: source ?? "unknown",
+      },
+      details: details,
+    });
   }
 
   formatCompletion(item: TableContextItem): Completion {

package/src/core/ai/context/providers/variable.ts

@@ -5,6 +5,7 @@ import { createVariableInfoElement } from "@/core/codemirror/completion/variable
 import type { DatasetTablesMap } from "@/core/datasets/data-source-connections";
 import type { Variable, Variables } from "@/core/variables/types";
 import { type AIContextItem, AIContextProvider } from "../registry";
+import { contextToXml } from "../utils";
 import { Boosts } from "./common";
 
 export interface VariableContextItem extends AIContextItem {
@@ -65,8 +66,16 @@ export class VariableContextProvider extends AIContextProvider<VariableContextIt
   }
 
   formatContext(item: VariableContextItem): string {
-    const { uri: id, data } = item;
+    const { data } = item;
     const { variable } = data;
-    return `Variable: ${id}\nType: ${variable.dataType || "unknown"}\nPreview: ${JSON.stringify(variable.value)}`;
+    return contextToXml({
+      type: this.contextType,
+      data: {
+        name: variable.name,
+        dataType: variable.dataType,
+      },
+      details:
+        variable.value == null ? undefined : JSON.stringify(variable.value),
+    });
   }
 }

package/src/core/ai/context/utils.ts

@@ -0,0 +1,49 @@
+/* Copyright 2024 Marimo. All rights reserved. */
+
+export interface AiContextPayload {
+  type: string;
+  data: Record<string, unknown>;
+  details?: string;
+}
+
+// XML escaping utility
+function escapeXml(unsafe: string): string {
+  return (
+    unsafe
+      // We don't escape these characters because this is for an LLM and they can interpret this just fine.
+      // .replaceAll("&", "&amp;")
+      // .replaceAll('"', "&quot;")
+      // .replaceAll("'", "&#39;")
+      .replaceAll("<", "&lt;")
+      .replaceAll(">", "&gt;")
+  );
+}
+
+// Convert a single Context object to XML
+export function contextToXml(context: AiContextPayload): string {
+  const { type, data, details } = context;
+
+  // Start with opening tag
+  let xml = `<${type}`;
+
+  // Add data as attributes
+  for (const [key, value] of Object.entries(data)) {
+    const escapedValue = escapeXml(String(value));
+    if (value !== undefined) {
+      xml += ` ${key}="${escapedValue}"`;
+    }
+  }
+
+  // Close the opening tag
+  xml += ">";
+
+  // Add details as content if present
+  if (details) {
+    xml += escapeXml(details);
+  }
+
+  // Add closing tag
+  xml += `</${type}>`;
+
+  return xml;
+}

package/src/core/cells/cells.ts

@@ -1489,7 +1489,7 @@ export const canUndoDeletesAtom = atom((get) =>
 
 export const needsRunAtom = atom((get) => notebookNeedsRun(get(notebookAtom)));
 
-const cellErrorsAtom = atom((get) => {
+export const cellErrorsAtom = atom((get) => {
   const { cellIds, cellRuntime, cellData } = get(notebookAtom);
   const errors = cellIds.inOrderIds
     .map((cellId) => {

package/dist/assets/_baseMap-u5TY8JMW.js

@@ -1 +0,0 @@
-import{b as m}from"./_baseEach-DJPM6ZzY.js";import{x as s}from"./index-CQFSXxPp.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-BIZI32eZ.js

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

package/dist/assets/classDiagram-3BZAVTQC-DfPpF-gd.js

@@ -1 +0,0 @@
-import{s as a,c as s,a as t,C as o}from"./chunk-JBRWN2VN-CB8a_SQv.js";import{_ as e}from"./mermaid-CUkYs7pX.js";import"./transform-B8bpuzxV.js";import"./chunk-GLLZNHP4-CRWkmhQj.js";import"./chunk-WVR4S24B-ZnYlHppS.js";import"./chunk-NRVI72HA-nFtyi9pH.js";import"./index-CQFSXxPp.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-DfPpF-gd.js

@@ -1 +0,0 @@
-import{s as a,c as s,a as t,C as o}from"./chunk-JBRWN2VN-CB8a_SQv.js";import{_ as e}from"./mermaid-CUkYs7pX.js";import"./transform-B8bpuzxV.js";import"./chunk-GLLZNHP4-CRWkmhQj.js";import"./chunk-WVR4S24B-ZnYlHppS.js";import"./chunk-NRVI72HA-nFtyi9pH.js";import"./index-CQFSXxPp.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-yHYNh3QG.js

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

package/dist/assets/infoDiagram-6WOFNB3A-BQOY1VMN.js

@@ -1,2 +0,0 @@
-import{_ as a,l as i,I as s,d as p,K as n}from"./mermaid-CUkYs7pX.js";import{p as m}from"./treemap-75Q7IDZK-BBbbCdps.js";import"./index-CQFSXxPp.js";import"./transform-B8bpuzxV.js";import"./timer-BwIYMJWC.js";import"./step-BwsUM5iJ.js";import"./_baseEach-DJPM6ZzY.js";import"./_baseUniq-CrXjB6JP.js";import"./min-Yuyj_o4k.js";import"./_baseMap-u5TY8JMW.js";import"./clone-yHYNh3QG.js";import"./_createAggregator-Cd0noqMc.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-DX5NRIHK.js

@@ -1 +0,0 @@
-import{s as t,b as a,a as e,S as o}from"./chunk-LXBSTHXV-YNRk01ob.js";import{_ as s}from"./mermaid-CUkYs7pX.js";import"./transform-B8bpuzxV.js";import"./chunk-WVR4S24B-ZnYlHppS.js";import"./chunk-NRVI72HA-nFtyi9pH.js";import"./index-CQFSXxPp.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};