🐞 fix(chatstream): 修复星火大模型和豆包吞字的bug

Author: dingdinglz

README.md

@@ -2,6 +2,8 @@
 
 golang的支持调用所有openai范式的ai的api的库
 
+[![Go Report Card](https://goreportcard.com/badge/github.com/dingdinglz/openai)](https://goreportcard.com/report/github.com/dingdinglz/openai)
+
 ## 特征
 
 - 支持任何符合openai范式的api，如deepseek、kimi等

chat.go

@@ -4,6 +4,7 @@ import (
 	"bufio"
 	"encoding/json"
 	"errors"
+	"strings"
 )
 
 type Message struct {
@@ -180,11 +181,20 @@ func (client Client) ChatStream(model string, messages []Message, during func(st
 			break
 		}
 		if initFlag {
-			resError, e := parseRealError([]byte(_res))
-			if e == nil {
+			inError := _res
+			if strings.HasPrefix(_res, "data:") {
+				inError = inError[6:]
+			}
+			resError, e := parseRealError([]byte(inError))
+			if e != nil {
+				return errors.New(_res)
+			}
+			if resError != "" {
 				return errors.New(resError)
 			}
 			initFlag = false
+		}
+		if len(_res) < 7 {
 			continue
 		}
 		_res = _res[6:]
@@ -299,11 +309,20 @@ func (client Client) ChatReasonStream(model string, messages []Message, think fu
 			break
 		}
 		if initFlag {
-			resError, e := parseRealError([]byte(_res))
-			if e == nil {
+			inError := _res
+			if strings.HasPrefix(_res, "data:") {
+				inError = inError[6:]
+			}
+			resError, e := parseRealError([]byte(inError))
+			if e != nil {
+				return errors.New(_res)
+			}
+			if resError != "" {
 				return errors.New(resError)
 			}
 			initFlag = false
+		}
+		if len(_res) < 7 {
 			continue
 		}
 		_res = _res[6:]

chat_test.go

@@ -60,6 +60,7 @@ func TestChatStream(t *testing.T) {
 	})
 	if e != nil {
 		t.Error(e.Error())
+		return
 	}
 	e = client.ChatStreamWithConfig(ChatRequest{
 		Model: "deepseek-chat",