fix: 移除流式响应结尾的[Done],以适应Gemini API的行为
This commit is contained in:
@@ -121,8 +121,8 @@ func GeminiTextGenerationStreamHandler(c *gin.Context, resp *http.Response, info
|
|||||||
usage.PromptTokensDetails.TextTokens = usage.PromptTokens
|
usage.PromptTokensDetails.TextTokens = usage.PromptTokens
|
||||||
usage.CompletionTokens = usage.TotalTokens - usage.PromptTokens
|
usage.CompletionTokens = usage.TotalTokens - usage.PromptTokens
|
||||||
|
|
||||||
// 结束流式响应
|
// 移除流式响应结尾的[Done],因为Gemini API没有发送Done的行为
|
||||||
helper.Done(c)
|
//helper.Done(c)
|
||||||
|
|
||||||
return usage, nil
|
return usage, nil
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user