Skip to content

Commit 99a6161

Browse files
committed
fix(agent-task view): display premium requests used
Signed-off-by: Babak K. Shandiz <babakks@github.com>
1 parent fe95cd8 commit 99a6161

2 files changed

Lines changed: 118 additions & 93 deletions

File tree

pkg/cmd/agent-task/view/view.go

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ import (
77
"net/http"
88
"net/url"
99
"strconv"
10+
"strings"
1011
"time"
1112

1213
"github.com/MakeNowJust/heredoc"
@@ -306,13 +307,19 @@ func printSession(opts *ViewOptions, session *capi.Session) {
306307
fmt.Fprintf(opts.IO.Out, "Started %s\n", text.FuzzyAgo(time.Now(), session.CreatedAt))
307308
}
308309

309-
var durationNote string
310+
additionalNotes := make([]string, 0, 2)
311+
312+
if session.PremiumRequests > 0 {
313+
s := strings.TrimSuffix(fmt.Sprintf("%.1f", session.PremiumRequests), ".0")
314+
additionalNotes = append(additionalNotes, fmt.Sprintf("Used %s premium request(s)", s))
315+
}
316+
310317
if session.CompletedAt.After(session.CreatedAt) {
311-
durationNote = fmt.Sprintf("Duration %s", session.CompletedAt.Sub(session.CreatedAt).Round(time.Second).String())
318+
additionalNotes = append(additionalNotes, fmt.Sprintf("Duration %s", session.CompletedAt.Sub(session.CreatedAt).Round(time.Second).String()))
312319
}
313320

314-
if durationNote != "" {
315-
fmt.Fprintf(opts.IO.Out, "%s\n", cs.Muted(durationNote))
321+
if len(additionalNotes) > 0 {
322+
fmt.Fprintf(opts.IO.Out, "%s\n", cs.Muted(strings.Join(additionalNotes, " • ")))
316323
}
317324

318325
if !opts.Log {

0 commit comments

Comments
 (0)