extractUsage function

  1. @visibleForTesting
GenerationUsage? extractUsage(
  1. UsageMetadata? metadata
)

Implementation

@visibleForTesting
GenerationUsage? extractUsage(gcl.UsageMetadata? metadata) {
  if (metadata == null) return null;
  return GenerationUsage(
    inputTokens: metadata.promptTokenCount,
    outputTokens: metadata.candidatesTokenCount,
    totalTokens: metadata.totalTokenCount,
    thoughtsTokens: metadata.thoughtsTokenCount,
    cachedContentTokens: metadata.cachedContentTokenCount,
    custom: {
      if (metadata.toolUsePromptTokenCount != null)
        'toolUsePromptTokenCount': metadata.toolUsePromptTokenCount,
    },
  );
}