diff --git a/service/doubao.go b/service/doubao.go index 17679e1..e76667c 100644 --- 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)