Skip to content

Commit 0b9cb85

Browse files
authored
chore(client): refactor streaming slightly to better future proof it
2 parents a473e04 + a764571 commit 0b9cb85

File tree

1 file changed

+1
-16
lines changed

1 file changed

+1
-16
lines changed

packages/ssestream/ssestream.go

Lines changed: 1 addition & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -136,13 +136,6 @@ func NewStream[T any](decoder Decoder, err error) *Stream[T] {
136136
}
137137
}
138138

139-
var eventPrefixesToBeParsed = []string{
140-
"response.",
141-
"image_generation.",
142-
"image_generation.",
143-
"transcript.",
144-
}
145-
146139
// Next returns false if the stream has ended or an error occurred.
147140
// Call Stream.Current() to get the current value.
148141
// Call Stream.Err() to get the error.
@@ -172,15 +165,7 @@ func (s *Stream[T]) Next() bool {
172165

173166
var nxt T
174167

175-
hasPrefixToParse := false
176-
for _, prefix := range eventPrefixesToBeParsed {
177-
if strings.HasPrefix(s.decoder.Event().Type, prefix) {
178-
hasPrefixToParse = true
179-
break
180-
}
181-
}
182-
183-
if s.decoder.Event().Type == "" || hasPrefixToParse {
168+
if s.decoder.Event().Type == "" || !strings.HasPrefix(s.decoder.Event().Type, "thread.") {
184169
ep := gjson.GetBytes(s.decoder.Event().Data, "error")
185170
if ep.Exists() {
186171
s.err = fmt.Errorf("received error while streaming: %s", ep.String())

0 commit comments

Comments
 (0)