ai-design-system 0.1.61 → 0.1.63

package/components/ai-elements/tool.tsx

@@ -40,7 +40,7 @@ export type ToolProps = ComponentProps<typeof Collapsible>;
 
 export const Tool = ({ className, ...props }: ToolProps) => (
   <Collapsible
-    className={cn("not-prose mb-4 w-full rounded-md border", className)}
+    className={cn("group not-prose mb-4 w-full rounded-md border overflow-hidden", className)}
     {...props}
   />
 );

package/components/blocks/AIConversation/AIConversation.tsx

@@ -267,6 +267,7 @@ export const AIConversation = React.memo<AIConversationProps>(
                           id: subAgent.id,
                           name: subAgent.subAgentName,
                           description: undefined,
+                          input: subAgent.input,
                           content: typeof subAgent.output === 'string' ? subAgent.output : (subAgent.output ? JSON.stringify(subAgent.output) : ''),
                           status: subAgent.status,
                           toolCalls: [],

package/components/composites/FileTreeExplorer/FileTreeExplorer.tsx

@@ -16,6 +16,11 @@ export type FileTreeNode = {
   children?: FileTreeNode[]
 }
 
+export interface FileDownloadResult {
+  blob: Blob
+  filename: string
+}
+
 export interface FileTreeExplorerProps {
   tree: FileTreeNode[]
   defaultExpanded?: Set<string>
@@ -24,6 +29,7 @@ export interface FileTreeExplorerProps {
   searchPlaceholder?: string
   onCreateClick?: () => void
   createButtonLabel?: string
+  onDownload?: () => Promise<FileDownloadResult | undefined>
   headerClassName?: string
   className?: string
 }
@@ -84,12 +90,28 @@ export const FileTreeExplorer = React.memo<FileTreeExplorerProps>(
     searchPlaceholder = "Filter files...",
     onCreateClick,
     createButtonLabel,
+    onDownload,
     headerClassName,
     className,
   }) => {
+    const downloadRef = React.useRef<HTMLAnchorElement>(null)
     const [searchQuery, setSearchQuery] = React.useState("")
     const [userExpanded, setUserExpanded] = React.useState<Set<string>>(defaultExpanded || new Set())
 
+    const handleDownloadClick = React.useCallback(async () => {
+      if (!onDownload) return
+      const result = await onDownload()
+      if (!result) return
+      const url = URL.createObjectURL(result.blob)
+      const anchor = downloadRef.current
+      if (anchor) {
+        anchor.href = url
+        anchor.download = result.filename
+        anchor.click()
+      }
+      URL.revokeObjectURL(url)
+    }, [onDownload])
+
     // Auto-expand parents when selectedPath changes
     React.useEffect(() => {
       if (selectedPath) {
@@ -145,6 +167,17 @@ export const FileTreeExplorer = React.memo<FileTreeExplorerProps>(
               {createButtonLabel ?? "New"}
             </button>
           ) : null}
+          {onDownload ? (
+            <button
+              type="button"
+              onClick={handleDownloadClick}
+              title="Download all files"
+              className="inline-flex h-7 w-7 items-center justify-center rounded-md text-muted-foreground transition-colors hover:bg-muted hover:text-foreground"
+            >
+              <Icon name="download" size="sm" />
+            </button>
+          ) : null}
+          <a ref={downloadRef} style={{ display: 'none' }} />
         </div>
         <div className="p-2">
           <FileTree

package/components/composites/FileTreeExplorer/index.ts

@@ -3,6 +3,7 @@ export {
 } from "./FileTreeExplorer"
 
 export type {
+  FileDownloadResult,
   FileTreeExplorerProps,
   FileTreeNode,
 } from "./FileTreeExplorer"

package/components/composites/SpecialistMessage/SpecialistMessage.tsx

@@ -1,12 +1,10 @@
 import * as React from "react"
 import {
-  Plan,
-  PlanHeader,
-  PlanTitle,
-  PlanDescription,
-  PlanTrigger,
-  PlanContent,
-} from "@/components/ai-elements/plan"
+  Tool,
+  ToolHeader,
+  ToolContent,
+  ToolInput,
+} from "@/components/ai-elements/tool"
 import { ToolCallDisplay, type ToolCall } from "@/components/composites/ToolCallDisplay"
 import { cn } from "@/lib/utils"
 import { Response } from "@/components/composites/response"
@@ -15,7 +13,7 @@ import { Response } from "@/components/composites/response"
  * SpecialistMessage Block
  *
  * A block component for displaying specialist agent messages with optional collapsible content.
- * Uses Plan AI element for consistent styling and optional space-efficient display.
+ * Uses Tool AI element for consistent styling with tool calls.
  */
 
 export interface SpecialistMessageData {
@@ -28,6 +26,7 @@ export interface SpecialistMessageData {
   status: "pending" | "active" | "completed" | "error"
   avatarSrc?: string
   avatarName?: string
+  input?: string | Record<string, unknown>
 }
 
 export interface SpecialistMessageProps {
@@ -56,6 +55,8 @@ export interface SpecialistMessageProps {
  */
 export const SpecialistMessage = React.memo<SpecialistMessageProps>(
   ({ message, isNested = false, defaultOpen = false, collapsible = true }) => {
+    const [isOpen, setIsOpen] = React.useState(collapsible ? defaultOpen : true)
+
     const hasContent = React.useMemo(
       () => message.content && message.content.trim() !== "",
       [message.content]
@@ -65,53 +66,63 @@ export const SpecialistMessage = React.memo<SpecialistMessageProps>(
       [message.toolCalls]
     )
 
-    const description = React.useMemo(() => {
-      if (message.description) return message.description;
-      if (message.status === "active") return "Thinking...";
-      return undefined;
-    }, [message.description, message.status]);
+    const toolState = React.useMemo(() => {
+      switch (message.status) {
+        case "pending":
+          return "input-streaming" as const
+        case "active":
+          return "input-available" as const
+        case "completed":
+          return "output-available" as const
+        case "error":
+          return "output-error" as const
+        default:
+          return "output-available" as const
+      }
+    }, [message.status])
 
     return (
-      <Plan
-        defaultOpen={collapsible ? defaultOpen : true}
-        className={cn(isNested && "ml-8")}
-        isStreaming={message.status === "active"}
+      <Tool
+        open={isOpen}
+        onOpenChange={collapsible ? setIsOpen : undefined}
       >
-        <PlanHeader>
-          <div>
-            <div className="mb-4 flex items-center gap-2">
-              {message.icon && message.icon}
-              <PlanTitle>{message.name}</PlanTitle>
-            </div>
-            {description && (
-              <PlanDescription>{description}</PlanDescription>
+        <ToolHeader
+          title={message.name}
+          type={`tool-${message.name}` as const}
+          state={toolState}
+        />
+
+        <ToolContent>
+          <div className="p-4 pt-0 flex flex-col gap-4">
+            {message.input && (
+              <ToolInput 
+                input={typeof message.input === 'string' ? { task: message.input } : message.input} 
+                className="p-0" 
+              />
             )}
-          </div>
-          {collapsible && <PlanTrigger />}
-        </PlanHeader>
 
-        <PlanContent>
-          {/* Message content */}
-          {hasContent && (
-            <Response
-              className="size-full [&>*:first-child]:mt-0 [&>*:last-child]:mb-0"
-              mode={message.status === "active" ? "streaming" : "static"}
-              isAnimating={message.status === "active"}
-            >
-              {message.content}
-            </Response>
-          )}
+            {/* Message content */}
+            {hasContent && (
+              <Response
+                className="size-full [&>*:first-child]:mt-0 [&>*:last-child]:mb-0"
+                mode={message.status === "active" ? "streaming" : "static"}
+                isAnimating={message.status === "active"}
+              >
+                {message.content}
+              </Response>
+            )}
 
-          {/* Tool calls */}
-          {hasToolCalls && (
-            <div className={cn("space-y-2", hasContent && "mt-4")}>
-              {message.toolCalls!.map((toolCall: ToolCall) => (
-                <ToolCallDisplay key={toolCall.id} toolCall={toolCall} />
-              ))}
-            </div>
-          )}
-        </PlanContent>
-      </Plan>
+            {/* Tool calls */}
+            {hasToolCalls && (
+              <div className={cn("space-y-2", hasContent && "mt-4")}>
+                {message.toolCalls!.map((toolCall: ToolCall) => (
+                  <ToolCallDisplay key={toolCall.id} toolCall={toolCall} />
+                ))}
+              </div>
+            )}
+          </div>
+        </ToolContent>
+      </Tool>
     )
   }
 )

package/components/features/TextEditor/TextEditor.tsx

@@ -37,7 +37,7 @@ import { DocumentEditorWithComments } from '@/components/blocks/DocumentEditorWi
 import { DocumentTabBar } from '@/components/composites/DocumentTabBar'
 import { AdjustableLayout } from '@/components/composites/AdjustableLayout'
 import { FileTreeExplorer } from '@/components/composites/FileTreeExplorer'
-import type { FileTreeNode } from '@/components/composites/FileTreeExplorer'
+import type { FileDownloadResult, FileTreeNode } from '@/components/composites/FileTreeExplorer'
 import { cn } from '@/lib/utils'
 import type { JSONContent } from '@tiptap/core'
 import type { Annotation, User } from '@/types/ai-editor/annotations'
@@ -105,6 +105,8 @@ export interface TextEditorMultiTabProps {
   fileTree?: FileTreeNode[]
   /** When true, hides the document tab bar — useful for single-file mode where the file tree provides navigation. */
   hideTabBar?: boolean
+  /** Callback when download all button is clicked in the file tree header */
+  onDownloadAllFiles?: () => Promise<FileDownloadResult | undefined>
 }
 
 /**
@@ -185,8 +187,7 @@ export const TextEditor = React.memo<TextEditorProps>(
     const currentDocument = useMemo(() => {
       if (isMultiTab && props.documents) {
         return (
-          props.documents.find((doc) => doc.file.id === props.activeDocumentId) ||
-          props.documents[0]
+          props.documents.find((doc) => doc.file.id === props.activeDocumentId) || null
         )
       }
       return null
@@ -213,7 +214,7 @@ export const TextEditor = React.memo<TextEditorProps>(
     )
     const fileTreeNodes = useMemo(() => {
       if (!isMultiTab) return []
-      
+
       const multiProps = props as TextEditorMultiTabProps
       if (multiProps.fileTree) {
         return multiProps.fileTree
@@ -263,9 +264,9 @@ export const TextEditor = React.memo<TextEditorProps>(
      */
     if (!isMultiTab) {
       const format = props.format || 'markdown'
-      const isMarkdown = format === 'markdown'
+      const isMarkdown = format === 'markdown' || format === 'md' || format === 'mdx'
       const isCode = !isMarkdown && typeof props.content === 'string'
-      
+
       if (isMarkdown || isCode) {
         const contentStr = props.content as string
         const content = isCode
@@ -300,7 +301,7 @@ export const TextEditor = React.memo<TextEditorProps>(
     /**
      * Multi-tab mode
      */
-    const { hideTabBar } = props as TextEditorMultiTabProps
+    const { hideTabBar, onDownloadAllFiles } = props as TextEditorMultiTabProps
 
     let editorPane: React.ReactNode
 
@@ -317,13 +318,13 @@ export const TextEditor = React.memo<TextEditorProps>(
             />
           )}
           <div className="flex-1 flex items-center justify-center p-6 text-muted-foreground">
-            No documents open
+            No documents selected
           </div>
         </div>
       )
     } else {
       const format = currentDocument.file.format || 'markdown'
-      const isMarkdownMulti = format === 'markdown'
+      const isMarkdownMulti = format === 'markdown' || format === 'md' || format === 'mdx'
       const isCodeMulti = !isMarkdownMulti && typeof currentDocument.content === 'string'
       const renderAsStreamdown = isMarkdownMulti || isCodeMulti
 
@@ -388,6 +389,7 @@ export const TextEditor = React.memo<TextEditorProps>(
                 tree={fileTreeNodes}
                 selectedPath={props.activeDocumentId}
                 onSelect={props.onTabSelect}
+                onDownload={onDownloadAllFiles}
               />
             ),
           },