@marimo-team/islands 0.16.4 → 0.17.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@marimo-team/islands",
-  "version": "0.16.4",
+  "version": "0.17.0",
   "main": "dist/main.js",
   "types": "dist/index.d.ts",
   "type": "module",
@@ -19,19 +19,19 @@
   "dependencies": {
     "@ai-sdk/react": "^2.0.30",
     "@anywidget/types": "^0.2.0",
-    "@codemirror/autocomplete": "^6.18.7",
-    "@codemirror/commands": "^6.8.1",
-    "@codemirror/lang-markdown": "^6.3.4",
+    "@codemirror/autocomplete": "^6.19.0",
+    "@codemirror/commands": "^6.9.0",
+    "@codemirror/lang-markdown": "^6.4.0",
     "@codemirror/lang-python": "^6.2.1",
-    "@codemirror/lang-sql": "^6.9.1",
+    "@codemirror/lang-sql": "^6.10.0",
     "@codemirror/language": "^6.11.3",
     "@codemirror/language-data": "^6.5.1",
-    "@codemirror/legacy-modes": "^6.5.1",
-    "@codemirror/lint": "^6.8.5",
+    "@codemirror/legacy-modes": "^6.5.2",
+    "@codemirror/lint": "^6.9.0",
     "@codemirror/search": "^6.5.11",
     "@codemirror/state": "^6.5.2",
     "@codemirror/theme-one-dark": "^6.1.3",
-    "@codemirror/view": "^6.38.2",
+    "@codemirror/view": "^6.38.5",
     "@dagrejs/dagre": "^1.1.5",
     "@date-fns/tz": "^1.4.1",
     "@dnd-kit/core": "^6.3.1",
@@ -50,7 +50,7 @@
     "@lezer/markdown": "^1.4.3",
     "@lezer/python": "^1.1.18",
     "@marimo-team/codemirror-ai": "^0.3.2",
-    "@marimo-team/codemirror-languageserver": "1.15.24",
+    "@marimo-team/codemirror-languageserver": "^1.16.0",
     "@marimo-team/codemirror-mcp": "^0.1.5",
     "@marimo-team/codemirror-sql": "^0.2.3",
     "@marimo-team/llm-info": "workspace:*",
@@ -95,7 +95,7 @@
     "@types/jsdom": "^21.1.7",
     "@types/react-grid-layout": "^1.3.5",
     "@uidotdev/usehooks": "^2.4.1",
-    "@uiw/codemirror-extensions-langs": "^4.25.1",
+    "@uiw/codemirror-extensions-langs": "^4.25.2",
     "@uiw/react-codemirror": "4.25.1",
     "@uwdata/flechette": "^1.1.2",
     "@valtown/codemirror-codeium": "^1.1.1",
@@ -138,8 +138,8 @@
     "plotly.js": "^2.35.3",
     "pyodide": "0.27.7",
     "react-arborist": "^3.4.3",
-    "react-aria": "^3.43.1",
-    "react-aria-components": "^1.12.1",
+    "react-aria": "^3.44.0",
+    "react-aria-components": "^1.13.0",
     "react-codemirror-merge": "4.25.1",
     "react-dropzone": "^14.3.8",
     "react-error-boundary": "^5.0.0",

package/src/__mocks__/requests.ts

@@ -70,6 +70,8 @@ export const MockRequestClient = {
       listSecretKeys: vi.fn().mockResolvedValue({ keys: [] }),
       writeSecret: vi.fn().mockResolvedValue({}),
       invokeAiTool: vi.fn().mockResolvedValue({}),
+      clearCache: vi.fn().mockResolvedValue(null),
+      getCacheInfo: vi.fn().mockResolvedValue(null),
       ...overrides,
     };
   },

package/src/__tests__/chat-history.test.ts

@@ -0,0 +1,123 @@
+/* Copyright 2024 Marimo. All rights reserved. */
+
+import { afterAll, beforeAll, describe, expect, it } from "vitest";
+import { groupChatsByDate } from "../components/chat/chat-history-utils";
+import type { Chat } from "../core/ai/state";
+
+// Mock current time for consistent testing
+const mockNow = new Date("2024-01-15T12:00:00Z").getTime();
+
+// Mock Date.now to return our fixed time
+const originalDateNow = Date.now;
+beforeAll(() => {
+  Date.now = () => mockNow;
+});
+
+afterAll(() => {
+  Date.now = originalDateNow;
+});
+
+describe("groupChatsByDate", () => {
+  const createMockChat = (daysAgo: number, title: string): Chat => ({
+    id: `chat-${daysAgo}` as Chat["id"],
+    title,
+    messages: [],
+    createdAt: mockNow - daysAgo * 24 * 60 * 60 * 1000,
+    updatedAt: mockNow - daysAgo * 24 * 60 * 60 * 1000,
+  });
+
+  it("should group chats correctly by date periods", () => {
+    const chats: Chat[] = [
+      createMockChat(0, "Today chat"),
+      createMockChat(1, "Yesterday chat"),
+      createMockChat(2, "2 days ago chat"),
+      createMockChat(3, "3 days ago chat"),
+      createMockChat(5, "5 days ago chat"), // Should go to "This week"
+      createMockChat(10, "10 days ago chat"), // Should go to "This month"
+      createMockChat(40, "40 days ago chat"), // Should go to "Older"
+    ];
+
+    const result = groupChatsByDate(chats);
+
+    // Should have 7 groups
+    expect(result).toHaveLength(chats.length);
+
+    // Check Today group
+    const todayGroup = result.find((g) => g.label === "Today");
+    expect(todayGroup?.chats).toHaveLength(1);
+    expect(todayGroup?.chats[0].title).toBe("Today chat");
+
+    // Check Yesterday group
+    const yesterdayGroup = result.find((g) => g.label === "Yesterday");
+    expect(yesterdayGroup?.chats).toHaveLength(1);
+    expect(yesterdayGroup?.chats[0].title).toBe("Yesterday chat");
+
+    // Check 2d ago group
+    const twoDaysGroup = result.find((g) => g.label === "2d ago");
+    expect(twoDaysGroup?.chats).toHaveLength(1);
+    expect(twoDaysGroup?.chats[0].title).toBe("2 days ago chat");
+
+    // Check 3d ago group
+    const threeDaysGroup = result.find((g) => g.label === "3d ago");
+    expect(threeDaysGroup?.chats).toHaveLength(1);
+    expect(threeDaysGroup?.chats[0].title).toBe("3 days ago chat");
+
+    // Check This week group (should include 5)
+    const thisWeekGroup = result.find((g) => g.label === "This week");
+    expect(thisWeekGroup?.chats).toHaveLength(1);
+    expect(thisWeekGroup?.chats.map((c) => c.title)).toContain(
+      "5 days ago chat",
+    );
+
+    // Check This month group (should include 40 days ago)
+    const thisMonthGroup = result.find((g) => g.label === "This month");
+    expect(thisMonthGroup?.chats).toHaveLength(1);
+    expect(thisMonthGroup?.chats[0].title).toBe("10 days ago chat");
+
+    // Check Older group (should include 40 days ago)
+    const olderGroup = result.find((g) => g.label === "Older");
+    expect(olderGroup?.chats).toHaveLength(1);
+    expect(olderGroup?.chats[0].title).toBe("40 days ago chat");
+  });
+
+  it("should include all chats in some group", () => {
+    const chats: Chat[] = [
+      createMockChat(0, "Today"),
+      createMockChat(1, "Yesterday"),
+      createMockChat(2, "2 days ago"),
+      createMockChat(3, "3 days ago"),
+      createMockChat(5, "5 days ago"),
+      createMockChat(10, "10 days ago"),
+      createMockChat(20, "20 days ago"),
+      createMockChat(40, "40 days ago"),
+      createMockChat(100, "100 days ago"),
+    ];
+
+    const result = groupChatsByDate(chats);
+
+    // Count total chats across all groups
+    const totalChatsInGroups = result.reduce(
+      (sum, group) => sum + group.chats.length,
+      0,
+    );
+    expect(totalChatsInGroups).toBe(chats.length);
+  });
+
+  it("should handle empty chat list", () => {
+    const result = groupChatsByDate([]);
+    expect(result).toHaveLength(0);
+  });
+
+  it("should filter out empty groups", () => {
+    const chats: Chat[] = [
+      createMockChat(0, "Today chat"),
+      createMockChat(40, "Old chat"),
+    ];
+
+    const result = groupChatsByDate(chats);
+
+    // Should only have Today and Older groups, not the empty ones in between
+    expect(result).toHaveLength(2);
+    expect(result.map((g) => g.label)).toEqual(["Today", "Older"]);
+  });
+});

package/src/components/app-config/ai-config.tsx

@@ -930,6 +930,29 @@ export const AiAssistConfig: React.FC<AiConfigProps> = ({
   return (
     <SettingGroup>
       <SettingSubtitle>AI Assistant</SettingSubtitle>
+
+      <FormField
+        control={form.control}
+        name="ai.inline_tooltip"
+        render={({ field }) => (
+          <div className="flex flex-col gap-y-1">
+            <FormItem className={formItemClasses}>
+              <FormLabel className="font-normal">AI Edit Tooltip</FormLabel>
+              <FormControl>
+                <Checkbox
+                  data-testid="inline-ai-checkbox"
+                  checked={field.value === true}
+                  onCheckedChange={field.onChange}
+                />
+              </FormControl>
+            </FormItem>
+            <FormDescription>
+              Enable "Edit with AI" tooltip when selecting code.
+            </FormDescription>
+          </div>
+        )}
+      />
+
       <FormErrorsBanner />
       <ModelSelector
         label="Chat Model"

package/src/components/app-config/mcp-config.tsx

@@ -1,6 +1,6 @@
 /* Copyright 2024 Marimo. All rights reserved. */
 
-import { CheckSquareIcon } from "lucide-react";
+import { CheckSquareIcon, Loader2, RefreshCwIcon } from "lucide-react";
 import React from "react";
 import type { UseFormReturn } from "react-hook-form";
 import {
@@ -12,6 +12,8 @@ import {
 } from "@/components/ui/card";
 import { FormField, FormItem } from "@/components/ui/form";
 import type { UserConfig } from "@/core/config/config-schema";
+import { useMCPRefresh, useMCPStatus } from "../mcp/hooks";
+import { McpStatusText } from "../mcp/mcp-status-indicator";
 import { Button } from "../ui/button";
 import { Kbd } from "../ui/kbd";
 import { SettingSubtitle } from "./common";
@@ -45,10 +47,48 @@ const PRESET_CONFIGS: PresetConfig[] = [
 
 export const MCPConfig: React.FC<MCPConfigProps> = ({ form, onSubmit }) => {
   const { handleClick } = useOpenSettingsToTab();
+  const { data: status, refetch, isFetching } = useMCPStatus();
+  const { refresh, isRefreshing } = useMCPRefresh();
+
+  const handleRefresh = async () => {
+    await refresh();
+    refetch();
+  };
 
   return (
     <div className="flex flex-col gap-4">
-      <SettingSubtitle>MCP Servers</SettingSubtitle>
+      <div className="flex items-center justify-between">
+        <SettingSubtitle>MCP Servers</SettingSubtitle>
+        <div className="flex items-center gap-2">
+          {status && <McpStatusText status={status.status} />}
+          <Button
+            variant="outline"
+            size="xs"
+            onClick={handleRefresh}
+            disabled={isRefreshing || isFetching}
+          >
+            {isRefreshing || isFetching ? (
+              <Loader2 className="h-3 w-3 animate-spin" />
+            ) : (
+              <RefreshCwIcon className="h-3 w-3" />
+            )}
+          </Button>
+        </div>
+      </div>
+      {status?.error && (
+        <div className="text-xs text-red-600 dark:text-red-400 bg-red-50 dark:bg-red-950/20 p-2 rounded">
+          {status.error}
+        </div>
+      )}
+      {status?.servers && (
+        <div className="text-xs text-muted-foreground">
+          {Object.entries(status.servers).map(([server, status]) => (
+            <div key={server}>
+              {server}: <McpStatusText status={status} />
+            </div>
+          ))}
+        </div>
+      )}
       <p className="text-sm text-muted-foreground">
         Enable Model Context Protocol (MCP) servers to provide additional
         capabilities and data sources for AI features.

package/src/components/app-config/user-config-form.tsx

@@ -559,6 +559,35 @@ export const UserConfigForm: React.FC = () => {
               />
             </SettingGroup>
 
+            <SettingGroup title="Diagnostics">
+              <FormField
+                control={form.control}
+                name="diagnostics.sql_linter"
+                render={({ field }) => (
+                  <div className="flex flex-col space-y-1">
+                    <FormItem className={formItemClasses}>
+                      <FormLabel>SQL Linter</FormLabel>
+                      <FormControl>
+                        <Checkbox
+                          data-testid="sql-linter-checkbox"
+                          checked={field.value}
+                          onCheckedChange={field.onChange}
+                        />
+                      </FormControl>
+                      <FormMessage />
+                      <IsOverridden
+                        userConfig={config}
+                        name="diagnostics.sql_linter"
+                      />
+                    </FormItem>
+                    <FormDescription>
+                      Better linting and autocompletions for SQL cells.
+                    </FormDescription>
+                  </div>
+                )}
+              />
+            </SettingGroup>
+
             <SettingGroup title="Keymap">
               <FormField
                 control={form.control}
@@ -1228,30 +1257,6 @@ export const UserConfigForm: React.FC = () => {
               notebook to take effect.
             </p>
 
-            <FormField
-              control={form.control}
-              name="experimental.inline_ai_tooltip"
-              render={({ field }) => (
-                <div className="flex flex-col gap-y-1">
-                  <FormItem className={formItemClasses}>
-                    <FormLabel className="font-normal">
-                      AI Edit Tooltip
-                    </FormLabel>
-                    <FormControl>
-                      <Checkbox
-                        data-testid="inline-ai-checkbox"
-                        checked={field.value === true}
-                        onCheckedChange={field.onChange}
-                      />
-                    </FormControl>
-                  </FormItem>
-                  <FormDescription>
-                    Enable experimental "Edit with AI" tooltip when selecting
-                    code.
-                  </FormDescription>
-                </div>
-              )}
-            />
             <FormField
               control={form.control}
               name="experimental.rtc_v2"
@@ -1305,31 +1310,6 @@ export const UserConfigForm: React.FC = () => {
                 </div>
               )}
             />
-            <FormField
-              control={form.control}
-              name="experimental.sql_linter"
-              render={({ field }) => (
-                <div className="flex flex-col gap-y-1">
-                  <FormItem className={formItemClasses}>
-                    <FormLabel className="font-normal">SQL Linter</FormLabel>
-                    <FormControl>
-                      <Checkbox
-                        data-testid="sql-linter-checkbox"
-                        checked={field.value === true}
-                        onCheckedChange={field.onChange}
-                      />
-                    </FormControl>
-                  </FormItem>
-                  <IsOverridden
-                    userConfig={config}
-                    name="experimental.sql_linter"
-                  />
-                  <FormDescription>
-                    Enable experimental SQL linting and autocompletion.
-                  </FormDescription>
-                </div>
-              )}
-            />
             <FormField
               control={form.control}
               name="experimental.external_agents"

package/src/components/chat/acp/__tests__/context-utils.test.ts

@@ -133,7 +133,7 @@ describe("parseContextFromPrompt", () => {
       type: "resource",
       resource: {
         uri: "context.md",
-        mimeType: "text/markdown",
+        mimeType: "text/plain",
         text: "formatted context",
       },
     });

package/src/components/chat/acp/agent-panel.tsx

@@ -867,7 +867,7 @@ const AgentPanel: React.FC = () => {
             type: "resource",
             resource: {
               uri: "marimo_rules.md",
-              mimeType: "text/markdown",
+              mimeType: "text/plain",
               text: getAgentPrompt(filename),
             },
           },

package/src/components/chat/acp/blocks.tsx

@@ -30,7 +30,6 @@ import {
 } from "@/components/ui/popover";
 import { logNever } from "@/utils/assertNever";
 import { cn } from "@/utils/cn";
-import { type Base64String, base64ToDataURL } from "@/utils/json/base64";
 import { Strings } from "@/utils/strings";
 import { MarkdownRenderer } from "../markdown-renderer";
 import { SimpleAccordion } from "./common";
@@ -289,7 +288,7 @@ export const PlansBlock = (props: { data: PlanNotificationEvent[] }) => {
 
 export const UserMessagesBlock = (props: { data: UserNotificationEvent[] }) => {
   return (
-    <div className="flex flex-col gap-2 text-muted-foreground border p-2 bg-background rounded break-words">
+    <div className="flex flex-col gap-2 text-muted-foreground border p-2 bg-background rounded break-words overflow-x-hidden">
       <ContentBlocks data={props.data.map((item) => item.content)} />
     </div>
   );
@@ -364,58 +363,28 @@ export const ResourceBlock = (props: { data: ContentBlockOf<"resource"> }) => {
     return (
       <Popover>
         <PopoverTrigger>
-          <span className="flex items-center gap-1">
+          <span className="flex items-center gap-1 hover:bg-muted rounded-md px-1">
             {props.data.resource.mimeType && (
               <MimeIcon mimeType={props.data.resource.mimeType} />
             )}
             {props.data.resource.uri}
           </span>
         </PopoverTrigger>
-        <PopoverContent className="max-h-96 overflow-y-auto scrollbar-thin">
-          <MarkdownRenderer content={props.data.resource.text} />
+        <PopoverContent className="max-h-96 overflow-y-auto scrollbar-thin whitespace-pre-wrap w-full max-w-[500px]">
+          <span className="text-muted-foreground text-xs mb-1 italic">
+            Formatted for agents, not humans.
+          </span>
+          {props.data.resource.mimeType === "text/plain" ? (
+            <pre className="text-xs whitespace-pre-wrap p-2 bg-muted rounded-md break-words">
+              {props.data.resource.text}
+            </pre>
+          ) : (
+            <MarkdownRenderer content={props.data.resource.text} />
+          )}
         </PopoverContent>
       </Popover>
     );
   }
-
-  if ("blob" in props.data.resource) {
-    if (props.data.resource.mimeType?.startsWith("image/")) {
-      return (
-        <ImageBlock
-          data={{
-            type: "image",
-            mimeType: props.data.resource.mimeType,
-            data: props.data.resource.blob,
-          }}
-        />
-      );
-    }
-    if (props.data.resource.mimeType?.startsWith("audio/")) {
-      return (
-        <AudioBlock
-          data={{
-            type: "audio",
-            mimeType: props.data.resource.mimeType,
-            data: props.data.resource.blob,
-          }}
-        />
-      );
-    }
-    const dataURL = base64ToDataURL(
-      props.data.resource.blob as Base64String,
-      props.data.resource.mimeType ?? "",
-    );
-    return (
-      <a href={dataURL} className="flex items-center gap-1" download={true}>
-        {props.data.resource.mimeType && (
-          <MimeIcon mimeType={props.data.resource.mimeType} />
-        )}
-        {props.data.resource.uri}
-      </a>
-    );
-  }
-  logNever(props.data.resource);
-  return null;
 };
 
 export const ResourceLinkBlock = (props: {
@@ -427,15 +396,38 @@ export const ResourceLinkBlock = (props: {
         href={props.data.uri}
         target="_blank"
         rel="noopener noreferrer"
-        className="text-link hover:underline"
+        className="text-link hover:underline px-1"
       >
         {props.data.name}
       </a>
     );
   }
 
+  // Show image in popover for image mime types
+  if (props.data.mimeType?.startsWith("image/")) {
+    return (
+      <div>
+        <Popover>
+          <PopoverTrigger>
+            <span className="flex items-center gap-1 hover:bg-muted rounded-md px-1 cursor-pointer">
+              <MimeIcon mimeType={props.data.mimeType} />
+              {props.data.name || props.data.title || props.data.uri}
+            </span>
+          </PopoverTrigger>
+          <PopoverContent className="w-auto max-w-[500px] p-2">
+            <img
+              src={props.data.uri}
+              alt={props.data.name || props.data.title || "Image"}
+              className="max-w-full max-h-96 object-contain"
+            />
+          </PopoverContent>
+        </Popover>
+      </div>
+    );
+  }
+
   return (
-    <span className="flex items-center gap-1">
+    <span className="flex items-center gap-1 px-1">
       {props.data.mimeType && <MimeIcon mimeType={props.data.mimeType} />}
       {props.data.name || props.data.title || props.data.uri}
     </span>
@@ -443,22 +435,23 @@ export const ResourceLinkBlock = (props: {
 };
 
 export const MimeIcon = (props: { mimeType: string }) => {
+  const classNames = "h-2 w-2 flex-shrink-0";
   if (props.mimeType.startsWith("image/")) {
-    return <FileImageIcon className="h-2 w-2" />;
+    return <FileImageIcon className={classNames} />;
   }
   if (props.mimeType.startsWith("audio/")) {
-    return <FileAudio2Icon className="h-2 w-2" />;
+    return <FileAudio2Icon className={classNames} />;
   }
   if (props.mimeType.startsWith("video/")) {
-    return <FileVideoCameraIcon className="h-2 w-2" />;
+    return <FileVideoCameraIcon className={classNames} />;
   }
   if (props.mimeType.startsWith("text/")) {
-    return <FileTextIcon className="h-2 w-2" />;
+    return <FileTextIcon className={classNames} />;
   }
   if (props.mimeType.startsWith("application/")) {
-    return <FileJsonIcon className="h-2 w-2" />;
+    return <FileJsonIcon className={classNames} />;
   }
-  return <FileIcon className="h-2 w-2" />;
+  return <FileIcon className={classNames} />;
 };
 
 export const SessionNotificationsBlock = <
@@ -533,7 +526,7 @@ export const ToolNotificationsBlock = (props: {
   const toolCalls = mergeToolCalls(props.data);
 
   return (
-    <div className="flex flex-col text-muted-foreground">
+    <div className="flex flex-col text-muted-foreground overflow-x-hidden">
       {toolCalls.map((item) => (
         <SimpleAccordion
           key={item.toolCallId}

package/src/components/chat/acp/common.tsx

@@ -68,7 +68,7 @@ export const SimpleAccordion: React.FC<SimpleAccordionProps> = ({
         >
           <span className="flex items-center gap-1">
             {getStatusIcon()}
-            <code className="font-mono text-xs">{title}</code>
+            <code className="font-mono text-xs truncate">{title}</code>
           </span>
         </AccordionTrigger>
         <AccordionContent className="p-2">

package/src/components/chat/acp/context-utils.ts

@@ -93,7 +93,7 @@ export async function parseContextFromPrompt(
         type: "resource",
         resource: {
           uri: "context.md",
-          mimeType: "text/markdown",
+          mimeType: "text/plain",
           text: contextString,
         },
       });

package/src/components/chat/acp/session-tabs.tsx

@@ -30,7 +30,7 @@ const SessionTab: React.FC<SessionTabProps> = memo(
       <div
         className={cn(
           "flex items-center gap-1 px-2 py-1 text-xs border-r border-border bg-muted/30 hover:bg-muted/50 cursor-pointer min-w-0",
-          isActive && "bg-background border-b-0 relative z-10",
+          isActive && "bg-background border-b-0 relative z-1",
         )}
         onClick={() => onSelect(session.tabId)}
       >