baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix error on ollama run with a non-existent model

Browse source
This commit is contained in:
Jeffrey Morgan 2024-02-01 23:11:52 -08:00
parent e135167484
commit 09a6f76f4c
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
cmd/cmd.go
View file

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