Replace with GetModelByName, output friendly name

cmd/run/run.go

@@ -227,10 +227,11 @@ func NewRunCommand() *cobra.Command {
 				modelName = args[0]
 			}
 
-			modelName, err = util.ValidateModelName(modelName, models)
+			model, err := util.GetModelByName(modelName, models)
 			if err != nil {
 				return err
 			}
+			modelName = model.Name
 
 			initialPrompt := ""
 			singleShot := false

cmd/view/view.go

@@ -61,12 +61,12 @@ func NewViewCommand() *cobra.Command {
 				modelName = args[0]
 			}
 
-			modelName, err = util.ValidateModelName(modelName, models)
+			model, err := util.GetModelByName(modelName, models)
 			if err != nil {
 				return err
 			}
 
-			io.WriteString(out, "You selected: "+modelName+"\n")
+			io.WriteString(out, "You selected: "+model.FriendlyName+"\n")
 			return nil
 		},
 	}

pkg/util/models.go

@@ -7,12 +7,12 @@ import (
 	"github.com/github/gh-models/internal/azure_models"
 )
 
-// ValidateModelName checks whether the given model name is a valid one, based on the provided list of models.
-func ValidateModelName(modelName string, models []*azure_models.ModelSummary) (string, error) {
+// GetModelByName returns the model with the specified name, or an error if no such model exists within the given list.
+func GetModelByName(modelName string, models []*azure_models.ModelSummary) (*azure_models.ModelSummary, error) {
 	for _, model := range models {
 		if strings.EqualFold(model.FriendlyName, modelName) || strings.EqualFold(model.Name, modelName) {
-			return model.Name, nil
+			return model, nil
 		}
 	}
-	return "", fmt.Errorf("the specified model name is not supported: %s", modelName)
+	return nil, fmt.Errorf("the specified model name is not supported: %s", modelName)
 }