@marimo-team/frontend 0.16.2 → 0.16.3

package/src/core/ai/__tests__/model-registry.test.ts

@@ -250,16 +250,6 @@ describe("AiModelRegistry", () => {
       const providers = listModelsByProvider.map(([provider]) => provider);
       expect(providers).toEqual(PROVIDER_SORT_ORDER);
     });
-
-    it("should include custom providers at the top", () => {
-      const customModels = ["openrouter/custom-gpt"];
-      const registry = AiModelRegistry.create({ customModels });
-      const listModelsByProvider = registry.getListModelsByProvider();
-      expect(listModelsByProvider).toHaveLength(4);
-
-      const providers = listModelsByProvider.map(([provider]) => provider);
-      expect(providers).toEqual(["openrouter", ...PROVIDER_SORT_ORDER]);
-    });
   });
 
   describe("getCustomModels", () => {

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

@@ -14,11 +14,11 @@ export interface ErrorContextItem extends AIContextItem {
   type: "error";
   data: {
     type: "all-errors";
-    errors: Array<{
+    errors: {
       cellId: CellId;
       cellName: string;
       errorData: MarimoError[];
-    }>;
+    }[];
   };
 }
 

package/src/core/ai/ids/ids.ts

@@ -11,6 +11,7 @@ export const PROVIDERS = [
   "deepseek",
   "azure",
   "github",
+  "openrouter",
   "marimo",
 ] as const;
 export type ProviderId = (typeof PROVIDERS)[number];

package/src/core/ai/model-registry.ts

@@ -19,6 +19,7 @@ export const PROVIDER_SORT_ORDER: ProviderId[] = [
   "openai",
   "google",
   "github",
+  "openrouter",
   "deepseek",
   "azure",
   "bedrock",
@@ -201,7 +202,7 @@ export class AiModelRegistry {
     return this.modelsByProviderMap;
   }
 
-  getListModelsByProvider(): Array<[ProviderId, AiModel[]]> {
+  getListModelsByProvider(): [ProviderId, AiModel[]][] {
     const modelsByProvider = this.getGroupedModelsByProvider();
     const arrayModels = [...modelsByProvider.entries()];
     const providerToOrderIdx = getProviderMap().providerToOrderIdx;

package/src/core/cells/cells.ts

@@ -84,14 +84,14 @@ export interface NotebookState {
    *
    * (CodeMirror types the serialized config as any.)
    */
-  history: Array<{
+  history: {
     name: string;
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     serializedEditorState: any;
     column: CellColumnId;
     index: CellIndex;
     isSetupCell: boolean;
-  }>;
+  }[];
   /**
    * Key of cell to scroll to; typically set by actions that re-order the cell
    * array. Call the SCROLL_TO_TARGET action to scroll to the specified cell
@@ -1128,10 +1128,10 @@ const {
 
         // Find the start/end of the collapsed ranges
         const nodes = [...column.nodes];
-        const rangeIndexes: Array<{
+        const rangeIndexes: {
           start: CellIndex;
           end: CellIndex;
-        }> = [];
+        }[] = [];
         const reversedCollapseRanges = [];
 
         // Iterate in reverse order (bottom-up) to process children first
@@ -1668,7 +1668,7 @@ export const getCellEditorView = (cellId: CellId) => {
 
 export function flattenTopLevelNotebookCells(
   state: NotebookState,
-): Array<CellData & CellRuntimeState> {
+): (CellData & CellRuntimeState)[] {
   const { cellIds, cellData, cellRuntime } = state;
   return cellIds.getColumns().flatMap((column) =>
     column.topLevelIds.map((cellId) => ({

package/src/core/cells/logs.ts

@@ -87,7 +87,7 @@ const CellLogLogger = {
           ? "red"
           : "orange";
     const status = payload.level.toUpperCase();
-    // eslint-disable-next-line no-console
+    /** biome-ignore lint/suspicious/noConsole: for debugging */
     console.log(
       `%c[${status}]`,
       `color:${color}; padding:2px 0; border-radius:2px; font-weight:bold`,

package/src/core/cells/types.ts

@@ -93,7 +93,7 @@ export interface CellRuntimeState {
   /** TOC outline */
   outline: Outline | null;
   /** messages encoding the cell's console outputs. */
-  consoleOutputs: Array<WithResponse<OutputMessage>>;
+  consoleOutputs: WithResponse<OutputMessage>[];
   /** current status of the cell */
   status: RuntimeState;
   /** whether the cell has stale inputs*/

package/src/core/codemirror/__tests__/format.test.ts

@@ -3,6 +3,7 @@
 import { python } from "@codemirror/lang-python";
 import { EditorState } from "@codemirror/state";
 import { EditorView } from "@codemirror/view";
+import { atom } from "jotai";
 import { beforeEach, describe, expect, it, vi } from "vitest";
 import { MockRequestClient } from "@/__mocks__/requests";
 import type { NotebookState } from "@/core/cells/cells";
@@ -34,6 +35,11 @@ vi.mock("@/core/cells/utils", () => ({
 
 vi.mock("@/core/config/config", () => ({
   getResolvedMarimoConfig: vi.fn(),
+  resolvedMarimoConfigAtom: atom({
+    display: {
+      theme: "light",
+    },
+  }),
 }));
 
 const updateCellCode = vi.fn();

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

@@ -29,7 +29,7 @@ function createErrorDecorations(state: EditorState, errors: TracebackInfos) {
 
   // Filter and sort errors by line number to ensure they're added in order
   const relevantErrors = errors
-    .filter((error) => error.cellId === cellId)
+    .filter((error) => error.kind === "cell" && error.cellId === cellId)
     .sort((a, b) => a.lineNumber - b.lineNumber);
 
   for (const error of relevantErrors) {

package/src/core/codemirror/editing/commands.ts

@@ -3,14 +3,14 @@ import { foldAll, unfoldAll } from "@codemirror/language";
 import type { Command, EditorView } from "@codemirror/view";
 import type { Nullable } from "vitest";
 
-export type BulkCommand = (targets: Array<Nullable<EditorView>>) => boolean;
+export type BulkCommand = (targets: Nullable<EditorView>[]) => boolean;
 
 /**
  * Make a bulk command from a single {@type Command} that applies
  * the given command to all targets.
  */
 export function makeBulkCommand(command: Command) {
-  return (targets: Array<Nullable<EditorView>>) => {
+  return (targets: Nullable<EditorView>[]) => {
     let changed = false;
     for (const target of targets) {
       if (target) {

package/src/core/codemirror/find-replace/navigate.ts

@@ -110,7 +110,7 @@ export const findPrev = findInDirection("prev");
  */
 export const replaceAll = searchCommand(({ query }) => {
   const views = getAllEditorViews();
-  const undoHandlers: Array<() => void> = [];
+  const undoHandlers: (() => void)[] = [];
   for (const view of views) {
     if (view.state.readOnly) {
       continue;

package/src/core/codemirror/language/__tests__/extension.test.ts

@@ -14,7 +14,7 @@ import {
   languageAdapterState,
   switchLanguage,
 } from "../extension";
-import { exportedForTesting as sqlValidationErrorsForTesting } from "../languages/sql/validation-errors";
+import { exportedForTesting as sqlValidationErrorsForTesting } from "../languages/sql/banner-validation-errors";
 import { languageMetadataField } from "../metadata";
 
 let view: EditorView | null = null;

package/src/core/codemirror/language/__tests__/sql-validation.test.ts

@@ -1,7 +1,7 @@
 /* Copyright 2024 Marimo. All rights reserved. */
 
 import { describe, expect, it } from "vitest";
-import { exportedForTesting } from "../languages/sql/validation-errors";
+import { exportedForTesting } from "../languages/sql/banner-validation-errors";
 
 describe("Error Message Splitting", () => {
   it("should handle error message splitting correctly", () => {