no prompt on empty line

This commit is contained in:
Michael Yang 2023-07-07 10:12:58 -07:00
parent 3d73ad0c56
commit 053739d19f
2 changed files with 31 additions and 26 deletions

View file

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

View file

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