@marimo-team/frontend 0.25.1-dev27 → 0.25.1-dev28

package/src/components/editor/output/ansi-reduce.ts

@@ -42,7 +42,7 @@ export class TerminalBuffer {
   }
 
   /** Write a visible character at the current cursor position. */
-  writeChar(ch: string) {
+  public writeChar(ch: string) {
     this.ensureLine(this.cursor.row);
     const line = this.lines[this.cursor.row];
     const padded = line.padEnd(this.cursor.col, " ");
@@ -52,7 +52,7 @@ export class TerminalBuffer {
   }
 
   /** Write a string of visible characters at the current cursor position (optimized batch write). */
-  writeString(str: string) {
+  public writeString(str: string) {
     if (str.length === 0) {
       return;
     }
@@ -67,7 +67,7 @@ export class TerminalBuffer {
   }
 
   /** Handle simple control characters (\n, \r, \t, \b, \v). */
-  control(ch: string) {
+  public control(ch: string) {
     switch (ch) {
       case "\n":
         this.cursor.row++;
@@ -95,7 +95,7 @@ export class TerminalBuffer {
    * Supports cursor movement and line erasing.
    * For other sequences (like color codes), preserve them by writing to the buffer.
    */
-  handleEscape(seq: string) {
+  public handleEscape(seq: string) {
     const match = TerminalBuffer.ESCAPE_REGEX.exec(seq);
     if (!match) {
       // If it doesn't match the cursor movement pattern, it might be:
@@ -172,7 +172,7 @@ export class TerminalBuffer {
   }
 
   /** Return the final rendered buffer as a single string. */
-  render(): string {
+  public render(): string {
     return this.lines.join("\n");
   }
 }
@@ -187,7 +187,7 @@ export class AnsiParser {
     "gu",
   );
 
-  parse(input: string): { type: "text" | "escape"; value: string }[] {
+  public parse(input: string): { type: "text" | "escape"; value: string }[] {
     const tokens: { type: "text" | "escape"; value: string }[] = [];
     let lastIndex = 0;
 
@@ -221,7 +221,7 @@ export class AnsiReducer {
    * Process the entire input string (replaces any previous state).
    * Use this for one-time processing or when starting fresh.
    */
-  reduce(input: string): string {
+  public reduce(input: string): string {
     this.reset();
     this.append(input);
     return this.render();
@@ -231,7 +231,7 @@ export class AnsiReducer {
    * Append new input to the existing buffer (for streaming/incremental updates).
    * This is efficient for streaming scenarios - only processes the new chunk.
    */
-  append(input: string): void {
+  public append(input: string): void {
     const tokens = this.parser.parse(input);
 
     // Fast path: if only one text token (no ANSI codes), handle directly
@@ -260,14 +260,14 @@ export class AnsiReducer {
    * Reset the buffer and cursor to initial state.
    * Use this when you want to start processing fresh input.
    */
-  reset(): void {
+  public reset(): void {
     this.buffer = new TerminalBuffer();
   }
 
   /**
    * Get the current rendered output without processing new input.
    */
-  render(): string {
+  public render(): string {
     return this.buffer.render();
   }
 
@@ -375,7 +375,7 @@ export class StatefulOutputMessage implements OutputMessage {
     return this._data;
   }
 
-  static create(message: StringOutputMessage): StatefulOutputMessage {
+  public static create(message: StringOutputMessage): StatefulOutputMessage {
     const ansiReducer = new AnsiReducer();
     ansiReducer.append(message.data);
     return new StatefulOutputMessage(
@@ -399,7 +399,7 @@ export class StatefulOutputMessage implements OutputMessage {
     this._data = this.ansiReducer.render();
   }
 
-  appendData(chunk: string): StatefulOutputMessage {
+  public appendData(chunk: string): StatefulOutputMessage {
     this.ansiReducer.append(chunk);
     return new StatefulOutputMessage(
       this.mimetype,
@@ -409,7 +409,7 @@ export class StatefulOutputMessage implements OutputMessage {
     );
   }
 
-  toJSON(): StringOutputMessage {
+  public toJSON(): StringOutputMessage {
     return {
       mimetype: this.mimetype,
       channel: this.channel,

package/src/components/slides/__tests__/minimap-actions.test.tsx

@@ -46,21 +46,21 @@ beforeAll(() => {
     };
   }
   global.IntersectionObserver ??= class {
-    observe() {
+    public observe() {
       /* noop */
     }
-    unobserve() {
+    public unobserve() {
       /* noop */
     }
-    disconnect() {
+    public disconnect() {
       /* noop */
     }
-    takeRecords() {
+    public takeRecords() {
       return [];
     }
-    root = null;
-    rootMargin = "";
-    thresholds = [];
+    public root = null;
+    public rootMargin = "";
+    public thresholds = [];
   } as unknown as typeof IntersectionObserver;
 });
 

package/src/components/slides/__tests__/minimap.test.ts

@@ -42,28 +42,28 @@ describe("useVisibleCellIds", () => {
     mutationObserveSpy = vi.fn();
 
     global.IntersectionObserver = class MockIntersectionObserver {
-      constructor(
+      public constructor(
         callback: IntersectionObserverCallback,
         _options?: IntersectionObserverInit,
       ) {
         intersectionCallback = callback;
       }
-      observe = observeSpy;
-      unobserve = vi.fn();
-      disconnect = intersectionDisconnectSpy;
-      root = null;
-      rootMargin = "";
-      thresholds = [0];
-      takeRecords = vi.fn(() => []);
+      public observe = observeSpy;
+      public unobserve = vi.fn();
+      public disconnect = intersectionDisconnectSpy;
+      public root = null;
+      public rootMargin = "";
+      public thresholds = [0];
+      public takeRecords = vi.fn(() => []);
     } as unknown as typeof IntersectionObserver;
 
     global.MutationObserver = class MockMutationObserver {
-      constructor(callback: MutationCallback) {
+      public constructor(callback: MutationCallback) {
         mutationCallback = callback;
       }
-      observe = mutationObserveSpy;
-      disconnect = mutationDisconnectSpy;
-      takeRecords = vi.fn(() => []);
+      public observe = mutationObserveSpy;
+      public disconnect = mutationDisconnectSpy;
+      public takeRecords = vi.fn(() => []);
     } as unknown as typeof MutationObserver;
   });
 

package/src/components/slides/__tests__/slide-scroll-hint.test.tsx

@@ -76,12 +76,12 @@ describe("SlideScrollContainer", () => {
     vi.stubGlobal(
       "ResizeObserver",
       class MockResizeObserver {
-        callback: ResizeObserverCallback;
-        observe = vi.fn();
-        unobserve = vi.fn();
-        disconnect = vi.fn();
+        public callback: ResizeObserverCallback;
+        public observe = vi.fn();
+        public unobserve = vi.fn();
+        public disconnect = vi.fn();
 
-        constructor(callback: ResizeObserverCallback) {
+        public constructor(callback: ResizeObserverCallback) {
           this.callback = callback;
           resizeObservers.push(this);
         }
@@ -90,9 +90,9 @@ describe("SlideScrollContainer", () => {
     vi.stubGlobal(
       "MutationObserver",
       class MockMutationObserver {
-        observe = vi.fn();
-        disconnect = vi.fn();
-        takeRecords = vi.fn(() => []);
+        public observe = vi.fn();
+        public disconnect = vi.fn();
+        public takeRecords = vi.fn(() => []);
       },
     );
   });

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

@@ -44,9 +44,9 @@ const mockAttachment2: FileUIPart = {
 
 // Concrete implementation of AIContextProvider for testing
 class MockContextProvider extends AIContextProvider<MockContextItem> {
-  readonly title = "Mock Items";
-  readonly mentionPrefix = "@";
-  readonly contextType = "mock";
+  public readonly title = "Mock Items";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "mock";
 
   private items: MockContextItem[] = [
     {
@@ -72,57 +72,60 @@ class MockContextProvider extends AIContextProvider<MockContextItem> {
     },
   ];
 
-  getItems(): MockContextItem[] {
+  public getItems(): MockContextItem[] {
     return this.items;
   }
 
-  formatContext(item: MockContextItem): string {
+  public formatContext(item: MockContextItem): string {
     return `Mock: ${item.name} (${item.data.value})`;
   }
 
-  formatCompletion(item: MockContextItem): Completion {
+  public formatCompletion(item: MockContextItem): Completion {
     return this.createBasicCompletion(item);
   }
 
   // Method to add items for testing
-  addItem(item: MockContextItem): void {
+  public addItem(item: MockContextItem): void {
     this.items.push(item);
   }
 
   // Method to clear items for testing
-  clearItems(): void {
+  public clearItems(): void {
     this.items = [];
   }
 }
 
 // Test provider that supports attachments
 class AttachmentContextProvider extends AIContextProvider<MockContextItem> {
-  readonly title = "Attachment Items";
-  readonly mentionPrefix = "@";
-  readonly contextType = "attachment";
+  public readonly title = "Attachment Items";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "attachment";
 
   private items: MockContextItem[];
   private attachments: FileUIPart[];
 
-  constructor(items: MockContextItem[] = [], attachments: FileUIPart[] = []) {
+  public constructor(
+    items: MockContextItem[] = [],
+    attachments: FileUIPart[] = [],
+  ) {
     super();
     this.items = items;
     this.attachments = attachments;
   }
 
-  getItems(): MockContextItem[] {
+  public getItems(): MockContextItem[] {
     return this.items;
   }
 
-  formatContext(item: MockContextItem): string {
+  public formatContext(item: MockContextItem): string {
     return `Attachment: ${item.name} (${item.data.value})`;
   }
 
-  formatCompletion(item: MockContextItem): Completion {
+  public formatCompletion(item: MockContextItem): Completion {
     return this.createBasicCompletion(item);
   }
 
-  override async getAttachments(
+  public override async getAttachments(
     items: MockContextItem[],
   ): Promise<FileUIPart[]> {
     // Return attachments for items that need them
@@ -142,9 +145,9 @@ interface FileContextItem extends AIContextItem {
 
 // Another mock provider with different prefix and type
 class FileContextProvider extends AIContextProvider<FileContextItem> {
-  readonly title = "Files";
-  readonly mentionPrefix = "@";
-  readonly contextType = "file";
+  public readonly title = "Files";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "file";
 
   private items: FileContextItem[] = [
     {
@@ -163,26 +166,26 @@ class FileContextProvider extends AIContextProvider<FileContextItem> {
     },
   ];
 
-  getItems(): FileContextItem[] {
+  public getItems(): FileContextItem[] {
     return this.items;
   }
 
-  formatCompletion(item: FileContextItem): Completion {
+  public formatCompletion(item: FileContextItem): Completion {
     return this.createBasicCompletion(item);
   }
 
-  formatContext(item: FileContextItem): string {
+  public formatContext(item: FileContextItem): string {
     return `File: ${item.uri}\nDescription: ${item.description}`;
   }
 }
 
 // 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";
+  public readonly title = "Secondary Mock Items";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "mock";
 
-  getItems(): MockContextItem[] {
+  public getItems(): MockContextItem[] {
     return [
       {
         type: "mock",
@@ -194,11 +197,11 @@ class SecondaryMockProvider extends AIContextProvider<MockContextItem> {
     ];
   }
 
-  formatContext(item: MockContextItem): string {
+  public formatContext(item: MockContextItem): string {
     return `Secondary: ${item.name}`;
   }
 
-  formatCompletion(item: MockContextItem): Completion {
+  public formatCompletion(item: MockContextItem): Completion {
     return this.createBasicCompletion(item);
   }
 }

package/src/core/ai/context/providers/cell-output.ts

@@ -80,16 +80,16 @@ function isMediaMimetype(
 }
 
 export class CellOutputContextProvider extends AIContextProvider<CellOutputContextItem> {
-  readonly title = "Cell Outputs";
-  readonly mentionPrefix = "@";
-  readonly contextType = "cell-output";
+  public readonly title = "Cell Outputs";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "cell-output";
   private store: JotaiStore;
-  constructor(store: JotaiStore) {
+  public constructor(store: JotaiStore) {
     super();
     this.store = store;
   }
 
-  getItems(): CellOutputContextItem[] {
+  public getItems(): CellOutputContextItem[] {
     const notebook = this.store.get(notebookAtom);
     const items: CellOutputContextItem[] = [];
 
@@ -116,7 +116,7 @@ export class CellOutputContextProvider extends AIContextProvider<CellOutputConte
     return items;
   }
 
-  formatCompletion(item: CellOutputContextItem): Completion {
+  public formatCompletion(item: CellOutputContextItem): Completion {
     const { cellOutput: data, cellName, cellCode } = item.data;
 
     return {
@@ -224,7 +224,7 @@ export class CellOutputContextProvider extends AIContextProvider<CellOutputConte
     };
   }
 
-  formatContext(item: CellOutputContextItem): string {
+  public formatContext(item: CellOutputContextItem): string {
     const { cellOutput: data, cellName, cellId, cellCode } = item.data;
 
     const contextData = {
@@ -253,7 +253,7 @@ export class CellOutputContextProvider extends AIContextProvider<CellOutputConte
   }
 
   /** Get attachments for cell output items that have shouldDownloadImage=true */
-  override async getAttachments(
+  public override async getAttachments(
     items: CellOutputContextItem[],
   ): Promise<FileUIPart[]> {
     const cellId = items[0].data.cellId;

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

@@ -46,13 +46,16 @@ export interface DatasourceContextItem extends AIContextItem {
 const CONTEXT_TYPE = "datasource";
 
 export class DatasourceContextProvider extends AIContextProvider<DatasourceContextItem> {
-  readonly title = "Datasource";
-  readonly mentionPrefix = "@";
-  readonly contextType = CONTEXT_TYPE;
+  public readonly title = "Datasource";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = CONTEXT_TYPE;
   private connectionsMap: ConnectionsMap;
   private dataframes: DataTable[];
 
-  constructor(connectionsMap: ConnectionsMap, tablesMap: DatasetTablesMap) {
+  public constructor(
+    connectionsMap: ConnectionsMap,
+    tablesMap: DatasetTablesMap,
+  ) {
     super();
     this.connectionsMap = connectionsMap;
     this.dataframes = [...tablesMap.values()].filter(
@@ -60,7 +63,7 @@ export class DatasourceContextProvider extends AIContextProvider<DatasourceConte
     );
   }
 
-  getItems(): DatasourceContextItem[] {
+  public getItems(): DatasourceContextItem[] {
     return [...this.connectionsMap.values()]
       .map((connection): DatasourceContextItem | null => {
         let description = "Database schema.";
@@ -92,7 +95,7 @@ export class DatasourceContextProvider extends AIContextProvider<DatasourceConte
       .filter(Boolean);
   }
 
-  formatContext(item: DatasourceContextItem): string {
+  public formatContext(item: DatasourceContextItem): string {
     const data = item.data;
     // Remove certain fields that are not needed in the context
     const { name, display_name, source, ...filteredDatasource } =
@@ -117,7 +120,7 @@ export class DatasourceContextProvider extends AIContextProvider<DatasourceConte
     });
   }
 
-  formatCompletion(item: DatasourceContextItem): Completion {
+  public formatCompletion(item: DatasourceContextItem): Completion {
     const datasource = item.data;
 
     const dataConnection = datasource.connection;

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

@@ -95,17 +95,17 @@ function errorContextName(entry: CellErrorEntry): string {
 
 const errorsTxt = new PluralWord("error", "errors");
 export class ErrorContextProvider extends AIContextProvider<ErrorContextItem> {
-  readonly title = "Errors";
-  readonly mentionPrefix = "@";
-  readonly contextType = "error";
+  public readonly title = "Errors";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "error";
   private store: JotaiStore;
 
-  constructor(store: JotaiStore) {
+  public constructor(store: JotaiStore) {
     super();
     this.store = store;
   }
 
-  getItems(): ErrorContextItem[] {
+  public getItems(): ErrorContextItem[] {
     const errors = getCellErrorEntries(this.store);
 
     if (errors.length === 0) {
@@ -141,7 +141,7 @@ export class ErrorContextProvider extends AIContextProvider<ErrorContextItem> {
     return items;
   }
 
-  formatCompletion(item: ErrorContextItem): Completion {
+  public formatCompletion(item: ErrorContextItem): Completion {
     if (item.data.type === "all-errors") {
       const errorCount = item.data.errors.length;
       return {
@@ -201,7 +201,7 @@ export class ErrorContextProvider extends AIContextProvider<ErrorContextItem> {
     };
   }
 
-  formatContext(item: ErrorContextItem): string {
+  public formatContext(item: ErrorContextItem): string {
     const entries =
       item.data.type === "all-errors" ? item.data.errors : [item.data.error];
 

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

@@ -38,14 +38,14 @@ const DEFAULT_FILE_SEARCH_CONFIG: FileSearchConfig = {
 };
 
 export class FileContextProvider extends AIContextProvider<FileContextItem> {
-  readonly title = "Files";
-  readonly mentionPrefix = "#";
-  readonly contextType = "file";
+  public readonly title = "Files";
+  public readonly mentionPrefix = "#";
+  public readonly contextType = "file";
 
   private apiRequests: EditRequests & RunRequests;
   private config: FileSearchConfig;
 
-  constructor(
+  public constructor(
     apiRequests: EditRequests & RunRequests,
     config: FileSearchConfig = DEFAULT_FILE_SEARCH_CONFIG,
   ) {
@@ -58,7 +58,7 @@ export class FileContextProvider extends AIContextProvider<FileContextItem> {
    * Create a dynamic completion source for file mentions
    * This bypasses the standard registry system to enable dynamic searching
    */
-  createCompletionSource(): CompletionSource {
+  public createCompletionSource(): CompletionSource {
     return async (
       context: CompletionContext,
     ): Promise<CompletionResult | null> => {
@@ -176,13 +176,13 @@ export class FileContextProvider extends AIContextProvider<FileContextItem> {
     }
   }
 
-  getItems(): FileContextItem[] {
+  public getItems(): FileContextItem[] {
     // Files are fetched dynamically, so return empty array
     // This provider relies on dynamic fetching via createCompletionSource()
     return [];
   }
 
-  formatCompletion(item: FileContextItem): Completion {
+  public formatCompletion(item: FileContextItem): Completion {
     const { data, name } = item;
     const icon = data.isDirectory ? "📁" : "📄";
 
@@ -254,7 +254,7 @@ export class FileContextProvider extends AIContextProvider<FileContextItem> {
     };
   }
 
-  formatContext(item: FileContextItem): string {
+  public formatContext(item: FileContextItem): string {
     const { data, name } = item;
     return contextToXml({
       type: this.contextType,

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

@@ -17,16 +17,16 @@ export interface TableContextItem extends AIContextItem {
 }
 
 export class TableContextProvider extends AIContextProvider<TableContextItem> {
-  readonly title = "Tables";
-  readonly mentionPrefix = "@";
-  readonly contextType = "data";
+  public readonly title = "Tables";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "data";
   private tablesMap: DatasetTablesMap;
-  constructor(tablesMap: DatasetTablesMap) {
+  public constructor(tablesMap: DatasetTablesMap) {
     super();
     this.tablesMap = tablesMap;
   }
 
-  getItems(): TableContextItem[] {
+  public getItems(): TableContextItem[] {
     return [...this.tablesMap.entries()].map(([tableName, table]) => ({
       uri: this.asURI(tableName),
       name: tableName,
@@ -36,7 +36,7 @@ export class TableContextProvider extends AIContextProvider<TableContextItem> {
     }));
   }
 
-  formatContext(item: TableContextItem): string {
+  public formatContext(item: TableContextItem): string {
     const { data } = item;
     const { columns, source, num_rows, num_columns, name, variable_name } =
       data;
@@ -90,7 +90,7 @@ export class TableContextProvider extends AIContextProvider<TableContextItem> {
     });
   }
 
-  formatCompletion(item: TableContextItem): Completion {
+  public formatCompletion(item: TableContextItem): Completion {
     const tableName = item.data.name;
     const table = item.data;
     return {

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

@@ -16,20 +16,20 @@ export interface VariableContextItem extends AIContextItem {
 }
 
 export class VariableContextProvider extends AIContextProvider<VariableContextItem> {
-  readonly title = "Variables";
-  readonly mentionPrefix = "@";
-  readonly contextType = "variable";
+  public readonly title = "Variables";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "variable";
 
   private variables: Variables;
   private tablesMap: DatasetTablesMap;
 
-  constructor(variables: Variables, tablesMap: DatasetTablesMap) {
+  public constructor(variables: Variables, tablesMap: DatasetTablesMap) {
     super();
     this.variables = variables;
     this.tablesMap = tablesMap;
   }
 
-  getItems(): VariableContextItem[] {
+  public getItems(): VariableContextItem[] {
     const ignore = new Set(this.tablesMap.keys());
 
     return Object.entries(this.variables).flatMap(([name, variable]) => {
@@ -51,7 +51,7 @@ export class VariableContextProvider extends AIContextProvider<VariableContextIt
     });
   }
 
-  formatCompletion(item: VariableContextItem): Completion {
+  public formatCompletion(item: VariableContextItem): Completion {
     const { data } = item;
     const { variable } = data;
     return {
@@ -66,7 +66,7 @@ export class VariableContextProvider extends AIContextProvider<VariableContextIt
     };
   }
 
-  formatContext(item: VariableContextItem): string {
+  public formatContext(item: VariableContextItem): string {
     const { data } = item;
     const { variable } = data;
     return contextToXml({