@@ -83,7 +83,7 @@ type Client interface {
8383
8484// ChatCompletion creates a completion with the Chat completion endpoint which
8585// is what powers the ChatGPT experience.
86- ChatCompletionStream (ctx context.Context ,request ChatCompletionRequest ,onData func (* ChatCompletionStreamResponse ))error
86+ ChatCompletionStream (ctx context.Context ,request ChatCompletionRequest ,onData func (* ChatCompletionStreamResponse )error )error
8787
8888// Completion creates a completion with the default engine. This is the main endpoint of the API
8989// which auto-completes based on the given prompt.
@@ -211,7 +211,7 @@ func (c *client) ChatCompletion(ctx context.Context, request ChatCompletionReque
211211func (c * client )ChatCompletionStream (
212212ctx context.Context ,
213213request ChatCompletionRequest ,
214- onData func (* ChatCompletionStreamResponse ))error {
214+ onData func (* ChatCompletionStreamResponse )error )error {
215215if request .Model == "" {
216216request .Model = GPT3Dot5Turbo
217217}
@@ -252,7 +252,9 @@ func (c *client) ChatCompletionStream(
252252if err := json .Unmarshal (line ,output );err != nil {
253253return fmt .Errorf ("invalid json stream data: %v" ,err )
254254}
255- onData (output )
255+ if err := onData (output );err != nil {
256+ return fmt .Errorf ("callback returned an error: %v" ,err )
257+ }
256258}
257259
258260return nil