lkn
/
ollie-core
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix: skip [0 0 tokens] when counts are zero

This commit is contained in:
Levi Neely 2026-04-01 12:56:35 +02:00
parent d60a1b9215
commit 71ff058b1a
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
agent/loop.go
View File

@ -107,14 +107,19 @@ func (l *Loop) Run(ctx context.Context, state State) error {
l.emit(OutputMsg{Role: "tool", Name: tc.Name, Content: content})
}
// Only emit the full assistant text if we did NOT stream it.
if !l.streamed && resp.Message.Content != "" {
l.emit(OutputMsg{Role: "assistant", Content: resp.Message.Content})
}
l.emit(OutputMsg{
Role: "usage",
Content: fmt.Sprintf("↑%d ↓%d tokens", resp.Usage.InputTokens, resp.Usage.OutputTokens),
})
// Emit usage only when there are actual token counts (skips
// intermediate steps and avoids [↑0 ↓0 tokens] noise).
if resp.Usage.InputTokens > 0 || resp.Usage.OutputTokens > 0 {
l.emit(OutputMsg{
Role: "usage",
Content: fmt.Sprintf("↑%d ↓%d tokens", resp.Usage.InputTokens, resp.Usage.OutputTokens),
})
}
if err := state.Update(resp.Message, results); err != nil {
return fmt.Errorf("step %d update: %w", step, err)
@ -131,7 +136,6 @@ func (l *Loop) Run(ctx context.Context, state State) error {
}
l.skippedCalls = make(map[string]bool)
l.streamed = true
return nil
}