chat.ts

import { startsWith } from './startsWith';

import type { ChatMessageBase, ChatToolMessage } from '../../components';

export const getTextContent = (message: ChatMessageBase) => {
  return message.parts
    .map((part) => ('text' in part ? part.text : ''))
    .join('');
};

export const hasTextContent = (message: ChatMessageBase) => {
  return getTextContent(message).trim() !== '';
};

export const getToolParts = (message: ChatMessageBase) => {
  return message.parts.filter((part) => startsWith(part.type, 'tool-'));
};

export const hasToolParts = (message: ChatMessageBase) => {
  return getToolParts(message).length > 0;
};

export const isPartText = (
  part: ChatMessageBase['parts'][number]
): part is Extract<ChatMessageBase['parts'][number], { type: 'text' }> => {
  return part.type === 'text';
};

export const isPartTool = (
  part: ChatMessageBase['parts'][number]
): part is ChatToolMessage => {
  return startsWith(part.type, 'tool-');
};

export const toolHasOutput = (message: ChatMessageBase) => {
  return message.parts.some(
    (part) => isPartTool(part) && part.state === 'output-available'
  );
};