more lint

internal/sse/stream.go

@@ -53,6 +53,7 @@ func (s *Streamer) connect(ctx context.Context) error {
 		}
 		s.attempt++
 		reconnectDelay := time.NewTimer(delay)
+		// once we only support go 1.23+, we can use time.After() here and simplify
 		defer reconnectDelay.Stop()
 		select {
 		case <-ctx.Done():
@@ -70,6 +71,7 @@ func (s *Streamer) connect(ctx context.Context) error {
 			req.Header.Set("Last-Event-ID", s.lastEventID)
 		}
 
+		//nolint:bodyclose
 		resp, err := s.c.Do(req)
 		if err != nil {
 			// try again

stream.go

@@ -298,16 +298,11 @@ func (f *fileStreamer) Close() error {
 	return f.s.Close()
 }
 
-type FileStreamer interface {
-	io.Closer
-	NextFile(ctx context.Context) (streaming.File, error)
-}
-
 // StreamPredictionFiles streams prediction file output via the replicate
 // streaming api.  It is the caller's responsibility to close the returned
 // FileStreamer to ensure connections and associated resources are cleaned up
 // appropriately.
-func (r *Client) StreamPredictionFiles(ctx context.Context, prediction *Prediction) (FileStreamer, error) {
+func (r *Client) StreamPredictionFiles(prediction *Prediction) (streaming.FileStreamer, error) {
 	url := prediction.URLs["stream"]
 	if url == "" {
 		return nil, errors.New("streaming not supported or not enabled for this prediction")

stream_test.go

@@ -174,7 +174,7 @@ event: done
 	c, err := replicate.NewClient(replicate.WithToken("test-token"))
 	require.NoError(t, err)
 
-	files, err := c.StreamPredictionFiles(ctx, p)
+	files, err := c.StreamPredictionFiles(p)
 
 	require.NoError(t, err)
 

streaming/file.go

@@ -5,6 +5,12 @@ import (
 	"io"
 )
 
+// FileStreamer represents a stream of output files from a model.
+type FileStreamer interface {
+	io.Closer
+	NextFile(ctx context.Context) (File, error)
+}
+
 // File represents a file output from a model over an SSE stream.  On the wire,
 // it might be a data URL or a regular http URL.  File abstracts over this and
 // provides a way to get the data regardless of the implementation.