(
messages: Array<{ role: string; content: string }>,
systemPrompt?: string,
)
| 77 | * 每条消息额外计 4 tokens 的格式开销(role + 分隔符)。 |
| 78 | */ |
| 79 | export function countMessagesTokens( |
| 80 | messages: Array<{ role: string; content: string }>, |
| 81 | systemPrompt?: string, |
| 82 | ): number { |
| 83 | let total = 0 |
| 84 | |
| 85 | if (systemPrompt) { |
| 86 | total += countTokens(systemPrompt) + 4 |
| 87 | } |
| 88 | |
| 89 | for (const msg of messages) { |
| 90 | total += countTokens(msg.content) + 4 |
| 91 | } |
| 92 | |
| 93 | // 回复引导 token |
| 94 | total += 3 |
| 95 | |
| 96 | return total |
| 97 | } |
no test coverage detected