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?.toDouble(),
    outputTokens: metadata.candidatesTokenCount?.toDouble(),
    totalTokens: metadata.totalTokenCount?.toDouble(),
    thoughtsTokens: metadata.thoughtsTokenCount?.toDouble(),
    cachedContentTokens: metadata.cachedContentTokenCount?.toDouble(),
    custom: {
      if (metadata.toolUsePromptTokenCount != null)
        'toolUsePromptTokenCount': metadata.toolUsePromptTokenCount,
    },
  );
}