add support to specify default model while running the binary

Author: Saumya40-codes

main.go

@@ -18,6 +18,7 @@ var (
 	FileLocation    string
 	HistoryLocation string
 	PromptText      = panes.PromptText
+	SelectedModel   = panes.Selected
 )
 
 func init() {
@@ -29,7 +30,7 @@ func init() {
 	err = os.MkdirAll(filepath.Dir(HistoryLocation), os.ModePerm)
 	checkNilErr(err)
 
-	logFile, _ := os.OpenFile("lazyai.log", os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0666)
+	logFile, _ := os.OpenFile("lazyai.log", os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0o666)
 	log.SetOutput(logFile)
 }
 
@@ -50,8 +51,17 @@ func setupUI(detachedMode *bool, app *tview.Application) {
 	}
 }
 
+func throwInvalidModelErr() {
+	fmt.Println("Invalid model used, available models:")
+	for _, model := range panes.AvailableModels {
+		fmt.Println(model)
+	}
+	os.Exit(0)
+}
+
 func main() {
 	detachedMode := flag.Bool("d", false, "Run in detached mode")
+	defaultModel := flag.String("m", "gemini-2.0-flash", "Set the default model to use")
 	defaultPrompt := flag.String("p", "", "Set the default prompt")
 	helpCommand := flag.Bool("help", false, "Show help commands")
 	flag.Parse()
@@ -70,6 +80,21 @@ func main() {
 		PromptText.PromptString = *defaultPrompt
 	}
 
+	// check if we support specified model
+	validModel := false
+	for _, v := range panes.AvailableModels {
+		if v.SelectedModel == *defaultModel {
+			validModel = true
+			break
+		}
+	}
+
+	if !validModel {
+		throwInvalidModelErr()
+	}
+
+	panes.Selected.SelectedModel = *defaultModel
+
 	app := tview.NewApplication().EnableMouse(true)
 
 	if !api.CheckCredentials(FileLocation, nil) {
@@ -80,7 +105,7 @@ func main() {
 				return
 			}
 
-			err := os.WriteFile(FileLocation, []byte(apiInput), 0644)
+			err := os.WriteFile(FileLocation, []byte(apiInput), 0o644)
 			checkNilErr(err)
 
 			log.Println("Starting clipboard monitoring after credential input.")

panes/modelsPane.go

@@ -12,7 +12,7 @@ var (
 	Selected  *internal.Model
 )
 
-var availableModels = map[string]*internal.Model{
+var AvailableModels = map[string]*internal.Model{
 	"Gemini 2.0 Flash":      {SelectedModel: "gemini-2.0-flash"},
 	"Gemini 2.0 Flash-Lite": {SelectedModel: "gemini-2.0-flash-lite"},
 	"Gemini Pro 1.5":        {SelectedModel: "gemini-1.5-pro"},
@@ -24,10 +24,10 @@ func init() {
 	ModelList.ShowSecondaryText(false).SetTitle(" Models ").SetBorder(true)
 	Selected = &internal.Model{}
 
-	SelectModel(availableModels["Gemini 2.0 Flash"].SelectedModel)
+	SelectModel(AvailableModels["Gemini 2.0 Flash"].SelectedModel)
 
 	// Add models to the list
-	for key, model := range availableModels {
+	for key, model := range AvailableModels {
 		currentModel := model
 		ModelList.AddItem(key, "", 0, func() {
 			SelectModel(currentModel.SelectedModel)
@@ -40,7 +40,7 @@ func init() {
 		case tcell.KeyEnter:
 			currentItem := ModelList.GetCurrentItem()
 			mainText, _ := ModelList.GetItemText(currentItem)
-			SelectModel(availableModels[mainText].SelectedModel)
+			SelectModel(AvailableModels[mainText].SelectedModel)
 		}
 		return event
 	})

panes/outputPane.go

@@ -18,10 +18,11 @@ var (
 )
 
 func init() {
-	var commands string = "\n## Commands:\n\n" +
+	commands := "\n## Commands:\n\n" +
 		"- run the application: `lazyAi`\n\n" +
 		"- run the application in detached mode: `lazyAi -d`\n" +
-		"- run the application with a default prompt: `lazyAi -p \"your prompt here\"`\n"
+		"- run the application with a default prompt: `lazyAi -p \"your prompt here\"`\n" +
+		"- run the application with a default model: `lazyAi -m \"gemini-2.0-flash\"`\n"
 
 	helpcmd := true
 	HelpCommands = MarkdownToTview(commands, &helpcmd)