Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

在function_call为空时不计费

Browse Source
This commit is contained in:
glzjin 2023-08-17 23:00:58 +08:00 committed by GitHub
parent 64217d1f50
commit bcd46f0cf7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
controller/relay-openai.go
View File

@ -52,9 +52,11 @@ func openaiStreamHandler(c *gin.Context, resp *http.Response, relayMode int) (*O
} }
for _, choice := range streamResponse.Choices { for _, choice := range streamResponse.Choices {
responseText += choice.Delta.Content responseText += choice.Delta.Content
if choice.Delta.FunctionCall != nil {
responseFunctionCallName += choice.Delta.FunctionCall.Name responseFunctionCallName += choice.Delta.FunctionCall.Name
responseFunctionCallArguments += choice.Delta.FunctionCall.Arguments responseFunctionCallArguments += choice.Delta.FunctionCall.Arguments
} }
}
case RelayModeCompletions: case RelayModeCompletions:
var streamResponse CompletionsStreamResponse var streamResponse CompletionsStreamResponse
err := json.Unmarshal([]byte(data), &streamResponse) err := json.Unmarshal([]byte(data), &streamResponse)