Function getSiblingToolUseIDs

src/utils/messages.ts:1115–1144

(
  message: NormalizedMessage,
  messages: Message[],
)

}

export function getSiblingToolUseIDs(
  message: NormalizedMessage,
  messages: Message[],
): Set<string> {
  const toolUseID = getToolUseID(message)
  if (!toolUseID) {
    return new Set()
  }

  const unnormalizedMessage = messages.find(
    (_): _ is AssistantMessage =>
      _.type === 'assistant' &&
      _.message.content.some(_ => _.type === 'tool_use' && _.id === toolUseID),
  )
  if (!unnormalizedMessage) {
    return new Set()
  }

  const messageID = unnormalizedMessage.message.id
  const siblingMessages = messages.filter(
    (_): _ is AssistantMessage =>
      _.type === 'assistant' && _.message.id === messageID,
  )

  return new Set(
    siblingMessages.flatMap(_ =>
      _.message.content.filter(_ => _.type === 'tool_use').map(_ => _.id),
    ),
  )
}

export type MessageLookups = {
  siblingToolUseIDs: Map<string, Set<string>>