Update go/genkit/x/genkit.go

apascal07 · gemini-code-assist[bot] · web-flow · commit 14df9d2a78c4 · 2026-02-11T10:39:22.000-08:00
Co-authored-by: gemini-code-assist[bot] &lt;176961590+gemini-code-assist[bot]@users.noreply.github.com&gt;
diff --git a/go/genkit/x/genkit.go b/go/genkit/x/genkit.go
@@ -92,9 +92,8 @@ func DefineStreamingFlow[In, Out, Stream any](g *genkit.Genkit, name string, fn
 				for range discardCh {
 				}
 			}()
-			output, err := fn(ctx, input, discardCh)
-			close(discardCh)
-			return output, err
+defer close(discardCh)
+return fn(ctx, input, discardCh)
 		}
 
 		// Create a cancellable context for the user function.

Original file line number	Diff line number	Diff line change
`@@ -92,9 +92,8 @@ func DefineStreamingFlow[In, Out, Stream any](g *genkit.Genkit, name string, fn`
`92`	`92`	`for range discardCh {`
`93`	`93`	`}`
`94`	`94`	`}()`
`95`		`- output, err := fn(ctx, input, discardCh)`
`96`		`- close(discardCh)`
`97`		`- return output, err`
	`95`	`+defer close(discardCh)`
	`96`	`+return fn(ctx, input, discardCh)`
`98`	`97`	`}`
`99`	`98`
`100`	`99`	`// Create a cancellable context for the user function.`