From 2ecc3a33c380cd7915a8541d629d7f21c2b8472a Mon Sep 17 00:00:00 2001 From: Quinn Slack Date: Sat, 26 Aug 2023 00:47:56 -0700 Subject: [PATCH] delete all models (not just 1st) in `ollama rm` (#415) Previously, `ollama rm model1 model2 modelN` would only delete `model1`. The other model command-line arguments would be silently ignored. Now, all models mentioned are deleted. --- cmd/cmd.go | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/cmd/cmd.go b/cmd/cmd.go index eab7bad8..5ce83326 100644 --- a/cmd/cmd.go +++ b/cmd/cmd.go @@ -220,11 +220,13 @@ func DeleteHandler(cmd *cobra.Command, args []string) error { return err } - req := api.DeleteRequest{Name: args[0]} - if err := client.Delete(context.Background(), &req); err != nil { - return err + for _, name := range args { + req := api.DeleteRequest{Name: name} + if err := client.Delete(context.Background(), &req); err != nil { + return err + } + fmt.Printf("deleted '%s'\n", name) } - fmt.Printf("deleted '%s'\n", args[0]) return nil }