fix: retry prompt tokens

This commit is contained in:
1808837298@qq.com
2025-01-02 16:33:00 +08:00
parent ba1aad8ac4
commit ef4c1a2e48
3 changed files with 2357 additions and 2785 deletions

View File

@@ -112,6 +112,7 @@ func TextHelper(c *gin.Context) (openaiErr *dto.OpenAIErrorWithStatusCode) {
var promptTokens int
if value, exists := c.Get("prompt_tokens"); exists {
promptTokens = value.(int)
relayInfo.PromptTokens = promptTokens
} else {
promptTokens, err = getPromptTokens(textRequest, relayInfo)
// count messages token error 计算promptTokens错误