Function drainStream

pkg/sessiontitle/generator.go:170–186

drainStream reads the entire content of a chat completion stream and returns the concatenated delta content. The stream is always closed before returning.

(stream chat.MessageStream)

// returns the concatenated delta content. The stream is always closed before
// returning.
func drainStream(stream chat.MessageStream) (string, error) {
	defer stream.Close()

	var content strings.Builder
	for {
		response, err := stream.Recv()
		if errors.Is(err, io.EOF) {
			return content.String(), nil
		}
		if err != nil {
			return "", err
		}
		if len(response.Choices) > 0 {
			content.WriteString(response.Choices[0].Delta.Content)
		}
	}
}

// buildPrompt formats the user messages into the system+user message pair
// sent to the model.