diff --git a/cmd/cmd.go b/cmd/cmd.go index bec5c3ea..bd3db19a 100644 --- a/cmd/cmd.go +++ b/cmd/cmd.go @@ -951,7 +951,8 @@ func generateInteractive(cmd *cobra.Command, opts generateOptions) error { cmd.SetContext(ctx) } if len(opts.Images) == 0 { - fmt.Println("This model requires you to add a jpeg, png, or svg image.\n") + fmt.Println("This model requires you to add a jpeg, png, or svg image.") + fmt.Println() prompt = "" continue } @@ -1069,7 +1070,7 @@ func getImageData(filePath string) ([]byte, error) { // Check if the file size exceeds 100MB var maxSize int64 = 100 * 1024 * 1024 // 100MB in bytes if info.Size() > maxSize { - return nil, fmt.Errorf("file size exceeds maximum limit (100MB).") + return nil, fmt.Errorf("file size exceeds maximum limit (100MB)") } buf = make([]byte, info.Size()) diff --git a/server/images_test.go b/server/images_test.go index 28cb39a3..17c433cd 100644 --- a/server/images_test.go +++ b/server/images_test.go @@ -81,7 +81,7 @@ func TestChat(t *testing.T) { Template: tt.template, } t.Run(tt.name, func(t *testing.T) { - got, err := m.ChatPrompt(tt.msgs) + got, _, err := m.ChatPrompt(tt.msgs) if tt.wantErr != "" { if err == nil { t.Errorf("ChatPrompt() expected error, got nil")