diff --git a/README.md b/README.md index 325adedf..bf19cdba 100644 --- a/README.md +++ b/README.md @@ -48,6 +48,14 @@ Get a helping hand. ollama run orca "Write an email to my boss." ``` +### 🔎 Ask questions about documents + +Send the contents of a document and ask questions about it. + +``` +ollama run nous-hermes "$(cat input.txt)", please summarize this story +``` + ### 📖 Storytelling Venture into the unknown. diff --git a/cmd/cmd.go b/cmd/cmd.go index 8fea7e1e..ad52853e 100644 --- a/cmd/cmd.go +++ b/cmd/cmd.go @@ -64,8 +64,10 @@ func pull(model string) error { } func RunGenerate(_ *cobra.Command, args []string) error { + // join all args into a single prompt + prompt := strings.Join(args[1:], " ") if len(args) > 1 { - return generateOneshot(args[0], args[1:]...) + return generate(args[0], prompt) } if term.IsTerminal(int(os.Stdin.Fd())) { @@ -114,17 +116,6 @@ func generate(model, prompt string) error { return nil } -func generateOneshot(model string, prompts ...string) error { - for _, prompt := range prompts { - fmt.Printf(">>> %s\n", prompt) - if err := generate(model, prompt); err != nil { - return err - } - } - - return nil -} - func generateInteractive(model string) error { fmt.Print(">>> ") scanner := bufio.NewScanner(os.Stdin)