some changes

openai/sse.go

@@ -0,0 +1,37 @@
+package openai
+
+import (
+	"bufio"
+	"io"
+	"iter"
+	"strings"
+)
+
+// Server-sent event
+
+func ReadSSE(r io.ReadCloser) iter.Seq[string] {
+	reader := bufio.NewReader(r)
+	return func(yield func(string) bool) {
+		for {
+			line, err := reader.ReadString('\n')
+			if err != nil {
+				return
+			}
+			if line == "" || line == "\n" {
+				continue
+			}
+			if strings.HasPrefix(line, "data: ") {
+				line = line[len("data: "):]
+			}
+			line = strings.TrimSpace(line)
+			line = strings.Trim(line, "\r")
+			line = strings.Trim(line, "\n")
+			if strings.HasPrefix(line, "[DONE]") {
+				return
+			}
+			if !yield(line) {
+				return
+			}
+		}
+	}
+}