修改流式的实现,按照流式的格式去输出,不直接return

This commit is contained in:
haochun
2024-10-09 14:46:03 +08:00
parent 5e3042752e
commit 9267c5f12e
2 changed files with 13 additions and 5 deletions

View File

@@ -202,7 +202,8 @@ func StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusC
// Check for known error codes and handle accordingly
if aliResponse.Code != "" {
return &model.ErrorWithStatusCode{
errorResponse := &model.ErrorWithStatusCode{
Error: model.Error{
Message: aliResponse.Message,
Type: aliResponse.Code,
@@ -210,7 +211,14 @@ func StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusC
Code: aliResponse.Code,
},
StatusCode: resp.StatusCode,
}, nil
}
err = render.ObjectData(c, errorResponse)
if err != nil {
logger.SysError(err.Error())
}
render.Done(c)
return nil, nil
}
if aliResponse.Usage.OutputTokens != 0 {