修复doubao未返回token信息导致错误

2025-03-30 14:21:59 +08:00 · 2025-03-30 14:21:59 +08:00 · 81e342b154
parent 77fc9a844a
commit 81e342b154
1 changed files with 3 additions and 0 deletions
--- a/service/doubao.go
+++ b/service/doubao.go
@ -93,6 +93,9 @@ func DouBaoV2(modelParam proto.ModelParam, imCtx *proto.IMParamContext) {
 	req := model.CreateChatCompletionRequest{
 		Model:    modelParam.Model,
 		Messages: messages,
+		StreamOptions: &model.StreamOptions{
+			IncludeUsage: true,
+		},
 	}

 	stream, err := doubaoClient.CreateChatCompletionStream(ctx, req)