mirror of
https://github.com/ollama/ollama.git
synced 2025-12-04 23:51:54 +01:00
take all args as one prompt
- parse all run arguments into one prompt - do not echo prompt back on one-shot - example of summarizing a document
This commit is contained in:
15
cmd/cmd.go
15
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)
|
||||
|
||||
Reference in New Issue
Block a user