diff --git a/runtime/loop.go b/runtime/loop.go index b6c8d6c..5155cab 100644 --- a/runtime/loop.go +++ b/runtime/loop.go @@ -352,7 +352,7 @@ func Run(ctx context.Context, cfg Config, user llm.Message) (Result, error) { return Result{}, err } messages = append(messages, last) - emit(cfg, Event{Type: "message_end", Message: last, PromptTokens: end.Usage.PromptTokens, CompletionTokens: end.Usage.CompletionTokens, TotalTokens: end.Usage.TotalTokens}) + emit(cfg, Event{Type: "message_end", Message: last, PromptTokens: end.Usage.PromptTokens, CompletionTokens: end.Usage.CompletionTokens, TotalTokens: end.Usage.TotalTokens, CacheReadTokens: end.Usage.CacheReadTokens, CacheWriteTokens: end.Usage.CacheWriteTokens}) if last.StopReason == llm.StopToolCalls && len(last.ToolCalls) > 0 { if reason := budgetExceeded(cfg.Budget, usage, toolCalls, time.Now()); reason != "" {