( usageMetadata: GenerateContentResponseUsageMetadata | undefined )
| 143 | * We include toolUsePromptTokenCount in input and thoughtsTokenCount in output for correct billing. |
| 144 | */ |
| 145 | export function convertUsageMetadata( |
| 146 | usageMetadata: GenerateContentResponseUsageMetadata | undefined |
| 147 | ): GeminiUsage { |
| 148 | const thoughtsTokenCount = usageMetadata?.thoughtsTokenCount ?? 0 |
| 149 | const toolUsePromptTokenCount = usageMetadata?.toolUsePromptTokenCount ?? 0 |
| 150 | const promptTokenCount = (usageMetadata?.promptTokenCount ?? 0) + toolUsePromptTokenCount |
| 151 | const candidatesTokenCount = (usageMetadata?.candidatesTokenCount ?? 0) + thoughtsTokenCount |
| 152 | return { |
| 153 | promptTokenCount, |
| 154 | candidatesTokenCount, |
| 155 | totalTokenCount: usageMetadata?.totalTokenCount ?? 0, |
| 156 | } |
| 157 | } |
| 158 | |
| 159 | /** |
| 160 | * Tool definition for Gemini format |
no outgoing calls
no test coverage detected