Merge pull request #333 from Calcium-Ion/suno

Update Suno
This commit is contained in:
Calcium-Ion authored and GitHub committed 2024-06-26 17:23:39 +08:00
commit 34eeac1cef
4 files changed
+1799 -1507

No files matched your search

+1 -1
View File
@@ -187,7 +187,7 @@ func OpenaiHandler(c *gin.Context, resp *http.Response, promptTokens int, model
return service.OpenAIErrorWrapper(err, "close_response_body_failed", http.StatusInternalServerError), nil
}
if simpleResponse.Usage.TotalTokens == 0 {
if simpleResponse.Usage.TotalTokens == 0 || (simpleResponse.Usage.PromptTokens == 0 && simpleResponse.Usage.CompletionTokens == 0) {
completionTokens := 0
for _, choice := range simpleResponse.Choices {
ctkm, _ := service.CountTokenText(string(choice.Message.Content), model)