Function MessagePart

packages/ai-react-ui/src/chat-message.tsx:139–269

({
  part,
  isThinkingComplete,
  textPartRenderer,
  thinkingPartRenderer,
  toolsRenderer,
  defaultToolRenderer,
  toolResultRenderer,
}: {
  part: UIMessage['parts'][number]
  isThinkingComplete?: boolean
  textPartRenderer?: ChatMessageProps['textPartRenderer']
  thinkingPartRenderer?: ChatMessageProps['thinkingPartRenderer']
  toolsRenderer?: ChatMessageProps['toolsRenderer']
  defaultToolRenderer?: ChatMessageProps['defaultToolRenderer']
  toolResultRenderer?: ChatMessageProps['toolResultRenderer']
})

}

function MessagePart({
  part,
  isThinkingComplete,
  textPartRenderer,
  thinkingPartRenderer,
  toolsRenderer,
  defaultToolRenderer,
  toolResultRenderer,
}: {
  part: UIMessage['parts'][number]
  isThinkingComplete?: boolean
  textPartRenderer?: ChatMessageProps['textPartRenderer']
  thinkingPartRenderer?: ChatMessageProps['thinkingPartRenderer']
  toolsRenderer?: ChatMessageProps['toolsRenderer']
  defaultToolRenderer?: ChatMessageProps['defaultToolRenderer']
  toolResultRenderer?: ChatMessageProps['toolResultRenderer']
}) {
  // Text part
  if (part.type === 'text') {
    if (textPartRenderer) {
      return <>{textPartRenderer({ content: part.content })}</>
    }
    return (
      <div data-part-type="text" data-part-content>
        {part.content}
      </div>
    )
  }

  // Thinking part
  if (part.type === 'thinking') {
    if (thinkingPartRenderer) {
      return (
        <>
          {thinkingPartRenderer({
            content: part.content,
            isComplete: isThinkingComplete,
          })}
        </>
      )
    }
    return (
      <ThinkingPart content={part.content} isComplete={isThinkingComplete} />
    )
  }

  // Tool call part
  if (part.type === 'tool-call') {
    const toolProps: ToolCallRenderProps = {
      id: part.id,
      name: part.name,
      arguments: part.arguments,
      state: part.state,
      approval: part.approval,
      output: part.output,
    }

    // Check if there's a specific renderer for this tool