prompt: Fix JobList Tokens (#85)

sunshineplan · Jul 12, 2024 · b3654ac · b3654ac
1 parent 7156a27
commit b3654ac
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/prompt/prompt.go b/prompt/prompt.go
@@ -154,9 +154,11 @@ func (prompt *Prompt) JobList(ctx context.Context, ai ai.AI, input []string, pre
  resp, err := chat(ai, prompt.d, r.Prompt)
  if err != nil {
  r.Result = nil
+ r.Tokens = 0
  r.Error = err
  } else {
  r.Result = resp.Results()
+ r.Tokens = resp.TokenCount().Total
  r.Error = nil
  }
  c <- r