fix formatting when exiting ollama run

ollama · Oct 28, 2023 · 9ec16f0 · 9ec16f0
1 parent 57a58db
commit 9ec16f0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/cmd/cmd.go b/cmd/cmd.go
@@ -561,6 +561,7 @@ func generateInteractive(cmd *cobra.Command, model string) error {
 		line, err := scanner.Readline()
 		switch {
 		case errors.Is(err, io.EOF):
+			fmt.Println()
 			return nil
 		case errors.Is(err, readline.ErrInterrupt):
 			if line == "" {

diff --git a/readline/readline.go b/readline/readline.go
@@ -76,14 +76,15 @@ func (i *Instance) Readline() (string, error) {
 		}
 
 		r, err := i.Terminal.Read()
-		if err != nil {
-			return "", io.EOF
-		}
 
 		if buf.IsEmpty() {
 			fmt.Print(ClearToEOL)
 		}
 
+		if err != nil {
+			return "", io.EOF
+		}
+
 		if escex {
 			escex = false