baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

take all args as one prompt

Browse source
This commit is contained in:
Bruce MacDonald 2023-07-10 06:05:09 -04:00 committed by GitHub
commit 07a4c1e3fb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
README.md
View file

@ -48,6 +48,14 @@ Get a helping hand.
ollama run orca "Write an email to my boss." 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 ### 📖 Storytelling
Venture into the unknown. Venture into the unknown.

15
cmd/cmd.go
View file

@ -64,8 +64,10 @@ func pull(model string) error {
} }
func RunGenerate(_ *cobra.Command, args []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 { if len(args) > 1 {
return generateOneshot(args[0], args[1:]...) return generate(args[0], prompt)
} }
if term.IsTerminal(int(os.Stdin.Fd())) { if term.IsTerminal(int(os.Stdin.Fd())) {
@ -114,17 +116,6 @@ func generate(model, prompt string) error {
return nil 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 { func generateInteractive(model string) error {
fmt.Print(">>> ") fmt.Print(">>> ")
scanner := bufio.NewScanner(os.Stdin) scanner := bufio.NewScanner(os.Stdin)