refactor: 重构敏感词

This commit is contained in:
1808837298@qq.com committed 2024-05-23 23:59:55 +08:00
1 parent e35ec03f5b
commit 313c16f360
12 files changed
+158 -91

No files matched your search

+1 -1
View File
@@ -190,7 +190,7 @@ func OpenaiHandler(c *gin.Context, resp *http.Response, promptTokens int, model
if simpleResponse.Usage.TotalTokens == 0 {
completionTokens := 0
for _, choice := range simpleResponse.Choices {
ctkm, _, _ := service.CountTokenText(string(choice.Message.Content), model, false)
ctkm, _ := service.CountTokenText(string(choice.Message.Content), model)
completionTokens += ctkm
}
simpleResponse.Usage = dto.Usage{