|
@@ -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)
|