Skip to content

Commit 4e27d4c

Browse files
committed
refactor: refactor codebase to improve efficiency and maintainability
- Remove the import of `strconv` - Simplify the token usage output by using `resp.Usage.String()` Signed-off-by: Bo-Yi Wu <[email protected]>
1 parent 33bf186 commit 4e27d4c

File tree

1 file changed

+2
-9
lines changed

1 file changed

+2
-9
lines changed

cmd/review.go

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
package cmd
22

33
import (
4-
"strconv"
54
"strings"
65

76
"github.com/appleboy/CodeGPT/core"
@@ -91,10 +90,7 @@ var reviewCmd = &cobra.Command{
9190
return err
9291
}
9392
summarizeMessage := resp.Content
94-
color.Magenta("PromptTokens: " + strconv.Itoa(resp.Usage.PromptTokens) +
95-
", CompletionTokens: " + strconv.Itoa(resp.Usage.CompletionTokens) +
96-
", TotalTokens: " + strconv.Itoa(resp.Usage.TotalTokens),
97-
)
93+
color.Magenta(resp.Usage.String())
9894

9995
if prompt.GetLanguage(viper.GetString("output.lang")) != prompt.DefaultLanguage {
10096
out, err = util.GetTemplateByString(
@@ -115,10 +111,7 @@ var reviewCmd = &cobra.Command{
115111
if err != nil {
116112
return err
117113
}
118-
color.Magenta("PromptTokens: " + strconv.Itoa(resp.Usage.PromptTokens) +
119-
", CompletionTokens: " + strconv.Itoa(resp.Usage.CompletionTokens) +
120-
", TotalTokens: " + strconv.Itoa(resp.Usage.TotalTokens),
121-
)
114+
color.Magenta(resp.Usage.String())
122115
summarizeMessage = resp.Content
123116
}
124117

0 commit comments

Comments
 (0)