@marimo-team/frontend 0.23.9-dev34 → 0.23.9-dev37

package/src/core/config/__tests__/config-schema.test.ts

@@ -46,6 +46,7 @@ test("default UserConfig - empty", () => {
     {
       "ai": {
         "custom_providers": {},
+        "enabled": true,
         "inline_tooltip": false,
         "mode": "manual",
         "models": {
@@ -118,6 +119,7 @@ test("default UserConfig - one level", () => {
     {
       "ai": {
         "custom_providers": {},
+        "enabled": true,
         "inline_tooltip": false,
         "mode": "manual",
         "models": {

package/src/core/config/config-schema.ts

@@ -158,6 +158,7 @@ export const UserConfigSchema = z
       .prefault({}),
     ai: z
       .looseObject({
+        enabled: z.boolean().prefault(true),
         rules: z.string().prefault(""),
         max_tokens: z.number().int().positive().nullable().optional(),
         mode: z.enum(COPILOT_MODES).prefault("manual"),

package/src/core/config/config.ts

@@ -78,6 +78,14 @@ export const aiEnabledAtom = atom<boolean>((get) => {
   return isAiEnabled(get(resolvedMarimoConfigAtom));
 });
 
+export const aiModelConfiguredAtom = atom<boolean>((get) => {
+  return isAiModelConfigured(get(resolvedMarimoConfigAtom));
+});
+
+export const aiFeaturesEnabledAtom = atom<boolean>((get) => {
+  return isAiFeatureEnabled(get(resolvedMarimoConfigAtom));
+});
+
 export const editorFontSizeAtom = atom<number>((get) => {
   return get(resolvedMarimoConfigAtom).display.code_editor_font_size;
 });
@@ -87,6 +95,10 @@ export const localeAtom = atom<string | null | undefined>((get) => {
 });
 
 export function isAiEnabled(config: UserConfig) {
+  return config.ai?.enabled !== false;
+}
+
+export function isAiModelConfigured(config: UserConfig) {
   return (
     Boolean(config.ai?.models?.chat_model) ||
     Boolean(config.ai?.models?.edit_model) ||
@@ -94,6 +106,10 @@ export function isAiEnabled(config: UserConfig) {
   );
 }
 
+export function isAiFeatureEnabled(config: UserConfig) {
+  return isAiEnabled(config) && isAiModelConfigured(config);
+}
+
 /**
  * Atom for storing the app config.
  */

package/src/utils/__tests__/json-parser.test.ts

@@ -1,10 +1,6 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 import { expect, it } from "vitest";
-import {
-  jsonParseWithSpecialChar,
-  jsonToMarkdown,
-  jsonToTSV,
-} from "../json/json-parser";
+import { jsonParseWithSpecialChar, jsonToMarkdown } from "../json/json-parser";
 
 it("can jsonParseWithSpecialChar happy path", () => {
   expect(jsonParseWithSpecialChar('"hello"')).toEqual("hello");
@@ -72,70 +68,6 @@ it("can parse bigInts", () => {
   });
 });
 
-it("can convert json to tsv with en-US locale", () => {
-  const locale = "en-US";
-
-  expect(jsonToTSV([], locale)).toEqual("");
-
-  expect(jsonToTSV([{ a: 1, b: 2 }], locale)).toEqual("a\tb\n1\t2");
-
-  expect(
-    jsonToTSV(
-      [
-        { a: 1, b: 2 },
-        { a: 3, b: 4 },
-      ],
-      locale,
-    ),
-  ).toEqual("a\tb\n1\t2\n3\t4");
-
-  // Does not handle sparse arrays
-  expect(jsonToTSV([{ a: 1 }, { a: 2, b: 3 }], locale)).toMatchInlineSnapshot(
-    '"a\n1\n2"',
-  );
-
-  // Handles special characters
-  expect(
-    jsonToTSV([{ a: "hello\tworld", b: "new\nline" }], locale),
-  ).toMatchInlineSnapshot('"a\tb\nhello\tworld\tnew\nline"');
-
-  // Handles floats with en-US locale (uses . as decimal separator)
-  expect(jsonToTSV([{ a: 1.5, b: 2.7 }], locale)).toEqual("a\tb\n1.5\t2.7");
-});
-
-it("can convert json to tsv with de-DE locale", () => {
-  const locale = "de-DE";
-
-  // Handles floats with de-DE locale (uses , as decimal separator)
-  expect(jsonToTSV([{ a: 1.5, b: 2.7 }], locale)).toEqual("a\tb\n1,5\t2,7");
-
-  // Handles integers (no change)
-  expect(jsonToTSV([{ a: 1, b: 2 }], locale)).toEqual("a\tb\n1\t2");
-});
-
-it("can convert json to tsv with fr-FR locale", () => {
-  const locale = "fr-FR";
-
-  // Handles floats with fr-FR locale (uses , as decimal separator)
-  expect(jsonToTSV([{ a: 3.14, b: 2.123_45 }], locale)).toEqual(
-    "a\tb\n3,14\t2,12345",
-  );
-});
-
-it("handles null and undefined values in TSV", () => {
-  const locale = "en-US";
-
-  expect(jsonToTSV([{ a: null, b: undefined, c: 1 }], locale)).toEqual(
-    "a\tb\tc\n\t\t1",
-  );
-});
-
-it("handles NaN values in TSV", () => {
-  const locale = "en-US";
-
-  expect(jsonToTSV([{ a: Number.NaN, b: 1 }], locale)).toEqual("a\tb\nNaN\t1");
-});
-
 it("can convert json to markdown - basic table", () => {
   expect(jsonToMarkdown([])).toMatchInlineSnapshot(`""`);
 

package/src/utils/json/json-parser.ts

@@ -66,36 +66,6 @@ export function jsonParseWithSpecialChar<T = unknown>(
   }
 }
 
-/**
- * Formats a value for TSV export, respecting user's locale for numbers
- */
-function formatValueForTSV(value: unknown, locale: string): string {
-  if (value === null || value === undefined) {
-    return "";
-  }
-  if (typeof value === "number" && !Number.isNaN(value)) {
-    // Use toLocaleString to format numbers according to user's locale
-    // This will use the appropriate decimal separator (e.g., "," in European locales)
-    return value.toLocaleString(locale, {
-      useGrouping: false,
-      maximumFractionDigits: 20,
-    });
-  }
-  return String(value);
-}
-
-export function jsonToTSV(json: Record<string, unknown>[], locale: string) {
-  if (json.length === 0) {
-    return "";
-  }
-
-  const keys = Object.keys(json[0]);
-  const values = json.map((row) =>
-    keys.map((key) => formatValueForTSV(row[key], locale)).join("\t"),
-  );
-  return `${keys.join("\t")}\n${values.join("\n")}`;
-}
-
 /**
  * Converts JSON data to a Markdown table format
  * Detects URLs and converts them to markdown links [url](url)