This commit is contained in:
Zhang Minghan 2024-02-18 18:10:30 +08:00
parent 082bd0695a
commit a3b7efd611

View File

@ -144,12 +144,12 @@ func sendTranshipmentResponse(c *gin.Context, form RelayForm, messages []globals
}) })
} }
func getFinishReason(data *globals.Chunk, end bool) interface{} { func getFinishReason(buffer *utils.Buffer, end bool) interface{} {
if end { if !end {
return nil return nil
} }
if data.ToolCall != nil { if buffer.IsFunctionCalling() {
return ReasonToolsCall return ReasonToolsCall
} }
@ -171,7 +171,7 @@ func getStreamTranshipmentForm(id string, created int64, form RelayForm, data *g
ToolCalls: data.ToolCall, ToolCalls: data.ToolCall,
FunctionCall: data.FunctionCall, FunctionCall: data.FunctionCall,
}, },
FinishReason: getFinishReason(data, end), FinishReason: getFinishReason(buffer, end),
}, },
}, },
Usage: Usage{ Usage: Usage{