Merge pull request #54 from jmorganca/empty-line

no prompt on empty line

cmd/cmd.go

@@ -9,6 +9,7 @@ import (
 	"net"
 	"os"
 	"path"
+	"strings"
 	"time"
 
 	"github.com/schollz/progressbar/v3"
@@ -75,6 +76,7 @@ func RunGenerate(_ *cobra.Command, args []string) error {
 }
 
 func generate(model, prompt string) error {
+	if len(strings.TrimSpace(prompt)) > 0 {
 		client := api.NewClient()
 
 		spinner := progressbar.NewOptions(-1,
@@ -107,6 +109,8 @@ func generate(model, prompt string) error {
 
 		fmt.Println()
 		fmt.Println()
+	}
+
 	return nil
 }
 

server/templates/wizardlm.prompt

@@ -1,2 +1,3 @@
 {{ .Prompt }}
+
 ### Response: