From f262460e2fc38531301cfce84df14de3fd026834 Mon Sep 17 00:00:00 2001
From: Justin McCarthy <justinamccarthy@gmail.com>
Date: Mon, 16 Jun 2025 14:07:41 -0700
Subject: [PATCH] fix: check disconnect error

---
 internal/client.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/internal/client.go b/internal/client.go
index 9ff100e56..bc3807b3f 100644
--- a/internal/client.go
+++ b/internal/client.go
@@ -31,7 +31,11 @@ func (c *Client) Query(ctx context.Context, prompt string, opts *model.Options)
 	go func() {
 		defer close(out)
 		defer close(errCh)
-		defer c.Transport.Disconnect()
+		defer func() {
+			if err := c.Transport.Disconnect(); err != nil {
+				errCh <- err
+			}
+		}()
 		for data := range rawCh {
 			if errVal, ok := data["error"]; ok {
 				if e, ok2 := errVal.(error); ok2 {