@marimo-team/islands 0.24.1-dev42 → 0.24.1-dev43

package/dist/{useDateFormatter-6ImHK1ln.js → useDateFormatter-DjRKOAkR.js}

@@ -1,7 +1,7 @@
 import { a as e } from "./rolldown-runtime-Cb76FRpK.js";
 import { t } from "./react-DMWcHin-.js";
 import { u as n } from "./dist-DTSuOdDc.js";
-import { E as r } from "./strings-CoQ5sA4r.js";
+import { E as r } from "./select-Bakm9EUs.js";
 var i = n("chart-pie", [["path", {
 	d: "M21 12c.552 0 1.005-.449.95-.998a10 10 0 0 0-8.953-8.951c-.55-.055-.998.398-.998.95v8a1 1 0 0 0 1 1z",
 	key: "pzmjnu"

package/dist/{vega-component-DlnGFDeY.js → vega-component-A2xO4BUa.js}

@@ -4,17 +4,17 @@ import { t as i } from "./react-DMWcHin-.js";
 import { t as a } from "./compiler-runtime-CAH9RLw2.js";
 import { _ as o, c as s } from "./toDate-C4ZHOUTi.js";
 import { t as c } from "./jsx-runtime-CpoKNRgu.js";
-import { n as l } from "./error-banner-BbxpEiD5.js";
+import { n as l } from "./error-banner-BT4uIIQ2.js";
 import { t as u } from "./tooltip-rwx3Qbw6.js";
 import { t as d } from "./debounce-C3jyATpc.js";
 import { A as f, n as p } from "./useTheme-CLjV5H1i.js";
 import { c as m } from "./arrays-CN1djC4k.js";
 import { t as ee } from "./objects-CzUgoMIF.js";
-import { a as h, i as te, n as ne, t as re } from "./formats-l5hZdDGj.js";
-import { A as ie, n as g } from "./vega-loader.browser-DEggZzni.js";
-import { t as ae } from "./react-vega-DSD5aT7_.js";
-import { a as oe, n as se } from "./loader-CgRtoLc5.js";
-import { t as _ } from "./useAsyncData-BPIoRmNG.js";
+import { t as h } from "./useAsyncData-BPIoRmNG.js";
+import { a as te, i as ne, n as re, t as g } from "./formats-l5hZdDGj.js";
+import { A as ie, n as ae } from "./vega-loader.browser-DEggZzni.js";
+import { t as oe } from "./react-vega-DSD5aT7_.js";
+import { a as se, n as _ } from "./loader-CgRtoLc5.js";
 import { t as ce } from "./useDeepCompareMemoize-rqhnGZ6D.js";
 import { t as le } from "./semaphore-89jsoHtS.js";
 import { t as ue } from "./use-vega-container-remeasure-DUW4tRxq.js";
@@ -490,7 +490,7 @@ async function U(e) {
 		} catch {
 			return e;
 		}
-		let a = e.data.format, o = await n.run(() => se(i.href, a));
+		let a = e.data.format, o = await n.run(() => _(i.href, a));
 		return t[i.pathname] = o, {
 			...e,
 			data: { name: i.pathname }
@@ -504,11 +504,11 @@ async function U(e) {
 //#endregion
 //#region src/plugins/impl/vega/vega-component.tsx
 var W = c();
-g("arrow", re);
+ae("arrow", g);
 var G = (e) => {
 	let t = (0, de.c)(12), { value: n, setValue: r, chartSelection: i, fieldSelection: a, spec: o, embedOptions: s } = e, c, u;
 	t[0] === o ? (c = t[1], u = t[2]) : (c = async () => U(o), u = [o], t[0] = o, t[1] = c, t[2] = u);
-	let { data: d, error: f } = _(c, u);
+	let { data: d, error: f } = h(c, u);
 	if (f) {
 		let e;
 		return t[3] === f ? e = t[4] : (e = /* @__PURE__ */ (0, W.jsx)(l, { error: f }), t[3] = f, t[4] = e), e;
@@ -524,7 +524,7 @@ var G = (e) => {
 		embedOptions: s
 	}), t[5] = i, t[6] = s, t[7] = a, t[8] = d, t[9] = r, t[10] = n, t[11] = p) : p = t[11], p;
 }, K = (e) => {
-	let i = (0, de.c)(50), { value: a, setValue: s, chartSelection: c, fieldSelection: l, spec: m, embedOptions: h } = e, { theme: re } = p(), g = (0, v.useRef)(null), se = (0, v.useRef)(void 0), [_, le] = (0, v.useState)(), y;
+	let i = (0, de.c)(50), { value: a, setValue: s, chartSelection: c, fieldSelection: l, spec: m, embedOptions: h } = e, { theme: te } = p(), g = (0, v.useRef)(null), ae = (0, v.useRef)(void 0), [_, le] = (0, v.useState)(), y;
 	bb0: {
 		if (h && "actions" in h) {
 			y = h.actions;
@@ -570,7 +570,7 @@ var G = (e) => {
 	}, i[17] = w, i[18] = F);
 	let I = f(F), L;
 	i[19] === Symbol.for("react.memo_cache_sentinel") ? (L = (e) => {
-		t.debug("[Vega view] created", e), se.current = e, le(void 0);
+		t.debug("[Vega view] created", e), ae.current = e, le(void 0);
 	}, i[19] = L) : L = i[19];
 	let R = f(L), z;
 	i[20] === D ? z = i[21] : (z = () => {
@@ -585,12 +585,12 @@ var G = (e) => {
 			children: /* @__PURE__ */ (0, W.jsx)(o, { className: "absolute bottom-1 right-0 m-2 h-4 w-4 cursor-help text-muted-foreground hover:text-foreground" })
 		});
 	}, i[20] = D, i[21] = z);
-	let B = z, V = re === "dark" ? "dark" : void 0, H;
+	let B = z, V = te === "dark" ? "dark" : void 0, H;
 	i[22] !== b || i[23] !== V ? (H = {
 		theme: V,
 		actions: b,
 		mode: "vega-lite",
-		tooltip: oe.call,
+		tooltip: se.call,
 		renderer: "canvas"
 	}, i[22] = b, i[23] = V, i[24] = H) : H = i[24];
 	let U;
@@ -601,7 +601,7 @@ var G = (e) => {
 		onError: I,
 		onEmbed: R
 	}, i[25] = I, i[26] = R, i[27] = w, i[28] = H, i[29] = U) : U = i[29];
-	let G = ae(U), K;
+	let G = oe(U), K;
 	i[30] === w ? K = i[31] : (K = ie(w), i[30] = w, i[31] = K);
 	let q = K;
 	ue(g, q === "container");
@@ -626,9 +626,9 @@ var G = (e) => {
 		});
 	}), Y = [G, P], i[32] = G, i[33] = P, i[34] = J, i[35] = Y) : (J = i[34], Y = i[35]), (0, v.useEffect)(J, Y);
 	let X;
-	i[36] === _ ? X = i[37] : (X = _ && /* @__PURE__ */ (0, W.jsxs)(ne, {
+	i[36] === _ ? X = i[37] : (X = _ && /* @__PURE__ */ (0, W.jsxs)(re, {
 		variant: "destructive",
-		children: [/* @__PURE__ */ (0, W.jsx)(te, { children: _.message }), /* @__PURE__ */ (0, W.jsx)("div", {
+		children: [/* @__PURE__ */ (0, W.jsx)(ne, { children: _.message }), /* @__PURE__ */ (0, W.jsx)("div", {
 			className: "text-md",
 			children: _.stack
 		})]
@@ -655,7 +655,7 @@ function me(e) {
 	return e instanceof Set ? [...e] : e;
 }
 function he(e) {
-	return Array.isArray(e) ? e.map((e) => e instanceof Date && h(e) ? new Date(e).getTime() : e) : e;
+	return Array.isArray(e) ? e.map((e) => e instanceof Date && te(e) ? new Date(e).getTime() : e) : e;
 }
 function ge(e, t) {
 	return /* @__PURE__ */ (0, W.jsxs)("div", { children: [

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@marimo-team/islands",
-  "version": "0.24.1-dev42",
+  "version": "0.24.1-dev43",
   "main": "dist/main.js",
   "types": "dist/index.d.ts",
   "type": "module",

package/src/components/datasources/datasources.tsx

@@ -3,7 +3,7 @@
 import { CommandList } from "cmdk";
 import { atom, useAtom, useAtomValue, useSetAtom } from "jotai";
 import { atomWithStorage } from "jotai/utils";
-import { PlusIcon, PlusSquareIcon, XIcon } from "lucide-react";
+import { PlusSquareIcon, XIcon } from "lucide-react";
 import React from "react";
 import { dbDisplayName } from "@/components/databases/display";
 import { EngineVariable } from "@/components/databases/engine-variable";
@@ -64,7 +64,7 @@ import { ErrorBoundary } from "../editor/boundary/ErrorBoundary";
 import { PythonIcon } from "../editor/cell/code/icons";
 import { useAddCodeToNewCell } from "../editor/cell/useAddCell";
 import { PanelEmptyState } from "../editor/chrome/panels/empty-state";
-import { AddConnectionDialog } from "../editor/connections/add-connection-dialog";
+import { AddConnectionButton } from "../editor/connections/add-connection-button";
 import { DatasetColumnPreview } from "./column-preview";
 import {
   ColumnName,
@@ -250,12 +250,12 @@ export const DataSources: React.FC = () => {
         title="No tables found"
         description="Any datasets/dataframes in the global scope will be shown here."
         action={
-          <AddConnectionDialog>
-            <Button variant="outline" size="sm">
-              Add database or catalog
-              <PlusIcon className="h-4 w-4 ml-2" />
-            </Button>
-          </AddConnectionDialog>
+          <AddConnectionButton
+            group="database"
+            label="Add database or catalog"
+            variant="outline"
+            size="sm"
+          />
         }
         icon={<DatabaseIcon />}
       />
@@ -301,15 +301,14 @@ export const DataSources: React.FC = () => {
           className="px-2 rounded-none focus-visible:outline-hidden"
         />
 
-        <AddConnectionDialog>
-          <Button
-            variant="ghost"
-            size="sm"
-            className="px-2 rounded-none focus-visible:outline-hidden"
-          >
-            <PlusIcon className="h-4 w-4" />
-          </Button>
-        </AddConnectionDialog>
+        <AddConnectionButton
+          group="database"
+          label="Add database or catalog"
+          compact={true}
+          variant="ghost"
+          size="sm"
+          className="px-2 rounded-none focus-visible:outline-hidden"
+        />
       </div>
 
       <CommandList className="flex flex-col">

package/src/components/editor/actions/useNotebookActions.tsx

@@ -73,6 +73,7 @@ import { useRequestClient } from "@/core/network/requests";
 import { useFilename } from "@/core/saving/filename";
 import { createShareableLink } from "@/core/wasm/share";
 import { isWasm } from "@/core/wasm/utils";
+import { useDetectedDataSources } from "@/hooks/useDataSourceDiscovery";
 import { copyToClipboard } from "@/utils/copy";
 import { Objects } from "@/utils/objects";
 import { Strings } from "@/utils/strings";
@@ -81,6 +82,7 @@ import { useRunAllCells } from "../cell/useRunCells";
 import { useChromeActions, useChromeState } from "../chrome/state";
 import { isPanelHidden, PANELS } from "../chrome/types";
 import { AddConnectionDialogContent } from "../connections/add-connection-dialog";
+import { useAddDetectedDataSource } from "../connections/components";
 import { keyboardShortcutsAtom } from "../controls/keyboard-shortcuts";
 import { commandPaletteAtom } from "../controls/state";
 import { displayLayoutName, getLayoutIcon } from "../renderers/layout-select";
@@ -135,6 +137,9 @@ export function useNotebookActions({
   const setKeyboardShortcutsOpen = useSetAtom(keyboardShortcutsAtom);
   const setExportOptions = useSetAtom(exportOptionsAtom);
   const { readCode, saveCellConfig } = useRequestClient();
+  const detectedDatabaseSources = useDetectedDataSources("database");
+  const detectedStorageSources = useDetectedDataSources("storage");
+  const addDetectedDataSource = useAddDetectedDataSource();
 
   const hasDisabledCells = useAtomValue(hasDisabledCellsAtom);
   const canUndoDeletes = useAtomValue(canUndoDeletesAtom);
@@ -455,6 +460,27 @@ export function useNotebookActions({
       handle: () => {
         openModal(<AddConnectionDialogContent onClose={closeModal} />);
       },
+      dropdown:
+        detectedDatabaseSources.length === 0
+          ? undefined
+          : [
+              ...detectedDatabaseSources.map((source) => ({
+                icon: <SparklesIcon size={14} strokeWidth={1.5} />,
+                label: `Add ${source.displayName}`,
+                description: "Detected in your environment",
+                handle: () => addDetectedDataSource(source),
+              })),
+              {
+                divider: true,
+                icon: <DatabaseIcon size={14} strokeWidth={1.5} />,
+                label: "Browse all connections",
+                handle: () => {
+                  openModal(
+                    <AddConnectionDialogContent onClose={closeModal} />,
+                  );
+                },
+              },
+            ],
     },
     {
       icon: <HardDrive size={14} strokeWidth={1.5} />,
@@ -467,6 +493,30 @@ export function useNotebookActions({
           />,
         );
       },
+      dropdown:
+        detectedStorageSources.length === 0
+          ? undefined
+          : [
+              ...detectedStorageSources.map((source) => ({
+                icon: <SparklesIcon size={14} strokeWidth={1.5} />,
+                label: `Add ${source.displayName}`,
+                description: "Detected in your environment",
+                handle: () => addDetectedDataSource(source),
+              })),
+              {
+                divider: true,
+                icon: <HardDrive size={14} strokeWidth={1.5} />,
+                label: "Browse all connections",
+                handle: () => {
+                  openModal(
+                    <AddConnectionDialogContent
+                      defaultTab="storage"
+                      onClose={closeModal}
+                    />,
+                  );
+                },
+              },
+            ],
     },
     {
       icon: <Undo2Icon size={14} strokeWidth={1.5} />,

package/src/components/editor/chrome/panels/components.tsx

@@ -105,7 +105,7 @@ const DiscoveredSourcesBadge = ({
   return (
     <Tooltip content={content}>
       <span>
-        <PanelBadge className="gap-0.5 border-(--blue-6) bg-(--blue-3) text-(--blue-11)">
+        <PanelBadge className="gap-0.5 border-(--blue-6) bg-(--blue-3) text-(--blue-11) hover:bg-(--blue-4) dark:border-(--blue-8) dark:bg-(--blue-3) dark:hover:bg-(--blue-4)">
           <SparklesIcon className="h-2.5 w-2.5" />
           {count}
         </PanelBadge>

package/src/components/editor/connections/__tests__/add-connection-button.test.tsx

@@ -0,0 +1,129 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+import { fireEvent, render, screen } from "@testing-library/react";
+import { Provider } from "jotai";
+import type React from "react";
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import { ModalProvider } from "@/components/modal/ImperativeModal";
+import { TooltipProvider } from "@/components/ui/tooltip";
+import type { DetectedDataSource } from "@/core/datasets/data-source-discovery";
+import { store } from "@/core/state/jotai";
+import { AddConnectionButton } from "../add-connection-button";
+
+const mocks = vi.hoisted(() => ({
+  addDetectedDataSource: vi.fn(),
+  sources: [] as DetectedDataSource[],
+}));
+
+const MYSQL_SOURCE: DetectedDataSource = {
+  id: "mysql-environment",
+  integration: "mysql",
+  category: "database",
+  displayName: "MySQL",
+  confidence: "high",
+  origins: [{ type: "environment", label: "Kernel environment" }],
+  configuration: [],
+  code: "engine = create_engine()",
+  hidesWhen: { kind: "dialect", substrings: ["mysql"] },
+};
+
+vi.mock("@/hooks/useDataSourceDiscovery", () => ({
+  useDetectedDataSources: () => mocks.sources,
+}));
+
+vi.mock("../components", async (importOriginal) => ({
+  ...(await importOriginal<typeof import("../components")>()),
+  useAddDetectedDataSource: () => mocks.addDetectedDataSource,
+}));
+
+function wrapper({ children }: { children: React.ReactNode }) {
+  return (
+    <Provider store={store}>
+      <TooltipProvider>
+        <ModalProvider>{children}</ModalProvider>
+      </TooltipProvider>
+    </Provider>
+  );
+}
+
+describe("AddConnectionButton", () => {
+  beforeEach(() => {
+    mocks.sources = [];
+    mocks.addDetectedDataSource.mockClear();
+    vi.stubGlobal("PointerEvent", MouseEvent);
+  });
+
+  afterEach(() => vi.unstubAllGlobals());
+
+  it("opens the connection dialog directly without suggestions", () => {
+    render(
+      <AddConnectionButton
+        group="database"
+        label="Add database or catalog"
+        variant="outline"
+      />,
+      { wrapper },
+    );
+
+    fireEvent.click(
+      screen.getByRole("button", { name: "Add database or catalog" }),
+    );
+
+    expect(
+      screen.getByRole("dialog", { name: "Add Connection" }),
+    ).toBeVisible();
+  });
+
+  it("opens an ordered suggestion menu when sources are detected", async () => {
+    mocks.sources = [MYSQL_SOURCE];
+    render(
+      <AddConnectionButton
+        group="database"
+        label="Add database or catalog"
+        variant="outline"
+      />,
+      { wrapper },
+    );
+
+    fireEvent.pointerDown(
+      screen.getByRole("button", {
+        name: "View detected database connections",
+      }),
+      { button: 0, ctrlKey: false },
+    );
+
+    const suggestion = await screen.findByRole("menuitem", {
+      name: "Add MySQL",
+    });
+    const browseAll = screen.getByRole("menuitem", {
+      name: "Browse all connections",
+    });
+    expect(
+      suggestion.compareDocumentPosition(browseAll) &
+        Node.DOCUMENT_POSITION_FOLLOWING,
+    ).toBeTruthy();
+
+    fireEvent.click(suggestion);
+    expect(mocks.addDetectedDataSource).toHaveBeenCalledWith(MYSQL_SOURCE);
+  });
+
+  it("keeps the primary action direct when sources are detected", () => {
+    mocks.sources = [MYSQL_SOURCE];
+    render(
+      <AddConnectionButton
+        group="database"
+        label="Add database or catalog"
+        variant="outline"
+      />,
+      { wrapper },
+    );
+
+    fireEvent.click(
+      screen.getByRole("button", { name: "Add database or catalog" }),
+    );
+
+    expect(
+      screen.getByRole("dialog", { name: "Add Connection" }),
+    ).toBeVisible();
+  });
+});

package/src/components/editor/connections/add-connection-button.tsx

@@ -0,0 +1,132 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+import {
+  DatabaseIcon,
+  HardDriveIcon,
+  PlusIcon,
+  SparklesIcon,
+} from "lucide-react";
+import type React from "react";
+import { useImperativeModal } from "@/components/modal/ImperativeModal";
+import { Button } from "@/components/ui/button";
+import {
+  DropdownMenu,
+  DropdownMenuContent,
+  DropdownMenuItem,
+  DropdownMenuSeparator,
+  DropdownMenuTrigger,
+} from "@/components/ui/dropdown-menu";
+import { Tooltip } from "@/components/ui/tooltip";
+import type { DataSourceDiscoveryGroup } from "@/core/datasets/data-source-discovery";
+import { useDetectedDataSources } from "@/hooks/useDataSourceDiscovery";
+import { cn } from "@/utils/cn";
+import {
+  AddConnectionDialog,
+  AddConnectionDialogContent,
+} from "./add-connection-dialog";
+import { useAddDetectedDataSource } from "./components";
+
+type AddConnectionButtonProps = Omit<
+  React.ComponentProps<typeof Button>,
+  "children"
+> & {
+  group: DataSourceDiscoveryGroup;
+  label: string;
+  compact?: boolean;
+};
+
+/**
+ * Opens the connection dialog directly and, when useful suggestions exist,
+ * adds a secondary menu for detected sources.
+ */
+export const AddConnectionButton: React.FC<AddConnectionButtonProps> = ({
+  group,
+  label,
+  compact = false,
+  className,
+  ...buttonProps
+}) => {
+  const sources = useDetectedDataSources(group);
+  const addDetectedDataSource = useAddDetectedDataSource();
+  const { openModal, closeModal } = useImperativeModal();
+  const hasSuggestions = sources.length > 0;
+  const defaultTab = group === "storage" ? "storage" : "databases";
+
+  const primaryButton = (
+    <Button
+      {...buttonProps}
+      className={cn(className, hasSuggestions && !compact && "rounded-r-none")}
+      aria-label={compact ? label : buttonProps["aria-label"]}
+    >
+      {!compact && label}
+      <PlusIcon className={cn("h-4 w-4", !compact && "ml-2")} />
+    </Button>
+  );
+
+  const primaryAction = (
+    <AddConnectionDialog defaultTab={defaultTab}>
+      {primaryButton}
+    </AddConnectionDialog>
+  );
+
+  if (!hasSuggestions) {
+    return primaryAction;
+  }
+
+  const BrowseIcon = group === "storage" ? HardDriveIcon : DatabaseIcon;
+  const suggestionsLabel =
+    group === "storage"
+      ? "View detected remote storage connections"
+      : "View detected database connections";
+
+  return (
+    <div className="inline-flex">
+      {primaryAction}
+      <DropdownMenu>
+        <Tooltip content={suggestionsLabel}>
+          <span className="inline-flex">
+            <DropdownMenuTrigger asChild={true}>
+              <Button
+                variant={buttonProps.variant}
+                size={buttonProps.size}
+                disabled={buttonProps.disabled}
+                className={cn(
+                  "rounded-l-none border-l-0 px-2 text-(--blue-10) hover:bg-(--blue-3)",
+                  compact && "rounded-none",
+                )}
+                aria-label={suggestionsLabel}
+              >
+                <SparklesIcon className="h-4 w-4" />
+              </Button>
+            </DropdownMenuTrigger>
+          </span>
+        </Tooltip>
+        <DropdownMenuContent align="end">
+          {sources.map((source) => (
+            <DropdownMenuItem
+              key={source.id}
+              onSelect={() => addDetectedDataSource(source)}
+            >
+              <SparklesIcon className="mr-2 h-4 w-4 text-(--blue-9)" />
+              Add {source.displayName}
+            </DropdownMenuItem>
+          ))}
+          <DropdownMenuSeparator />
+          <DropdownMenuItem
+            onSelect={() =>
+              openModal(
+                <AddConnectionDialogContent
+                  defaultTab={defaultTab}
+                  onClose={closeModal}
+                />,
+              )
+            }
+          >
+            <BrowseIcon className="mr-2 h-4 w-4" />
+            Browse all connections
+          </DropdownMenuItem>
+        </DropdownMenuContent>
+      </DropdownMenu>
+    </div>
+  );
+};

package/src/components/editor/connections/components.tsx

@@ -21,6 +21,7 @@ import { maybeAddMarimoImport } from "@/core/cells/add-missing-import";
 import { useCellActions } from "@/core/cells/cells";
 import { useLastFocusedCellId } from "@/core/cells/focus";
 import { autoInstantiateAtom } from "@/core/config/config";
+import type { DetectedDataSource } from "@/core/datasets/data-source-discovery";
 import {
   ENV_RENDERER,
   SECRET_TEXTAREA_RENDERER,
@@ -134,6 +135,16 @@ export function useInsertCode() {
   };
 }
 
+/** Insert the configured cell supplied by a detected data source. */
+export function useAddDetectedDataSource(onAdd?: () => void) {
+  const insertCode = useInsertCode();
+
+  return (source: DetectedDataSource) => {
+    insertCode(source.code);
+    onAdd?.();
+  };
+}
+
 /**
  * Generic connection form: Zod-driven form with secrets support, a library
  * picker, and Back/Add buttons. Used by both database and storage forms.

package/src/components/editor/connections/quick-add-data-sources.tsx

@@ -1,13 +1,14 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 
 import { PlusIcon, SparklesIcon } from "lucide-react";
+import { useId } from "react";
 import { Tooltip, TooltipProvider } from "@/components/ui/tooltip";
 import type {
   DataSourceDiscoveryGroup,
   DetectedDataSource,
 } from "@/core/datasets/data-source-discovery";
 import { useDetectedDataSources } from "@/hooks/useDataSourceDiscovery";
-import { useInsertCode } from "./components";
+import { useAddDetectedDataSource } from "./components";
 import { cn } from "@/utils/cn";
 
 export const QuickAddDataSources: React.FC<{
@@ -15,22 +16,24 @@ export const QuickAddDataSources: React.FC<{
   sources: DetectedDataSource[];
   onAdd: (source: DetectedDataSource) => void;
 }> = ({ className, sources, onAdd }) => {
+  const titleId = useId();
+
   if (sources.length === 0) {
     return null;
   }
 
   return (
     <section
-      aria-labelledby="quick-add-data-sources-title"
+      aria-labelledby={titleId}
       className={cn(
-        "rounded-full bg-[linear-gradient(135deg,var(--blue-2),var(--purple-3))] px-3 py-2",
+        "rounded-full bg-[linear-gradient(135deg,var(--blue-2),var(--purple-3))] px-3 py-2 dark:bg-[linear-gradient(135deg,var(--blue-2),var(--purple-2))]",
         className,
       )}
     >
       <div className="flex flex-wrap items-center gap-2">
         <div className="mr-1 flex items-center gap-1.5">
-          <SparklesIcon className="h-3.5 w-3.5 text-(--blue-9)" />
-          <h3 id="quick-add-data-sources-title" className="text-sm">
+          <SparklesIcon className="h-3.5 w-3.5 text-(--blue-9) dark:text-(--blue-10)" />
+          <h3 id={titleId} className="text-sm">
             Quick add
           </h3>
         </div>
@@ -44,7 +47,7 @@ export const QuickAddDataSources: React.FC<{
               <button
                 type="button"
                 aria-label={`Add ${source.displayName} connection`}
-                className="inline-flex items-center gap-1 rounded-full border border-border/60 bg-background px-2.5 py-1 text-xs font-medium text-foreground/90 transition-colors hover:border-(--blue-7) focus:outline-hidden focus:ring-2 focus:ring-ring focus:ring-offset-2"
+                className="inline-flex items-center gap-1 rounded-full border border-border/60 bg-background px-2.5 py-1 text-xs font-medium text-foreground/90 transition-colors hover:border-(--blue-7) focus:outline-hidden focus:ring-2 focus:ring-ring focus:ring-offset-2 dark:border-border dark:bg-background/60 dark:hover:bg-accent/80"
                 onClick={() => onAdd(source)}
               >
                 <PlusIcon className="h-3 w-3 text-muted-foreground" />
@@ -101,17 +104,14 @@ export const AutoDiscoveredDataSources: React.FC<{
   className?: string;
   group?: DataSourceDiscoveryGroup;
 }> = ({ onSubmit, className, group }) => {
-  const insertCode = useInsertCode();
+  const addDetectedDataSource = useAddDetectedDataSource(onSubmit);
   const sources = useDetectedDataSources(group);
 
   return (
     <QuickAddDataSources
       className={className}
       sources={sources}
-      onAdd={(source) => {
-        insertCode(source.code);
-        onSubmit?.();
-      }}
+      onAdd={addDetectedDataSource}
     />
   );
 };