check for expected EOF

deepgram · Jan 19, 2024 · 3c6e7e5 · 3c6e7e5
1 parent a4d89b2
commit 3c6e7e5
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pkg/client/live/client.go b/pkg/client/live/client.go
@@ -262,7 +262,11 @@ func (c *Client) Stream(r io.Reader) error {
 			return nil
 		default:
 			bytesRead, err := r.Read(chunk)
-			if err != nil {
+			if err == io.EOF && !c.retry {
+				klog.V(3).Infof("stream object EOF\n")
+				klog.V(6).Infof("live.Stream() LEAVE\n")
+				return nil
+			} else if err != nil {
 				klog.V(4).Infof("r.Read encountered EOF. Err: %v\n", err)
 				klog.V(6).Infof("live.Stream() LEAVE\n")
 				return err