浏览代码

return tool calls finish reason for openai (#5995)

* hot fix

* backend stream support

* clean up

* finish reason

* move to openai
royjhan 9 月之前
父节点
当前提交
365431d406
共有 1 个文件被更改,包括 3 次插入0 次删除
  1. 3 0
      openai/openai.go

+ 3 - 0
openai/openai.go

@@ -218,6 +218,9 @@ func toChatCompletion(id string, r api.ChatResponse) ChatCompletion {
 			Index:   0,
 			Index:   0,
 			Message: Message{Role: r.Message.Role, Content: r.Message.Content, ToolCalls: toolCalls},
 			Message: Message{Role: r.Message.Role, Content: r.Message.Content, ToolCalls: toolCalls},
 			FinishReason: func(reason string) *string {
 			FinishReason: func(reason string) *string {
+				if len(toolCalls) > 0 {
+					reason = "tool_calls"
+				}
 				if len(reason) > 0 {
 				if len(reason) > 0 {
 					return &reason
 					return &reason
 				}
 				}