From a07e39ded8e7313accaeaa1eec98f34d1f430149 Mon Sep 17 00:00:00 2001
From: shenlan <manbuzhe2009@qq.com>
Date: Sat, 9 Aug 2025 22:40:51 +0800
Subject: [PATCH] fix: prevent nil pointer in CLI upsert

---
 client/main.go | 46 +++++++++++++++++++++++++---------------------
 1 file changed, 25 insertions(+), 21 deletions(-)

diff --git a/client/main.go b/client/main.go
index 9249c56..7f2f954 100644
--- a/client/main.go
+++ b/client/main.go
@@ -114,27 +114,31 @@ func main() {
 		if err != nil {
 			log.Fatalf("marshal docs: %v", err)
 		}
-		var resp *http.Response
-		for i := 0; i < 3; i++ {
-			req, err := http.NewRequestWithContext(ctx, http.MethodPost, baseURL+"/api/rag/upsert", bytes.NewReader(b))
-			if err != nil {
-				log.Fatalf("create request: %v", err)
-			}
-			req.Header.Set("Content-Type", "application/json")
-			resp, err = http.DefaultClient.Do(req)
-			if err == nil {
-				break
-			}
-			time.Sleep(time.Second * time.Duration(i+1))
-		}
-		if err != nil {
-			log.Fatalf("upsert request: %v", err)
-		}
-		defer resp.Body.Close()
-		if resp.StatusCode != http.StatusOK {
-			body, _ := io.ReadAll(resp.Body)
-			log.Fatalf("upsert failed: %s: %s", resp.Status, strings.TrimSpace(string(body)))
-		}
+               var resp *http.Response
+               var req *http.Request
+               for i := 0; i < 3; i++ {
+                       req, err = http.NewRequestWithContext(ctx, http.MethodPost, baseURL+"/api/rag/upsert", bytes.NewReader(b))
+                       if err != nil {
+                               log.Fatalf("create request: %v", err)
+                       }
+                       req.Header.Set("Content-Type", "application/json")
+                       resp, err = http.DefaultClient.Do(req)
+                       if err == nil {
+                               break
+                       }
+                       time.Sleep(time.Second * time.Duration(i+1))
+               }
+               if err != nil {
+                       log.Fatalf("upsert request: %v", err)
+               }
+               if resp == nil {
+                       log.Fatalf("upsert request returned no response")
+               }
+               defer resp.Body.Close()
+               if resp.StatusCode != http.StatusOK {
+                       body, _ := io.ReadAll(resp.Body)
+                       log.Fatalf("upsert failed: %s: %s", resp.Status, strings.TrimSpace(string(body)))
+               }
 		log.Printf("ingested %d chunks for %s", len(rows), rel)
 		return
 	}