(usage?: PiUsage)
| 56 | let toolRounds = 0 |
| 57 | |
| 58 | const addPiUsage = (usage?: PiUsage) => { |
| 59 | if (!usage) return |
| 60 | totalUsage.promptTokens += usage.input || 0 |
| 61 | totalUsage.completionTokens += usage.output || 0 |
| 62 | totalUsage.totalTokens += usage.totalTokens || usage.input + usage.output || 0 |
| 63 | totalUsage.cacheReadTokens += usage.cacheRead || 0 |
| 64 | totalUsage.cacheWriteTokens += usage.cacheWrite || 0 |
| 65 | } |
| 66 | |
| 67 | if (abortSignal?.aborted) { |
| 68 | return { usage: totalUsage, finalMessages: [], toolsUsed: [], toolRounds: 0 } |