From 09a6f76f4c30fb8a9708680c519d08feeb504197 Mon Sep 17 00:00:00 2001
From: Jeffrey Morgan <jmorganca@gmail.com>
Date: Thu, 1 Feb 2024 23:11:52 -0800
Subject: [PATCH] fix error on `ollama run` with a non-existent model

---
 cmd/cmd.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/cmd/cmd.go b/cmd/cmd.go
index c393a9e2..66909c2c 100644
--- a/cmd/cmd.go
+++ b/cmd/cmd.go
@@ -147,6 +147,7 @@ func RunHandler(cmd *cobra.Command, args []string) error {
 	}
 
 	name := args[0]
+
 	// check if the model exists on the server
 	show, err := client.Show(cmd.Context(), &api.ShowRequest{Name: name})
 	var statusError api.StatusError
@@ -155,6 +156,11 @@ func RunHandler(cmd *cobra.Command, args []string) error {
 		if err := PullHandler(cmd, []string{name}); err != nil {
 			return err
 		}
+
+		show, err = client.Show(cmd.Context(), &api.ShowRequest{Name: name})
+		if err != nil {
+			return err
+		}
 	case err != nil:
 		return err
 	}