|
@@ -102,16 +102,14 @@ 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
|
|
_, err = client.Show(context.Background(), &api.ShowRequest{Name: name})
|
|
_, err = client.Show(context.Background(), &api.ShowRequest{Name: name})
|
|
- if err != nil {
|
|
|
|
- var statusError api.StatusError
|
|
|
|
- switch {
|
|
|
|
- case errors.As(err, &statusError) && statusError.StatusCode == http.StatusNotFound:
|
|
|
|
- if err := PullHandler(cmd, args); err != nil {
|
|
|
|
- return err
|
|
|
|
- }
|
|
|
|
- case err != nil:
|
|
|
|
|
|
+ var statusError api.StatusError
|
|
|
|
+ switch {
|
|
|
|
+ case errors.As(err, &statusError) && statusError.StatusCode == http.StatusNotFound:
|
|
|
|
+ if err := PullHandler(cmd, args); err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
|
|
+ case err != nil:
|
|
|
|
+ return err
|
|
}
|
|
}
|
|
|
|
|
|
return RunGenerate(cmd, args)
|
|
return RunGenerate(cmd, args)
|