feat: --ask-model/-M

config.go

@@ -33,6 +33,7 @@ var help = map[string]string{
 	"apis":              "Aliases and endpoints for OpenAI compatible REST API.",
 	"http-proxy":        "HTTP proxy to use for API requests.",
 	"model":             "Default model (gpt-3.5-turbo, gpt-4, ggml-gpt4all-j...).",
+	"ask-model":         "Shows a view to select the model to use.",
 	"max-input-chars":   "Default character limit on input to model.",
 	"format":            "Ask for the response to be formatted as markdown unless otherwise set.",
 	"format-text":       "Text to append when using the -f flag.",
@@ -141,6 +142,7 @@ type Config struct {
 	FormatText        FormatText `yaml:"format-text" env:"FORMAT_TEXT"`
 	HTTPProxy         string     `yaml:"http-proxy" env:"HTTP_PROXY"`
 	APIs              APIs       `yaml:"apis"`
+	AskModel          bool
 	FormatAs          string
 	API               string
 	Models            map[string]Model

main.go

@@ -84,36 +84,15 @@ var (
 			}
 
 			if isNoArgs() && isInputTTY() {
-				var opts []huh.Option[string]
-				for _, api := range config.APIs {
-					for model := range api.Models {
-						opts = append(opts, huh.Option[string]{
-							Key:   model,
-							Value: model,
-						})
-					}
+				var groups []*huh.Group
+				if config.AskModel {
+					groups = append(groups, huh.NewGroup(newModelSelect()))
 				}
-
-				if err := huh.Run(
-					huh.NewSelect[string]().
-						Title("Model").
-						Options(opts...).
-						Value(&config.Model),
-				); err != nil {
+				groups = append(groups, huh.NewGroup(newPromptInput()))
+				if err := huh.NewForm(groups...).Run(); err != nil {
 					return modsError{
 						err:    err,
-						reason: "Failed to ask for model.",
-					}
-				}
-				if err := huh.Run(
-					huh.NewText().
-						Title(fmt.Sprintf("Write a prompt for %s:", config.Model)).
-						Lines(4). //nolint: gomnd
-						Value(&config.Prefix),
-				); err != nil {
-					return modsError{
-						err:    err,
-						reason: "No prompt provided and failed to ask for one.",
+						reason: "Prompt failed",
 					}
 				}
 			}
@@ -217,6 +196,7 @@ var (
 func initFlags() {
 	flags := rootCmd.Flags()
 	flags.StringVarP(&config.Model, "model", "m", config.Model, stdoutStyles().FlagDesc.Render(help["model"]))
+	flags.BoolVarP(&config.AskModel, "ask-model", "M", config.AskModel, stdoutStyles().FlagDesc.Render(help["ask-model"]))
 	flags.StringVarP(&config.API, "api", "a", config.API, stdoutStyles().FlagDesc.Render(help["api"]))
 	flags.StringVarP(&config.HTTPProxy, "http-proxy", "x", config.HTTPProxy, stdoutStyles().FlagDesc.Render(help["http-proxy"]))
 	flags.BoolVarP(&config.Format, "format", "f", config.Format, stdoutStyles().FlagDesc.Render(help["format"]))
@@ -563,3 +543,29 @@ func isNoArgs() bool {
 		!config.Dirs &&
 		!config.ResetSettings
 }
+
+func newPromptInput() *huh.Text {
+	title := fmt.Sprintf("Write a prompt for %s:", config.Model)
+	if config.AskModel {
+		title = "Write a prompt:"
+	}
+	return huh.NewText().
+		Title(title).
+		Value(&config.Prefix)
+}
+
+func newModelSelect() *huh.Select[string] {
+	var opts []huh.Option[string]
+	for _, api := range config.APIs {
+		for model := range api.Models {
+			opts = append(opts, huh.Option[string]{
+				Key:   model,
+				Value: model,
+			})
+		}
+	}
+	return huh.NewSelect[string]().
+		Title("Select the model:").
+		Options(opts...).
+		Value(&config.Model)
+}