From c344da4c5a09521fce0c825dcef4df1c155b150d Mon Sep 17 00:00:00 2001
From: Patrick Devine <pdevine@sonic.net>
Date: Tue, 14 May 2024 15:17:04 -0700
Subject: [PATCH] fix keepalive for non-interactive mode (#4438)

---
 cmd/cmd.go | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/cmd/cmd.go b/cmd/cmd.go
index f1429e1c..cae35f51 100644
--- a/cmd/cmd.go
+++ b/cmd/cmd.go
@@ -905,14 +905,15 @@ func generate(cmd *cobra.Command, opts runOptions) error {
 	}
 
 	request := api.GenerateRequest{
-		Model:    opts.Model,
-		Prompt:   opts.Prompt,
-		Context:  generateContext,
-		Images:   opts.Images,
-		Format:   opts.Format,
-		System:   opts.System,
-		Template: opts.Template,
-		Options:  opts.Options,
+		Model:     opts.Model,
+		Prompt:    opts.Prompt,
+		Context:   generateContext,
+		Images:    opts.Images,
+		Format:    opts.Format,
+		System:    opts.System,
+		Template:  opts.Template,
+		Options:   opts.Options,
+		KeepAlive: opts.KeepAlive,
 	}
 
 	if err := client.Generate(ctx, &request, fn); err != nil {