Worked

npx shadcn@latest add @simple-ai/worked

import {
  splitWorkedParts,
  Worked,
  WorkedContent,
  WorkedTrigger,
} from "@/components/ui/worked"

export function AssistantParts({
  parts,
  isStreaming,
  duration,
  renderPart,
}: {
  parts: Array<{ type: string }>
  isStreaming: boolean
  duration?: number
  renderPart: (part: { type: string }, index: number) => React.ReactNode
}) {
  return splitWorkedParts(parts).map((segment) => {
    if (segment.kind === "worked") {
      return (
        <Worked
          duration={duration}
          isStreaming={isStreaming}
          key={segment.items[0]?.index}
        >
          <WorkedTrigger />
          <WorkedContent>
            {segment.items.map((item) => renderPart(item.part, item.index))}
          </WorkedContent>
        </Worked>
      )
    }
    return renderPart(segment.item.part, segment.item.index)
  })
}