X-Git-Url: http://git.giacomolli.uk/onechat/blobdiff_plain/a3b9849a54a6cd8a4cc5b1ae03a39aab31322f7b..522d7fd117285e4e4ce3ea5ec9b5a1722521b681:/build_run.go diff --git a/build_run.go b/build_run.go index 9c6b767..da75679 100644 --- a/build_run.go +++ b/build_run.go @@ -151,8 +151,10 @@ func buildReadUsage(path string) (records []ModelUsageRecord, err error) { scanner := bufio.NewScanner(file) scanner.Buffer(make([]byte, 64*1024), buildLineBytes) + for scanner.Scan() { var record ModelUsageRecord + err = json.Unmarshal(scanner.Bytes(), &record) if err != nil { return nil, fmt.Errorf("decode usage: %w", err) @@ -176,6 +178,7 @@ func buildReadUsage(path string) (records []ModelUsageRecord, err error) { // buildWriteReport writes the result of a build run. func buildWriteReport(out io.Writer, appended int, stop string, records []ModelUsageRecord, sizes []int) error { var input, cached, output, failed int + for _, record := range records { input += record.InputTokens cached += record.CachedTokens @@ -186,6 +189,7 @@ func buildWriteReport(out io.Writer, appended int, stop string, records []ModelU } var lines []string + if stop != "" { lines = append(lines, "stopped early: "+stop) } @@ -288,6 +292,7 @@ func buildTimeText(records []ModelUsageRecord) string { } var events []event + seconds := make([]float64, 0, len(records)) total := 0.0 for _, record := range records { @@ -324,6 +329,7 @@ func buildTimeText(records []ModelUsageRecord) string { } wall := events[len(events)-1].at.Sub(events[0].at).Seconds() + slices.Sort(seconds) return fmt.Sprintf("time: %.0f s in all, %.1f calls per minute, %.1f s per call (median %.1f, max %.1f), calls at once: at most %d, %.1f on average", wall, 60*float64(len(seconds))/max(wall, 1e-9), total/float64(len(seconds)), seconds[len(seconds)/2], seconds[len(seconds)-1], most, total/max(wall, 1e-9))