(messages: Message[])
| 136 | } |
| 137 | |
| 138 | export function getCurrentUsage(messages: Message[]): { |
| 139 | input_tokens: number |
| 140 | output_tokens: number |
| 141 | cache_creation_input_tokens: number |
| 142 | cache_read_input_tokens: number |
| 143 | } | null { |
| 144 | for (let i = messages.length - 1; i >= 0; i--) { |
| 145 | const message = messages[i] |
| 146 | const usage = message ? getTokenUsage(message) : undefined |
| 147 | if (usage) { |
| 148 | return { |
| 149 | input_tokens: usage.input_tokens, |
| 150 | output_tokens: usage.output_tokens, |
| 151 | cache_creation_input_tokens: usage.cache_creation_input_tokens ?? 0, |
| 152 | cache_read_input_tokens: usage.cache_read_input_tokens ?? 0, |
| 153 | } |
| 154 | } |
| 155 | } |
| 156 | return null |
| 157 | } |
| 158 | |
| 159 | export function doesMostRecentAssistantMessageExceed200k( |
| 160 | messages: Message[], |
no test coverage detected