38 lines
626 B
Go
38 lines
626 B
Go
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
|
|
}
|
|
}
|
|
}
|
|
}
|