Skip to content

GitLab

Unverified Commit 0fea50cd authored by Michael Yang's avatar Michael Yang Committed by GitHub
Browse files

Merge pull request #54 from jmorganca/empty-line 

no prompt on empty line
parents c3168935 053739d1
Show whitespace changes
Inline Side-by-side
Showing with 32 additions and 27 deletions
cmd/cmd.go
View file @ 0fea50cd
...@@ -9,6 +9,7 @@ import ( ...@@ -9,6 +9,7 @@ import (
"net" "net"
"os" "os"
"path" "path"
"strings"
"time" "time"
"github.com/schollz/progressbar/v3" "github.com/schollz/progressbar/v3"
...@@ -75,6 +76,7 @@ func RunGenerate(_ *cobra.Command, args []string) error { ...@@ -75,6 +76,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,
...@@ -107,6 +109,8 @@ func generate(model, prompt string) error { ...@@ -107,6 +109,8 @@ func generate(model, prompt string) error {
fmt.Println() fmt.Println()
fmt.Println() fmt.Println()
}
return nil return nil
} }
......
server/templates/wizardlm.prompt
View file @ 0fea50cd
{{ .Prompt }} {{ .Prompt }}
### Response: ### Response:
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment