refactor streamReader

client.go

@@ -154,13 +154,15 @@ func sendRequestStream[T streamable](client *Client, req *http.Request) (*stream
 	if isFailureStatusCode(resp) {
 		return new(streamReader[T]), client.handleErrorResp(resp)
 	}
-	return &streamReader[T]{
+	reader := &streamReader[T]{
 		emptyMessagesLimit: client.config.EmptyMessagesLimit,
 		reader:             bufio.NewReader(resp.Body),
 		response:           resp,
 		errAccumulator:     utils.NewErrorAccumulator(),
 		unmarshaler:        &utils.JSONUnmarshaler{},
-	}, nil
+	}
+	reader.SetHeader(resp.Header)
+	return reader, nil
 }
 
 func (c *Client) setCommonHeaders(req *http.Request) {

stream_reader.go

@@ -27,6 +27,8 @@ type streamReader[T streamable] struct {
 	response       *http.Response
 	errAccumulator utils.ErrorAccumulator
 	unmarshaler    utils.Unmarshaler
+
+	httpHeader
 }
 
 func (stream *streamReader[T]) Recv() (response T, err error) {
@@ -109,11 +111,3 @@ func (stream *streamReader[T]) unmarshalError() (errResp *ErrorResponse) {
 func (stream *streamReader[T]) Close() {
 	stream.response.Body.Close()
 }
-
-func (stream *streamReader[T]) Header() http.Header {
-	return stream.response.Header
-}
-
-func (stream *streamReader[T]) GetRateLimitHeaders() RateLimitHeaders {
-	return NewRateLimitHeaders(stream.Header())
-}