Skip to content

phsl / new-api

Go to a project
Commit 13ea5cac by CalciumIon
Options

feat: update stream options

parent efd18779
Showing with 2 additions and 6 deletions
relay/channel/openai/relay-openai.go
...@@ -78,9 +78,7 @@ func OpenaiStreamHandler(c *gin.Context, resp *http.Response, info *relaycommon. ...@@ -78,9 +78,7 @@ func OpenaiStreamHandler(c *gin.Context, resp *http.Response, info *relaycommon.
if err == nil { if err == nil {
if streamResponse.Usage != nil { if streamResponse.Usage != nil {
if streamResponse.Usage.TotalTokens != 0 { if streamResponse.Usage.TotalTokens != 0 {
usage.PromptTokens += streamResponse.Usage.PromptTokens usage = *streamResponse.Usage
usage.CompletionTokens += streamResponse.Usage.CompletionTokens
usage.TotalTokens += streamResponse.Usage.TotalTokens
} }
} }
for _, choice := range streamResponse.Choices { for _, choice := range streamResponse.Choices {
...@@ -101,9 +99,7 @@ func OpenaiStreamHandler(c *gin.Context, resp *http.Response, info *relaycommon. ...@@ -101,9 +99,7 @@ func OpenaiStreamHandler(c *gin.Context, resp *http.Response, info *relaycommon.
for _, streamResponse := range streamResponses { for _, streamResponse := range streamResponses {
if streamResponse.Usage != nil { if streamResponse.Usage != nil {
if streamResponse.Usage.TotalTokens != 0 { if streamResponse.Usage.TotalTokens != 0 {
usage.PromptTokens += streamResponse.Usage.PromptTokens usage = *streamResponse.Usage
usage.CompletionTokens += streamResponse.Usage.CompletionTokens
usage.TotalTokens += streamResponse.Usage.TotalTokens
} }
} }
for _, choice := range streamResponse.Choices { for _, choice := range streamResponse.Choices {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment