prompt: Execute return result count (#22)

sunshineplan · Mar 19, 2024 · 87c6832 · 87c6832
1 parent a0e3fd9
commit 87c6832
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/prompt/prompt.go b/prompt/prompt.go
@@ -122,12 +122,13 @@ type Result struct {
  Error error
 }
 
-func (prompt *Prompt) Execute(ai ai.AI, input []string, prefix string) (<-chan Result, error) {
+func (prompt *Prompt) Execute(ai ai.AI, input []string, prefix string) (<-chan Result, int, error) {
  prompts, err := prompt.execute(input, prefix)
  if err != nil {
- return nil, err
+ return nil, 0, err
  }
- c := make(chan Result, len(prompts))
+ n := len(prompts)
+ c := make(chan Result, n)
  go func() {
  workers.RunSlice(prompt.workers, prompts, func(i int, p string) {
  ctx, cancel := context.WithTimeout(context.Background(), prompt.d)
@@ -141,5 +142,5 @@ func (prompt *Prompt) Execute(ai ai.AI, input []string, prefix string) (<-chan R
  })
  close(c)
  }()
- return c, nil
+ return c, n, nil
 }