@gram-ai/elements 1.18.4 → 1.18.6

package/src/components/assistant-ui/tool-fallback.tsx

@@ -0,0 +1,111 @@
+import { cn } from '@/lib/utils'
+import {
+  useAssistantState,
+  type ToolCallMessagePartComponent,
+} from '@assistant-ui/react'
+import { useToolApproval } from '@/hooks/useToolApproval'
+import {
+  ToolUI,
+  type ToolStatus,
+  type ContentItem,
+} from '@/components/ui/tool-ui'
+
+export const ToolFallback: ToolCallMessagePartComponent = ({
+  toolName,
+  toolCallId,
+  status,
+  result,
+  args,
+}) => {
+  const {
+    pendingApprovals,
+    whitelistTool,
+    confirmPendingApproval,
+    rejectPendingApproval,
+  } = useToolApproval()
+
+  // Check if this specific tool call has a pending approval
+  const pendingApproval = pendingApprovals.get(toolCallId)
+  const message = useAssistantState(({ message }) => message)
+  const toolParts = message.parts.filter((part) => part.type === 'tool-call')
+  const matchingMessagePartIndex = toolParts.findIndex(
+    (part) => part.toolName === toolName
+  )
+
+  const handleApproveOnce = () => {
+    confirmPendingApproval(toolCallId)
+  }
+
+  const handleApproveForSession = () => {
+    whitelistTool(toolName)
+    confirmPendingApproval(toolCallId)
+  }
+
+  const handleDeny = () => {
+    rejectPendingApproval(toolCallId)
+  }
+
+  // Map assistant-ui status to ToolUI status
+  const getToolStatus = (): ToolStatus => {
+    if (pendingApproval) return 'approval'
+    if (status.type === 'incomplete') return 'error'
+    if (status.type === 'complete') {
+      // Check if the result indicates an error (e.g., tool was denied)
+      if (
+        result &&
+        typeof result === 'object' &&
+        'isError' in result &&
+        result.isError
+      ) {
+        return 'error'
+      }
+      return 'complete'
+    }
+    return 'running'
+  }
+
+  // Parse result to structured content if possible
+  const getResult = ():
+    | string
+    | Record<string, unknown>
+    | { content: ContentItem[] }
+    | undefined => {
+    if (result === undefined) return undefined
+    // Check if it's structured content with a content array
+    if (
+      typeof result === 'object' &&
+      result !== null &&
+      'content' in result &&
+      Array.isArray((result as { content: unknown }).content)
+    ) {
+      return result as { content: ContentItem[] }
+    }
+    // Otherwise return as-is (string or object)
+    if (typeof result === 'string') return result
+    return result as Record<string, unknown>
+  }
+
+  return (
+    <div
+      className={cn(
+        'aui-tool-fallback-root flex w-full flex-col',
+        matchingMessagePartIndex !== -1 &&
+          matchingMessagePartIndex !== toolParts.length - 1 &&
+          'border-b'
+      )}
+    >
+      <ToolUI
+        name={toolName}
+        status={getToolStatus()}
+        request={args as Record<string, unknown>}
+        result={getResult()}
+        onApproveOnce={pendingApproval ? handleApproveOnce : undefined}
+        onApproveForSession={
+          pendingApproval ? handleApproveForSession : undefined
+        }
+        onDeny={pendingApproval ? handleDeny : undefined}
+        className="rounded-none border-0"
+      />
+    </div>
+  )
+}

package/src/components/assistant-ui/tool-group.tsx

@@ -0,0 +1,59 @@
+import { cn } from '@/lib/utils'
+import { useAssistantState } from '@assistant-ui/react'
+import { useMemo, type FC, type PropsWithChildren } from 'react'
+import { useElements } from '@/hooks/useElements'
+import { humanizeToolName } from '@/lib/humanize'
+import { ToolUIGroup } from '@/components/ui/tool-ui'
+
+export const ToolGroup: FC<
+  PropsWithChildren<{ startIndex: number; endIndex: number }>
+> = ({ children }) => {
+  const parts = useAssistantState(({ message }) => message).parts
+  const toolCallParts = parts.filter((part) => part.type === 'tool-call')
+  const anyMessagePartsAreRunning = toolCallParts.some(
+    (part) => part.status?.type === 'running'
+  )
+
+  const { config } = useElements()
+  const defaultExpanded = config.tools?.expandToolGroupsByDefault ?? false
+
+  const groupTitle = useMemo(() => {
+    const toolParts = parts.filter((part) => part.type === 'tool-call')
+
+    if (toolParts.length === 0) return 'No tools called'
+    if (toolParts.length === 1)
+      return `Calling ${humanizeToolName(toolParts[0].toolName)}...`
+    return anyMessagePartsAreRunning
+      ? `Calling ${toolParts.length} tools...`
+      : `Executed ${toolParts.length} tools`
+  }, [parts, anyMessagePartsAreRunning])
+
+  // If there's a custom component for the single tool, render children directly
+  if (config.tools?.components?.[toolCallParts[0]?.toolName]) {
+    return children
+  }
+
+  // For single tool calls, render without the group wrapper
+  if (toolCallParts.length === 1) {
+    return (
+      <div className={cn('my-4 w-full max-w-xl')}>
+        <div className="border-border bg-card overflow-hidden rounded-lg border">
+          {children}
+        </div>
+      </div>
+    )
+  }
+
+  // For multiple tool calls, use the group component
+  return (
+    <div className="my-4 w-full max-w-xl">
+      <ToolUIGroup
+        title={groupTitle}
+        status={anyMessagePartsAreRunning ? 'running' : 'complete'}
+        defaultExpanded={defaultExpanded}
+      >
+        {children}
+      </ToolUIGroup>
+    </div>
+  )
+}

package/src/components/assistant-ui/tooltip-icon-button.tsx

@@ -0,0 +1,57 @@
+'use client'
+
+import { ComponentPropsWithRef, forwardRef } from 'react'
+import { Slottable } from '@radix-ui/react-slot'
+
+import {
+  Tooltip,
+  TooltipContent,
+  TooltipTrigger,
+} from '@/components/ui/tooltip'
+import { Button } from '@/components/ui/button'
+import { cn } from '@/lib/utils'
+
+type TooltipIconButtonProps = ComponentPropsWithRef<typeof Button> & {
+  tooltip: string
+  side?: 'top' | 'bottom' | 'left' | 'right'
+  align?: 'start' | 'center' | 'end'
+}
+
+export const TooltipIconButton = forwardRef<
+  HTMLButtonElement,
+  TooltipIconButtonProps
+>(
+  (
+    {
+      children,
+      tooltip,
+      side = 'bottom',
+      align = 'center',
+      className,
+      ...rest
+    },
+    ref
+  ) => {
+    return (
+      <Tooltip>
+        <TooltipTrigger asChild>
+          <Button
+            variant="ghost"
+            size="icon"
+            {...rest}
+            className={cn('aui-button-icon size-6 p-1', className)}
+            ref={ref}
+          >
+            <Slottable>{children}</Slottable>
+            <span className="aui-sr-only sr-only">{tooltip}</span>
+          </Button>
+        </TooltipTrigger>
+        <TooltipContent side={side} align={align}>
+          {tooltip}
+        </TooltipContent>
+      </Tooltip>
+    )
+  }
+)
+
+TooltipIconButton.displayName = 'TooltipIconButton'

package/src/components/ui/avatar.tsx

@@ -0,0 +1,51 @@
+import * as React from 'react'
+import * as AvatarPrimitive from '@radix-ui/react-avatar'
+
+import { cn } from '@/lib/utils'
+
+function Avatar({
+  className,
+  ...props
+}: React.ComponentProps<typeof AvatarPrimitive.Root>) {
+  return (
+    <AvatarPrimitive.Root
+      data-slot="avatar"
+      className={cn(
+        'relative flex size-8 shrink-0 overflow-hidden rounded-full',
+        className
+      )}
+      {...props}
+    />
+  )
+}
+
+function AvatarImage({
+  className,
+  ...props
+}: React.ComponentProps<typeof AvatarPrimitive.Image>) {
+  return (
+    <AvatarPrimitive.Image
+      data-slot="avatar-image"
+      className={cn('aspect-square size-full', className)}
+      {...props}
+    />
+  )
+}
+
+function AvatarFallback({
+  className,
+  ...props
+}: React.ComponentProps<typeof AvatarPrimitive.Fallback>) {
+  return (
+    <AvatarPrimitive.Fallback
+      data-slot="avatar-fallback"
+      className={cn(
+        'bg-muted flex size-full items-center justify-center rounded-full',
+        className
+      )}
+      {...props}
+    />
+  )
+}
+
+export { Avatar, AvatarImage, AvatarFallback }

package/src/components/ui/button.tsx

@@ -0,0 +1,27 @@
+import * as React from 'react'
+import { Slot } from '@radix-ui/react-slot'
+import { type VariantProps } from 'class-variance-authority'
+
+import { cn } from '@/lib/utils'
+import { buttonVariants } from './buttonVariants'
+
+const Button = React.forwardRef<
+  HTMLButtonElement,
+  React.ComponentProps<'button'> &
+    VariantProps<typeof buttonVariants> & {
+      asChild?: boolean
+    }
+>(({ className, variant, size, asChild = false, ...props }, ref) => {
+  const Comp = asChild ? Slot : 'button'
+  return (
+    <Comp
+      data-slot="button"
+      className={cn(buttonVariants({ variant, size, className }))}
+      ref={ref}
+      {...props}
+    />
+  )
+})
+Button.displayName = 'Button'
+
+export { Button }

package/src/components/ui/buttonVariants.ts

@@ -0,0 +1,33 @@
+import { cva } from 'class-variance-authority'
+
+export const buttonVariants = cva(
+  "inline-flex items-center justify-center gap-2 whitespace-nowrap rounded-md text-sm font-medium transition-all disabled:pointer-events-none disabled:opacity-50 [&_svg]:pointer-events-none [&_svg:not([class*='size-'])]:size-4 shrink-0 [&_svg]:shrink-0 outline-none focus-visible:border-ring focus-visible:ring-ring/50 focus-visible:ring-[3px] aria-invalid:ring-destructive/20 dark:aria-invalid:ring-destructive/40 aria-invalid:border-destructive",
+  {
+    variants: {
+      variant: {
+        default: 'bg-primary text-primary-foreground hover:bg-primary/90',
+        destructive:
+          'bg-destructive text-white hover:bg-destructive/90 focus-visible:ring-destructive/20 dark:focus-visible:ring-destructive/40 dark:bg-destructive/60',
+        outline:
+          'border bg-background shadow-xs hover:bg-accent hover:text-accent-foreground dark:bg-input/30 dark:border-input dark:hover:bg-input/50',
+        secondary:
+          'bg-secondary text-secondary-foreground hover:bg-secondary/80',
+        ghost:
+          'hover:bg-accent hover:text-accent-foreground dark:hover:bg-accent/50',
+        link: 'text-primary underline-offset-4 hover:underline',
+      },
+      size: {
+        default: 'h-9 px-4 py-2 has-[>svg]:px-3',
+        sm: 'h-8 rounded-md gap-1.5 px-3 has-[>svg]:px-2.5',
+        lg: 'h-10 rounded-md px-6 has-[>svg]:px-4',
+        icon: 'size-9',
+        'icon-sm': 'size-8',
+        'icon-lg': 'size-10',
+      },
+    },
+    defaultVariants: {
+      variant: 'default',
+      size: 'default',
+    },
+  }
+)

package/src/components/ui/collapsible.tsx

@@ -0,0 +1,31 @@
+import * as CollapsiblePrimitive from '@radix-ui/react-collapsible'
+
+function Collapsible({
+  ...props
+}: React.ComponentProps<typeof CollapsiblePrimitive.Root>) {
+  return <CollapsiblePrimitive.Root data-slot="collapsible" {...props} />
+}
+
+function CollapsibleTrigger({
+  ...props
+}: React.ComponentProps<typeof CollapsiblePrimitive.CollapsibleTrigger>) {
+  return (
+    <CollapsiblePrimitive.CollapsibleTrigger
+      data-slot="collapsible-trigger"
+      {...props}
+    />
+  )
+}
+
+function CollapsibleContent({
+  ...props
+}: React.ComponentProps<typeof CollapsiblePrimitive.CollapsibleContent>) {
+  return (
+    <CollapsiblePrimitive.CollapsibleContent
+      data-slot="collapsible-content"
+      {...props}
+    />
+  )
+}
+
+export { Collapsible, CollapsibleTrigger, CollapsibleContent }

package/src/components/ui/dialog.tsx

@@ -0,0 +1,141 @@
+import * as React from 'react'
+import * as DialogPrimitive from '@radix-ui/react-dialog'
+import { XIcon } from 'lucide-react'
+
+import { cn } from '@/lib/utils'
+
+function Dialog({
+  ...props
+}: React.ComponentProps<typeof DialogPrimitive.Root>) {
+  return <DialogPrimitive.Root data-slot="dialog" {...props} />
+}
+
+function DialogTrigger({
+  ...props
+}: React.ComponentProps<typeof DialogPrimitive.Trigger>) {
+  return <DialogPrimitive.Trigger data-slot="dialog-trigger" {...props} />
+}
+
+function DialogPortal({
+  ...props
+}: React.ComponentProps<typeof DialogPrimitive.Portal>) {
+  return <DialogPrimitive.Portal data-slot="dialog-portal" {...props} />
+}
+
+function DialogClose({
+  ...props
+}: React.ComponentProps<typeof DialogPrimitive.Close>) {
+  return <DialogPrimitive.Close data-slot="dialog-close" {...props} />
+}
+
+function DialogOverlay({
+  className,
+  ...props
+}: React.ComponentProps<typeof DialogPrimitive.Overlay>) {
+  return (
+    <DialogPrimitive.Overlay
+      data-slot="dialog-overlay"
+      className={cn(
+        'data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 fixed inset-0 z-50 bg-black/50',
+        className
+      )}
+      {...props}
+    />
+  )
+}
+
+function DialogContent({
+  className,
+  children,
+  showCloseButton = true,
+  ...props
+}: React.ComponentProps<typeof DialogPrimitive.Content> & {
+  showCloseButton?: boolean
+}) {
+  return (
+    <DialogPortal data-slot="dialog-portal">
+      <DialogOverlay />
+      <DialogPrimitive.Content
+        data-slot="dialog-content"
+        className={cn(
+          'bg-background data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 fixed top-[50%] left-[50%] z-50 grid w-full max-w-[calc(100%-2rem)] translate-x-[-50%] translate-y-[-50%] gap-4 rounded-lg border p-6 shadow-lg duration-200 sm:max-w-lg',
+          className
+        )}
+        {...props}
+      >
+        {children}
+        {showCloseButton && (
+          <DialogPrimitive.Close
+            data-slot="dialog-close"
+            className="ring-offset-background focus:ring-ring data-[state=open]:bg-accent data-[state=open]:text-muted-foreground absolute top-4 right-4 rounded-xs opacity-70 transition-opacity hover:opacity-100 focus:ring-2 focus:ring-offset-2 focus:outline-hidden disabled:pointer-events-none [&_svg]:pointer-events-none [&_svg]:shrink-0 [&_svg:not([class*='size-'])]:size-4"
+          >
+            <XIcon />
+            <span className="sr-only">Close</span>
+          </DialogPrimitive.Close>
+        )}
+      </DialogPrimitive.Content>
+    </DialogPortal>
+  )
+}
+
+function DialogHeader({ className, ...props }: React.ComponentProps<'div'>) {
+  return (
+    <div
+      data-slot="dialog-header"
+      className={cn('flex flex-col gap-2 text-center sm:text-left', className)}
+      {...props}
+    />
+  )
+}
+
+function DialogFooter({ className, ...props }: React.ComponentProps<'div'>) {
+  return (
+    <div
+      data-slot="dialog-footer"
+      className={cn(
+        'flex flex-col-reverse gap-2 sm:flex-row sm:justify-end',
+        className
+      )}
+      {...props}
+    />
+  )
+}
+
+function DialogTitle({
+  className,
+  ...props
+}: React.ComponentProps<typeof DialogPrimitive.Title>) {
+  return (
+    <DialogPrimitive.Title
+      data-slot="dialog-title"
+      className={cn('text-lg leading-none font-semibold', className)}
+      {...props}
+    />
+  )
+}
+
+function DialogDescription({
+  className,
+  ...props
+}: React.ComponentProps<typeof DialogPrimitive.Description>) {
+  return (
+    <DialogPrimitive.Description
+      data-slot="dialog-description"
+      className={cn('text-muted-foreground text-sm', className)}
+      {...props}
+    />
+  )
+}
+
+export {
+  Dialog,
+  DialogClose,
+  DialogContent,
+  DialogDescription,
+  DialogFooter,
+  DialogHeader,
+  DialogOverlay,
+  DialogPortal,
+  DialogTitle,
+  DialogTrigger,
+}

package/src/components/ui/popover.tsx

@@ -0,0 +1,46 @@
+import * as React from 'react'
+import * as PopoverPrimitive from '@radix-ui/react-popover'
+
+import { cn } from '@/lib/utils'
+
+function Popover({
+  ...props
+}: React.ComponentProps<typeof PopoverPrimitive.Root>) {
+  return <PopoverPrimitive.Root data-slot="popover" {...props} />
+}
+
+function PopoverTrigger({
+  ...props
+}: React.ComponentProps<typeof PopoverPrimitive.Trigger>) {
+  return <PopoverPrimitive.Trigger data-slot="popover-trigger" {...props} />
+}
+
+function PopoverContent({
+  className,
+  align = 'center',
+  sideOffset = 4,
+  ...props
+}: React.ComponentProps<typeof PopoverPrimitive.Content>) {
+  return (
+    <PopoverPrimitive.Portal>
+      <PopoverPrimitive.Content
+        data-slot="popover-content"
+        align={align}
+        sideOffset={sideOffset}
+        className={cn(
+          'bg-popover text-popover-foreground data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 data-[side=bottom]:slide-in-from-top-2 data-[side=left]:slide-in-from-right-2 data-[side=right]:slide-in-from-left-2 data-[side=top]:slide-in-from-bottom-2 z-20 w-72 origin-(--radix-popover-content-transform-origin) rounded-md border p-4 shadow-md outline-hidden',
+          className
+        )}
+        {...props}
+      />
+    </PopoverPrimitive.Portal>
+  )
+}
+
+function PopoverAnchor({
+  ...props
+}: React.ComponentProps<typeof PopoverPrimitive.Anchor>) {
+  return <PopoverPrimitive.Anchor data-slot="popover-anchor" {...props} />
+}
+
+export { Popover, PopoverTrigger, PopoverContent, PopoverAnchor }

package/src/components/ui/skeleton.tsx

@@ -0,0 +1,13 @@
+import { cn } from '@/lib/utils'
+
+function Skeleton({ className, ...props }: React.ComponentProps<'div'>) {
+  return (
+    <div
+      data-slot="skeleton"
+      className={cn('bg-accent animate-pulse rounded-md', className)}
+      {...props}
+    />
+  )
+}
+
+export { Skeleton }