Merge 80c5528075 into 8df4a2670b

relay/adaptor/openai/main.go

@@ -19,7 +19,7 @@ import (
 )
 
 const (
-	dataPrefix       = "data: "
+	dataPrefix       = "data:"
 	done             = "[DONE]"
 	dataPrefixLength = len(dataPrefix)
 )
@@ -40,12 +40,16 @@ func StreamHandler(c *gin.Context, resp *http.Response, relayMode int) (*model.E
 		}
 		if data[:dataPrefixLength] != dataPrefix && data[:dataPrefixLength] != done {
 			continue
-		}
+        } else {
-		if strings.HasPrefix(data[dataPrefixLength:], done) {
+            payload := strings.TrimLeft(data[dataPrefixLength:], " ")
-			render.StringData(c, data)
+            // 这里处理标准的data:开头，标准化为 data: + 单空格格式
-			doneRendered = true
+            data = dataPrefix + " " + strings.TrimLeft(payload, " ")
-			continue
+            if strings.HasPrefix(payload, done) {
-		}
+                render.StringData(c, data)
+                doneRendered = true
+                continue
+            }
+        }
 		switch relayMode {
 		case relaymode.ChatCompletions:
 			var streamResponse ChatCompletionsStreamResponse