@marimo-team/islands 0.23.14-dev19 → 0.23.14-dev21

package/dist/{process-output-Dw6We6B6.js → process-output-B9ysqk1y.js}

@@ -1,6 +1,6 @@
 import { s as __toESM } from "./chunk-BNovOVIE.js";
 import { t as require_compiler_runtime } from "./compiler-runtime-CEbnTgxf.js";
-import { at as parseHtmlContent, it as ansiToPlainText } from "./html-to-image-DBHfshED.js";
+import { at as parseHtmlContent, it as ansiToPlainText } from "./html-to-image-DLSOS-bE.js";
 import { u as createLucideIcon } from "./dist--2Bqjvs0.js";
 import { t as Strings } from "./strings-Dq_j3Rxw.js";
 import { t as require_jsx_runtime } from "./jsx-runtime-DebpN0FN.js";

package/dist/{reveal-component-tJhA77Am.js → reveal-component-BJwt0B2_.js}

@@ -6,10 +6,10 @@ import { s as __toESM } from "./chunk-BNovOVIE.js";
 import { _ as Logger, g as cn, h as Events, l as useEventListener, t as Button } from "./button-BacYv-bE.js";
 import { t as require_react } from "./react-DA-nE2FX.js";
 import { t as require_compiler_runtime } from "./compiler-runtime-CEbnTgxf.js";
-import { lt as kioskModeAtom } from "./html-to-image-DBHfshED.js";
+import { lt as kioskModeAtom } from "./html-to-image-DLSOS-bE.js";
 import "./chunk-5FQGJX7Z-BbqSm5gU.js";
 import { u as createLucideIcon } from "./dist--2Bqjvs0.js";
-import { a as DEFAULT_SLIDE_TYPE, c as Slide, ct as PanelGroup, i as DEFAULT_DECK_TRANSITION, lt as PanelResizeHandle, on as EyeOff, s as SlideSidebar, sn as Expand, st as Panel, t as useNotebookCodeAvailable, un as Code } from "./code-visibility-CxDvzm2S.js";
+import { a as DEFAULT_SLIDE_TYPE, c as Slide, ct as PanelGroup, i as DEFAULT_DECK_TRANSITION, lt as PanelResizeHandle, on as EyeOff, s as SlideSidebar, sn as Expand, st as Panel, t as useNotebookCodeAvailable, un as Code } from "./code-visibility-CPWEmIAX.js";
 import { X as useDebouncedCallback } from "./input-BSdZp5Ng.js";
 import "./toDate-D1Z7ZXWh.js";
 import "./react-dom-BTJzcVJ9.js";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@marimo-team/islands",
-  "version": "0.23.14-dev19",
+  "version": "0.23.14-dev21",
   "main": "dist/main.js",
   "types": "dist/index.d.ts",
   "type": "module",
@@ -56,7 +56,7 @@
     "@lezer/python": "^1.1.18",
     "@marimo-team/codemirror-ai": "^0.3.7",
     "@marimo-team/codemirror-languageserver": "^1.16.12",
-    "@marimo-team/codemirror-mcp": "^0.1.5",
+    "@marimo-team/codemirror-mcp": "^0.1.7",
     "@marimo-team/codemirror-sql": "^0.2.8",
     "@marimo-team/llm-info": "workspace:*",
     "@marimo-team/marimo-api": "workspace:*",

package/src/components/editor/errors/__tests__/auto-fix.test.ts

@@ -1,119 +1,23 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 
-import { beforeEach, describe, expect, it, vi } from "vitest";
+import { describe, expect, it } from "vitest";
 import { cellId } from "@/__tests__/branded";
-import type { MarimoError } from "@/core/kernel/messages";
+import { buildFixInChatPrompt } from "../auto-fix";
 
-const getDatasourceContext = vi.fn<(id: unknown) => string | null>(() => null);
-
-vi.mock("@/core/ai/context/providers/datasource", () => ({
-  getDatasourceContext: (id: unknown) => getDatasourceContext(id),
-}));
-
-const { buildFixPrompt, buildFixPromptFromText } = await import("../auto-fix");
-
-describe("buildFixPromptFromText", () => {
-  beforeEach(() => {
-    getDatasourceContext.mockReset();
-    getDatasourceContext.mockReturnValue(null);
-  });
-
-  it("includes the cell id in the header when provided", () => {
-    const prompt = buildFixPromptFromText("boom", cellId("cell-1"));
-    expect(prompt).toBe(
-      "My cell (id: cell-1) produced the following error. Please fix it:\n\nboom",
-    );
-  });
-
-  it("uses a generic header when no cell id is provided", () => {
-    const prompt = buildFixPromptFromText("boom");
-    expect(prompt).toBe(
-      "My code gives the following error. Please fix it:\n\nboom",
+describe("buildFixInChatPrompt", () => {
+  it("references the per-cell error context URI", () => {
+    expect(buildFixInChatPrompt(cellId("cell-1"))).toBe(
+      "@error://cell-1\n\nPlease fix this error.",
     );
   });
 
-  it("appends datasource context when explicitly requested for the cell", () => {
-    getDatasourceContext.mockReturnValue("@datasource://my_db");
-    const prompt = buildFixPromptFromText("boom", cellId("cell-1"), {
-      includeDatasourceContext: true,
-    });
-    expect(prompt).toBe(
-      "My cell (id: cell-1) produced the following error. Please fix it:\n\nboom\n\nDatabase schema: @datasource://my_db",
+  it("falls back to inline error text when no cell id", () => {
+    expect(buildFixInChatPrompt(undefined, "ValueError: boom")).toBe(
+      "My code gives the following error. Please fix it:\n\nValueError: boom",
     );
   });
 
-  it("does not append datasource context by default", () => {
-    getDatasourceContext.mockReturnValue("@datasource://my_db");
-    const prompt = buildFixPromptFromText("boom", cellId("cell-1"));
-    expect(prompt).toBe(
-      "My cell (id: cell-1) produced the following error. Please fix it:\n\nboom",
-    );
-    expect(getDatasourceContext).not.toHaveBeenCalled();
-  });
-
-  it("does not look up datasource context without a cell id", () => {
-    buildFixPromptFromText("boom", undefined, {
-      includeDatasourceContext: true,
-    });
-    expect(getDatasourceContext).not.toHaveBeenCalled();
-  });
-});
-
-describe("buildFixPrompt", () => {
-  beforeEach(() => {
-    getDatasourceContext.mockReset();
-    getDatasourceContext.mockReturnValue(null);
-  });
-
-  it("uses the error message when present", () => {
-    const errors: MarimoError[] = [
-      { type: "sql-error", msg: "syntax error", sql_statement: "SELECT" },
-    ];
-    expect(buildFixPrompt(errors, cellId("cell-1"))).toBe(
-      "My cell (id: cell-1) produced the following error. Please fix it:\n\nsyntax error",
-    );
-  });
-
-  it("joins multiple errors with newlines", () => {
-    const errors: MarimoError[] = [
-      { type: "syntax", msg: "bad syntax" },
-      {
-        type: "exception",
-        exception_type: "ValueError",
-        msg: "bad value",
-        raising_cell: null,
-      },
-    ];
-    expect(buildFixPrompt(errors, cellId("cell-1"))).toBe(
-      "My cell (id: cell-1) produced the following error. Please fix it:\n\nbad syntax\nbad value",
-    );
-  });
-
-  it("falls back to the error type when there is no message", () => {
-    const errors: MarimoError[] = [
-      { type: "multiple-defs", name: "foo", cells: [cellId("foo")] },
-    ];
-    expect(buildFixPrompt(errors, cellId("cell-1"))).toBe(
-      "My cell (id: cell-1) produced the following error. Please fix it:\n\nmultiple-defs",
-    );
-  });
-
-  it("appends datasource context for SQL errors", () => {
-    getDatasourceContext.mockReturnValue("@datasource://my_db");
-    const errors: MarimoError[] = [
-      { type: "sql-error", msg: "syntax error", sql_statement: "SELECT" },
-    ];
-    expect(buildFixPrompt(errors, cellId("cell-1"))).toBe(
-      "My cell (id: cell-1) produced the following error. Please fix it:\n\nsyntax error\n\nDatabase schema: @datasource://my_db",
-    );
-  });
-
-  it("does not append datasource context for non-SQL errors", () => {
-    getDatasourceContext.mockReturnValue("@datasource://my_db");
-    const errors: MarimoError[] = [{ type: "syntax", msg: "bad syntax" }];
-    expect(buildFixPrompt(errors, cellId("cell-1"))).toBe(
-      "My cell (id: cell-1) produced the following error. Please fix it:\n\nbad syntax",
-    );
-    expect(getDatasourceContext).not.toHaveBeenCalled();
+  it("uses a generic prompt when no cell id and no error text", () => {
+    expect(buildFixInChatPrompt(undefined)).toBe("Please fix this error.");
   });
 });

package/src/components/editor/errors/auto-fix.tsx

@@ -21,44 +21,24 @@ import { aiCompletionCellAtom } from "@/core/ai/state";
 import { notebookAtom, useCellActions } from "@/core/cells/cells";
 import type { CellId } from "@/core/cells/ids";
 import { aiFeaturesEnabledAtom } from "@/core/config/config";
-import { getDatasourceContext } from "@/core/ai/context/providers/datasource";
 import { getAutoFixes } from "@/core/errors/errors";
 import type { MarimoError } from "@/core/kernel/messages";
 import { cn } from "@/utils/cn";
 import { useOpenAiAssistant } from "../chrome/wrapper/useOpenAiAssistant";
 import { type FixMode, useFixMode } from "./fix-mode";
 
-export function buildFixPromptFromText(
-  errorText: string,
-  cellId?: CellId,
-  {
-    includeDatasourceContext = false,
-  }: { includeDatasourceContext?: boolean } = {},
+export function buildFixInChatPrompt(
+  cellId: CellId | undefined,
+  fallbackErrorText?: string,
 ): string {
-  const header =
-    cellId != null
-      ? `My cell (id: ${cellId}) produced the following error. Please fix it:`
-      : "My code gives the following error. Please fix it:";
-  let prompt = `${header}\n\n${errorText}`;
-  if (cellId != null && includeDatasourceContext) {
-    const datasourceContext = getDatasourceContext(cellId);
-    if (datasourceContext) {
-      prompt += `\n\nDatabase schema: ${datasourceContext}`;
-    }
+  if (cellId != null) {
+    return `@error://${cellId}\n\nPlease fix this error.`;
   }
-  return prompt;
-}
-
-export function buildFixPrompt(errors: MarimoError[], cellId: CellId): string {
-  const errorText = errors
-    .map((error) => ("msg" in error && error.msg ? error.msg : error.type))
-    .join("\n");
-  const includeDatasourceContext = errors.some(
-    (error) => error.type === "sql-error",
-  );
-  return buildFixPromptFromText(errorText, cellId, {
-    includeDatasourceContext,
-  });
+  const errorText = fallbackErrorText?.trim();
+  if (errorText) {
+    return `My code gives the following error. Please fix it:\n\n${errorText}`;
+  }
+  return "Please fix this error.";
 }
 
 export const AutoFixButton = ({
@@ -112,7 +92,7 @@ export const AutoFixButton = ({
 
   const openAISidebar = () => {
     openAiAssistant({
-      prompt: buildFixPrompt(errors, cellId),
+      prompt: buildFixInChatPrompt(cellId),
     });
   };
 

package/src/components/editor/output/MarimoTracebackOutput.tsx

@@ -42,7 +42,7 @@ import {
   getTracebackInfo,
 } from "@/utils/traceback";
 import { useOpenAiAssistant } from "../chrome/wrapper/useOpenAiAssistant";
-import { AIFixButton, buildFixPromptFromText } from "../errors/auto-fix";
+import { AIFixButton, buildFixInChatPrompt } from "../errors/auto-fix";
 import { MangledSegments } from "../errors/mangled-local-chip";
 import { CellLinkTraceback } from "../links/cell-link";
 import type { OnRefactorWithAI } from "../Output";
@@ -101,7 +101,7 @@ export const MarimoTracebackOutput = ({
 
   const openAISidebar = () => {
     openAiAssistant({
-      prompt: buildFixPromptFromText(lastTracebackLine, cellId),
+      prompt: buildFixInChatPrompt(cellId, lastTracebackLine),
     });
   };
 

package/src/core/ai/context/__tests__/registry.test.ts

@@ -176,6 +176,33 @@ class FileContextProvider extends AIContextProvider<FileContextItem> {
   }
 }
 
+// Second mock provider with the same context type as MockContextProvider
+class SecondaryMockProvider extends AIContextProvider<MockContextItem> {
+  readonly title = "Secondary Mock Items";
+  readonly mentionPrefix = "@";
+  readonly contextType = "mock";
+
+  getItems(): MockContextItem[] {
+    return [
+      {
+        type: "mock",
+        uri: this.asURI("secondary-item"),
+        name: "Secondary Item",
+        description: "From the second mock provider",
+        data: { value: "secondary" },
+      },
+    ];
+  }
+
+  formatContext(item: MockContextItem): string {
+    return `Secondary: ${item.name}`;
+  }
+
+  formatCompletion(item: MockContextItem): Completion {
+    return this.createBasicCompletion(item);
+  }
+}
+
 describe("AIContextProvider", () => {
   let provider: MockContextProvider;
 
@@ -411,6 +438,95 @@ describe("AIContextRegistry", () => {
     });
   });
 
+  describe("resolveItems", () => {
+    let mockGetItems: ReturnType<typeof vi.spyOn>;
+    let fileGetItems: ReturnType<typeof vi.spyOn>;
+
+    beforeEach(() => {
+      registry.register(mockProvider);
+      registry.register(fileProvider);
+      mockGetItems = vi.spyOn(mockProvider, "getItems");
+      fileGetItems = vi.spyOn(fileProvider, "getItems");
+    });
+
+    it("should resolve valid IDs from the matching providers only", () => {
+      const resolved = registry.resolveItems([
+        "mock://item1",
+        "file://config.py",
+      ] as ContextLocatorId[]);
+
+      expect(resolved).toHaveLength(2);
+      expect(resolved.map((item) => item.uri)).toEqual([
+        "mock://item1",
+        "file://config.py",
+      ]);
+      expect(mockGetItems).toHaveBeenCalledTimes(1);
+      expect(fileGetItems).toHaveBeenCalledTimes(1);
+    });
+
+    it("should not call unrelated providers", () => {
+      registry.resolveItems(["mock://item1"] as ContextLocatorId[]);
+
+      expect(mockGetItems).toHaveBeenCalledTimes(1);
+      expect(fileGetItems).not.toHaveBeenCalled();
+    });
+
+    it("should ignore unknown or invalid URIs", () => {
+      const resolved = registry.resolveItems([
+        "mock://nonexistent",
+        "unknown://item",
+        "malformed" as ContextLocatorId,
+      ] as ContextLocatorId[]);
+
+      expect(resolved).toEqual([]);
+      expect(mockGetItems).toHaveBeenCalledTimes(1);
+      expect(fileGetItems).not.toHaveBeenCalled();
+    });
+
+    it("should return empty array for empty input", () => {
+      expect(registry.resolveItems([])).toEqual([]);
+      expect(mockGetItems).not.toHaveBeenCalled();
+      expect(fileGetItems).not.toHaveBeenCalled();
+    });
+
+    it("should preserve requested order when providers are interleaved", () => {
+      const resolved = registry.resolveItems([
+        "mock://item1",
+        "file://config.py",
+        "mock://item2",
+        "file://utils/helpers.py",
+      ] as ContextLocatorId[]);
+
+      expect(resolved.map((item) => item.uri)).toEqual([
+        "mock://item1",
+        "file://config.py",
+        "mock://item2",
+        "file://utils/helpers.py",
+      ]);
+      // Each provider is still queried only once, regardless of interleaving.
+      expect(mockGetItems).toHaveBeenCalledTimes(1);
+      expect(fileGetItems).toHaveBeenCalledTimes(1);
+    });
+
+    it("should resolve items from all providers that share a context type", () => {
+      const secondaryProvider = new SecondaryMockProvider();
+      const secondaryGetItems = vi.spyOn(secondaryProvider, "getItems");
+      registry.register(secondaryProvider);
+
+      const resolved = registry.resolveItems([
+        "mock://item1",
+        "mock://secondary-item",
+      ] as ContextLocatorId[]);
+
+      expect(resolved.map((item) => item.uri)).toEqual([
+        "mock://item1",
+        "mock://secondary-item",
+      ]);
+      expect(mockGetItems).toHaveBeenCalledTimes(1);
+      expect(secondaryGetItems).toHaveBeenCalledTimes(1);
+    });
+  });
+
   describe("getContextInfo", () => {
     beforeEach(() => {
       registry.register(mockProvider);
@@ -531,6 +647,19 @@ describe("AIContextRegistry", () => {
       expect(formatted).toContain("Mock: Item 1 (value1)");
       expect(formatted.split("\n\n")).toHaveLength(1);
     });
+
+    it("should format items from the provider that owns them", () => {
+      registry.register(new SecondaryMockProvider());
+
+      const formatted = registry.formatContextForAI([
+        "mock://item1",
+        "mock://secondary-item",
+      ] as ContextLocatorId[]);
+
+      expect(formatted).toBe(
+        "Mock: Item 1 (value1)\n\nSecondary: Secondary Item",
+      );
+    });
   });
 
   describe("edge cases and error handling", () => {
@@ -621,7 +750,7 @@ describe("AIContextRegistry", () => {
       expect(provider).toBe(errorProvider);
 
       const items = registry.getAllItems();
-      expect(items).toHaveLength(1);
+      expect(items).toHaveLength(3);
       expect(items[0].type).toBe("error");
       expect(items[0].name).toBe("Errors");
     });
@@ -654,10 +783,11 @@ describe("AIContextRegistry", () => {
       expect(provider).toBeDefined();
 
       const items = provider!.getItems();
-      expect(items).toHaveLength(1);
+      expect(items).toHaveLength(3);
 
       const completion = provider!.formatCompletion(items[0]);
       expect(completion.label).toBe("@Errors");
+      expect(completion.apply).toBe("@error://all");
       expect(completion.type).toBe("error");
     });
 
@@ -690,7 +820,7 @@ describe("AIContextRegistry", () => {
       const errorItems = items.filter((item) => item.type === "error");
       const mockItems = items.filter((item) => item.type === "mock");
 
-      expect(errorItems).toHaveLength(1);
+      expect(errorItems).toHaveLength(3);
       expect(mockItems).toHaveLength(3);
     });
 

package/src/core/ai/context/providers/__tests__/datasource.test.ts

@@ -1,6 +1,7 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 /* oxlint-disable typescript/no-explicit-any */
 
+import { createStore } from "jotai";
 import { beforeEach, describe, expect, it } from "vitest";
 import { cellId, variableName } from "@/__tests__/branded";
 import type {
@@ -634,7 +635,8 @@ describe("DatasourceContextProvider", () => {
 
 describe("getDatasourceContext", () => {
   it("should return null if no cell ID is found", () => {
-    const context = getDatasourceContext(cellId("1"));
+    const store = createStore();
+    const context = getDatasourceContext(cellId("1"), store);
     expect(context).toBeNull();
   });
 });