Function flattenModalityTokenCounts

packages/ai-gemini/src/usage.ts:30–67

(
  modalities?: Array<ModalityTokenCount>,
)

 * normalized structure.
 */
export function flattenModalityTokenCounts(
  modalities?: Array<ModalityTokenCount>,
): FlattenedModalityTokens {
  if (!modalities || modalities.length === 0) {
    return {}
  }

  const result: FlattenedModalityTokens = {}

  for (const item of modalities) {
    if (!item.modality || item.tokenCount === undefined) {
      continue
    }

    const modality = item.modality.toUpperCase()
    const count = item.tokenCount

    switch (modality) {
      case 'TEXT':
        result.textTokens = (result.textTokens ?? 0) + count
        break
      case 'IMAGE':
        result.imageTokens = (result.imageTokens ?? 0) + count
        break
      case 'AUDIO':
        result.audioTokens = (result.audioTokens ?? 0) + count
        break
      case 'VIDEO':
        result.videoTokens = (result.videoTokens ?? 0) + count
        break
      case 'DOCUMENT':
        result.documentTokens = (result.documentTokens ?? 0) + count
        break
    }
  }

  return result
}

/**
 * Checks if a FlattenedModalityTokens object has any values set.