@marimo-team/islands 0.25.1-dev26 → 0.25.1-dev28

package/src/core/ai/staged-cells.ts

@@ -305,11 +305,11 @@ class StagedCellReconciler {
   private readonly actions: StagedCellReconcilerActions;
   private marimoImportCellId: CellId | null = null;
 
-  constructor(actions: StagedCellReconcilerActions) {
+  public constructor(actions: StagedCellReconcilerActions) {
     this.actions = actions;
   }
 
-  reconcile(completionCells: GeneratedCell[]) {
+  public reconcile(completionCells: GeneratedCell[]) {
     for (const [idx, cell] of completionCells.entries()) {
       if (idx < this.createdCells.length) {
         const existingCell = this.createdCells[idx];
@@ -336,7 +336,7 @@ class StagedCellReconciler {
     this.syncMarimoImport(completionCells);
   }
 
-  discard(stagedCells: StagedAICells) {
+  public discard(stagedCells: StagedAICells) {
     for (const { cellId } of this.createdCells.toReversed()) {
       if (stagedCells.has(cellId)) {
         this.actions.deleteStagedCell(cellId);
@@ -350,17 +350,17 @@ class StagedCellReconciler {
     this.marimoImportCellId = null;
   }
 
-  accept() {
+  public accept() {
     for (const cellId of this.stagedCellIds()) {
       this.actions.removeStagedCell(cellId);
     }
   }
 
-  hasStagedCells(stagedCells: StagedAICells) {
+  public hasStagedCells(stagedCells: StagedAICells) {
     return this.stagedCellIds().some((cellId) => stagedCells.has(cellId));
   }
 
-  stagedCellIds() {
+  public stagedCellIds() {
     const cellIds = this.createdCells.map(({ cellId }) => cellId);
     if (this.marimoImportCellId) {
       cellIds.push(this.marimoImportCellId);

package/src/core/ai/tools/base.ts

@@ -26,12 +26,12 @@ export type StatusValue = "success" | "error" | "warning";
  * );
  */
 export class ToolExecutionError extends Error {
-  readonly code: string;
-  readonly isRetryable: boolean;
-  readonly suggestedFix?: string;
-  readonly meta?: Record<string, unknown>;
+  public readonly code: string;
+  public readonly isRetryable: boolean;
+  public readonly suggestedFix?: string;
+  public readonly meta?: Record<string, unknown>;
 
-  constructor(
+  public constructor(
     message: string,
     code = "TOOL_ERROR",
     isRetryable = false,
@@ -46,7 +46,7 @@ export class ToolExecutionError extends Error {
     this.meta = meta;
   }
 
-  toStructuredString(): string {
+  public toStructuredString(): string {
     const stringError = JSON.stringify({
       message: this.message,
       code: this.code,

package/src/core/ai/tools/edit-notebook-tool.ts

@@ -76,13 +76,13 @@ export class EditNotebookTool implements AiTool<
   EditNotebookInput,
   ToolOutputBase
 > {
-  readonly name = "edit_notebook_tool";
-  readonly description = description;
-  readonly schema = editNotebookSchema;
-  readonly outputSchema = toolOutputBaseSchema;
-  readonly mode: CopilotMode[] = ["agent"];
+  public readonly name = "edit_notebook_tool";
+  public readonly description = description;
+  public readonly schema = editNotebookSchema;
+  public readonly outputSchema = toolOutputBaseSchema;
+  public readonly mode: CopilotMode[] = ["agent"];
 
-  handler = async (
+  public handler = async (
     { edit }: EditNotebookInput,
     toolContext: ToolNotebookContext,
   ): Promise<ToolOutputBase> => {

package/src/core/ai/tools/registry.ts

@@ -36,11 +36,11 @@ export class FrontendToolRegistry {
   /** All registered tools */
   private tools = new Map<string, StoredTool>();
 
-  constructor(tools: StoredTool[] = []) {
+  public constructor(tools: StoredTool[] = []) {
     this.tools = new Map(tools.map((tool) => [tool.name, tool]));
   }
 
-  has(toolName: string) {
+  public has(toolName: string) {
     return this.tools.has(toolName);
   }
 
@@ -52,7 +52,7 @@ export class FrontendToolRegistry {
     return tool;
   }
 
-  async invoke<TName extends string>({
+  public async invoke<TName extends string>({
     toolName,
     rawArgs,
     toolContext,
@@ -127,7 +127,7 @@ export class FrontendToolRegistry {
   }
 
   @Memoize()
-  getToolSchemas(mode: CopilotMode): FrontendToolDefinition[] {
+  public getToolSchemas(mode: CopilotMode): FrontendToolDefinition[] {
     const tools = [...this.tools.values()].filter((tool) =>
       tool.mode.includes(mode),
     );

package/src/core/ai/tools/run-cells-tool.ts

@@ -54,10 +54,10 @@ export class RunStaleCellsTool implements AiTool<
   EmptyToolInput,
   RunStaleCellsOutput
 > {
-  readonly name = "run_stale_cells_tool";
-  readonly description = description;
-  readonly schema = z.object({});
-  readonly outputSchema = toolOutputBaseSchema.extend({
+  public readonly name = "run_stale_cells_tool";
+  public readonly description = description;
+  public readonly schema = z.object({});
+  public readonly outputSchema = toolOutputBaseSchema.extend({
     cellsToOutput: z
       .record(
         z.string(),
@@ -72,15 +72,15 @@ export class RunStaleCellsTool implements AiTool<
       )
       .optional(),
   }) satisfies z.ZodType<RunStaleCellsOutput>;
-  readonly mode: CopilotMode[] = ["agent"];
+  public readonly mode: CopilotMode[] = ["agent"];
 
   private readonly postExecutionDelay: number;
 
-  constructor(opts?: { postExecutionDelay?: number }) {
+  public constructor(opts?: { postExecutionDelay?: number }) {
     this.postExecutionDelay = opts?.postExecutionDelay ?? POST_EXECUTION_DELAY;
   }
 
-  handler = async (
+  public handler = async (
     _args: EmptyToolInput,
     toolContext: ToolNotebookContext,
   ): Promise<RunStaleCellsOutput> => {

package/src/core/ai/tools/sample-tool.ts

@@ -36,18 +36,18 @@ interface Output extends ToolOutputBase {
 
 /** A sample frontend tool that demonstrates real tool output structure */
 export class TestFrontendTool implements AiTool<Input, Output> {
-  readonly name = "test_frontend_tool";
-  readonly description = description;
-  readonly schema = z.object({ name: z.string() });
-  readonly outputSchema = toolOutputBaseSchema.extend({
+  public readonly name = "test_frontend_tool";
+  public readonly description = description;
+  public readonly schema = z.object({ name: z.string() });
+  public readonly outputSchema = toolOutputBaseSchema.extend({
     data: z.object({
       greeting: z.string(),
       timestamp: z.string(),
     }),
   });
-  readonly mode: CopilotMode[] = ["ask"];
+  public readonly mode: CopilotMode[] = ["ask"];
 
-  async handler({ name }: Input): Promise<Output> {
+  public async handler({ name }: Input): Promise<Output> {
     // Example: Validate input and throw ToolExecutionError on invalid data
     if (!name.trim()) {
       throw new ToolExecutionError(

package/src/core/codemirror/cells/debugger-decorations.ts

@@ -48,9 +48,12 @@ function createLineDecoration(
 
 class CurrentLineHighlighter implements PluginValue {
   private unsubscribe: () => void;
-  decorations: DecorationSet;
+  public decorations: DecorationSet;
 
-  constructor(view: EditorView, lineObservable: Observable<number | null>) {
+  public constructor(
+    view: EditorView,
+    lineObservable: Observable<number | null>,
+  ) {
     this.decorations = createLineDecoration(view.state, lineObservable.get());
     this.unsubscribe = lineObservable.sub((line) => {
       this.decorations = createLineDecoration(view.state, line);
@@ -59,11 +62,11 @@ class CurrentLineHighlighter implements PluginValue {
     });
   }
 
-  update(update: ViewUpdate) {
+  public update(update: ViewUpdate) {
     this.decorations = this.decorations.map(update.changes);
   }
 
-  destroy() {
+  public destroy() {
     this.unsubscribe();
   }
 }
@@ -88,7 +91,7 @@ export function debuggerLineHighlighter(
 // --- Breakpoint gutter ------------------------------------------------------
 
 class BreakpointMarker extends GutterMarker {
-  override toDOM() {
+  public override toDOM() {
     const dot = document.createElement("div");
     dot.className = "cm-breakpoint-marker";
     return dot;
@@ -130,7 +133,10 @@ function buildBreakpointMarkers(state: EditorState): RangeSet<GutterMarker> {
 class BreakpointSync implements PluginValue {
   private unsubscribe: () => void;
 
-  constructor(view: EditorView, observable: Observable<ReadonlySet<number>>) {
+  public constructor(
+    view: EditorView,
+    observable: Observable<ReadonlySet<number>>,
+  ) {
     const apply = (lines: ReadonlySet<number>) => {
       view.dispatch({ effects: setBreakpointLines.of(lines) });
     };
@@ -144,7 +150,7 @@ class BreakpointSync implements PluginValue {
     this.unsubscribe = observable.sub(apply);
   }
 
-  destroy() {
+  public destroy() {
     this.unsubscribe();
   }
 }

package/src/core/codemirror/cells/line-timing-decorations.ts

@@ -42,16 +42,16 @@ class LineTimerWidget extends WidgetType {
 
   private readonly startedAtMs: number;
 
-  constructor(startedAtMs: number) {
+  public constructor(startedAtMs: number) {
     super();
     this.startedAtMs = startedAtMs;
   }
 
-  override eq(other: LineTimerWidget): boolean {
+  public override eq(other: LineTimerWidget): boolean {
     return other.startedAtMs === this.startedAtMs;
   }
 
-  override toDOM(): HTMLElement {
+  public override toDOM(): HTMLElement {
     const el = document.createElement("span");
     el.className = "cm-line-timer";
     el.setAttribute("aria-hidden", "true");
@@ -68,7 +68,7 @@ class LineTimerWidget extends WidgetType {
     return el;
   }
 
-  override destroy(dom: HTMLElement): void {
+  public override destroy(dom: HTMLElement): void {
     const interval = LineTimerWidget.intervals.get(dom);
     if (interval !== undefined) {
       window.clearInterval(interval);
@@ -76,7 +76,7 @@ class LineTimerWidget extends WidgetType {
     }
   }
 
-  override ignoreEvent(): boolean {
+  public override ignoreEvent(): boolean {
     return true;
   }
 }
@@ -112,9 +112,9 @@ function createTimingDecorations(
 
 class ActiveLineTimer implements PluginValue {
   private unsubscribe: () => void;
-  decorations: DecorationSet;
+  public decorations: DecorationSet;
 
-  constructor(
+  public constructor(
     view: EditorView,
     infoObservable: Observable<ActiveLineInfo | null>,
   ) {
@@ -129,11 +129,11 @@ class ActiveLineTimer implements PluginValue {
     });
   }
 
-  update(update: ViewUpdate) {
+  public update(update: ViewUpdate) {
     this.decorations = this.decorations.map(update.changes);
   }
 
-  destroy() {
+  public destroy() {
     this.unsubscribe();
   }
 }

package/src/core/codemirror/cells/traceback-decorations.ts

@@ -91,9 +91,12 @@ function unfoldErrorLines(view: EditorView, errors: TracebackInfos) {
  */
 class ErrorHighlighter implements PluginValue {
   private unsubscribe: () => void;
-  decorations: DecorationSet;
+  public decorations: DecorationSet;
 
-  constructor(view: EditorView, errorsObservable: Observable<TracebackInfos>) {
+  public constructor(
+    view: EditorView,
+    errorsObservable: Observable<TracebackInfos>,
+  ) {
     const errors = errorsObservable.get();
     this.decorations = createErrorDecorations(view.state, errors);
     unfoldErrorLines(view, errors);
@@ -113,11 +116,11 @@ class ErrorHighlighter implements PluginValue {
     });
   }
 
-  update(update: ViewUpdate) {
+  public update(update: ViewUpdate) {
     this.decorations = this.decorations.map(update.changes);
   }
 
-  destroy() {
+  public destroy() {
     this.unsubscribe();
   }
 }

package/src/core/codemirror/code-lens/extension.ts

@@ -56,12 +56,12 @@ class LensHoverController {
   private showTimer: number | undefined;
   private hideTimer: number | undefined;
 
-  constructor(view: EditorView) {
+  public constructor(view: EditorView) {
     this.view = view;
   }
 
   /** Pointer entered a lens icon */
-  enterLens(spec: CodeLensSpec, icon: HTMLElement): void {
+  public enterLens(spec: CodeLensSpec, icon: HTMLElement): void {
     this.clearTimers();
     dismissEditorHoverTooltips(this.view, icon);
     const hovered = this.view.state.field(codeLensHoverField, false);
@@ -76,7 +76,7 @@ class LensHoverController {
   }
 
   /** Pointer left a lens icon or the popover */
-  leave(): void {
+  public leave(): void {
     this.clearTimers();
     this.hideTimer = window.setTimeout(() => {
       this.hideTimer = undefined;
@@ -85,19 +85,19 @@ class LensHoverController {
   }
 
   /** Pointer entered the popover */
-  enterPopover(): void {
+  public enterPopover(): void {
     window.clearTimeout(this.hideTimer);
     this.hideTimer = undefined;
   }
 
-  hide(): void {
+  public hide(): void {
     this.clearTimers();
     if (this.view.state.field(codeLensHoverField, false)) {
       this.view.dispatch({ effects: setHoveredLens.of(null) });
     }
   }
 
-  destroy(): void {
+  public destroy(): void {
     this.clearTimers();
   }
 
@@ -135,18 +135,18 @@ function dismissEditorHoverTooltips(
 class CodeLensWidget extends WidgetType {
   private readonly spec: CodeLensSpec;
 
-  constructor(spec: CodeLensSpec) {
+  public constructor(spec: CodeLensSpec) {
     super();
     this.spec = spec;
   }
 
-  override eq(other: CodeLensWidget): boolean {
+  public override eq(other: CodeLensWidget): boolean {
     // `pos` is captured by the DOM hover/click handlers, so a reused widget
     // whose anchor moved must not be treated as equal
     return specEquals(this.spec, other.spec);
   }
 
-  override toDOM(view: EditorView): HTMLElement {
+  public override toDOM(view: EditorView): HTMLElement {
     const { spec } = this;
     const element = document.createElement("span");
     element.className = "mo-code-lens";
@@ -188,7 +188,7 @@ class CodeLensWidget extends WidgetType {
     return element;
   }
 
-  override ignoreEvent(): boolean {
+  public override ignoreEvent(): boolean {
     // The widget handles its own events
     return true;
   }
@@ -281,7 +281,7 @@ class CodeLensPlugin {
   private readonly debounceMs = 300;
   private readonly scheduleUpdate: DebouncedFunc<() => void>;
 
-  constructor(view: EditorView, cellId: CellId, includeCache: boolean) {
+  public constructor(view: EditorView, cellId: CellId, includeCache: boolean) {
     this.view = view;
     this.cellId = cellId;
     this.includeCache = includeCache;
@@ -297,7 +297,7 @@ class CodeLensPlugin {
     this.scheduleUpdate();
   }
 
-  update(update: ViewUpdate) {
+  public update(update: ViewUpdate) {
     // Recompute on edits, and when the cell's language changes (e.g. Python ->
     // SQL) so stale Python-only icons are cleared even if the text is unchanged
     const adapterChanged =
@@ -308,7 +308,7 @@ class CodeLensPlugin {
     }
   }
 
-  destroy() {
+  public destroy() {
     this.scheduleUpdate.cancel();
     for (const unsubscribe of this.unsubscribes) {
       unsubscribe();

package/src/core/codemirror/copilot/__tests__/language-server.test.ts

@@ -22,11 +22,11 @@ vi.mock("../state", async (importOriginal) => {
 
 class MockTransport implements Transport {
   private readonly handlers = new Set<(message: JSONRPCMessage) => void>();
-  readonly sent: JSONRPCMessage[] = [];
+  public readonly sent: JSONRPCMessage[] = [];
 
-  connect = vi.fn().mockResolvedValue(undefined);
-  close = vi.fn();
-  send = vi.fn((message: JSONRPCMessage) => {
+  public connect = vi.fn().mockResolvedValue(undefined);
+  public close = vi.fn();
+  public send = vi.fn((message: JSONRPCMessage) => {
     this.sent.push(message);
     if (isRequest(message) && message.method === "initialize") {
       queueMicrotask(() => {
@@ -39,14 +39,14 @@ class MockTransport implements Transport {
     }
   });
 
-  onMessage(handler: (message: JSONRPCMessage) => void): () => void {
+  public onMessage(handler: (message: JSONRPCMessage) => void): () => void {
     this.handlers.add(handler);
     return () => {
       this.handlers.delete(handler);
     };
   }
 
-  emit(message: JSONRPCMessage) {
+  public emit(message: JSONRPCMessage) {
     for (const handler of this.handlers) {
       handler(message);
     }

package/src/core/codemirror/copilot/__tests__/transport.test.ts

@@ -8,24 +8,24 @@ vi.mock("@/utils/Logger", () => ({
 }));
 
 class FakeWebSocket {
-  static readonly OPEN = 1;
-  static readonly CLOSED = 3;
-  static readonly instances: FakeWebSocket[] = [];
+  public static readonly OPEN = 1;
+  public static readonly CLOSED = 3;
+  public static readonly instances: FakeWebSocket[] = [];
 
-  readonly sent: string[] = [];
-  readyState = 0;
-  readonly url: string;
+  public readonly sent: string[] = [];
+  public readyState = 0;
+  public readonly url: string;
   private readonly listeners = new Map<
     string,
     Set<(event: Record<string, never>) => void>
   >();
 
-  constructor(url: string) {
+  public constructor(url: string) {
     this.url = url;
     FakeWebSocket.instances.push(this);
   }
 
-  addEventListener(
+  public addEventListener(
     type: string,
     listener: (event: Record<string, never>) => void,
   ) {
@@ -34,21 +34,21 @@ class FakeWebSocket {
     this.listeners.set(type, listeners);
   }
 
-  send(frame: string) {
+  public send(frame: string) {
     this.sent.push(frame);
   }
 
-  close() {
+  public close() {
     this.readyState = FakeWebSocket.CLOSED;
     this.emit("close");
   }
 
-  open() {
+  public open() {
     this.readyState = FakeWebSocket.OPEN;
     this.emit("open");
   }
 
-  fail() {
+  public fail() {
     this.emit("error");
   }
 

package/src/core/codemirror/copilot/language-server.ts

@@ -117,7 +117,7 @@ export class CopilotLanguageServerClient extends LanguageServerClient {
   private lastOpenedDocument: DidOpenTextDocumentParams | undefined;
   private copilotSettings: Record<string, unknown> = {};
 
-  constructor(
+  public constructor(
     options: ConstructorParameters<typeof LanguageServerClient>[0] & {
       copilotSettings?: Record<string, unknown>;
     },
@@ -140,7 +140,7 @@ export class CopilotLanguageServerClient extends LanguageServerClient {
    * Re-run the LSP initialize handshake and send configuration.
    * Called by the transport's onReconnect callback after reconnecting.
    */
-  async reInitialize(): Promise<void> {
+  public async reInitialize(): Promise<void> {
     logger.log("#reInitialize: Re-initializing LSP connection");
     this.initializePromise = this.initialize();
     await this.initializePromise;
@@ -186,7 +186,7 @@ export class CopilotLanguageServerClient extends LanguageServerClient {
     return this.notify(method, params);
   }
 
-  override getInitializationOptions() {
+  public override getInitializationOptions() {
     const info = {
       name: "marimo",
       version: "0.1.0",
@@ -208,7 +208,7 @@ export class CopilotLanguageServerClient extends LanguageServerClient {
     return !isCopilotEnabled();
   }
 
-  override async textDocumentDidOpen(
+  public override async textDocumentDidOpen(
     params: DidOpenTextDocumentParams,
   ): Promise<boolean> {
     if (this.isDisabled()) {
@@ -223,7 +223,7 @@ export class CopilotLanguageServerClient extends LanguageServerClient {
     return super.textDocumentDidOpen(params);
   }
 
-  override async textDocumentDidClose(
+  public override async textDocumentDidClose(
     params: DidCloseTextDocumentParams,
   ): Promise<void> {
     if (this.openDocumentCount === 0) {
@@ -233,14 +233,14 @@ export class CopilotLanguageServerClient extends LanguageServerClient {
     return super.textDocumentDidClose(params);
   }
 
-  override async textDocumentCompletion(
+  public override async textDocumentCompletion(
     params: CompletionParams,
   ): Promise<CompletionList | CompletionItem[]> {
     // Not used in Copilot
     return [];
   }
 
-  override async textDocumentDidChange(
+  public override async textDocumentDidChange(
     params: DidChangeTextDocumentParams,
   ): Promise<void> {
     if (this.isDisabled()) {
@@ -304,17 +304,17 @@ export class CopilotLanguageServerClient extends LanguageServerClient {
     });
   }
 
-  override textDocumentHover(params: HoverParams): Promise<Hover> {
+  public override textDocumentHover(params: HoverParams): Promise<Hover> {
     // Not used in Copilot
     return Promise.resolve({ contents: [] });
   }
 
   // AUTH
-  signOut() {
+  public signOut() {
     return this._request("signOut", {});
   }
 
-  async signInInitiate() {
+  public async signInInitiate() {
     logger.log("#signInInitiate: Starting sign-in flow");
     try {
       const result = await this._request("signIn", {});
@@ -326,7 +326,7 @@ export class CopilotLanguageServerClient extends LanguageServerClient {
     }
   }
 
-  async signInConfirm(params: GitHubCopilotSignInConfirmParams) {
+  public async signInConfirm(params: GitHubCopilotSignInConfirmParams) {
     logger.log("#signInConfirm: Confirming sign-in");
     try {
       const result = await this._request("signInConfirm", params);
@@ -338,7 +338,7 @@ export class CopilotLanguageServerClient extends LanguageServerClient {
     }
   }
 
-  async signedIn() {
+  public async signedIn() {
     try {
       const { status } = await this._request("checkStatus", {});
       logger.log("#checkStatus: Status check completed", { status });
@@ -375,7 +375,7 @@ export class CopilotLanguageServerClient extends LanguageServerClient {
     200,
   );
 
-  async getCompletion(
+  public async getCompletion(
     params: InlineCompletionParams,
   ): Promise<InlineCompletionList | InlineCompletionItem[] | null> {
     if (this.isDisabled()) {

package/src/core/codemirror/copilot/transport.ts

@@ -38,7 +38,7 @@ export interface LazyWebsocketTransportOptions {
  * Copilot-specific readiness checks, retries, and user-facing errors.
  */
 export class LazyWebsocketTransport extends ReconnectingWebSocketTransport {
-  constructor(options: LazyWebsocketTransportOptions) {
+  public constructor(options: LazyWebsocketTransportOptions) {
     super({
       getWsUrl: options.getWsUrl,
       waitForConnection: options.waitForReady,

package/src/core/codemirror/find-replace/search-highlight.ts

@@ -90,15 +90,15 @@ const HighlightMargin = 250;
 
 export const searchHighlighter = ViewPlugin.fromClass(
   class {
-    decorations: DecorationSet;
-    readonly view: EditorView;
+    public decorations: DecorationSet;
+    public readonly view: EditorView;
 
-    constructor(view: EditorView) {
+    public constructor(view: EditorView) {
       this.view = view;
       this.decorations = this.highlight(view.state.field(searchState));
     }
 
-    update(update: ViewUpdate) {
+    public update(update: ViewUpdate) {
       const state = update.state.field(searchState);
       if (
         state !== update.startState.field(searchState) ||
@@ -110,7 +110,7 @@ export const searchHighlighter = ViewPlugin.fromClass(
       }
     }
 
-    highlight(query: QueryType) {
+    public highlight(query: QueryType) {
       if (!query.spec.valid) {
         return Decoration.none;
       }