prompt: Close channel when done (#21)

sunshineplan · Mar 19, 2024 · a0e3fd9 · a0e3fd9
1 parent d1b4928
commit a0e3fd9
Showing 1 changed file with 13 additions and 10 deletions.
diff --git a/prompt/prompt.go b/prompt/prompt.go
@@ -128,15 +128,18 @@ func (prompt *Prompt) Execute(ai ai.AI, input []string, prefix string) (<-chan R
 		return nil, err
 	}
 	c := make(chan Result, len(prompts))
-	go workers.RunSlice(prompt.workers, prompts, func(i int, p string) {
-		ctx, cancel := context.WithTimeout(context.Background(), prompt.d)
-		defer cancel()
-		resp, err := ai.Chat(ctx, p)
-		if err != nil {
-			c <- Result{i, p, nil, err}
-		} else {
-			c <- Result{i, p, resp.Results(), nil}
-		}
-	})
+	go func() {
+		workers.RunSlice(prompt.workers, prompts, func(i int, p string) {
+			ctx, cancel := context.WithTimeout(context.Background(), prompt.d)
+			defer cancel()
+			resp, err := ai.Chat(ctx, p)
+			if err != nil {
+				c <- Result{i, p, nil, err}
+			} else {
+				c <- Result{i, p, resp.Results(), nil}
+			}
+		})
+		close(c)
+	}()
 	return c, nil
 }