(&mut self, usage: &TokenUsage)
| 97 | } |
| 98 | |
| 99 | pub(super) fn record_usage(&mut self, usage: &TokenUsage) { |
| 100 | self.total_usage.prompt_tokens += usage.prompt_tokens; |
| 101 | self.total_usage.completion_tokens += usage.completion_tokens; |
| 102 | self.total_usage.total_tokens += usage.total_tokens; |
| 103 | } |
| 104 | |
| 105 | pub(super) fn record_tool_call(&mut self) { |
| 106 | self.tool_calls_count += 1; |