This commit is contained in:
Zheng Kai
2023-03-31 12:18:23 +08:00
parent a0fe0b26ee
commit d2bede241b
5 changed files with 19 additions and 5 deletions

View File

@@ -6,6 +6,7 @@ import (
"project/metrics"
"project/pb"
"project/util"
"project/zj"
)
func doMetrics(ab []byte, cached bool, r *http.Request) {
@@ -22,6 +23,9 @@ func doMetrics(ab []byte, cached bool, r *http.Request) {
}
metrics.RspToken(u.PromptTokens, u.TotalTokens, cached)
if !cached {
zj.J(`token`, u.PromptTokens, u.TotalTokens)
}
ip, err := util.GetIP(r)
sip := ip.String()